优化

internal/server/handler/export.go

@@ -22,15 +22,26 @@ func NewExport(service *service.ExportService) *Export {
 	return &Export{service: service}
 }
 
-func exportParams(values url.Values) map[string]string {
+func exportParams(values url.Values) (map[string]string, error) {
 	out := map[string]string{}
-	nested, _ := url.ParseQuery(values.Get("params"))
+	nested, err := url.ParseQuery(values.Get("params"))
+	if err != nil {
+		return nil, err
+	}
 	for key, items := range nested {
 		if len(items) > 0 {
 			out[key] = items[0]
 		}
 	}
-	return out
+	return out, nil
+}
+
+func exportIssueParams(values url.Values, blank bool) (map[string]string, error) {
+	params, err := exportParams(values)
+	if err != nil && blank {
+		return map[string]string{}, nil
+	}
+	return params, err
 }
 
 func (h *Export) Create(c *gin.Context) {
@@ -133,7 +144,8 @@ func (h *Export) Preview(c *gin.Context) {
 		httpx.Fail(c, "模板ID不能为空")
 		return
 	}
-	sql, err := h.service.Preview(c.Request.Context(), templateID, exportParams(c.Request.URL.Query()))
+	params, _ := exportParams(c.Request.URL.Query())
+	sql, err := h.service.Preview(c.Request.Context(), templateID, params)
 	if err != nil {
 		httpx.Fail(c, "获取失败")
 		return
@@ -148,7 +160,12 @@ func (h *Export) Issue(blank bool) gin.HandlerFunc {
 			httpx.Fail(c, "模板ID不能为空")
 			return
 		}
-		token, err := h.service.IssueToken(c.Request.Context(), templateID, exportParams(c.Request.URL.Query()), blank)
+		params, err := exportIssueParams(c.Request.URL.Query(), blank)
+		if err != nil {
+			httpx.Fail(c, "解析 params 参数失败")
+			return
+		}
+		token, err := h.service.IssueToken(c.Request.Context(), templateID, params, blank)
 		if err != nil {
 			httpx.Fail(c, "导出令牌创建失败")
 			return