From 94d12d55c67b00995986ff576a19ea1eda260d1f Mon Sep 17 00:00:00 2001 From: Space-Banane Date: Wed, 20 May 2026 21:57:37 +0200 Subject: [PATCH] Add optional DB write probe to readiness endpoint --- backend/main.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/backend/main.py b/backend/main.py index 84a6624..db5608e 100644 --- a/backend/main.py +++ b/backend/main.py @@ -51,12 +51,16 @@ def healthz(): @app.get("/readyz") -def readyz(): +def readyz(write_test: bool = False): try: with db() as c: with c.cursor() as cur: cur.execute("select 1 as ok") cur.fetchone() + if write_test: + cur.execute("create temporary table if not exists readyz_probe(id int)") + cur.execute("insert into readyz_probe(id) values (1)") + cur.execute("truncate table readyz_probe") except Exception: raise HTTPException(503, "Database not ready") return {"ok": True}