fix server subscriber

Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
2023-08-24 09:25:03 +03:00
parent fdffd31ed3
commit a34c33e25b
1 changed files with 1 additions and 0 deletions
--- a/requestid.go
+++ b/requestid.go
@@ -122,6 +122,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 {