From 0993f6bbf52f85f137afe17d1c05aa3fe7bd72cd Mon Sep 17 00:00:00 2001 From: Shulhan Date: Mon, 16 Feb 2026 15:22:59 +0700 Subject: [DO-NOT-MERGE] all: add option shutdown-idle The shutdown-idle set the duration to automatically shutdown the HTTP server when no request after specific duration. This is to complement the socket based activation to minimize the resources on local environment. --- internal/middleware/timeout/timeout.go | 10 ++++++++++ 1 file changed, 10 insertions(+) (limited to 'internal/middleware/timeout/timeout.go') diff --git a/internal/middleware/timeout/timeout.go b/internal/middleware/timeout/timeout.go index 93d992da..391919c4 100644 --- a/internal/middleware/timeout/timeout.go +++ b/internal/middleware/timeout/timeout.go @@ -10,11 +10,21 @@ import ( "time" ) +var ShutdownIdleDuration time.Duration +var ShutdownIdleTimer *time.Timer + // Timeout returns a new Middleware that times out each request after the given // duration. func Timeout(d time.Duration) func(http.Handler) http.Handler { return func(h http.Handler) http.Handler { return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { + if ShutdownIdleTimer != nil { + ok := ShutdownIdleTimer.Reset(ShutdownIdleDuration) + if !ok { + // Timer had reached or closed + return + } + } ctx, cancel := context.WithTimeout(r.Context(), d) defer cancel() h.ServeHTTP(w, r.WithContext(ctx)) -- cgit v1.3