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

Compare commits

base: self:0e1dd5d44e4c7fbcf5a8a233da9eccc2cb32a997
Branches Tags
self:master
self:dev
self:prod
..
compare: self:1d92f9564953690ccd6db90c0e49d22dcbe7095e
Branches Tags
self:dev
self:prod
self:master

No commits in common. "0e1dd5d44e4c7fbcf5a8a233da9eccc2cb32a997" and "1d92f9564953690ccd6db90c0e49d22dcbe7095e" have entirely different histories.
0e1dd5d44e ... 1d92f95649

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

4
app/routers/take.py
View File

@ -1,5 +1,3 @@
import asyncio
import collections
import fastapi import fastapi
import pydantic import pydantic
import typing import typing
@ -7,7 +5,6 @@ import typing
from app.storage.mongo import tasks from app.storage.mongo import tasks
locks = collections.defaultdict(asyncio.Lock)
router = fastapi.APIRouter() router = fastapi.APIRouter()
@ -23,7 +20,6 @@ class Response(pydantic.BaseModel):
@router.get('/api/v1/take', responses={404: {'description': 'Not found'}}) @router.get('/api/v1/take', responses={404: {'description': 'Not found'}})
async def execute(queue: typing.Annotated[str, fastapi.Header()]) -> Response: async def execute(queue: typing.Annotated[str, fastapi.Header()]) -> Response:
async with locks[queue]:
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)