Call AccessHandler's f also on panic.

rs · Oct 23, 2023 · c79fa09 · c79fa09
1 parent 6c1c8bf
commit c79fa09
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/hlog/hlog.go b/hlog/hlog.go
@@ -294,8 +294,10 @@ func AccessHandler(f func(r *http.Request, status, size int, duration time.Durat
 		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 			start := time.Now()
 			lw := mutil.WrapWriter(w)
+			defer func() {
+				f(r, lw.Status(), lw.BytesWritten(), time.Since(start))
+			}()
 			next.ServeHTTP(lw, r)
-			f(r, lw.Status(), lw.BytesWritten(), time.Since(start))
 		})
 	}
 }