tt

hotgo-server/app/middleware/handler_response.go

@@ -0,0 +1,74 @@
+//
+// @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)
+}