blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-04-29 05:38:45 +08:00
Code Issues Packages Projects Releases Wiki Activity

Error code standardization

Browse Source
This commit is contained in:
skiffer-git 2023-01-03 14:08:29 +08:00
parent 20d56aefa8
commit bf747e538b
1 changed files with 28 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
pkg/common/trace_log/ctx.go
View File

@ -11,35 +11,37 @@ import (
) )
const TraceLogKey = "trace_log" const TraceLogKey = "trace_log"
const GinContextKey = "gin_context"
func NewCtx(c *gin.Context, api string) context.Context { func NewCtx(c *gin.Context, api string) context.Context {
req := ApiInfo{ApiName: api, GinCtx: c} req := &ApiInfo{ApiName: api, GinCtx: c, Funcs: &[]FuncInfo{}}
req.OperationID = c.PostForm("operationID") return context.WithValue(c, TraceLogKey, req)
return context.WithValue(c, GinContextKey, req) }
func SetOperationID(ctx context.Context, operationID string) {
ctx.Value(TraceLogKey).(*ApiInfo).OperationID = operationID
} }
func ShowLog(ctx context.Context) { func ShowLog(ctx context.Context) {
t := ctx.Value(TraceLogKey).(ApiInfo) t := ctx.Value(TraceLogKey).(*ApiInfo)
log.Info(t.OperationID, "api: ", t.ApiName) log.Info(t.OperationID, "funcs", len(*t.Funcs), "api:", t.ApiName)
for _, v := range t.Funcs { for _, v := range *t.Funcs {
if v.Err != nil { if v.Err != nil {
log.Error(v.FuncName, v.Err, v.Args) log.Error(t.OperationID, "func: ", v.FuncName, " args: ", v.Args, v.Err.Error())
} else { } else {
log.Info(v.FuncName, v.Args) log.Info(t.OperationID, "func: ", v.FuncName, " args: ", v.Args)
} }
} }
} }
func WriteErrorResponse(ctx context.Context, funcName string, err error) { func WriteErrorResponse(ctx context.Context, funcName string, err error, args ...interface{}) {
SetContextInfo(ctx, funcName, err) SetContextInfo(ctx, funcName, err, args)
e := new(constant.ErrInfo) e := new(constant.ErrInfo)
switch { switch {
case errors.As(err, &e): case errors.As(err, &e):
ctx.Value(GinContextKey).(ApiInfo).GinCtx.JSON(http.StatusOK, gin.H{"errCode": e.ErrCode, "errMsg": e.ErrMsg}) ctx.Value(TraceLogKey).(*ApiInfo).GinCtx.JSON(http.StatusOK, gin.H{"errCode": e.ErrCode, "errMsg": e.ErrMsg})
return return
default: default:
ctx.Value(GinContextKey).(ApiInfo).GinCtx.JSON(http.StatusOK, gin.H{"errCode": constant.ErrDefaultOther.ErrCode, "errMsg": constant.ErrDefaultOther.ErrMsg}) ctx.Value(TraceLogKey).(*ApiInfo).GinCtx.JSON(http.StatusOK, gin.H{"errCode": constant.ErrDefault.ErrCode, "errMsg": constant.ErrDefault.ErrMsg})
return return
} }
} }
@ -47,7 +49,7 @@ func WriteErrorResponse(ctx context.Context, funcName string, err error) {
type ApiInfo struct { type ApiInfo struct {
ApiName string ApiName string
OperationID string OperationID string
Funcs []FuncInfo Funcs *[]FuncInfo
GinCtx *gin.Context GinCtx *gin.Context
} }
@ -58,21 +60,30 @@ type FuncInfo struct {
} }
func SetContextInfo(ctx context.Context, funcName string, err error, args ...interface{}) { func SetContextInfo(ctx context.Context, funcName string, err error, args ...interface{}) {
t := ctx.Value(TraceLogKey).(ApiInfo) t := ctx.Value(TraceLogKey).(*ApiInfo)
var funcInfo FuncInfo var funcInfo FuncInfo
funcInfo.Args = make(map[string]interface{}) funcInfo.Args = make(map[string]interface{})
argsHandle(args, funcInfo.Args) argsHandle(args, funcInfo.Args)
funcInfo.FuncName = funcName funcInfo.FuncName = funcName
funcInfo.Err = err funcInfo.Err = err
t.Funcs = append(t.Funcs, funcInfo) *t.Funcs = append(*t.Funcs, funcInfo)
}
func SetSuccess(ctx context.Context, funcName string, data interface{}) {
SetContextInfo(ctx, funcName, nil, "data", data)
ctx.Value(TraceLogKey).(*ApiInfo).GinCtx.JSON(http.StatusOK, gin.H{"errCode": 0, "errMsg": "", "data": data})
} }
func argsHandle(args []interface{}, fields map[string]interface{}) { func argsHandle(args []interface{}, fields map[string]interface{}) {
for i := 0; i < len(args); i += 2 { for i := 0; i < len(args); i += 2 {
if i+1 < len(args) { if i+1 < len(args) {
fields[fmt.Sprintf("%v", args[i])] = args[i+1] fields[fmt.Sprintf("%v", args[i])] = fmt.Sprintf("%+v", args[i+1])
} else { } else {
fields[fmt.Sprintf("%v", args[i])] = "" fields[fmt.Sprintf("%v", args[i])] = ""
} }
} }
} }
func GetApiErr(errCode int32, errMsg string) api_struct.ErrInfo {
return api_struct.ErrInfo{ErrCode: errCode, ErrMsg: errMsg}
}