opt: reduce downloader writes

Scighost · Aug 8, 2024 · 05208d3 · 05208d3
1 parent 86d2fc2
commit 05208d3
Showing 1 changed file with 6 additions and 13 deletions.
diff --git a/src/Starward/Services/Download/InstallGameService.cs b/src/Starward/Services/Download/InstallGameService.cs
@@ -1211,25 +1211,18 @@ protected async Task DownloadItemAsync(InstallGameItem item, CancellationToken c
                 {
                     if (InstallGameManager.IsEnableSpeedLimit)
                     {
-                        int totalWritten = 0;
-                        while (totalWritten < length)
+                        int totalTokens = 0;
+                        while (totalTokens < length)
                         {
-                            int remaining = length - totalWritten;
+                            int remaining = length - totalTokens;
                             if (TokenBucketRateLimiterExtension.TryAcquire(InstallGameManager.RateLimiter, remaining, out int tokensAcquired, out _))
                                 await Task.Delay(1, cancellationToken).ConfigureAwait(false);
                             else
-                            {
-                                await fs.WriteAsync(buffer.AsMemory(totalWritten, tokensAcquired), cancellationToken).ConfigureAwait(false);
-                                totalWritten += tokensAcquired;
-                                Interlocked.Add(ref _finishBytes, tokensAcquired);
-                            }
+                                totalTokens += tokensAcquired;
                         }
                     }
-                    else
-                    {
-                        await fs.WriteAsync(buffer.AsMemory(0, length), cancellationToken).ConfigureAwait(false);
-                        Interlocked.Add(ref _finishBytes, length);
-                    }
+                    await fs.WriteAsync(buffer.AsMemory(0, length), cancellationToken).ConfigureAwait(false);
+                    Interlocked.Add(ref _finishBytes, length);
                 }
             }
         }