More respect stop event even in _consume (#35)

b-scheduler · Sep 3, 2024 · bed83f2 · bed83f2
1 parent 55b9a82
commit bed83f2
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/brq/consumer.py b/brq/consumer.py
@@ -173,6 +173,8 @@ async def _set_retry_timeout(self):
  await self.redis.set(self.retry_cooldown_key, "1", ex=self.retry_cooldown_time)
 
  async def _consume(self):
+ if self._stop_event.is_set():
+ return
  try:
  if not await self._is_retry_cooldown() and await self._acquire_retry_lock():
  await self._process_unacked_job()
@@ -181,6 +183,8 @@ async def _consume(self):
  finally:
  await self._release_retry_lock()
 
+ if self._stop_event.is_set():
+ return
  if self.run_parallel:
  await self._pool_job_prallel()
  else: