Fix goroutine leak bug in DoDownloadTimeout

Signed-off-by: Ting Yuan <[email protected]>
dragonflyoss · Nov 1, 2021 · aa4e2c3 · aa4e2c3
1 parent 1bb1d3e
commit aa4e2c3
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/dfget/core/downloader/downloader.go b/dfget/core/downloader/downloader.go
@@ -53,7 +53,12 @@ func DoDownloadTimeout(downloader Downloader, timeout time.Duration) error {
 
 	var ch = make(chan error)
 	go func() {
-		ch <- downloader.Run(ctx)
+		select {
+		case <- ctx.Done():
+			return
+		case ch <- downloader.Run(ctx):
+			return
+		}
 	}()
 	defer cancel()