[GH-29]Fix cpu bound when queue is empty (#30)

the event is not cleared when queue is empty which cause high cpu consumption
peter-wangxu · Oct 31, 2017 · 71ded7c · 71ded7c
1 parent d26561e
commit 71ded7c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/persistqueue/sqlqueue.py b/persistqueue/sqlqueue.py
@@ -66,6 +66,7 @@ def get(self, block=True, timeout=None):
             # block until a put event.
             pickled = self._pop()
             while not pickled:
+                self.put_event.clear()
                 self.put_event.wait(TICK_FOR_WAIT)
                 pickled = self._pop()
         elif timeout < 0:
@@ -75,6 +76,7 @@ def get(self, block=True, timeout=None):
             endtime = _time.time() + timeout
             pickled = self._pop()
             while not pickled:
+                self.put_event.clear()
                 remaining = endtime - _time.time()
                 if remaining <= 0.0:
                     raise Empty