From b03154dce554fdaeae5742facce837e8dc0a03a2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Zolt=C3=A1n=20Papp?= <zoltan.pmail@gmail.com>
Date: Tue, 18 Nov 2025 17:25:46 +0100
Subject: [PATCH] Use dedicated ctx in stream

---
 shared/management/client/grpc.go | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/shared/management/client/grpc.go b/shared/management/client/grpc.go
index 3f2ea3441..be3d09ccf 100644
--- a/shared/management/client/grpc.go
+++ b/shared/management/client/grpc.go
@@ -164,6 +164,9 @@ func (c *GrpcClient) handleJobStream(
 	serverPubKey wgtypes.Key,
 	msgHandler func(msg *proto.JobRequest) *proto.JobResponse,
 ) error {
+	ctx, cancelStream := context.WithCancel(ctx)
+	defer cancelStream()
+
 	stream, err := c.realClient.Job(ctx)
 	if err != nil {
 		log.WithContext(ctx).Errorf("failed to open job stream: %v", err)