Module: Sniffer::Adapters::EventMachineAdapter::Client

Defined in:
lib/sniffer/adapters/eventmachine_adapter.rb

Overview

Overrides #send_request, #parse_response, #on_body_data

Class Method Summary collapse

Instance Method Summary collapse

Class Method Details

.included(base) ⇒ Object 



9
10
11
12
13
14
15
16
17
18
 
# 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

Instance Method Details

#on_body_data_with_sniffer(data) ⇒ Object 



50
51
52
53
54
55
56
57
 
# File 'lib/sniffer/adapters/eventmachine_adapter.rb', line 50

def on_body_data_with_sniffer(data)
  if Sniffer.enabled?
    @data_item.response.body = data
    @data_item.log
  end

  on_body_data_without_sniffer(data)
end

#parse_response_header_with_sniffer(header, version, status) ⇒ Object

rubocop:enable Metrics/AbcSize,Metrics/MethodLength



40
41
42
43
44
45
46
47
48
 
# 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) ⇒ Object

rubocop:disable Metrics/AbcSize,Metrics/MethodLength



21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
 
# 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