#736 use shared.Context insead of creating a new context

smallnest · Jan 31, 2023 · 09f6359 · 09f6359
1 parent 28b118c
commit 09f6359
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/client/client.go b/client/client.go
@@ -343,7 +343,11 @@ func (client *Client) call(ctx context.Context, servicePath, serviceMethod strin
 
 // SendRaw sends raw messages. You don't care args and replies.
 func (client *Client) SendRaw(ctx context.Context, r *protocol.Message) (map[string]string, []byte, error) {
-	ctx = context.WithValue(ctx, seqKey{}, r.Seq())
+	if sharedCtx, ok := ctx.(*share.Context); ok {
+		sharedCtx.SetValue(seqKey{}, r.Seq())
+	} else {
+		ctx = context.WithValue(ctx, seqKey{}, r.Seq())
+	}
 
 	call := new(Call)
 	call.Raw = true