v2.0

server/internal/logic/middleware/response.go

@@ -0,0 +1,74 @@
+// Package middleware
+// @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/gogf/gf/v2/errors/gcode"
+	"github.com/gogf/gf/v2/frame/g"
+	"github.com/gogf/gf/v2/net/ghttp"
+	"hotgo/internal/consts"
+	"hotgo/internal/library/contexts"
+	"hotgo/internal/library/response"
+	"hotgo/utility/charset"
+)
+
+// ResponseHandler HTTP响应预处理
+func (s *sMiddleware) ResponseHandler(r *ghttp.Request) {
+	r.Middleware.Next()
+
+	var (
+		ctx         = r.Context()
+		comResponse = contexts.Get(ctx).Response
+		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("服务器居然开小差了，请稍后再试吧！")
+	}
+
+	// 已存在响应内容，且是comResponse返回的时，中断运行
+	if r.Response.BufferLength() > 0 && comResponse != nil {
+		return
+	}
+
+	if err = r.GetError(); err != nil {
+		// 记录到自定义错误日志文件
+		g.Log("exception").Print(r.Context(), "exception:", err)
+
+		code = consts.CodeInternalError
+		message = "服务器居然开小差了，请稍后再试吧！"
+
+		// 是否输出错误到页面
+		if debug, _ := g.Cfg().Get(ctx, "hotgo.debug", true); debug.Bool() {
+			data = charset.GetStack(err)
+			message = err.Error()
+		}
+
+		//} 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()
+		//}
+
+	} else {
+		data = r.GetHandlerResponse()
+	}
+
+	// 返回固定的友好信息
+	response.RJson(r, code, message, data)
+}