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

fix
All checks were successful
Deploy Dev / Build (pull_request) Successful in 18s
Details
Deploy Dev / Push (pull_request) Successful in 8s
Details
Deploy Dev / Deploy dev (pull_request) Successful in 20s
Details

Browse Source
This commit is contained in:
Egor Matveev 2024-12-28 13:37:07 +03:00
parent 9dfc115224
commit 8ee3573d9a
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'))