Use grpc's forceful stop instead of the graceful stop.

The graceful stop seems to deadlock when there are streaming RPCs.
google · Sep 11, 2020 · 4ddd08c · 4ddd08c
1 parent 0553055
commit 4ddd08c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gapis/server/grpc.go b/gapis/server/grpc.go
@@ -138,7 +138,7 @@ func (s *grpcServer) stopIfIdle(ctx context.Context, server *grpc.Server, idleTi
 	stoppedSignal, stopped := task.NewSignal()
 	defer func() {
 		stop()
-		server.GracefulStop()
+		server.Stop()
 		stopped(ctx)
 	}()
 
@@ -171,7 +171,7 @@ func (s *grpcServer) stopOnInterrupt(ctx context.Context, server *grpc.Server, s
 	stoppedSignal, stopped := task.NewSignal()
 	defer func() {
 		stop()
-		server.GracefulStop()
+		server.Stop()
 		stopped(ctx)
 	}()