benchmark: exit from UnconstrainedStreamingCall() when stream is canc…

…eled (#3623)
grpc · May 18, 2020 · f62d68d · f62d68d
1 parent 36493bc
commit f62d68d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/benchmark/benchmark.go b/benchmark/benchmark.go
@@ -113,6 +113,7 @@ func (s *testServer) UnconstrainedStreamingCall(stream testpb.BenchmarkService_U
 			err := stream.RecvMsg(in)
 			switch status.Code(err) {
 			case codes.Canceled:
+				return
 			case codes.OK:
 			default:
 				log.Fatalf("server recv error: %v", err)
@@ -125,6 +126,7 @@ func (s *testServer) UnconstrainedStreamingCall(stream testpb.BenchmarkService_U
 			err := stream.Send(response)
 			switch status.Code(err) {
 			case codes.Unavailable:
+				return
 			case codes.OK:
 			default:
 				log.Fatalf("server send error: %v", err)