change response error with data logic

Author: ArronDou

api/gogen/default-logic.tpl

@@ -3,6 +3,8 @@ package logic
 import (
 	"context"
     "net/http"
+	"encoding/json"
+	"fmt"
     "github.com/zeromicro/go-zero/core/logx"
 	{{.imports}}
 )
@@ -13,6 +15,7 @@ type {{.logic}} struct {
     ctx        context.Context
     reqCtx     *http.Request
     respWriter *http.ResponseWriter
+    NewErrorResponse func(code int, resp any) *ResponseDataError
 }
 
 func New{{.logic}}(ctx context.Context, svcCtx *svc.ServiceContext, reqCtx *http.Request, respWriter *http.ResponseWriter) *{{.logic}} {
@@ -22,5 +25,26 @@ func New{{.logic}}(ctx context.Context, svcCtx *svc.ServiceContext, reqCtx *http
 		ctx:            ctx,
 		reqCtx:         reqCtx,
 		respWriter:     respWriter,
+		NewErrorResponse: NewErrorResponse,
+	}
+}
+
+type ResponseDataError struct {
+	Data any `json:"data"`
+	Code int `json:"code"`
+}
+
+func (e *ResponseDataError) Error() string {
+	jsonStr, err := json.Marshal(e.Data)
+	if err != nil {
+		return fmt.Sprint(e.Data)
+	}
+	return string(jsonStr)
+}
+
+func NewErrorResponse(code int, resp any) *ResponseDataError {
+	return &ResponseDataError{
+		Data: resp,
+		Code: code,
 	}
 }

api/gogen/handler.tpl

@@ -20,8 +20,12 @@ func {{.HandlerName}}(svcCtx *svc.ServiceContext) http.HandlerFunc {
 		{{end}}l := {{.LogicName}}.New{{.LogicType}}(r.Context(), svcCtx, r, &w)
 		{{if .HasResp}}resp, {{end}}err := l.{{.Call}}({{if .HasRequest}}&req{{end}})
 		if err != nil {
-			l.Error(err)
-			httpx.ErrorCtx(r.Context(), w, err)
+			if respErr, ok := err.(*logic.ResponseDataError); ok {
+                httpx.WriteJson(w, respErr.Code, respErr.Data)
+            } else {
+                l.Error(err)
+                httpx.ErrorCtx(r.Context(), w, err)
+            }
 		} else {
 			{{if .HasResp}}httpx.OkJsonCtx(r.Context(), w, resp){{else}}httpx.Ok(w){{end}}
 		}

api/gogen/main.tpl

@@ -3,6 +3,7 @@ package main
 import (
 	"flag"
 	"fmt"
+	"os"
 	"github.com/zeromicro/go-zero/core/logx"
 
 	{{.importPackages}}