influxdb-rails 1.0.0.beta3 → 1.0.1.beta3

data/lib/influxdb/rails/context.rb

@@ -1,41 +1,25 @@
 module InfluxDB
   module Rails
-    class Context # rubocop:disable Style/Documentation
-      def controller
-        Thread.current[:_influxdb_rails_controller]
-      end
-
-      def controller=(value)
-        Thread.current[:_influxdb_rails_controller] = value
-      end
-
-      def action
-        Thread.current[:_influxdb_rails_action]
-      end
-
-      def action=(value)
-        Thread.current[:_influxdb_rails_action] = value
+    class Context
+      def reset
+        Thread.current[:_influxdb_rails_tags] = {}
+        Thread.current[:_influxdb_rails_values] = {}
       end
 
-      def location
-        [
-          controller,
-          action,
-        ].reject(&:blank?).join("#")
+      def tags
+        Thread.current[:_influxdb_rails_tags].to_h
       end
 
-      def reset
-        Thread.current[:_influxdb_rails_controller] = nil
-        Thread.current[:_influxdb_rails_action] = nil
-        Thread.current[:_influxdb_rails_tags] = nil
+      def tags=(tags)
+        Thread.current[:_influxdb_rails_tags] = self.tags.merge(tags)
       end
 
-      def tags
-        Thread.current[:_influxdb_rails_tags] || {}
+      def values
+        Thread.current[:_influxdb_rails_values].to_h
       end
 
-      def tags=(tags)
-        Thread.current[:_influxdb_rails_tags] = tags
+      def values=(values)
+        Thread.current[:_influxdb_rails_values] = self.values.merge(values)
       end
     end
   end

data/lib/influxdb/rails/helpers/rspec_matchers.rb

@@ -0,0 +1,48 @@
+require_relative "../test_client"
+require "launchy"
+
+module InfluxDB
+  module Rails
+    module Matchers
+      def expect_metric(name: "rails", **options)
+        expect(metrics).to include(
+          a_hash_including(options.merge(name: name))
+        )
+      end
+
+      def expect_no_metric(name: "rails", **options)
+        expect(metrics).not_to include(
+          a_hash_including(options.merge(name: name))
+        )
+      end
+
+      def save_and_open_metrics
+        dir = File.join(File.dirname(__FILE__), "..", "..", "tmp")
+        FileUtils.mkdir_p(dir)
+        file_path = File.join(dir, "metrics.json")
+        output = JSON.pretty_generate(metrics)
+        File.write(file_path, output, mode: "wb")
+        ::Launchy.open(file_path)
+      end
+
+      def metrics
+        TestClient.metrics
+      end
+
+      RSpec.configure do |config|
+        config.before :each do
+          InfluxDB::Rails.instance_variable_set :@configuration, nil
+          InfluxDB::Rails.configure
+
+          allow(InfluxDB::Rails).to receive(:client).and_return(InfluxDB::Rails::TestClient.new)
+          allow_any_instance_of(InfluxDB::Rails::Configuration)
+            .to receive(:ignored_environments).and_return(%w[development])
+
+          InfluxDB::Rails::TestClient.metrics.clear
+        end
+
+        config.include InfluxDB::Rails::Matchers
+      end
+    end
+  end
+end

data/lib/influxdb/rails/metric.rb

@@ -0,0 +1,39 @@
+require "influxdb/rails/values"
+require "influxdb/rails/tags"
+
+module InfluxDB
+  module Rails
+    class Metric
+      def initialize(configuration:, timestamp:, tags: {}, values: {})
+        @configuration = configuration
+        @timestamp = timestamp
+        @tags = tags
+        @values = values
+      end
+
+      def write
+        client.write_point configuration.measurement_name, options
+      end
+
+      private
+
+      attr_reader :configuration, :tags, :values, :timestamp
+
+      def options
+        {
+          values:    Values.new(values: values).to_h,
+          tags:      Tags.new(tags: tags, config: configuration).to_h,
+          timestamp: timestamp_with_precision,
+        }
+      end
+
+      def timestamp_with_precision
+        InfluxDB.convert_timestamp(timestamp.utc, configuration.client.time_precision)
+      end
+
+      def client
+        InfluxDB::Rails.client
+      end
+    end
+  end
+end

data/lib/influxdb/rails/middleware/action_mailer_subscriber.rb

@@ -0,0 +1,22 @@
+require "influxdb/rails/middleware/subscriber"
+
+module InfluxDB
+  module Rails
+    module Middleware
+      class ActionMailerSubscriber < Subscriber # :nodoc:
+        private
+
+        def values
+          { value: 1 }
+        end
+
+        def tags
+          {
+            hook:   "deliver",
+            mailer: payload[:mailer],
+          }
+        end
+      end
+    end
+  end
+end

data/lib/influxdb/rails/middleware/active_job_subscriber.rb

@@ -0,0 +1,67 @@
+require "influxdb/rails/middleware/subscriber"
+
+module InfluxDB
+  module Rails
+    module Middleware
+      class ActiveJobSubscriber < Subscriber # :nodoc:
+        private
+
+        JOB_STATE = {
+          "enqueue"       => "queued",
+          "perform_start" => "running",
+          "perform"       => "succeeded",
+        }.freeze
+        private_constant :JOB_STATE
+
+        def values
+          {
+            value: value,
+          }
+        end
+
+        def tags
+          {
+            hook:  short_hook_name,
+            state: job_state,
+            job:   job.class.name,
+            queue: job.queue_name,
+          }
+        end
+
+        def job_state
+          return "failed" if failed?
+
+          JOB_STATE[short_hook_name]
+        end
+
+        def measure_performance?
+          short_hook_name == "perform"
+        end
+
+        def short_hook_name
+          @short_hook_name ||= fetch_short_hook_name
+        end
+
+        def fetch_short_hook_name
+          return "enqueue" if hook_name.include?("enqueue")
+
+          "perform"
+        end
+
+        def job
+          @job ||= payload[:job]
+        end
+
+        def value
+          return duration if measure_performance?
+
+          1
+        end
+
+        def failed?
+          payload[:aborted]
+        end
+      end
+    end
+  end
+end

data/lib/influxdb/rails/middleware/active_record_subscriber.rb

@@ -0,0 +1,26 @@
+require "influxdb/rails/middleware/subscriber"
+require "influxdb/rails/sql/query"
+
+module InfluxDB
+  module Rails
+    module Middleware
+      class ActiveRecordSubscriber < Subscriber # :nodoc:
+        private
+
+        def values
+          {
+            value:        duration,
+            record_count: payload[:record_count],
+          }
+        end
+
+        def tags
+          {
+            hook:       "instantiation",
+            class_name: payload[:class_name],
+          }
+        end
+      end
+    end
+  end
+end

data/lib/influxdb/rails/middleware/block_instrumentation_subscriber.rb

@@ -0,0 +1,24 @@
+require "influxdb/rails/middleware/subscriber"
+
+module InfluxDB
+  module Rails
+    module Middleware
+      class BlockInstrumentationSubscriber < Subscriber
+        private
+
+        def values
+          {
+            value: duration,
+          }.merge(payload[:values].to_h)
+        end
+
+        def tags
+          {
+            hook: "block_instrumentation",
+            name: payload[:name],
+          }.merge(payload[:tags].to_h)
+        end
+      end
+    end
+  end
+end

data/lib/influxdb/rails/middleware/render_subscriber.rb

@@ -1,24 +1,30 @@
-require "influxdb/rails/middleware/simple_subscriber"
+require "influxdb/rails/middleware/subscriber"
 
 module InfluxDB
   module Rails
     module Middleware
-      class RenderSubscriber < SimpleSubscriber # :nodoc:
+      class RenderSubscriber < Subscriber # :nodoc:
         private
 
-        def values(started, finished, payload)
-          super(started, finished, payload).merge(
+        def values
+          {
+            value:      duration,
             count:      payload[:count],
-            cache_hits: payload[:cache_hits]
-          ).reject { |_, value| value.nil? }
+            cache_hits: payload[:cache_hits],
+          }
         end
 
-        def tags(payload)
-          tags = {
-            location: location,
+        def tags
+          {
+            hook:     short_hook_name,
             filename: payload[:identifier],
           }
-          super(tags)
+        end
+
+        def short_hook_name
+          return "render_template" if hook_name.include?("render_template")
+          return "render_partial" if hook_name.include?("render_partial")
+          return "render_collection" if hook_name.include?("render_collection")
         end
       end
     end

data/lib/influxdb/rails/middleware/request_subscriber.rb

@@ -4,46 +4,39 @@ module InfluxDB
   module Rails
     module Middleware
       class RequestSubscriber < Subscriber # :nodoc:
-        def call(_name, start, finish, _id, payload) # rubocop:disable Metrics/MethodLength
-          return unless enabled?
-
-          ts = InfluxDB.convert_timestamp(finish.utc, configuration.time_precision)
-          tags = tags(payload)
-          begin
-            series(payload, start, finish).each do |series_name, value|
-              InfluxDB::Rails.client.write_point \
-                series_name,
-                values:    { value: value },
-                tags:      tags,
-                timestamp: ts
-            end
-          rescue StandardError => e
-            log :error, "[InfluxDB::Rails] Unable to write points: #{e.message}"
-          ensure
-            InfluxDB::Rails.current.reset
-          end
+        def write
+          super
+        ensure
+          InfluxDB::Rails.current.reset
         end
 
         private
 
-        def series(payload, start, finish)
+        def tags
           {
-            configuration.series_name_for_controller_runtimes => ((finish - start) * 1000).ceil,
-            configuration.series_name_for_view_runtimes       => (payload[:view_runtime] || 0).ceil,
-            configuration.series_name_for_db_runtimes         => (payload[:db_runtime] || 0).ceil,
-          }
-        end
-
-        def tags(payload)
-          tags = {
             method:      "#{payload[:controller]}##{payload[:action]}",
+            hook:        "process_action",
             status:      payload[:status],
             format:      payload[:format],
             http_method: payload[:method],
-            server:      Socket.gethostname,
-            app_name:    configuration.application_name,
+            exception:   payload[:exception]&.first,
+          }
+        end
+
+        def values
+          {
+            controller: duration,
+            view:       (payload[:view_runtime] || 0).ceil,
+            db:         (payload[:db_runtime] || 0).ceil,
+            started:    started,
           }
-          super(tags)
+        end
+
+        def started
+          InfluxDB.convert_timestamp(
+            start.utc,
+            configuration.client.time_precision
+          )
         end
       end
     end

data/lib/influxdb/rails/middleware/sql_subscriber.rb

@@ -1,31 +1,35 @@
-require "influxdb/rails/middleware/simple_subscriber"
+require "influxdb/rails/middleware/subscriber"
 require "influxdb/rails/sql/query"
 
 module InfluxDB
   module Rails
     module Middleware
-      class SqlSubscriber < SimpleSubscriber # :nodoc:
-        def call(_name, started, finished, _unique_id, payload)
-          return unless InfluxDB::Rails::Sql::Query.new(payload).track?
-
-          super
-        end
-
+      class SqlSubscriber < Subscriber # :nodoc:
         private
 
-        def values(started, finished, payload)
-          super.merge(sql: InfluxDB::Rails::Sql::Normalizer.new(payload[:sql]).perform)
+        def values
+          {
+            value: duration,
+            sql:   InfluxDB::Rails::Sql::Normalizer.new(payload[:sql]).perform,
+          }
         end
 
-        def tags(payload)
-          query = InfluxDB::Rails::Sql::Query.new(payload)
-          tags = {
-            location:   location,
+        def tags
+          {
+            hook:       "sql",
             operation:  query.operation,
             class_name: query.class_name,
             name:       query.name,
+            location:   :raw,
           }
-          super(tags)
+        end
+
+        def disabled?
+          super || !query.track?
+        end
+
+        def query
+          @query ||= InfluxDB::Rails::Sql::Query.new(payload)
         end
       end
     end