feat: add async job queue
This commit is contained in:
15
app/queue.py
Normal file
15
app/queue.py
Normal file
@@ -0,0 +1,15 @@
|
||||
import asyncio
|
||||
|
||||
|
||||
class JobQueue:
|
||||
def __init__(self) -> None:
|
||||
self._queue: asyncio.Queue[str] = asyncio.Queue()
|
||||
|
||||
async def enqueue(self, job_id: str) -> None:
|
||||
await self._queue.put(job_id)
|
||||
|
||||
async def dequeue(self) -> str:
|
||||
return await self._queue.get()
|
||||
|
||||
def size(self) -> int:
|
||||
return self._queue.qsize()
|
||||
Reference in New Issue
Block a user