@@ -118,8 +118,8 @@ apacheMiddleware ala app req sendResponse = app req $ \res -> do
118118customMiddleware :: Callback -> IO ZonedDate -> OutputFormatter -> Middleware
119119customMiddleware cb getdate formatter app req sendResponse = app req $ \res -> do
120120 date <- liftIO getdate
121− -- We use Nothing for the response size since we generally don't know it
122− liftIO $ cb $ formatter date req (responseStatus res) Nothing
121+ let msize = contentLength (responseHeaders res)
122+ liftIO $ cb $ formatter date req (responseStatus res) msize
123123 sendResponse res
124124
125125customMiddlewareWithDetails :: Callback -> IO ZonedDate -> OutputFormatterWithDetails -> Middleware
@@ -129,12 +129,12 @@ customMiddlewareWithDetails cb getdate formatter app req sendResponse = do
129129 app req' $ \res -> do
130130 t1 <- getCurrentTime
131131 date <- liftIO getdate
132− -- We use Nothing for the response size since we generally don't know it
132+ let msize = contentLength (responseHeaders res)
133133 builderIO <- newIORef $ B.byteString ""
134134 res' <- recordChunks builderIO res
135135 rspRcv <- sendResponse res'
136136 _ <- liftIO . cb .
137− formatter date req' (responseStatus res') Nothing (t1 `diffUTCTime` t0) reqBody =<<
137+ formatter date req' (responseStatus res') msize (t1 `diffUTCTime` t0) reqBody =<<
138138 readIORef builderIO
139139 return rspRcv
140140