From 5e9364eedcbbda5501c138338d1b38f4737c4073 Mon Sep 17 00:00:00 2001
From: David Perl <perl_d@x06da-cons-01.psi.ch>
Date: Tue, 25 Aug 2026 14:24:26 +0200
Subject: [PATCH] fix: add harder timeouts to uvicorn.run

---
 src/aare/daq/server.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/aare/daq/server.py b/src/aare/daq/server.py
index cb3b4ed2..94a6c9bc 100644
--- a/src/aare/daq/server.py
+++ b/src/aare/daq/server.py
@@ -3,6 +3,7 @@ import hmac
 import importlib
 import json
 import os
+import threading
 import time
 from collections.abc import AsyncGenerator
 from contextlib import asynccontextmanager
@@ -133,8 +134,8 @@ async def lifespan(application: FastAPI):
     yield  # ── application serves requests here ──
 
     # Shutdown: add cleanup here if needed
-    daq.shutdown()
     logger.info(f"Worker {os.getpid()} shutting down.")
+    daq.shutdown()
 
 
 app = FastAPI(lifespan=lifespan)
@@ -2531,6 +2532,8 @@ def main():
         proxy_headers=False,
         log_config=get_uvicorn_logging_config(),
         timeout_worker_healthcheck=30,
+        timeout_graceful_shutdown=1,
+        timeout_keep_alive=30,
     )