From 21b9b288f6e83360df6f5c916c52ae215243a29c Mon Sep 17 00:00:00 2001
From: Astro <astro@spaceboyz.net>
Date: Sun, 22 Jan 2023 23:27:27 +0100
Subject: [PATCH] butcher/trend_setter: fill from languages if queue is empty

---
 butcher/src/trend_setter.rs | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/butcher/src/trend_setter.rs b/butcher/src/trend_setter.rs
index 5636aee..0bfced1 100644
--- a/butcher/src/trend_setter.rs
+++ b/butcher/src/trend_setter.rs
@@ -71,7 +71,6 @@ pub fn start(mut store: Store) -> Tx {
 
         loop {
             let mut next_run = queue.keys().cloned().next();
-            // tracing::trace!("next_run in {:?}", next_run - Instant::now());
             if let Some(next_run_) = next_run {
                 let now = Instant::now();
                 if next_run_ <= now {
@@ -82,6 +81,14 @@ pub fn start(mut store: Store) -> Tx {
 
                     // update with next in queue
                     next_run = queue.keys().cloned().next();
+                } else {
+                    tracing::trace!("next_run in {:?}", next_run_ - now);
+                }
+            } else {
+                let languages = store.get_languages().await.unwrap();
+                tracing::info!("queue empty, filling from {} languages", languages.len());
+                for language in languages.into_iter() {
+                    enqueue(Some(language.clone()), &mut queue, &mut buffer, HashSet::new());
                 }
             }