From 39e288494e24bc456694e5e56c6f6eca13464278 Mon Sep 17 00:00:00 2001 From: Daniel Kim Date: Wed, 30 Sep 2026 13:31:41 -0700 Subject: [PATCH] Track remote gateway work before auto-restart --- src/commands/up.rs | 2 +- src/commands/up_remote.rs | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/src/commands/up.rs b/src/commands/up.rs index f6739744..78160d39 100644 --- a/src/commands/up.rs +++ b/src/commands/up.rs @@ -4318,7 +4318,7 @@ impl Drop for Active { } } -async fn track_active(request: axum::extract::Request, next: Next) -> Response { +pub(crate) async fn track_active(request: axum::extract::Request, next: Next) -> Response { // The dashboard holds its event stream open for as long as it is open. if request.uri().path() == "/api/events" { return next.run(request).await; diff --git a/src/commands/up_remote.rs b/src/commands/up_remote.rs index e65fc62d..01bf3490 100644 --- a/src/commands/up_remote.rs +++ b/src/commands/up_remote.rs @@ -1124,6 +1124,7 @@ fn gateway_router(session: Arc) -> Router { ) .fallback(gateway_fallback) .layer(DefaultBodyLimit::max(64 * 1024 * 1024)) + .layer(middleware::from_fn(super::up::track_active)) .layer(middleware::from_fn(gateway_loopback_guard)) .with_state(session) }