添加返回log输出

common/apis/api.go

@@ -1,7 +1,9 @@
 package apis
 
 import (
+	"encoding/json"
 	"github.com/gin-gonic/gin"
+	"go-admin/common/log"
 	"gorm.io/gorm"
 	"net/http"
 
@@ -27,7 +29,9 @@ func (e *Api) Error(c *gin.Context, code int, err error, msg string) {
 		res.Msg = msg
 	}
 	res.RequestId = tools.GenerateMsgIDFromContext(c)
-	c.AbortWithStatusJSON(http.StatusOK, res.ReturnError(code))
+	Return := res.ReturnError(code)
+	setResult(c, Return, res.RequestId, code)
+	c.AbortWithStatusJSON(http.StatusOK, Return)
 }
 
 // OK 通常成功数据处理
@@ -38,7 +42,9 @@ func (e *Api) OK(c *gin.Context, data interface{}, msg string) {
 		res.Msg = msg
 	}
 	res.RequestId = tools.GenerateMsgIDFromContext(c)
-	c.AbortWithStatusJSON(http.StatusOK, res.ReturnOK())
+	Return := res.ReturnOK()
+	setResult(c, Return, res.RequestId, 200)
+	c.AbortWithStatusJSON(http.StatusOK, Return)
 }
 
 // PageOK 分页数据处理
@@ -53,5 +59,18 @@ func (e *Api) PageOK(c *gin.Context, result interface{}, count int, pageIndex in
 
 // Custom 兼容函数
 func (e *Api) Custom(c *gin.Context, data gin.H) {
-	c.AbortWithStatusJSON(http.StatusOK, data)
+	msgID := tools.GenerateMsgIDFromContext(c)
+	Return := data
+	setResult(c, Return, msgID, 200)
+	c.AbortWithStatusJSON(http.StatusOK, Return)
+}
+
+func setResult(c *gin.Context, Return interface{}, msgID string, status int) {
+	var jsonStr []byte
+	jsonStr, err := json.Marshal(Return)
+	if err != nil {
+		log.Debugf("MsgID[%s] setResult error: %#v", msgID, err.Error())
+	}
+	c.Set("result", string(jsonStr))
+	c.Set("status", status)
 }