Merge pull request #56 from geofmureithi/milestone/v0.4

Fix runtime error
geofmureithi · Mar 28, 2023 · 56d5ece · 56d5ece
2 parents a42a4c7 + f1f9907
commit 56d5ece
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/packages/apalis-core/Cargo.toml b/packages/apalis-core/Cargo.toml
@@ -27,7 +27,7 @@ sentry-core =  { version= "0.25.0", optional = true, default-features = false }
 metrics = { version = "0.20", optional = true, default-features = false }
 metrics-exporter-prometheus = { version = "0.11", optional = true, default-features = false }
 async-trait = { version = "0.1" }
-graceful-shutdown = { version = "0.2", features = ["stream", "async-io-timeout" ] }
+graceful-shutdown = { version = "0.2", features = ["stream", "tokio-timeout" ] }
 uuid = { version = "0.8", features = ["serde", "v4"] }
 async-stream = "0.3"
 

diff --git a/packages/apalis-core/src/monitor.rs b/packages/apalis-core/src/monitor.rs
@@ -167,7 +167,7 @@ impl<E: Executor + Send + 'static> Monitor<E> {
         //     .map(|h| (h.0, h.1.is_finished()))
         //     .collect();
         if let Some(timeout) = self.timeout {
-            if self.shutdown.with_timer(timeout).await {
+            if self.shutdown.with_timeout(timeout).await {
                 warn!("Shutdown timeout reached. Exiting forcefully");
                 return Err(std::io::Error::new(
                     std::io::ErrorKind::TimedOut,