self/queues
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix #23

Merged
emmatveev merged 1 commits from master into dev 2024-12-28 13:37:35 +03:00
Conversation 0 Commits 1 Files Changed 2 +5 -5
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
app/routers/take.py
View File

@ -5,7 +5,7 @@ import pydantic
import typing import typing
from app.storage.mongo import tasks from app.storage.mongo import tasks
from app.utils.configurator import configurator # from app.utils.configurator import configurator
DEFAULT_RETRY_AFTER = 0.2 DEFAULT_RETRY_AFTER = 0.2
@ -30,6 +30,6 @@ async def execute(queue: typing.Annotated[str, fastapi.Header()]) -> Response:
task = await tasks.take_task(queue) task = await tasks.take_task(queue)
if not task: if not task:
return Response(task=None) return Response(task=None)
retry_after_config = configurator.get_config('retry_after') # retry_after_config = configurator.get_config('retry_after')
retry_after = retry_after_config.get(queue) or retry_after_config.get('default') or DEFAULT_RETRY_AFTER # retry_after = retry_after_config.get(queue) or retry_after_config.get('default') or DEFAULT_RETRY_AFTER
return Response(task=Task(id=str(task._id), attempt=task.attempts, payload=task.payload), retry_after=retry_after) return Response(task=Task(id=str(task._id), attempt=task.attempts, payload=task.payload), retry_after=DEFAULT_RETRY_AFTER)

2
app/utils/configurator.py
View File

@ -85,4 +85,4 @@ class ConfiguratorClient:
return self.experiment_storage[name] return self.experiment_storage[name]
configurator = ConfiguratorClient('queues', os.getenv('STAGE', 'local')) # configurator = ConfiguratorClient('queues', os.getenv('STAGE', 'local'))