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

1

Browse Source
This commit is contained in:
withchao 2023-01-12 10:07:08 +08:00
parent d142988390
commit 2841109f31
2 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
internal/api_to_rpc/api.go
View File

@ -11,13 +11,13 @@ import (
func ApiToRpc(c *gin.Context, apiReq, apiResp interface{}, rpcName string, rpcClientFunc interface{}, rpcFuncName string, tokenFunc func(token string, operationID string) (string, error)) {
operationID := c.GetHeader("operationID")
nCtx := trace_log.NewCtx(c, rpcFuncName)
nCtx := trace_log.NewCtx1(c, rpcFuncName, operationID)
//trace_log.SetOperationID(nCtx, operationID)
defer trace_log.ShowLog(nCtx)
if err := c.BindJSON(apiReq); err != nil {
trace_log.WriteErrorResponse(nCtx, "BindJSON", err)
return
}
trace_log.SetOperationID(nCtx, operationID)
trace_log.SetContextInfo(nCtx, "BindJSON", nil, "params", apiReq)
etcdConn, err := getcdv3.GetConn(c, rpcName)
if err != nil {

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

@ -20,6 +20,11 @@ func NewCtx(c *gin.Context, api string) context.Context {
return context.WithValue(c, TraceLogKey, req)
}
func NewCtx1(c *gin.Context, api, operationID string) context.Context {
req := &ApiInfo{ApiName: api, GinCtx: c, OperationID: operationID, Funcs: &[]FuncInfo{}}
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)