This commit is contained in:
2024-12-02 20:10:32 +03:00
commit 6e8a1540b4
16 changed files with 392 additions and 0 deletions

0
app/routers/__init__.py Normal file
View File

23
app/routers/acquire.py Normal file
View File

@@ -0,0 +1,23 @@
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)

19
app/routers/release.py Normal file
View File

@@ -0,0 +1,19 @@
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/release', status_code=fastapi.status.HTTP_202_ACCEPTED)
async def execute(body: RequestBody):
await locks.release(locks.Lock(name=body.name, locked_until=time.now() + datetime.timedelta(seconds=body.ttl)))