mirror of
https://github.com/bufanyun/hotgo.git
synced 2025-08-28 09:15:50 +08:00
75 lines
2.0 KiB
Go
75 lines
2.0 KiB
Go
//
|
||
// @Link https://github.com/bufanyun/hotgo
|
||
// @Copyright Copyright (c) 2022 HotGo CLI
|
||
// @Author Ms <133814250@qq.com>
|
||
// @License https://github.com/bufanyun/hotgo/blob/master/LICENSE
|
||
//
|
||
package middleware
|
||
|
||
import (
|
||
"github.com/bufanyun/hotgo/app/com"
|
||
"github.com/bufanyun/hotgo/app/consts"
|
||
"github.com/bufanyun/hotgo/app/utils"
|
||
"github.com/gogf/gf/v2/errors/gcode"
|
||
"github.com/gogf/gf/v2/errors/gerror"
|
||
"github.com/gogf/gf/v2/frame/g"
|
||
"github.com/gogf/gf/v2/net/ghttp"
|
||
)
|
||
|
||
//
|
||
// @Title 全局响应中间件
|
||
// @Description
|
||
// @Author Ms <133814250@qq.com>
|
||
// @Param r
|
||
//
|
||
func (s *sMiddleware) HandlerResponse(r *ghttp.Request) {
|
||
r.Middleware.Next()
|
||
|
||
var (
|
||
ctx = r.Context()
|
||
comResponse = com.Context.Get(ctx).ComResponse
|
||
code = gcode.CodeOK.Code()
|
||
message = "操作成功"
|
||
data interface{}
|
||
err error
|
||
)
|
||
|
||
if err := r.GetError(); err != nil {
|
||
g.Log().Print(ctx, err)
|
||
// 记录到自定义错误日志文件
|
||
//g.Log("exception").Error(err)
|
||
////返回固定的友好信息
|
||
//r.Response.ClearBuffer()
|
||
//r.Response.Writeln("服务器居然开小差了，请稍后再试吧！")
|
||
}
|
||
|
||
// TODO 已存在响应内容，且是comResponse返回的时，中断运行
|
||
if r.Response.BufferLength() > 0 && comResponse != nil {
|
||
return
|
||
}
|
||
|
||
if err = r.GetError(); err != nil {
|
||
// TODO 记录到自定义错误日志文件
|
||
g.Log("exception").Print(r.Context(), "exception:", err)
|
||
|
||
code = consts.CodeInternalError
|
||
message = "服务器居然开小差了，请稍后再试吧！"
|
||
|
||
// TODO 是否输出错误到页面
|
||
if debug, _ := g.Cfg().Get(ctx, "hotgo.debug", true); debug.Bool() {
|
||
data = utils.Charset.GetStack(err)
|
||
}
|
||
|
||
} else if data, err = r.GetHandlerResponse(); err != nil {
|
||
errCode := gerror.Code(err)
|
||
if errCode == gcode.CodeNil {
|
||
errCode = gcode.CodeInternalError
|
||
}
|
||
code = errCode.Code()
|
||
message = err.Error()
|
||
}
|
||
|
||
// TODO 返回固定的友好信息
|
||
com.Response.RJson(r, code, message, data)
|
||
}
|