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: