feat: deadlock clear

This commit is contained in:
2024-10-22 10:02:26 +03:00
parent 52b3f6e6ee
commit 1806f5dec3
3 changed files with 20 additions and 2 deletions

View File

@@ -1,8 +1,11 @@
import asyncio
from typing import List, Union
from celery.signals import worker_ready
import background.update
from background import celery
from limiter import BatchLimiter
@celery.task(name='process_update')
@@ -27,3 +30,10 @@ def update_marketplace_products(marketplace_id: int, product_ids: list[int]):
def update_marketplaces(marketplace_ids: Union[List[int], None]):
loop = asyncio.get_event_loop()
return loop.run_until_complete(background.update.update_marketplaces(marketplace_ids))
@worker_ready.connect
def worker_is_ready(sender, **kwargs):
loop = asyncio.get_event_loop()
limiter = BatchLimiter()
loop.run_until_complete(limiter.clear_locks())