added Queue subclass where put_nowait honors mutex lock, and that can be flushed

This commit is contained in:
2023-06-01 13:59:58 +02:00
parent bab3232c82
commit cc4fc3268a
2 changed files with 27 additions and 3 deletions

24
bstrd/prodqueue.py Normal file
View File

@ -0,0 +1,24 @@
from queue import Queue, Full
class ProdQueue(Queue):
def put_nowait(self, *args, **kwargs):
if self.mutex.locked():
raise Locked
super().put_nowait(*args, **kwargs)
def clear(self):
with self.mutex:
self.queue.clear()
class Locked(Exception):
"Exception raised by ProdQueue.put_nowait() if mutex is locked."
pass