viaduct_metric_catcher 1.0.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: efe9f3767f8a61f166f873b6a9a62fbbab64c476
+  data.tar.gz: d4acc7d0ba5644d00a1e15eadf8d99e2ac7d68db
+SHA512:
+  metadata.gz: 2fcca53f5dc5686f4501ab8ac18e2af3a14ecb9c0d83d2942b654035cfcdc3fec8f7cefb797d1b29dde517fcb9362dfc5c925e6f5049c411f7271f28798d0024
+  data.tar.gz: 2d209baed32f52a0897e9e51bde9f34a2bb33655bd8dc1e6baa577d71397e650ff667f3abb69650702180df658b26956c9ec8f2d5d02c8d03195397180e3bbec

data/lib/viaduct/metric_catcher.rb

@@ -0,0 +1,57 @@
+require 'viaduct/metric_catcher/railtie' if defined?(Rails)
+require 'viaduct/metric_catcher/client'
+require 'viaduct/metric_catcher/error'
+require 'socket'
+require 'logger'
+
+module Viaduct
+  module MetricCatcher
+    class << self
+
+      attr_accessor :framework, :hostname
+      attr_reader :client
+      attr_writer :logger
+
+      def logger
+        @logger ||= Logger.new(STDOUT)
+      end
+
+      def token=(token)
+        if token =~ /\A[a-zA-Z0-9]{20}\z/
+          @client = Client.new(token)
+        elsif token.nil?
+          @client = nil
+        else
+          raise Error, "Invalid token provided. Tokens must be 20 characters and only include letters & numbers."
+        end
+      end
+
+      def hostname
+        hostname_for_viaduct_process || hostname_from_socket
+      end
+
+      def hostname_for_viaduct_process
+        @hostname_for_viaduct_process ||= ENV['VDT_PROC'] ? "#{ENV['VDT_PROC']}/#{ENV['VDT_DEPLOYMENT']}" : nil
+      end
+
+      def hostname_from_socket
+        @hostname_from_socket ||= Socket.gethostname rescue nil
+      end
+
+      def send_payload(event, payload)
+        if self.client
+          hash = {
+            'event' => event.to_s,
+            'framework' => self.framework,
+            'hostname' => self.hostname,
+            'payload' => payload
+          }
+          self.client.send_to_udp(hash)
+        else
+          false
+        end
+      end
+
+    end
+  end
+end

data/lib/viaduct/metric_catcher/client.rb

@@ -0,0 +1,43 @@
+require 'socket'
+require 'json'
+
+module Viaduct
+  module MetricCatcher
+    class Client
+
+      class << self
+        attr_writer :host, :port
+
+        def host
+          @host ||= "metrics.viaduct.io"
+        end
+
+        def port
+          @port ||= 21100
+        end
+
+        def socket
+          @socket ||= UDPSocket.new
+        end
+      end
+
+      attr_reader :token
+
+      def initialize(token)
+        @token = token
+      end
+
+      def pack_hash_for_udp(hash)
+        time = Time.now.utc
+        [1, time.to_i, time.nsec, @token, hash.to_json].pack('NNNa20a*')
+      end
+
+      def send_to_udp(hash)
+        self.class.socket.send(pack_hash_for_udp(hash), 0, self.class.host, self.class.port)
+      rescue => e
+        Viaduct::MetricCatcher.logger.warn "[Viaduct Metrics] Exception while sending metrics. #{e.class} #{e.message}"
+      end
+
+    end
+  end
+end

data/lib/viaduct/metric_catcher/error.rb

@@ -0,0 +1,6 @@
+module Viaduct
+  module MetricCatcher
+    class Error < StandardError
+    end
+  end
+end

data/lib/viaduct/metric_catcher/railtie.rb

@@ -0,0 +1,42 @@
+module Viaduct
+  module MetricCatcher
+    class Railtie < ::Rails::Railtie
+
+      initializer 'viaduct.metric-catcher.notifications' do
+        # Set the framework to 'rails'
+        Viaduct::MetricCatcher.framework = 'rails'
+        # Set the logger to use the standard Rails logger
+        Viaduct::MetricCatcher.logger = Rails.logger
+        # Subscribe to all requests and send request details whenever a request
+        # happens.
+        ActiveSupport::Notifications.subscribe "process_action.action_controller" do |*args|
+          event = ActiveSupport::Notifications::Event.new(*args)
+          Viaduct::MetricCatcher.send_payload(:request, {
+            'method' => event.payload[:method],
+            'path' => event.payload[:path],
+            'status' => event.payload[:status],
+            'time' => event.duration,
+            'extra' => {
+              'db_runtime' => event.payload[:db_runtime],
+              'view_runtime' => event.payload[:view_runtime],
+              'controller' => event.payload[:controller],
+              'action' => event.payload[:action]
+            }
+          })
+        end
+        # Subscribe to all database queries and send quer details whenever a
+        # query is executed.
+        ActiveSupport::Notifications.subscribe "sql.active_record" do |*args|
+          event = ActiveSupport::Notifications::Event.new(*args)
+          Viaduct::MetricCatcher.send_payload(:database_query, {
+            'query' => event.payload[:sql],
+            'time' => event.duration,
+            'extra' => {
+              'name' => event.payload[:name]
+            }
+          })
+        end
+      end
+    end
+  end
+end

data/lib/viaduct_metric_catcher.rb

@@ -0,0 +1 @@
+require 'viaduct/metric_catcher'

metadata

@@ -0,0 +1,49 @@
+--- !ruby/object:Gem::Specification
+name: viaduct_metric_catcher
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+platform: ruby
+authors:
+- Adam Cooke
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-03-20 00:00:00.000000000 Z
+dependencies: []
+description: Automatically catch metrics and send them to the Viaduct Metrics service
+email:
+- adam@viaduct.io
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/viaduct/metric_catcher.rb
+- lib/viaduct/metric_catcher/client.rb
+- lib/viaduct/metric_catcher/error.rb
+- lib/viaduct/metric_catcher/railtie.rb
+- lib/viaduct_metric_catcher.rb
+homepage: https://github.com/viaduct/metric-catcher
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.5
+signing_key: 
+specification_version: 4
+summary: Automatically catch metrics and send them to the Viaduct Metrics service
+test_files: []