1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
# frozen_string_literal: true
module Labkit
##
# Prepend to Ruby's Net/HTTP standard HTTP library to publish a notification
# whenever a HTTP request is triggered. Net::HTTP has different class methods
# for each http method. Those methods are delegated to corresponding instance
# methods. Eventually, `request` method is call to dispatch the HTTP request.
# Therefore, a prepender that override `request` method covers all HTTP
# calls.
#
# For more information:
# https://github.com/ruby/ruby/blob/9b9cbbbc17bb5840581c7da37fd0feb0a7d4c1f3/lib/net/http.rb#L1510
#
# Note: some use cases to take care of
# - Create a request from input URI
# - Create a request from input host, port, and path string
# - Create a singular request and closes the connection immediately
# - Create a persistent connection and perform multiple HTTP requests
# - Notification payload must separate URI components
# - Create a post request with a body
# - Create a post request with form data
# - Create a request with basic authentication
# - Make a request via a proxy server
# - Streaming
module NetHttpPublisher
@prepend_mutex = Mutex.new
def self.labkit_prepend!
@prepend_mutex.synchronize do
return if @prepended
require "net/http"
Net::HTTP.prepend(self)
@prepended = true
end
end
def request(request, *args, &block)
return super unless started?
start_time = ::Labkit::System.monotonic_time
ActiveSupport::Notifications.instrument ::Labkit::EXTERNAL_HTTP_NOTIFICATION_TOPIC, create_request_payload(request) do |payload|
response =
begin
super
ensure
payload[:duration] = (::Labkit::System.monotonic_time - start_time).to_f
end
payload[:code] = response.code
response
end
end
private
def create_request_payload(request)
payload = {
method: request.method,
}
if request.uri.nil?
path_uri = URI(request.path)
payload[:host] = address
payload[:path] = path_uri.path
payload[:port] = port
payload[:scheme] = use_ssl? ? "https" : "http"
payload[:query] = path_uri.query
payload[:fragment] = path_uri.fragment
else
payload[:host] = request.uri.host
payload[:path] = request.uri.path
payload[:port] = request.uri.port
payload[:scheme] = request.uri.scheme
payload[:query] = request.uri.query
payload[:fragment] = request.uri.fragment
end
if proxy?
payload[:proxy_host] = proxy_address
payload[:proxy_port] = proxy_port
end
payload
end
end
end
|
