diff --git a/lib/librato/metrics/middleware/expects_status.rb b/lib/librato/metrics/middleware/expects_status.rb index a75552d..e448889 100644 --- a/lib/librato/metrics/middleware/expects_status.rb +++ b/lib/librato/metrics/middleware/expects_status.rb @@ -24,12 +24,12 @@ def on_complete(env) def sanitize_request(env) { - status: env.status, - url: env.url.to_s, - user_agent: env.request_headers["User-Agent"], + status: env[:status], + url: env[:url].to_s, + user_agent: (env[:request_headers] || {})["User-Agent"], request_body: env[:request_body], - response_headers: env.response_headers, - response_body: env.body + response_headers: env[:response_headers], + response_body: env[:body] } end end