lib/sniffer/adapters/httpclient_adapter.rb in sniffer-0.3.2 vs lib/sniffer/adapters/httpclient_adapter.rb in sniffer-0.4.0

@@ -28,15 +28,13 @@
         end
 
         retryable_response = nil
 
         bm = Benchmark.realtime do
-          begin
-            do_get_block_without_sniffer(req, proxy, conn, &block)
-          rescue HTTPClient::RetryableResponse => e
-            retryable_response = e
-          end
+          do_get_block_without_sniffer(req, proxy, conn, &block)
+        rescue HTTPClient::RetryableResponse => e
+          retryable_response = e
         end
 
         if Sniffer.enabled?
           res = conn.pop
           data_item.response = Sniffer::DataItem::Response.new(status: res.status_code.to_i,
@@ -44,16 +42,16 @@
                                                                body: res.body,
                                                                timing: bm)
 
           conn.push(res)
 
-          data_item.log
+          Sniffer.notify_response(data_item)
         end
 
         raise retryable_response unless retryable_response.nil?
       end
       # rubocop:enable Metrics/AbcSize, Metrics/MethodLength
     end
   end
 end
 
-HTTPClient.send(:include, Sniffer::Adapters::HTTPClientAdapter) if defined?(::HTTPClient)
+HTTPClient.include Sniffer::Adapters::HTTPClientAdapter if defined?(::HTTPClient)