Fix

src/bots/approved_bot/modules/download.rs

@@ -11,7 +11,7 @@ use teloxide::{
     prelude::*,
     types::*,
 };
-use tokio::time::sleep;
+use tokio::time::{self};
 use tokio_util::compat::FuturesAsyncReadCompatExt;
 use tracing::log;
 
@@ -507,7 +507,11 @@ async fn wait_archive(
     task_id: String,
     message: Message,
 ) -> BotHandlerInternal {
+    let mut interval = time::interval(Duration::from_secs(5));
+
     let task = loop {
+        interval.tick().await;
+
         let task = match get_task(task_id.clone()).await {
             Ok(v) => v,
             Err(err) => {
@@ -535,8 +539,6 @@ async fn wait_archive(
             .reply_markup(get_check_keyboard(task.id))
             .send()
             .await?;
-
-        sleep(Duration::from_secs(5)).await;
     };
 
     if task.status != TaskStatus::Complete {

src/bots_manager/mod.rs

@@ -24,7 +24,7 @@ use tokio::sync::RwLock;
 use smallvec::SmallVec;
 use teloxide::adaptors::throttle::Limits;
 use teloxide::types::{BotCommand, UpdateKind};
-use tokio::time::{sleep, Duration};
+use tokio::time::{sleep, Duration, self};
 use tower_http::trace::TraceLayer;
 
 use teloxide::prelude::*;
@@ -300,21 +300,19 @@ impl BotsManager {
 
         manager.start_axum_server().await;
 
-        let mut i = 0;
+        let mut interval = time::interval(Duration::from_secs(5));
 
         loop {
-            if !running.load(Ordering::SeqCst) {
+            for _i in 0..30 {
-                manager.stop_all().await;
+                interval.tick().await;
-                return;
-            };
 
-            if i == 0 {
+                if !running.load(Ordering::SeqCst) {
-                manager.check().await;
+                    manager.stop_all().await;
+                    return;
+                };
             }
 
-            sleep(Duration::from_secs(1)).await;
+            manager.check().await;
-
-            i = (i + 1) % 30;
         }
     }
 }