close the stream when we get the done signal

astriaorg · Nov 27, 2024 · 3dd1c14 · 3dd1c14
1 parent 57dfe08
commit 3dd1c14
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/grpc/optimistic/server.go b/grpc/optimistic/server.go
@@ -156,6 +156,8 @@ func (o *OptimisticServiceV1Alpha1) ExecuteOptimisticBlockStream(stream optimist
 			return status.Error(codes.DeadlineExceeded, "timed out waiting for mempool to clear after optimistic block execution")
 		case err := <-mempoolClearingEvent.Err():
 			return status.Errorf(codes.Internal, "error waiting for mempool clearing event: %v", err)
+		case err := <-stream.Context().Done():
+			return status.Errorf(codes.Internal, "stream closed with error: %v", err)
 		}
 	}
 }