Error code standardization

pkg/common/trace_log/ctx.go

@@ -17,6 +17,13 @@ func NewCtx(c *gin.Context, api string) context.Context {
 	return context.WithValue(c, TraceLogKey, req)
 }
 
+func NewRpcCtx(c context.Context, rpc string, operationID string) context.Context {
+	req := &ApiInfo{ApiName: rpc, Funcs: &[]FuncInfo{}}
+	ctx := context.WithValue(c, TraceLogKey, req)
+	SetOperationID(ctx, operationID)
+	return ctx
+}
+
 func SetOperationID(ctx context.Context, operationID string) {
 	ctx.Value(TraceLogKey).(*ApiInfo).OperationID = operationID
 }
@@ -69,6 +76,28 @@ func SetContextInfo(ctx context.Context, funcName string, err error, args ...int
 	*t.Funcs = append(*t.Funcs, funcInfo)
 }
 
+func SetRpcReqInfo(ctx context.Context, funcName string, req string) {
+	t := ctx.Value(TraceLogKey).(*ApiInfo)
+	var funcInfo FuncInfo
+	funcInfo.Args = make(map[string]interface{})
+	var args []interface{}
+	args = append(args, " rpc req ", req)
+	argsHandle(args, funcInfo.Args)
+	funcInfo.FuncName = funcName
+	*t.Funcs = append(*t.Funcs, funcInfo)
+}
+
+func SetRpcRespInfo(ctx context.Context, funcName string, resp string) {
+	t := ctx.Value(TraceLogKey).(*ApiInfo)
+	var funcInfo FuncInfo
+	funcInfo.Args = make(map[string]interface{})
+	var args []interface{}
+	args = append(args, " rpc resp ", resp)
+	argsHandle(args, funcInfo.Args)
+	funcInfo.FuncName = funcName
+	*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})