全局异常处理

This commit is contained in:
2026-02-05 22:38:02 +08:00
parent 8e3141d3c9
commit b88198f0f8
4 changed files with 81 additions and 2 deletions

View File

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

View File

@@ -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,
})
}

View File

@@ -2,4 +2,5 @@ export interface ApiResponse {
code?: number; code?: number;
message?: string; message?: string;
data?: any; data?: any;
err?: string;
} }

View File

@@ -227,6 +227,11 @@ const fetchPosts = async () => {
title: searchText.value || undefined title: searchText.value || undefined
}) })
const data = (res as any).data || res || {} const data = (res as any).data || res || {}
const errMsg = (res as any).err
if (errMsg) {
message.error(errMsg)
return
}
posts.value = data.list || [] posts.value = data.list || []
total.value = data.total || 0 total.value = data.total || 0
} catch (error) { } catch (error) {
@@ -266,13 +271,20 @@ const handleCreatePost = async () => {
createLoading.value = true createLoading.value = true
try { try {
await postApiForumPostsCreate({ const res = await postApiForumPostsCreate({
boardId: Number("1"),//默认板块IDtodo
userId: currentUserId.value, userId: currentUserId.value,
title: createForm.title, title: createForm.title,
content: createForm.content, content: createForm.content,
coverImage: createForm.coverImage, coverImage: createForm.coverImage,
status: 'NORMAL' status: 'NORMAL'
}) })
const data = (res as any).data || res || {}
const errMsg = (res as any).err
if (errMsg) {
message.error(errMsg)
return
}
message.success('情报发布成功') message.success('情报发布成功')
createModalVisible.value = false createModalVisible.value = false
fetchPosts() fetchPosts()