diff options
author | Eugen Rochko <eugen@zeonfederated.com> | 2017-01-26 19:08:05 +0100 |
---|---|---|
committer | Eugen Rochko <eugen@zeonfederated.com> | 2017-01-26 19:08:05 +0100 |
commit | d567f21d4f4c3feb09a08443cf75eb583904093b (patch) | |
tree | 79fdba0e0d0db6fc80ca57bffc7a55cc57bcb6c6 /config/initializers | |
parent | 7329fbd8a453bbd2fcbec8bd63b1390bfe7bc7b8 (diff) |
Improve StatsD instrumentation
Diffstat (limited to 'config/initializers')
-rw-r--r-- | config/initializers/instrumentation.rb | 18 | ||||
-rw-r--r-- | config/initializers/statsd.rb | 19 |
2 files changed, 24 insertions, 13 deletions
diff --git a/config/initializers/instrumentation.rb b/config/initializers/instrumentation.rb new file mode 100644 index 000000000..8483f2be2 --- /dev/null +++ b/config/initializers/instrumentation.rb @@ -0,0 +1,18 @@ +# frozen_string_literal: true + +instrumentation_hostname = ENV.fetch('INSTRUMENTATION_HOSTNAME') { 'localhost' } + +ActiveSupport::Notifications.subscribe(/process_action.action_controller/) do |*args| + event = ActiveSupport::Notifications::Event.new(*args) + controller = event.payload[:controller] + action = event.payload[:action] + format = event.payload[:format] || 'all' + format = 'all' if format == '*/*' + status = event.payload[:status] + key = "#{controller}.#{action}.#{format}.#{instrumentation_hostname}" + + ActiveSupport::Notifications.instrument :performance, action: :measure, measurement: "#{key}.total_duration", value: event.duration + ActiveSupport::Notifications.instrument :performance, action: :measure, measurement: "#{key}.db_time", value: event.payload[:db_runtime] + ActiveSupport::Notifications.instrument :performance, action: :measure, measurement: "#{key}.view_time", value: event.payload[:view_runtime] + ActiveSupport::Notifications.instrument :performance, measurement: "#{key}.status.#{status}" +end diff --git a/config/initializers/statsd.rb b/config/initializers/statsd.rb index c9c754e7f..4e772a5ed 100644 --- a/config/initializers/statsd.rb +++ b/config/initializers/statsd.rb @@ -3,18 +3,11 @@ StatsD.prefix = 'mastodon' StatsD.default_sample_rate = 1 -StatsDMonitor.extend(StatsD::Instrument) -StatsDMonitor.statsd_measure(:call, 'request.duration') +ActiveSupport::Notifications.subscribe(/performance/) do |name, _start, _finish, _id, payload| + action = payload[:action] || :increment + measurement = payload[:measurement] + value = payload[:value] + key_name = "#{name}.#{measurement}" -STATSD_REQUEST_METRICS = { - 'request.status.success' => 200, - 'request.status.not_found' => 404, - 'request.status.too_many_requests' => 429, - 'request.status.internal_server_error' => 500, -}.freeze - -STATSD_REQUEST_METRICS.each do |name, code| - StatsDMonitor.statsd_count_if(:call, name) do |status, _env, _body| - status.to_i == code - end + StatsD.send(action.to_s, key_name, (value || 1)) end |