全局异常处理

Backend/internal/cmd/cmd.go

@@ -3,6 +3,7 @@ package cmd
 import (
 	"TrangleAgent/internal/controller"
 	"TrangleAgent/internal/controller/websocket"
+	"TrangleAgent/internal/middleware"
 	"context"
 
 	"github.com/gogf/gf/v2/frame/g"
@@ -43,7 +44,7 @@ var (
 
 			// 注册 API 路由组
 			s.Group("/api", func(group *ghttp.RouterGroup) {
-				group.Middleware(ghttp.MiddlewareHandlerResponse)
+				group.Middleware(middleware.ResponseHandler)
 				controller.RegisterControllers(group)
 			})
 

Backend/internal/middleware/response.go

@@ -0,0 +1,65 @@
+package middleware
+
+import (
+	"net/http"
+
+	"github.com/gogf/gf/v2/errors/gcode"
+	"github.com/gogf/gf/v2/errors/gerror"
+	"github.com/gogf/gf/v2/net/ghttp"
+)
+
+// ResponseHandler custom response handler to add "err" field
+func ResponseHandler(r *ghttp.Request) {
+	r.Middleware.Next()
+
+	// There's custom buffer content, it then exits current handler.
+	if r.Response.BufferLength() > 0 {
+		return
+	}
+
+	var (
+		msg  string
+		err  error
+		code gcode.Code = gcode.CodeOK
+		res             = r.GetHandlerResponse()
+	)
+	if err = r.GetError(); err != nil {
+		code = gerror.Code(err)
+		if code == gcode.CodeNil {
+			code = gcode.CodeInternalError
+		}
+		msg = err.Error()
+	} else {
+		if r.Response.Status > 0 && r.Response.Status != 200 {
+			msg = http.StatusText(r.Response.Status)
+			switch r.Response.Status {
+			case 404:
+				code = gcode.CodeNotFound
+			case 403:
+				code = gcode.CodeNotAuthorized
+			default:
+				code = gcode.CodeInternalError
+			}
+		}
+	}
+
+	// Custom response structure
+	type Response struct {
+		Code    int         `json:"code"    dc:"Error code"`
+		Message string      `json:"message" dc:"Error message"`
+		Data    interface{} `json:"data"    dc:"Result data for certain request according to model definition"`
+		Err     string      `json:"err"     dc:"Error details"`
+	}
+
+	errStr := ""
+	if err != nil {
+		errStr = err.Error()
+	}
+
+	r.Response.WriteJson(Response{
+		Code:    code.Code(),
+		Message: msg,
+		Data:    res,
+		Err:     errStr,
+	})
+}