Factor out common recv code; throw QueueEmpty exception in get_nowait
This commit is contained in:
@@ -344,23 +344,22 @@ class GmiiSink(Reset):
|
||||
|
||||
self._init_reset(reset, reset_active_level)
|
||||
|
||||
async def recv(self, compact=True):
|
||||
frame = await self.queue.get()
|
||||
def _recv(self, frame, compact=True):
|
||||
if self.queue.empty():
|
||||
self.active_event.clear()
|
||||
self.queue_occupancy_bytes -= len(frame)
|
||||
self.queue_occupancy_frames -= 1
|
||||
if compact:
|
||||
frame.compact()
|
||||
return frame
|
||||
|
||||
async def recv(self, compact=True):
|
||||
frame = await self.queue.get()
|
||||
return self._recv(frame, compact)
|
||||
|
||||
def recv_nowait(self, compact=True):
|
||||
if not self.queue.empty():
|
||||
frame = self.queue.get_nowait()
|
||||
if self.queue.empty():
|
||||
self.active_event.clear()
|
||||
self.queue_occupancy_bytes -= len(frame)
|
||||
self.queue_occupancy_frames -= 1
|
||||
return frame
|
||||
return None
|
||||
frame = self.queue.get_nowait()
|
||||
return self._recv(frame, compact)
|
||||
|
||||
def count(self):
|
||||
return self.queue.qsize()
|
||||
|
||||
Reference in New Issue
Block a user