From 17e269202ba01e8467a87163395a1e6ac8261e02 Mon Sep 17 00:00:00 2001 From: JP Appel Date: Wed, 9 Oct 2024 16:03:12 -0400 Subject: Refactor middleware to separate package --- middleware/timeout.go | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) create mode 100644 middleware/timeout.go (limited to 'middleware/timeout.go') diff --git a/middleware/timeout.go b/middleware/timeout.go new file mode 100644 index 0000000..6a3f5de --- /dev/null +++ b/middleware/timeout.go @@ -0,0 +1,17 @@ +package middleware + +import ( + "context" + "net/http" + "time" +) + +// Middleware to timeout requests after a given duration +func Timeout(next http.Handler, duration time.Duration) http.Handler { + return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { + ctx, cancel := context.WithTimeout(r.Context(), duration) + defer cancel() + + next.ServeHTTP(w, r.WithContext(ctx)) + }) +} -- cgit v1.2.3