ensure context is cancelled to prevent goroutine leaks from grpc.newC…

…lientStream
moby · Jan 7, 2020 · 31d21fb · 31d21fb
1 parent f7cf482
commit 31d21fb
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/util/flightcontrol/flightcontrol.go b/util/flightcontrol/flightcontrol.go
@@ -116,7 +116,9 @@ func newCall(fn func(ctx context.Context) (interface{}, error)) *call {
 
 func (c *call) run() {
 	defer c.closeProgressWriter()
-	v, err := c.fn(c.ctx)
+	ctx, cancel := context.WithCancel(c.ctx)
+	defer cancel()
+	v, err := c.fn(ctx)
 	c.mu.Lock()
 	c.result = v
 	c.err = err