Add request context to make sure client closed connection

ella-to · May 2, 2024 · f719514 · f719514
1 parent d9809f3
commit f719514
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/client/client.go b/client/client.go
@@ -162,7 +162,7 @@ func (c *Client) Get(ctx context.Context, consumerOpts ...bus.ConsumerOpt) iter.
 	consumer.QueueName = header.Get("Consumer-Queue")
 	isAutoAck := header.Get("Consumer-Ack-Strategy") == "auto"
 
-	ctx, cancel := context.WithCancel(ctx)
+	ctx, cancel := context.WithCancel(req.Context())
 
 	incomings := sse.Receive(ctx, resp.Body)