Skip to content

Commit bed83f2

Browse files
authored
More respect stop event even in _consume (#35)
1 parent 55b9a82 commit bed83f2

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

brq/consumer.py

+4
Original file line numberDiff line numberDiff line change
@@ -173,6 +173,8 @@ async def _set_retry_timeout(self):
173173
await self.redis.set(self.retry_cooldown_key, "1", ex=self.retry_cooldown_time)
174174

175175
async def _consume(self):
176+
if self._stop_event.is_set():
177+
return
176178
try:
177179
if not await self._is_retry_cooldown() and await self._acquire_retry_lock():
178180
await self._process_unacked_job()
@@ -181,6 +183,8 @@ async def _consume(self):
181183
finally:
182184
await self._release_retry_lock()
183185

186+
if self._stop_event.is_set():
187+
return
184188
if self.run_parallel:
185189
await self._pool_job_prallel()
186190
else:

0 commit comments

Comments
 (0)