75 lines
1.9 KiB
Go
75 lines
1.9 KiB
Go
package middleware
|
||
|
||
import (
|
||
"github.com/gogf/gf/v2/frame/g"
|
||
"github.com/gogf/gf/v2/net/ghttp"
|
||
"nl-video-api/utility/response"
|
||
)
|
||
|
||
// ResponseHandler 全局响应处理中间件
|
||
// 确保所有响应都返回HTTP 200状态码，错误信息通过code和message字段传递
|
||
func ResponseHandler(r *ghttp.Request) {
|
||
// 设置默认响应头
|
||
r.Response.Header().Set("Content-Type", "application/json")
|
||
|
||
// 继续执行后续中间件和处理函数
|
||
r.Middleware.Next()
|
||
|
||
// 检查响应状态码，如果不是200，则转换为统一格式
|
||
if r.Response.Status != 200 {
|
||
// 获取原始状态码
|
||
originalStatus := r.Response.Status
|
||
|
||
// 根据HTTP状态码映射到业务错误码
|
||
var businessCode int
|
||
var message string
|
||
|
||
switch originalStatus {
|
||
case 400:
|
||
businessCode = response.CodeInvalidParam
|
||
message = "请求参数错误"
|
||
case 401:
|
||
businessCode = response.CodeUnauthorized
|
||
message = "未授权访问"
|
||
case 403:
|
||
businessCode = response.CodeForbidden
|
||
message = "权限不足"
|
||
case 404:
|
||
businessCode = response.CodeNotFound
|
||
message = "资源不存在"
|
||
case 500:
|
||
businessCode = response.CodeServerError
|
||
message = "服务器内部错误"
|
||
default:
|
||
businessCode = response.CodeError
|
||
message = "请求处理失败"
|
||
}
|
||
|
||
// 清空原有响应内容
|
||
r.Response.ClearBuffer()
|
||
|
||
// 返回统一格式的错误响应，HTTP状态码为200
|
||
response.Error(r, businessCode, message)
|
||
}
|
||
}
|
||
|
||
// ErrorHandler 错误处理中间件
|
||
// 捕获panic和其他未处理的错误，统一返回格式
|
||
func ErrorHandler(r *ghttp.Request) {
|
||
defer func() {
|
||
if err := recover(); err != nil {
|
||
// 记录错误日志
|
||
g.Log().Errorf(r.Context(), "Panic recovered: %v", err)
|
||
|
||
// 清空响应缓冲区
|
||
r.Response.ClearBuffer()
|
||
|
||
// 返回统一格式的错误响应
|
||
response.Error(r, response.CodeServerError, "系统内部错误")
|
||
}
|
||
}()
|
||
|
||
// 继续执行后续中间件和处理函数
|
||
r.Middleware.Next()
|
||
}
|