diff --git a/requestid.go b/requestid.go index 72f2593..e45fd44 100644 --- a/requestid.go +++ b/requestid.go @@ -123,6 +123,7 @@ func NewServerSubscriberWrapper() server.SubscriberWrapper { if id, ok := msg.Header()[DefaultMetadataKey]; ok { imd.Set(DefaultMetadataKey, id) omd.Set(DefaultMetadataKey, id) + ctx = context.WithValue(ctx, XRequestIDKey, id) ctx = metadata.NewIncomingContext(ctx, imd) ctx = metadata.NewOutgoingContext(ctx, omd) } else if ctx, err = DefaultMetadataFunc(ctx); err != nil {