module Sniffer::Adapters::EventMachineAdapter::Client

Overrides send_request, parse_response, on_body_data

Public Class Methods

included(base) click to toggle source
# File lib/sniffer/adapters/eventmachine_adapter.rb, line 9
def self.included(base)
  base.class_eval do
    alias_method :send_request_without_sniffer, :send_request
    alias_method :send_request, :send_request_with_sniffer
    alias_method :parse_response_header_without_sniffer, :parse_response_header
    alias_method :parse_response_header, :parse_response_header_with_sniffer
    alias_method :on_body_data_without_sniffer, :on_body_data
    alias_method :on_body_data, :on_body_data_with_sniffer
  end
end

Public Instance Methods

on_body_data_with_sniffer(data) click to toggle source
# File lib/sniffer/adapters/eventmachine_adapter.rb, line 50
def on_body_data_with_sniffer(data)
  if Sniffer.enabled?
    @data_item.response.body = data
    Sniffer.notify_response(@data_item)
  end

  on_body_data_without_sniffer(data)
end
parse_response_header_with_sniffer(header, version, status) click to toggle source

rubocop:enable Metrics/AbcSize,Metrics/MethodLength

# File lib/sniffer/adapters/eventmachine_adapter.rb, line 40
def parse_response_header_with_sniffer(header, version, status)
  if Sniffer.enabled?
    @data_item.response.timing = Time.now - @start_time
    @data_item.response.status = status
    @data_item.response.headers = header
  end

  parse_response_header_without_sniffer(header, version, status)
end
send_request_with_sniffer(head, body) click to toggle source

rubocop:disable Metrics/AbcSize,Metrics/MethodLength

# File lib/sniffer/adapters/eventmachine_adapter.rb, line 21
def send_request_with_sniffer(head, body)
  if Sniffer.enabled?
    @data_item = Sniffer::DataItem.new
    @data_item.response = Sniffer::DataItem::Response.new
    @data_item.request = Sniffer::DataItem::Request.new(host: @req.host,
                                                        method: @req.method.to_sym,
                                                        query: encode_query(@req.uri, @req.query),
                                                        headers: @req.headers,
                                                        body: @req.body.to_s,
                                                        port: @req.port)
    Sniffer.store(@data_item)

    @start_time = Time.now
  end

  send_request_without_sniffer(head, body)
end