diff options
Diffstat (limited to 'weed/pb/grpc_client_server.go')
| -rw-r--r-- | weed/pb/grpc_client_server.go | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/weed/pb/grpc_client_server.go b/weed/pb/grpc_client_server.go index ba0d7d0cc..f179cc3a7 100644 --- a/weed/pb/grpc_client_server.go +++ b/weed/pb/grpc_client_server.go @@ -127,8 +127,8 @@ func requestIDUnaryInterceptor() grpc.UnaryServerInterceptor { info *grpc.UnaryServerInfo, handler grpc.UnaryHandler, ) (interface{}, error) { - md, _ := metadata.FromIncomingContext(ctx) - idList := md.Get(util.RequestIDKey) + incomingMd, _ := metadata.FromIncomingContext(ctx) + idList := incomingMd.Get(util.RequestIDKey) var reqID string if len(idList) > 0 { reqID = idList[0] @@ -137,6 +137,11 @@ func requestIDUnaryInterceptor() grpc.UnaryServerInterceptor { reqID = uuid.New().String() } + ctx = metadata.NewOutgoingContext(ctx, + metadata.New(map[string]string{ + util.RequestIDKey: reqID, + })) + ctx = util.WithRequestID(ctx, reqID) grpc.SetTrailer(ctx, metadata.Pairs(util.RequestIDKey, reqID)) |