优化cancel

nilaoda · Oct 29, 2022 · a13d3c1 · a13d3c1
1 parent 80f6aa7
commit a13d3c1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/N_m3u8DL-RE/Util/DownloadUtil.cs b/src/N_m3u8DL-RE/Util/DownloadUtil.cs
@@ -64,7 +64,6 @@ public static async Task<DownloadResult> DownloadToFileAsync(string url, string
                     if (speedContainer.ShouldStop)
                     {
                         cancellationTokenSource.Cancel();
-                        speedContainer.ResetLowSpeedCount();
                         Logger.DebugMarkUp("Cancel...");
                         break;
                     }
@@ -105,7 +104,7 @@ public static async Task<DownloadResult> DownloadToFileAsync(string url, string
                 while ((size = await responseStream.ReadAsync(buffer, cancellationTokenSource.Token)) > 0)
                 {
                     speedContainer.Add(size);
-                    await stream.WriteAsync(buffer, 0, size);
+                    await stream.WriteAsync(buffer, 0, size, cancellationTokenSource.Token);
                 }
 
                 return new DownloadResult()
@@ -117,6 +116,7 @@ public static async Task<DownloadResult> DownloadToFileAsync(string url, string
             }
             catch (OperationCanceledException oce) when (oce.CancellationToken == cancellationTokenSource.Token)
             {
+                speedContainer.ResetLowSpeedCount();
                 throw new Exception("Download speed too slow!");
             }
         }