From dfb91b3fc0f95939144e2f74c82cdd9478768925 Mon Sep 17 00:00:00 2001 From: withchao <993506633@qq.com> Date: Thu, 12 Jan 2023 10:50:35 +0800 Subject: [PATCH 1/2] 1 --- pkg/common/middleware/rpc.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pkg/common/middleware/rpc.go b/pkg/common/middleware/rpc.go index 2311cec2d..4c6c28a6d 100644 --- a/pkg/common/middleware/rpc.go +++ b/pkg/common/middleware/rpc.go @@ -12,7 +12,6 @@ import ( "google.golang.org/grpc/metadata" "google.golang.org/grpc/status" "google.golang.org/protobuf/types/known/wrapperspb" - "path" "runtime/debug" ) @@ -23,7 +22,8 @@ func RpcServerInterceptor(ctx context.Context, req interface{}, info *grpc.Unary log.NewError(operationID, info.FullMethod, "type:", fmt.Sprintf("%T", r), "panic:", r, "stack:", string(debug.Stack())) } }() - funcName := path.Base(info.FullMethod) + //funcName := path.Base(info.FullMethod) + funcName := info.FullMethod md, ok := metadata.FromIncomingContext(ctx) if !ok { return nil, status.New(codes.InvalidArgument, "missing metadata").Err() From 5e020bb514af757f3c01c5bb3d565bea62ac89fb Mon Sep 17 00:00:00 2001 From: withchao <993506633@qq.com> Date: Thu, 12 Jan 2023 10:52:41 +0800 Subject: [PATCH 2/2] 1 --- pkg/common/middleware/rpc.go | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/pkg/common/middleware/rpc.go b/pkg/common/middleware/rpc.go index 4c6c28a6d..de4f9a46d 100644 --- a/pkg/common/middleware/rpc.go +++ b/pkg/common/middleware/rpc.go @@ -41,7 +41,7 @@ func RpcServerInterceptor(ctx context.Context, req interface{}, info *grpc.Unary } ctx = trace_log.NewRpcCtx(ctx, funcName, operationID) defer trace_log.ShowLog(ctx) - trace_log.SetContextInfo(ctx, funcName, err, "opUserID", opUserID, "rpcReq", req.(interface{ String() string }).String()) + trace_log.SetContextInfo(ctx, funcName, err, "opUserID", opUserID, "rpcReq", rpcString(req)) resp, err = handler(ctx, req) if err != nil { trace_log.SetContextInfo(ctx, funcName, err) @@ -58,7 +58,7 @@ func RpcServerInterceptor(ctx context.Context, req interface{}, info *grpc.Unary } return nil, sta.Err() } - trace_log.SetContextInfo(ctx, funcName, nil, "rpcResp", resp.(interface{ String() string }).String()) + trace_log.SetContextInfo(ctx, funcName, nil, "rpcResp", rpcString(resp)) return } @@ -77,3 +77,10 @@ func RpcClientInterceptor(ctx context.Context, method string, req, reply interfa md := metadata.Pairs("operationID", operationID, "opUserID", opUserID) return invoker(metadata.NewOutgoingContext(ctx, md), method, req, reply, cc, opts...) } + +func rpcString(v interface{}) string { + if s, ok := v.(interface{ String() string }); ok { + return s.String() + } + return fmt.Sprintf("%+v", v) +}