apunkt/dograh
1
0
Fork 0
mirror of https://github.com/dograh-hq/dograh.git synced 2026-07-25 12:01:04 +02:00
Code Issues Projects Releases Packages Wiki Activity

chore: drain active calls before rolling updates

Browse source
This commit is contained in:
Abhishek Kumar 2026-06-27 17:53:31 +05:30
parent 557de72b9c
commit f53102dabf
5 changed files with 183 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

19
api/routes/main.py
View file

@ -125,3 +125,22 @@ async def health() -> HealthResponse:
STACK_PUBLISHABLE_CLIENT_KEY if is_stack else None
),
)
class ActiveCallsResponse(BaseModel):
active_calls: int
@router.get("/health/active-calls", response_model=ActiveCallsResponse)
async def active_calls() -> ActiveCallsResponse:
"""In-flight call count for THIS worker — the drain signal for deploys.
A deploy orchestrator polls this per worker and waits for zero before
sending SIGTERM, because uvicorn force-closes live call WebSockets (close
code 1012) on SIGTERM and would cut calls mid-conversation otherwise. The
count is per-process: one uvicorn per VM port (scripts/rolling_update.sh)
or per Kubernetes pod (preStop hook). See api/services/pipecat/active_calls.py.
"""
from api.services.pipecat.active_calls import active_call_count
return ActiveCallsResponse(active_calls=active_call_count())