k-space/camera-motion-detect
10
0
Fork 0
Code Issues 2 Pull Requests Projects Activity

Reorder readiness check

Browse Source
This commit is contained in:
Lauri Võsandi 2022-08-24 23:02:25 +03:00
parent eebd2f3d96
commit e54ad1ddb7
1 changed files with 10 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
camdetect.py
View File

@ -484,6 +484,16 @@ async def stream_wrapper(request):
@app.route("/readyz") @app.route("/readyz")
async def ready_check(request): async def ready_check(request):
if app.ctx.mask is None:
return response.text("Not enough frames", status=503)
# Check if Mongo is accessible
try:
async for i in app.ctx.coll.find().limit(1):
break
except pymongo.errors.ServerSelectionTimeoutError:
return response.text("MongoDB server selection timeout", status=503)
# Check if S3 is accessible # Check if S3 is accessible
session = aioboto3.Session() session = aioboto3.Session()
async with session.resource("s3", async with session.resource("s3",
@ -492,15 +502,6 @@ async def ready_check(request):
endpoint_url=S3_ENDPOINT_URL) as s3: endpoint_url=S3_ENDPOINT_URL) as s3:
bucket = await s3.Bucket(S3_BUCKET_NAME) bucket = await s3.Bucket(S3_BUCKET_NAME)
await bucket.upload_fileobj(io.BytesIO(b"test"), "test") await bucket.upload_fileobj(io.BytesIO(b"test"), "test")
# Check if Mongo is accessible
try:
async for i in app.ctx.coll.find().limit(1):
break
except pymongo.errors.ServerSelectionTimeoutError:
return response.text("MongoDB server selection timeout", status=503)
if app.ctx.mask is None:
return response.text("Not enough frames", status=503)
return response.text("OK") return response.text("OK")