diff --git a/internal/rpc/auth/auth.go b/internal/rpc/auth/auth.go index ae31af027..33d302802 100644 --- a/internal/rpc/auth/auth.go +++ b/internal/rpc/auth/auth.go @@ -39,6 +39,10 @@ func Start(client discoveryregistry.SvcDiscoveryRegistry, server *grpc.Server) e func (s *authServer) UserToken(ctx context.Context, req *pbAuth.UserTokenReq) (*pbAuth.UserTokenResp, error) { log.Info("", "rpc come UserToken") + operationID, ok := ctx.Value(constant.OperationID).(string) + if !ok { + log.Error("2222", "ctx missing operationID", operationID) + } resp := pbAuth.UserTokenResp{} if _, err := s.userCheck.GetUserInfo(ctx, req.UserID); err != nil { log.Info("", "UserToken err:", err.Error()) diff --git a/internal/startrpc/start.go b/internal/startrpc/start.go index c1e442e4f..2e2f7d519 100644 --- a/internal/startrpc/start.go +++ b/internal/startrpc/start.go @@ -31,7 +31,7 @@ func Start(rpcPort int, rpcRegisterName string, prometheusPort int, rpcFn func(c return utils.Wrap1(err) } defer zkClient.Close() - zkClient.AddOption(grpc.WithTransportCredentials(insecure.NewCredentials())) + zkClient.AddOption(mw.GrpcClient(), grpc.WithTransportCredentials(insecure.NewCredentials())) registerIP, err := network.GetRpcRegisterIP(config.Config.RpcRegisterIP) if err != nil { return err diff --git a/pkg/common/db/relation/user_model.go b/pkg/common/db/relation/user_model.go index 697024e39..026417bd2 100644 --- a/pkg/common/db/relation/user_model.go +++ b/pkg/common/db/relation/user_model.go @@ -45,7 +45,7 @@ func (u *UserGorm) Find(ctx context.Context, userIDs []string) (users []*relatio defer func() { tracelog.SetCtxDebug(ctx, utils.GetFuncName(1), err, "userIDs", userIDs, "users", users) }() - err = utils.Wrap(u.DB.Where("user_id in (?)", userIDs).Find(&users).Error, "") + err = utils.Wrap(u.DB.Debug().Where("user_id in ?", userIDs).Find(&users).Error, "") return users, err } diff --git a/pkg/common/mw/rpc.go b/pkg/common/mw/rpc.go index 1ee9c8aa3..7e5c8e647 100644 --- 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 { @@ -58,6 +60,7 @@ func rpcClientInterceptor(ctx context.Context, method string, req, reply interfa } operationID, ok := ctx.Value(constant.OperationID).(string) if !ok { + log.Error("1111", "ctx missing operationID") return errs.ErrArgs.Wrap("ctx missing operationID") } md := metadata.Pairs(constant.OperationID, operationID)