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

Compare commits

base: self:8575e4695dab042ada8125b64b4faeaa512d5ecb
Branches Tags
self:master
self:dev
self:prod
..
compare: self:80c5d424fea5cc2808479b0d4b177811a9804158
Branches Tags
self:dev
self:prod
self:master

No commits in common. "8575e4695dab042ada8125b64b4faeaa512d5ecb" and "80c5d424fea5cc2808479b0d4b177811a9804158" have entirely different histories.
8575e4695d ... 80c5d424fe

2 changed files with 5 additions and 5 deletions
Show Stats Expand all files Collapse all files

8
app/routers/take.py
View File

@ -5,7 +5,7 @@ import pydantic
import typing
from app.storage.mongo import tasks
# from app.utils.configurator import configurator
from app.utils.configurator import configurator
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)
if not task:
return Response(task=None)
# 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
return Response(task=Task(id=str(task._id), attempt=task.attempts, payload=task.payload), retry_after=DEFAULT_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
return Response(task=Task(id=str(task._id), attempt=task.attempts, payload=task.payload), retry_after=retry_after)

2
app/utils/configurator.py
View File

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