feat: add timeout to reqwest for retry

AOSC-Dev · Apr 6, 2024 · 2a54ea0 · 2a54ea0
1 parent 102688e
commit 2a54ea0
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/worker/src/build.rs b/worker/src/build.rs
@@ -293,7 +293,10 @@ async fn build_worker_inner(args: &Args) -> anyhow::Result<()> {
 
     info!("Receiving new messages");
 
-    let client = reqwest::Client::new();
+    let client = reqwest::Client::builder()
+        .timeout(Duration::from_secs(30))
+        .build()
+        .unwrap();
     let req = WorkerPollRequest {
         hostname: gethostname::gethostname().to_string_lossy().to_string(),
         arch: args.arch.clone(),

diff --git a/worker/src/heartbeat.rs b/worker/src/heartbeat.rs
@@ -4,7 +4,10 @@ use log::{info, warn};
 use std::time::Duration;
 
 pub async fn heartbeat_worker_inner(args: &Args) -> anyhow::Result<()> {
-    let client = reqwest::Client::new();
+    let client = reqwest::Client::builder()
+        .timeout(Duration::from_secs(30))
+        .build()
+        .unwrap();
     loop {
         // info!("Sending heartbeat");
         client