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

Compare commits

base: self:875294bdd760d1e7308e396a65ca2ab3473b5fac
Branches Tags
self:master
self:dev
self:prod
..
compare: self:6bcd8375e8ca9c4134a8466d27fe9aabd794453d
Branches Tags
self:dev
self:prod
self:master

No commits in common. "875294bdd760d1e7308e396a65ca2ab3473b5fac" and "6bcd8375e8ca9c4134a8466d27fe9aabd794453d" have entirely different histories.
875294bdd7 ... 6bcd8375e8

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

2
Dockerfile
View File

@ -8,6 +8,4 @@ RUN pip install -r requirements.txt
COPY . . COPY . .
ENV PYTHONUNBUFFERED 1
ENTRYPOINT ["python", "main.py"] ENTRYPOINT ["python", "main.py"]

10
app/routers/take.py
View File

@ -1,3 +1,5 @@
import asyncio
import collections
import fastapi import fastapi
import pydantic import pydantic
import typing import typing
@ -7,6 +9,7 @@ from app.storage.mongo import tasks
DEFAULT_RETRY_AFTER = 0.2 DEFAULT_RETRY_AFTER = 0.2
locks = collections.defaultdict(asyncio.Lock)
router = fastapi.APIRouter() router = fastapi.APIRouter()
@ -20,13 +23,10 @@ class Response(pydantic.BaseModel):
task: Task|None task: Task|None
@router.get('/api/v1/take') @router.get('/api/v1/take', responses={500: {'detail': 'Internal server error'}})
async def execute(queue: typing.Annotated[str, fastapi.Header()]) -> Response: async def execute(queue: typing.Annotated[str, fastapi.Header()]) -> Response:
try: async with locks[queue]:
task = await tasks.take_task(queue) task = await tasks.take_task(queue)
except Exception as e:
print('GOT ERROR', e)
return Response(task=None)
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')