firehose: Add timeout to stream_blocks request (graphprotocol#3877)

kaiwetlesen · Sep 7, 2022 · 49b06f7 · 49b06f7
1 parent e679c5e
commit 49b06f7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/graph/src/blockchain/firehose_block_stream.rs b/graph/src/blockchain/firehose_block_stream.rs
@@ -241,7 +241,8 @@ fn stream_blocks<C: Blockchain, F: FirehoseMapper<C>>(
             }
 
             let mut connect_start = Instant::now();
-            let result = endpoint.clone().stream_blocks(request).await;
+            let req = endpoint.clone().stream_blocks(request);
+            let result = tokio::time::timeout(Duration::from_secs(30), req).await.map_err(|x| x.into()).and_then(|x| x);
 
             match result {
                 Ok(stream) => {