Merge pull request 'fix server subscriber' (#103) from idfix into master

Reviewed-on: #103
2023-08-24 09:32:34 +03:00
parent fdffd31ed3 a34c33e25b
commit c069636bf3
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 {