Merge pull request 'fix server subscriber' (#102) from idfix into v3

Reviewed-on: #102
2023-08-24 09:25:28 +03:00 · 2023-08-24 09:25:28 +03:00 · c1fa0f991d
commit c1fa0f991d
parent f163a2de72 a90681e18d
1 changed files with 1 additions and 0 deletions
--- 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 {