Use interval instead of manual duration calculation

Kuadrant · Mar 25, 2024 · eab1166 · eab1166
1 parent 57d8f81
commit eab1166
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/limitador/src/storage/redis/redis_cached.rs b/limitador/src/storage/redis/redis_cached.rs
@@ -225,20 +225,17 @@ impl CachedRedisStorage {
         let batcher = Arc::new(Mutex::new(Default::default()));
         if let Some(flushing_period) = flushing_period {
             let batcher_flusher = batcher.clone();
+            let mut interval = tokio::time::interval(flushing_period);
             tokio::spawn(async move {
                 loop {
-                    let time_start = Instant::now();
                     let counters = {
                         let mut batch = batcher_flusher.lock().unwrap();
                         std::mem::take(&mut *batch)
                     };
                     for (counter, delta) in counters {
                         storage.update_counter(&counter, delta).await.unwrap();
                     }
-                    let sleep_time = flushing_period
-                        .checked_sub(time_start.elapsed())
-                        .unwrap_or_else(|| Duration::from_secs(0));
-                    tokio::time::sleep(sleep_time).await;
+                    interval.tick().await;
                 }
             });
         }