vx-instrumentation 0.1.2 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 8e38ae361a2d539a48365d64eaeb9cc905934c7b
-  data.tar.gz: 8eb4ea7ba4eacceb47cda77958dc1e4f4f3919c8
+  metadata.gz: 9178f6d2a8a76a723de0d0ae4c88937f290e6e66
+  data.tar.gz: 9718fd7ca3ab06054ad1f98fe8c6f98bef81eaf3
 SHA512:
-  metadata.gz: a836d117ec05ce5200a9093ee69ad74d0e16cd86ac8e9109f6a3f14636e81721566e961a74ace09274959340ee62cabe9acd33827943046a374c5b4fce53ac4e
-  data.tar.gz: 3649718e03c7c01187f0553634da00ebae0965df4c2db2658587ce3585e1bcebcf2eb0b018a9a06d53b94c953b8ce792eb737faa83ca92f3081950769b6afbf4
+  metadata.gz: 259a62ac21ff478952e099b121fde6a0a34fd9860b9727790c34aa506fba8af14523fcbce6cac07fbc81b2d1e2be5e0ebbb9c22097fb08efb1a380a88fd62660
+  data.tar.gz: 6cad4995cff09cd4c4c15b219dd904ed5b8487c3af53b68b05f58ca5e8d28976151bde9d7cf8e120e878c66546bd4f762cb4afedd5f755300d20b6744a64eaa6

data/lib/vx/instrumentation.rb

@@ -21,6 +21,13 @@ module Vx
       Instrumentation::Logger.logger.level = log_level
     end
 
+    def activate(*probes)
+      probes.each do |probe|
+        require File.expand_path("../instrumentation/probe/#{probe}", __FILE__)
+        $stdout.puts " --> activate instrumentations for #{probe}"
+      end
+    end
+
     def with(new_keys)
       old_keys = Thread.current[THREAD_KEY]
       begin

data/lib/vx/instrumentation/probe/action_controller.rb

@@ -0,0 +1,6 @@
+require 'active_support/notifications'
+
+ActiveSupport::Notifications.subscribe(/\.action_controller$/) do |event, started, finished, _, payload|
+  Vx::Instrumentation.delivery event, payload, event.split("."), started, finished
+end
+

data/lib/vx/instrumentation/probe/action_dispatch.rb

@@ -0,0 +1,15 @@
+require 'active_support/notifications'
+
+ActiveSupport::Notifications.subscribe(/\.action_dispatch$/) do |event, started, finished, _, payload|
+  req = payload[:request]
+  payload = {
+    path:           req.fullpath,
+    ip:             req.remote_ip,
+    method:         req.method,
+    referer:        req.referer,
+    content_length: req.content_length,
+    user_agent:     req.user_agent
+  }
+  Vx::Instrumentation.delivery event, payload, event.split("."), started, finished
+end
+

data/lib/vx/instrumentation/probe/action_mailer.rb

@@ -0,0 +1,6 @@
+require 'active_support/notifications'
+
+ActiveSupport::Notifications.subscribe(/\.action_mailer$/) do |event, started, finished, _, payload|
+  Vx::Instrumentation.delivery event, payload, event.split("."), started, finished
+end
+

data/lib/vx/instrumentation/probe/action_view.rb

@@ -0,0 +1,8 @@
+require 'active_support/notifications'
+require 'vx/instrumentation'
+
+ActiveSupport::Notifications.subscribe(/\.action_view$/) do |event, started, finished, _, payload|
+  if event[0] != "!"
+    Vx::Instrumentation.delivery event, payload, event.split("."), started, finished
+  end
+end

data/lib/vx/instrumentation/probe/active_record.rb

@@ -0,0 +1,26 @@
+require 'active_support/notifications'
+
+ActiveSupport::Notifications.subscribe(/\.active_record$/) do |event, started, finished, _, payload|
+
+  binds =
+    (payload[:binds] || []).map do |column, value|
+      if column
+        if column.binary?
+          value = "<#{value.bytesize} bytes of binary data>"
+        end
+        [column.name, value]
+      else
+        [nil, value]
+      end
+    end.inspect
+
+  payload = {
+    sql:      payload[:sql],
+    binds:    binds,
+    name:     payload[:name],
+    duration: payload[:duration]
+  }
+
+  Vx::Instrumentation.delivery event, payload, event.split("."), started, finished
+end
+

data/lib/vx/instrumentation/probe/active_support.rb

@@ -0,0 +1,6 @@
+require 'active_support/notifications'
+
+ActiveSupport::Notifications.subscribe(/\.active_support$/) do |event, started, finished, _, payload|
+  Vx::Instrumentation.delivery event, payload, event.split("."), started, finished
+end
+

data/lib/vx/instrumentation/probe/faraday.rb

@@ -0,0 +1,24 @@
+require 'active_support/notifications'
+
+ActiveSupport::Notifications.subscribe(/\.faraday$/) do |event, started, finished, _, payload|
+
+  render_http_header = ->(headers) {
+    (headers || []).map do |key,value|
+      if %{ PRIVATE-TOKEN Authorization }.include?(key)
+        value = value.gsub(/./, "*")
+      end
+      "#{key}: #{value}"
+    end.join("\n")
+  }
+
+  payload = {
+    method:           payload[:method],
+    url:              payload[:url].to_s,
+    status:           payload[:status],
+    response_headers: render_http_header.call(payload[:response_headers]),
+    request_headers:  render_http_header.call(payload[:request_headers])
+  }
+
+  Vx::Instrumentation.delivery event, payload, event.split("."), started, finished
+end
+

data/lib/vx/instrumentation/version.rb

@@ -1,5 +1,5 @@
 module Vx
   module Instrumentation
-    VERSION = "0.1.2"
+    VERSION = "0.1.3"
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: vx-instrumentation
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.3
 platform: ruby
 authors:
 - Dmitry Galinsky
@@ -82,6 +82,13 @@ files:
 - Rakefile
 - lib/vx/instrumentation.rb
 - lib/vx/instrumentation/logger.rb
+- lib/vx/instrumentation/probe/action_controller.rb
+- lib/vx/instrumentation/probe/action_dispatch.rb
+- lib/vx/instrumentation/probe/action_mailer.rb
+- lib/vx/instrumentation/probe/action_view.rb
+- lib/vx/instrumentation/probe/active_record.rb
+- lib/vx/instrumentation/probe/active_support.rb
+- lib/vx/instrumentation/probe/faraday.rb
 - lib/vx/instrumentation/rack/handle_exceptions_middleware.rb
 - lib/vx/instrumentation/stderr.rb
 - lib/vx/instrumentation/version.rb