Use existence test + get w/expires storage.

Fixes #531
coleifer · Jun 18, 2020 · 335c5dc · 335c5dc
1 parent b9cb866
commit 335c5dc
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/huey/storage.py b/huey/storage.py
@@ -516,8 +516,11 @@ def put_data(self, key, value, is_result=False):
             self.conn.set(self.result_key(key), value)
 
     def peek_data(self, key):
-        res = self.conn.get(self.result_key(key))
-        return res if res is not None else EmptyData
+        pipe = self.conn.pipeline()
+        pipe.exists(self.result_key(key))
+        pipe.get(self.result_key(key))
+        exists, val = pipe.execute()
+        return EmptyData if not exists else val
 
     # Here we explicitly prevent result items from being removed by using the
     # same implementation for "pop" (get and delete) as we do for "peek"