Call AccessHandler's f also on panic.

rs · Oct 29, 2023 · 1b298bf · 1b298bf
1 parent dfd2071
commit 1b298bf
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))
 		})
 	}
 }