diff --git a/coordinator/src/handlers/data_layer.rs b/coordinator/src/handlers/data_layer.rs index 4b50f38b..8f4cf5e3 100644 --- a/coordinator/src/handlers/data_layer.rs +++ b/coordinator/src/handlers/data_layer.rs @@ -88,6 +88,7 @@ impl DataLayerHandlerImpl { pub fn connect(runner_url: &str) -> anyhow::Result { let channel = Channel::from_shared(runner_url.to_string()) .context("Runner URL is invalid")? + .rate_limit(1, std::time::Duration::from_secs(5)) .connect_lazy(); let client = DataLayerClient::new(channel); diff --git a/coordinator/src/lifecycle.rs b/coordinator/src/lifecycle.rs index 3d29967e..204c1189 100644 --- a/coordinator/src/lifecycle.rs +++ b/coordinator/src/lifecycle.rs @@ -699,6 +699,8 @@ mod tests { #[tokio::test] async fn restarts_unhealthy_stream() { + tokio::time::pause(); + let config = IndexerConfig::default(); let mut state = IndexerState { lifecycle_state: LifecycleState::Running, @@ -953,6 +955,8 @@ mod tests { #[tokio::test] async fn restarts_unhealthy_executor() { + tokio::time::pause(); + let config = IndexerConfig::default(); let mut state = IndexerState { lifecycle_state: LifecycleState::Running,