influx_reporter 1.1.0 → 1.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3ddddb2aec52aed72cd27b0068fd4c293b9f3300e4ad01955068aa335a2bca73
-  data.tar.gz: ae5677b7955e961a06f2e95927fb112474e02457ac506b8f84f16009518afad7
+  metadata.gz: cd77289b6d02bb08bdc655882de791a0e45f3964b4b89aeeac780213b41fd0d3
+  data.tar.gz: a52df2325d7c556d944417befe98266e4cbbb8fa4e1d1741654c79a85a5d3d06
 SHA512:
-  metadata.gz: 73ab7f2528040eee3c65135adb3e5e2192705378741b3c3863c3d5d9ee1e34e4e9a617f577448eca2396919433e45db21488c5b39c893f1d774af071b5941578
-  data.tar.gz: 6f6013c5f538cd50c46a07ccb3b5facef447af48dc1891e924cd9cdaab47f8e2085d6b19dd2987e339cf3bc1ea94f516e3a920e3447c9909546543ab2986f93a
+  metadata.gz: 42f7c94c2992dbef8de3e6cc191d89c1238e08d8fe568ccc28e1403423c41e59e336585bac53bcebb4e44c01fdb597d388fa5b704dabf127a3fc3d7ac1efc120
+  data.tar.gz: b97402fbe910ee75aef5f72573ac0fdd0ddaa7b363eaa23402a109175a60178e4c9935dc555d1167de2dc9855b12c47763fec4d899f0948096e8aa326e18a379

data/README.md

@@ -156,6 +156,11 @@ Adding tags & values is also possible:
 InfluxReporter.report_event 'event_name', extra: { tags: { key: 'tag' }, values: { key: 'value' } }
 ```
 
+Finally, events might generate lots of keys and you may want to use a specific database just for this purpose.
+```ruby
+InfluxReporter.report_event 'event_name', database: 'events_database'
+```
+
 ## Manual profiling
 
 It's easy to add performance tracking wherever you want using the `InfluxReporter` module.

data/lib/influx_reporter/client.rb

@@ -173,7 +173,7 @@ module InfluxReporter
       return if @pending_transactions.empty?
 
       data = @data_builders.transactions.build(@pending_transactions)
-      enqueue Worker::PostRequest.new('/transactions/', data)
+      enqueue Worker::PostRequest.new(resource_from_path('transactions', {}), data)
 
       @last_sent_transactions = Time.now.utc
       @pending_transactions = []
@@ -214,7 +214,7 @@ module InfluxReporter
       if error_message = ErrorMessage.from_exception(config, exception, opts)
         error_message.add_extra(@context) if @context
         data = @data_builders.error_message.build error_message
-        enqueue Worker::PostRequest.new('/errors/', data)
+        enqueue Worker::PostRequest.new(resource_from_path('errors', opts), data)
       end
     end
 
@@ -226,7 +226,7 @@ module InfluxReporter
       error_message = ErrorMessage.new(config, message, opts)
       error_message.add_extra(@context) if @context
       data = @data_builders.error_message.build error_message
-      enqueue Worker::PostRequest.new('/errors/', data)
+      enqueue Worker::PostRequest.new(resource_from_path('errors', opts), data)
     end
 
     def report_event(message, opts = {})
@@ -235,7 +235,7 @@ module InfluxReporter
       event = EventMessage.new(config, message, opts)
       event.add_extra(@context) if @context
       data = @data_builders.event.build event
-      enqueue Worker::PostRequest.new('/events/', data)
+      enqueue Worker::PostRequest.new(resource_from_path('events', opts), data)
     end
 
     def capture
@@ -264,6 +264,13 @@ module InfluxReporter
       @queue << request
     end
 
+    def resource_from_path(path, opts)
+      obj = { url: "/#{path}/" }
+      obj[:database] = opts[:database] if opts[:database]
+      obj
+    end
+
+
     def start_worker
       return if worker_running?
 

data/lib/influx_reporter/error_message.rb

@@ -45,6 +45,7 @@ module InfluxReporter
     attr_accessor :stacktrace
     attr_accessor :http
     attr_accessor :user
+    attr_accessor :database
 
     def self.from_exception(config, exception, opts = {})
       message = "#{exception.class}: #{exception.message}"

data/lib/influx_reporter/event_message.rb

@@ -24,6 +24,7 @@ module InfluxReporter
     attr_accessor :message
     attr_reader :timestamp
     attr_accessor :extra
+    attr_accessor :database
 
     def add_extra(info)
       @extra ||= {}

data/lib/influx_reporter/influx_db_client.rb

@@ -21,7 +21,7 @@ module InfluxReporter
     attr_reader :config
 
     def post(resource, data)
-      debug "POST #{resource}"
+      debug "POST #{resource[:url]}"
 
       unless state.should_try?
         info 'Temporarily skipping sending to InfluxReporter due to previous failure.'
@@ -30,7 +30,7 @@ module InfluxReporter
 
       begin
         data = [data] unless data.is_a?(Array)
-        client.write_points data
+        client.write_points data, nil, nil, resource.fetch(:database, nil)
       rescue StandardError => e
         debug { e.message }
         @state.fail!

data/lib/influx_reporter/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module InfluxReporter
-  VERSION = '1.1.0'
+  VERSION = '1.2.0'
 end

data/lib/influx_reporter/worker.rb

@@ -5,10 +5,10 @@ module InfluxReporter
   class Worker
     include Logging
 
-    class PostRequest < Struct.new(:path, :data)
+    class PostRequest < Struct.new(:resource, :data)
       # require all parameters
-      def initialize(path, data)
-        super(path, data)
+      def initialize(resource, data)
+        super(resource, data)
       end
     end
 
@@ -46,7 +46,7 @@ module InfluxReporter
       end
 
       begin
-        @influx_client.post(req.path, req.data)
+        @influx_client.post(req.resource, req.data)
       rescue => e
         fatal "Failed POST: #{e.inspect}"
         debug e.backtrace.join("\n")

data/spec/influx_reporter/worker_spec.rb

@@ -25,13 +25,13 @@ module InfluxReporter
         end
 
         it 'pops the queue' do
-          @queue << Worker::PostRequest.new('/errors/', { series: 'test', values: { id: 1 }, timestamp: 0})
-          @queue << Worker::PostRequest.new('/errors/', { series: 'test', values: { id: 1 }, timestamp: 1})
+          @queue << Worker::PostRequest.new({ url: '/errors/' }, { series: 'test', values: { id: 1 }, timestamp: 0})
+          @queue << Worker::PostRequest.new({ url: '/errors/', database: 'db2' }, { series: 'test', values: { id: 1 }, timestamp: 1})
 
           subject
 
           expect(WebMock).to have_requested(:post, %r{/write}).with(body: 'test id=1i 0')
-          expect(WebMock).to have_requested(:post, %r{/write}).with(body: 'test id=1i 1')
+          expect(WebMock).to have_requested(:post, %r{/write\?db=db2}).with(body: 'test id=1i 1')
         end
       end
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: influx_reporter
 version: !ruby/object:Gem::Version
-  version: 1.1.0
+  version: 1.2.0
 platform: ruby
 authors:
 - Kacper Kawecki
@@ -94,7 +94,6 @@ files:
 - lib/influx_reporter/error_message/user.rb
 - lib/influx_reporter/event_message.rb
 - lib/influx_reporter/filter.rb
-- lib/influx_reporter/http_client.rb
 - lib/influx_reporter/influx_db_client.rb
 - lib/influx_reporter/injections.rb
 - lib/influx_reporter/injections/json.rb

data/lib/influx_reporter/http_client.rb

@@ -1,140 +0,0 @@
-# frozen_string_literal: true
-
-require 'uri'
-require 'net/http'
-require 'json'
-
-module InfluxReporter
-  # @api private
-  class HttpClient
-    include Logging
-
-    USER_AGENT = "influx_reporter-ruby/#{InfluxReporter::VERSION}"
-
-    attr_reader :state
-    attr_reader :adapter
-
-    def initialize(config)
-      @config = config
-      @adapter = HTTPAdapter.new(config)
-      @state = ClientState.new config
-    end
-
-    attr_reader :config
-
-    def post(resource, body)
-      path = abs_path(resource)
-      debug "POST #{resource}"
-
-      unless state.should_try?
-        info 'Temporarily skipping sending to InfluxReporter due to previous failure.'
-        return
-      end
-
-      body = JSON.dump(body) if body.is_a?(Hash) || body.is_a?(Array)
-
-      request = adapter.post path do |req|
-        req['Authorization'] = auth_header
-        req['Content-Type'] = 'application/json'
-        req['Content-Length'] = body.bytesize.to_s
-        req['User-Agent'] = USER_AGENT
-        req.body = body
-      end
-
-      begin
-        response = adapter.perform_request request
-        unless response.code.to_i.between?(200, 299)
-          raise Error, "Error from InfluxReporter server (#{response.code}): #{response.body}"
-        end
-      rescue
-        debug { JSON.parse(body).inspect }
-        @state.fail!
-        raise
-      end
-
-      @state.success!
-
-      response
-    end
-
-    private
-
-    def auth_header
-      "Bearer #{@config.secret_token}"
-    end
-
-    def abs_path(path)
-      "/api/v1/organizations/#{@config.organization_id}" \
-        "/apps/#{@config.app_id}#{path}"
-    end
-
-    def encode(event)
-      event_hash = @filter.process_event_hash(event.to_hash)
-      event_hash.to_json
-    end
-
-    class HTTPAdapter
-      def initialize(conf)
-        @config = conf
-      end
-
-      def post(path)
-        req = Net::HTTP::Post.new path
-        yield req if block_given?
-        req
-      end
-
-      def perform_request(req)
-        http.start do |http|
-          http.request req
-        end
-      end
-
-      private
-
-      def http
-        return @http if @http
-
-        http = Net::HTTP.new server_uri.host, server_uri.port
-        http.use_ssl = @config.use_ssl
-        http.read_timeout = @config.timeout
-        http.open_timeout = @config.open_timeout
-
-        @http = http
-      end
-
-      def server_uri
-        @uri ||= URI(@config.server)
-      end
-    end
-
-    class ClientState
-      def initialize(config)
-        @config = config
-        @retry_number = 0
-        @last_check = Time.now.utc
-      end
-
-      def should_try?
-        return true if @status == :online
-
-        interval = ([@retry_number, 6].min**2) * @config.backoff_multiplier
-        return true if Time.now.utc - @last_check > interval
-
-        false
-      end
-
-      def fail!
-        @status = :error
-        @retry_number += 1
-        @last_check = Time.now.utc
-      end
-
-      def success!
-        @status = :online
-        @retry_number = 0
-        @last_check = nil
-      end
-    end
-  end
-end