24 lines
590 B
Python
24 lines
590 B
Python
import datetime
|
|
import fastapi
|
|
import pydantic
|
|
|
|
from app.storage.mongo import locks
|
|
from app.utils import time
|
|
|
|
|
|
class RequestBody(pydantic.BaseModel):
|
|
name: str
|
|
ttl: int
|
|
|
|
|
|
router = fastapi.APIRouter()
|
|
|
|
|
|
@router.post('/api/v1/acquire', status_code=fastapi.status.HTTP_202_ACCEPTED, responses={'409': {'description': 'Conflict'}})
|
|
async def execute(body: RequestBody):
|
|
try:
|
|
await locks.acquire(locks.Lock(name=body.name, locked_until=time.now() + datetime.timedelta(seconds=body.ttl)))
|
|
except Exception as e:
|
|
print(e)
|
|
raise fastapi.HTTPException(409)
|