You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
79 lines
1.9 KiB
79 lines
1.9 KiB
package trace_log
|
|
|
|
import (
|
|
"Open_IM/pkg/common/constant"
|
|
"Open_IM/pkg/common/log"
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"github.com/gin-gonic/gin"
|
|
"net/http"
|
|
)
|
|
|
|
const TraceLogKey = "trace_log"
|
|
const GinContextKey = "gin_context"
|
|
|
|
func NewCtx(c *gin.Context, api string) context.Context {
|
|
req := ApiInfo{ApiName: api, GinCtx: c}
|
|
req.OperationID = c.PostForm("operationID")
|
|
return context.WithValue(c, GinContextKey, req)
|
|
}
|
|
|
|
func ShowLog(ctx context.Context) {
|
|
t := ctx.Value(TraceLogKey).(ApiInfo)
|
|
log.Info(t.OperationID, "api: ", t.ApiName)
|
|
for _, v := range t.Funcs {
|
|
if v.Err != nil {
|
|
log.Error(v.FuncName, v.Err, v.Args)
|
|
} else {
|
|
log.Info(v.FuncName, v.Args)
|
|
}
|
|
}
|
|
}
|
|
|
|
func WriteErrorResponse(ctx context.Context, funcName string, err error) {
|
|
SetContextInfo(ctx, funcName, err)
|
|
e := new(constant.ErrInfo)
|
|
switch {
|
|
case errors.As(err, &e):
|
|
ctx.Value(GinContextKey).(ApiInfo).GinCtx.JSON(http.StatusOK, gin.H{"errCode": e.ErrCode, "errMsg": e.ErrMsg})
|
|
return
|
|
default:
|
|
ctx.Value(GinContextKey).(ApiInfo).GinCtx.JSON(http.StatusOK, gin.H{"errCode": constant.ErrDefaultOther.ErrCode, "errMsg": constant.ErrDefaultOther.ErrMsg})
|
|
return
|
|
}
|
|
}
|
|
|
|
type ApiInfo struct {
|
|
ApiName string
|
|
OperationID string
|
|
Funcs []FuncInfo
|
|
GinCtx *gin.Context
|
|
}
|
|
|
|
type FuncInfo struct {
|
|
FuncName string
|
|
Args map[string]interface{}
|
|
Err error
|
|
}
|
|
|
|
func SetContextInfo(ctx context.Context, funcName string, err error, args ...interface{}) {
|
|
t := ctx.Value(TraceLogKey).(ApiInfo)
|
|
var funcInfo FuncInfo
|
|
funcInfo.Args = make(map[string]interface{})
|
|
argsHandle(args, funcInfo.Args)
|
|
funcInfo.FuncName = funcName
|
|
funcInfo.Err = err
|
|
t.Funcs = append(t.Funcs, funcInfo)
|
|
}
|
|
|
|
func argsHandle(args []interface{}, fields map[string]interface{}) {
|
|
for i := 0; i < len(args); i += 2 {
|
|
if i+1 < len(args) {
|
|
fields[fmt.Sprintf("%v", args[i])] = args[i+1]
|
|
} else {
|
|
fields[fmt.Sprintf("%v", args[i])] = ""
|
|
}
|
|
}
|
|
}
|