conductivity.rb in faraday-conductivity-0.3.0

- old
+ new
@@ -7,17 +7,20 @@
 require "faraday/conductivity/request_id"
 require "faraday/conductivity/request_id_filter"
 require "faraday/conductivity/user_agent"
 require "faraday/conductivity/repeater"
 require "faraday/conductivity/selective_errors"
+require "faraday/conductivity/request_headers"
 
 module Faraday
   module Conductivity
   end
-  register_middleware :middleware,  :extended_logging => Faraday::Conductivity::ExtendedLogging
-  register_middleware :request,     :user_agent       => Faraday::Conductivity::UserAgent
-  register_middleware :request,     :request_id       => Faraday::Conductivity::RequestId
-  register_middleware :request,     :mimetype         => Faraday::Conductivity::Mimetype
-  register_middleware :middleware,  :repeater         => Faraday::Conductivity::Repeater
-  register_middleware :response,    :selective_errors => Faraday::Conductivity::SelectiveErrors
+
+  Faraday::Middleware.register_middleware :extended_logging => Faraday::Conductivity::ExtendedLogging
+  Faraday::Middleware.register_middleware :repeater         => Faraday::Conductivity::Repeater
+  Faraday::Request.register_middleware    :mimetype         => Faraday::Conductivity::Mimetype
+  Faraday::Request.register_middleware    :request_id       => Faraday::Conductivity::RequestId
+  Faraday::Request.register_middleware    :request_headers  => Faraday::Conductivity::RequestHeaders
+  Faraday::Request.register_middleware    :user_agent       => Faraday::Conductivity::UserAgent
+  Faraday::Response.register_middleware   :selective_errors => Faraday::Conductivity::SelectiveErrors
 end
 