initial
This commit is contained in:
0
app/routers/__init__.py
Normal file
0
app/routers/__init__.py
Normal file
23
app/routers/acquire.py
Normal file
23
app/routers/acquire.py
Normal 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
19
app/routers/release.py
Normal 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)))
|
||||
Reference in New Issue
Block a user