16 lines
351 B
Python
16 lines
351 B
Python
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()
|