cmd/torrent download: Handle interrupt when linear discarding

anacrolix · Feb 28, 2024 · edf59f0 · edf59f0
1 parent b8e03eb
commit edf59f0
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/cmd/torrent/download.go b/cmd/torrent/download.go
@@ -171,10 +171,18 @@ func addTorrents(
 					defer wg.Done()
 					waitForPieces(ctx, t, 0, t.NumPieces())
 				}()
-				if flags.LinearDiscard {
-					r := t.NewReader()
-					io.Copy(io.Discard, r)
-					r.Close()
+				done := make(chan struct{})
+				go func() {
+					defer close(done)
+					if flags.LinearDiscard {
+						r := t.NewReader()
+						io.Copy(io.Discard, r)
+						r.Close()
+					}
+				}()
+				select {
+				case <-done:
+				case <-ctx.Done():
 				}
 			} else {
 				for _, f := range t.Files() {