lib/patches/net_patches.rb in rack-mini-profiler-3.0.0 vs lib/patches/net_patches.rb in rack-mini-profiler-3.1.0

- old
+ new

@@ -1,26 +1,27 @@ # frozen_string_literal: true -if (defined?(Net) && defined?(Net::HTTP)) - - if defined?(Rack::MINI_PROFILER_PREPEND_NET_HTTP_PATCH) - module NetHTTPWithMiniProfiler - def request(request, *args, &block) - Rack::MiniProfiler.step("Net::HTTP #{request.method} #{request.path}") do - super +if ENV['RACK_MINI_PROFILER_PATCH_NET_HTTP'] != 'false' + if (defined?(Net) && defined?(Net::HTTP)) + if defined?(Rack::MINI_PROFILER_PREPEND_NET_HTTP_PATCH) + module NetHTTPWithMiniProfiler + def request(request, *args, &block) + Rack::MiniProfiler.step("Net::HTTP #{request.method} #{request.path}") do + super + end end end - end - Net::HTTP.prepend(NetHTTPWithMiniProfiler) - else - Net::HTTP.class_eval do - def request_with_mini_profiler(*args, &block) - request = args[0] - Rack::MiniProfiler.step("Net::HTTP #{request.method} #{request.path}") do - request_without_mini_profiler(*args, &block) + Net::HTTP.prepend(NetHTTPWithMiniProfiler) + else + Net::HTTP.class_eval do + def request_with_mini_profiler(*args, &block) + request = args[0] + Rack::MiniProfiler.step("Net::HTTP #{request.method} #{request.path}") do + request_without_mini_profiler(*args, &block) + end end + alias request_without_mini_profiler request + alias request request_with_mini_profiler end - alias request_without_mini_profiler request - alias request request_with_mini_profiler end end end