administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-05 10:45:21 +03:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Refactor put and get methods in LimitedQueue to handle queue size and blocking behavior more efficiently

Browse Source
This commit is contained in:
Sergey Krashevich 2023-07-02 03:12:13 +03:00
parent 4e2a529b0e
commit b682d835d0
No known key found for this signature in database
GPG Key ID: 625171324E7D3856
1 changed files with 14 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
frigate/util.py
View File

@ -1300,28 +1300,25 @@ class LimitedQueue(FFQueue):
self.size = multiprocessing.RawValue(
ctypes.c_int, 0
) # Add a counter for the number of items in the queue
self.lock = multiprocessing.Lock() # Add a lock
def put(self, x, block=True, timeout=DEFAULT_TIMEOUT):
with self.lock: # Acquire the lock
if self.maxsize > 0 and self.size.value >= self.maxsize:
if block:
start_time = time.time()
while self.size.value >= self.maxsize:
remaining = timeout - (time.time() - start_time)
if remaining <= 0.0:
raise Full
time.sleep(min(remaining, 0.1))
else:
raise Full
self.size.value += 1
if self.maxsize > 0 and self.size.value >= self.maxsize:
if block:
start_time = time.time()
while self.size.value >= self.maxsize:
remaining = timeout - (time.time() - start_time)
if remaining <= 0.0:
raise Full
time.sleep(min(remaining, 0.1))
else:
raise Full
self.size.value += 1
return super().put(x, block=block, timeout=timeout)
def get(self, block=True, timeout=DEFAULT_TIMEOUT):
with self.lock: # Acquire the lock
if self.size.value <= 0 and not block:
raise Empty
self.size.value -= 1
if self.size.value <= 0 and not block:
raise Empty
self.size.value -= 1
return super().get(block=block, timeout=timeout)
def qsize(self):