activerecord-honeycomb 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 19aa8e7d89c4d2ccd9bfebb896b4d36723f9d438
+  data.tar.gz: 587b8e665d2a259cb9f563a489470d5f2e6b4aa0
+SHA512:
+  metadata.gz: e765c841f53eaff2dc5c9e64c422f171a00c1528d191b62067c2e9de3a0ccb014576d916e70f4578a5a2a7683a73202714682a66d27e387e355cde7f95bda0ed
+  data.tar.gz: 320245b93cf1fb685182113a33a71f8ac65f8168dc61378dc093de1298d18d66af9944ab3d8e803e7a109f785ee7d1b54859660c857dcdcd40227f303cceae47

data/lib/active_record/connection_adapters/honeycomb_adapter.rb

@@ -0,0 +1,107 @@
+require 'active_record'
+require 'securerandom'
+
+module ActiveRecord
+  module ConnectionHandling
+    def honeycomb_connection(config)
+      real_config = config.merge(adapter: config.fetch(:real_adapter))
+
+      client = config[:honeycomb_client]
+
+      resolver = ConnectionAdapters::ConnectionSpecification::Resolver.new(Base.configurations)
+      spec = resolver.spec(real_config)
+
+      real_connection = ::ActiveRecord::Base.send(spec.adapter_method, spec.config)
+
+      unless real_connection.class.ancestors.include? ConnectionAdapters::HoneycombAdapter
+        real_connection.class.extend(Module.new do
+          private
+          define_method(:honeycomb_client) { client }
+        end) if client
+        real_connection.class.include ConnectionAdapters::HoneycombAdapter
+      end
+
+      real_connection
+    end
+  end
+
+  module ConnectionAdapters
+    module HoneycombAdapter
+      def self.included(klazz)
+        # TODO ugh clean this up
+        @_honeycomb ||= begin
+          if klazz.respond_to? :honeycomb_client, true # include private
+            klazz.send(:honeycomb_client)
+          elsif defined?(::Honeycomb.client)
+            ::Honeycomb.client
+          else
+            raise "Can't work without magic global Honeycomb.client at the moment"
+          end
+        end
+        klazz.class_exec(@_honeycomb) do |honeycomb_|
+          define_method(:honeycomb) { honeycomb_ }
+        end
+        super
+      end
+
+      def execute(sql, *args)
+        sending_honeycomb_event do |event|
+          event.add_field :sql, sql
+
+          adding_span_metadata_if_available(event, :statement) do
+            super
+          end
+        end
+      end
+
+      def exec_query(sql, *args)
+        sending_honeycomb_event do |event|
+          event.add_field :sql, sql
+
+          adding_span_metadata_if_available(event, :query) do
+            super
+          end
+        end
+      end
+
+      private
+      def sending_honeycomb_event
+        raise 'something went horribly wrong' unless honeycomb # TODO
+        event = honeycomb.event
+
+        start = Time.now
+        yield event
+      rescue Exception => e
+        if event
+          event.add_field :exception_class, e.class
+          event.add_field :exception_message, e.message
+        end
+        raise
+      ensure
+        if start && event
+          finish = Time.now
+          duration = finish - start
+          event.add_field :durationMs, duration * 1000
+          event.send
+        end
+      end
+
+      def adding_span_metadata_if_available(event, name)
+        return yield unless defined?(::Honeycomb.trace_id)
+
+        trace_id = ::Honeycomb.trace_id
+
+        event.add_field :traceId, trace_id if trace_id
+        span_id = SecureRandom.uuid
+        event.add_field :id, span_id
+        event.add_field :serviceName, 'active_record'
+        event.add_field :name, name
+
+        ::Honeycomb.with_span_id(span_id) do |parent_span_id|
+          event.add_field :parentId, parent_span_id
+          yield
+        end
+      end
+    end
+  end
+end

data/lib/active_record/honeycomb.rb

@@ -0,0 +1,15 @@
+require 'active_record'
+
+module ActiveRecord
+  module Honeycomb
+    def self.munge_config(config, client: nil, logger: nil)
+      munged = config.merge(
+        'adapter' => 'honeycomb',
+        'real_adapter' => config.fetch('adapter'),
+      )
+      munged['honeycomb_client'] = client if client
+      munged['honeycomb_logger'] = logger if logger
+      munged
+    end
+  end
+end

data/lib/active_record/honeycomb/version.rb

@@ -0,0 +1,6 @@
+module ActiveRecord
+  module Honeycomb
+    GEM_NAME = 'activerecord-honeycomb'
+    VERSION = '0.0.1'
+  end
+end

data/lib/activerecord-honeycomb.rb

@@ -0,0 +1,7 @@
+begin
+  gem 'activerecord'
+
+  require 'active_record/honeycomb'
+rescue LoadError
+  warn 'ActiveRecord not detected, not enabling activerecord-honeycomb'
+end

data/lib/activerecord-honeycomb/auto_install.rb

@@ -0,0 +1,25 @@
+module ActiveRecord
+  module Honeycomb
+    module AutoInstall
+      class << self
+        def available?(**_)
+          gem 'activerecord'
+        rescue Gem::LoadError
+          false
+        end
+
+        def auto_install!(honeycomb_client:, logger: nil)
+          require 'active_record'
+          require 'activerecord-honeycomb'
+
+          ActiveRecord::Base.extend(Module.new do
+            define_method :establish_connection do |config, *args|
+              munged_config = ActiveRecord::Honeycomb.munge_config(config, client: honeycomb_client, logger: logger)
+              super(munged_config, *args)
+            end
+          end)
+        end
+      end
+    end
+  end
+end

data/lib/activerecord/honeycomb.rb

@@ -0,0 +1 @@
+require 'active_record/honeycomb'

metadata

@@ -0,0 +1,134 @@
+--- !ruby/object:Gem::Specification
+name: activerecord-honeycomb
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Sam Stokes
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2018-05-07 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: libhoney
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: activerecord
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: bump
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: yard
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: "    TO DO *is* a description\n"
+email:
+- sam@honeycomb.io
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/active_record/connection_adapters/honeycomb_adapter.rb
+- lib/active_record/honeycomb.rb
+- lib/active_record/honeycomb/version.rb
+- lib/activerecord-honeycomb.rb
+- lib/activerecord-honeycomb/auto_install.rb
+- lib/activerecord/honeycomb.rb
+homepage: https://github.com/honeycombio/activerecord-honeycomb
+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.6.11
+signing_key: 
+specification_version: 4
+summary: Instrument your ActiveRecord queries with Honeycomb
+test_files: []