chore: ServerStreaming Benchmark stopped with IOException: The reques…

…t was aborted.
Cysharp · Nov 2, 2024 · 9007318 · 9007318
1 parent 0550e3a
commit 9007318
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/perf/BenchmarkApp/PerformanceTest.Client/ServerStreamingScenario.cs b/perf/BenchmarkApp/PerformanceTest.Client/ServerStreamingScenario.cs
@@ -24,9 +24,9 @@ public async ValueTask RunAsync(int connectionId, PerformanceTestRunningContext
         while (!cancellationToken.IsCancellationRequested)
         {
             ctx.Increment();
-            var begin = timeProvider.GetTimestamp();
             try
             {
+                var begin = timeProvider.GetTimestamp();
                 await stream.ResponseStream.MoveNext(cancellationToken);
                 ctx.Latency(connectionId, timeProvider.GetElapsedTime(begin));
             }
@@ -35,6 +35,11 @@ public async ValueTask RunAsync(int connectionId, PerformanceTestRunningContext
                 // canceling call is expected behavior.
                 break;
             }
+            catch (RpcException ex) when (ex.StatusCode == StatusCode.Unavailable && ex.InnerException is IOException)
+            {
+                // canceling call is expected behavior.
+                break;
+            }
         }
     }
 

diff --git a/perf/BenchmarkApp/PerformanceTest.Server/PerfTestService.cs b/perf/BenchmarkApp/PerformanceTest.Server/PerfTestService.cs
@@ -62,8 +62,8 @@ public async Task<ServerStreamingResult<SimpleResponse>> ServerStreamingAsync()
     {
         var response = SimpleResponse.Cached;
         var stream = GetServerStreamingContext<SimpleResponse>();
-
-        while(!stream.ServiceContext.CallContext.CancellationToken.IsCancellationRequested)
+        var ct = stream.ServiceContext.CallContext.CancellationToken;
+        while (!ct.IsCancellationRequested)
         {
             await stream.WriteAsync(response);
         }