defer f

rs · Nov 21, 2023 · 412c011 · 412c011
1 parent c1cdcd2
commit 412c011
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/hlog/hlog.go b/hlog/hlog.go
@@ -343,14 +343,16 @@ func AccessHandlerWithData(f func(data AccessHandlerData)) func(next http.Handle
 			body := mutil.NewByteCountReadCloser(r.Body)
 			r.Body = body
 
+			defer func() {
+				f(AccessHandlerData{
+					Request:      r,
+					Duration:     time.Since(start),
+					Status:       ww.Status(),
+					BytesWritten: ww.BytesWritten(),
+					BytesRead:    body.BytesRead(),
+				})
+			}()
 			next.ServeHTTP(ww, r)
-			f(AccessHandlerData{
-				Request:      r,
-				Duration:     time.Since(start),
-				Status:       ww.Status(),
-				BytesWritten: ww.BytesWritten(),
-				BytesRead:    body.BytesRead(),
-			})
 		})
 	}
 }