rpc add data

2 years ago · 8867c5e14a
parent cf29bd6227
commit 8867c5e14a
1 changed files with 2 additions and 0 deletions
--- a/pkg/common/mw/rpc.go
+++ b/pkg/common/mw/rpc.go
@ -41,6 +41,8 @@ func rpcServerInterceptor(ctx context.Context, req interface{}, info *grpc.Unary
 		opUserID = opts[0]
 	}
 	ctx = tracelog.SetFuncInfos(ctx, funcName, operationID)
+	ctx = context.WithValue(ctx, OperationID, operationID)
+	ctx = context.WithValue(ctx, OpUserID, opUserID)
 	tracelog.SetCtxInfo(ctx, funcName, err, "opUserID", opUserID, "rpcReq", rpcString(req))
 	resp, err = handler(ctx, req)
 	if err != nil {