Conditionally notify TTL background job (#44)

* Conditionally notify ttl job * Update src/store.rs Co-authored-by: Christian Gill <[email protected]> --------- Co-authored-by: Christian Gill <[email protected]>
ndelvalle · Jun 12, 2024 · 82918f3 · 82918f3
1 parent 10c907e
commit 82918f3
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/src/store.rs b/src/store.rs
@@ -59,9 +59,18 @@ impl InnerStore {
     }
 
     pub fn set2(&self, key: Key, value: NewValue) {
+        let has_ttl = value.ttl.is_some();
         let mut state = self.lock();
-        state.set2(key, value);
-        self.waker.notify_one();
+
+        state.set2(key.clone(), value);
+
+        if has_ttl {
+            let next_to_expire = state.ttls.iter().next().map(|(_, key)| key);
+            let expires_next = next_to_expire == Some(&key);
+            if expires_next {
+                self.waker.notify_one();
+            }
+        }
     }
 
     pub fn incr_by<T>(&self, key: &str, increment: T) -> Result<T, String>