Add timeout for waiting cache download

james58899 · Dec 18, 2023 · 8ae1942 · 8ae1942
1 parent 6a6882c
commit 8ae1942
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/route/cache.rs b/src/route/cache.rs
@@ -20,6 +20,7 @@ use tokio::{
     fs::{File, OpenOptions},
     io::{AsyncReadExt, AsyncSeekExt, AsyncWriteExt},
     sync::watch,
+    time::timeout,
 };
 
 use crate::{
@@ -215,7 +216,8 @@ async fn hath(
             let mut read_off = 0;
             let mut write_off = *rx.borrow();
 
-            'watch: while rx.changed().await.is_ok() || write_off != read_off {
+            let wait_time = Duration::from_secs(30);
+            'watch: while timeout(wait_time, rx.changed()).await.is_ok_and(|r| r.is_ok()) || write_off != read_off {
                 write_off = *rx.borrow();
 
                 while write_off > read_off {