flathttp: set ctx after timeout set

lithdew · May 24, 2020 · e7f95c7 · e7f95c7
1 parent fb7afd5
commit e7f95c7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/flathttp/middleware.go b/flathttp/middleware.go
@@ -21,6 +21,8 @@ func (m *Middleware) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 	if m.Timeout > 0 {
 		ctx, cancel := context.WithTimeout(r.Context(), m.Timeout)
 
+		r = r.WithContext(ctx)
+
 		defer func() {
 			cancel()