initial commit

.gitignore

@@ -0,0 +1,10 @@
+/.bundle/
+/.yardoc
+/Gemfile.lock
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/
+.idea

.rspec

@@ -0,0 +1,2 @@
+--format documentation
+--color

.travis.yml

@@ -0,0 +1,4 @@
+language: ruby
+rvm:
+  - 2.2.2
+before_install: gem install bundler -v 1.11.2

Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in lograge-sql.gemspec
+gemspec

README.md

@@ -0,0 +1,26 @@
+# Lograge::Sql
+
+Lograge::Sql is an extension to the famous [Lograge](https://github.com/roidrage/lograge) gem, which adds SQL queries to the Lograge Event and disable default ActiveRecord logging.
+This is extremely useful if you're using Lograge together with the ELK stack.
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'lograge-sql'
+```
+
+## Usage
+
+In order to enable SQL logging in your application, you'll simply need to add this on top of your lograge initializer:
+
+```ruby
+# config/initializers/lograge
+require 'lograge/sql/extension'
+```
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/iMacTia/lograge-sql.
+

Rakefile

@@ -0,0 +1,6 @@
+require 'bundler/gem_tasks'
+require 'rspec/core/rake_task'
+
+RSpec::Core::RakeTask.new(:spec)
+
+task :default => :spec

bin/console

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+
+require 'bundler/setup'
+require 'lograge/sql'
+
+# You can add fixtures and/or initialization code here to make experimenting
+# with your gem easier. You can also use a different console, if you like.
+
+# (If you use this, don't forget to add pry to your Gemfile!)
+# require "pry"
+# Pry.start
+
+require 'irb'
+IRB.start

bin/setup

@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+set -euo pipefail
+IFS=$'\n\t'
+set -vx
+
+bundle install
+
+# Do any other automated setup that you need to do here

lib/lograge/sql.rb

@@ -0,0 +1,7 @@
+require 'lograge/sql/version'
+
+module Lograge
+  module Sql
+
+  end
+end

lib/lograge/sql/extension.rb

@@ -0,0 +1,30 @@
+module Lograge
+  class RequestLogSubscriber
+    def sql(event)
+      ActiveRecord::LogSubscriber.runtime += event.duration
+      return if event.payload[:name] == 'SCHEMA'
+      Thread.current[:lograge_sql_queries] ||= []
+      Thread.current[:lograge_sql_queries] << ("#{event.payload[:name]} (#{event.duration.to_f.round(2)}) #{event.payload[:sql]}")
+    end
+
+    alias_method :old_extract_request, :extract_request
+    def extract_request(event, payload)
+      data = old_extract_request(event, payload)
+      data.merge!(extract_sql_queries)
+    end
+
+    def extract_sql_queries
+      sql_queries = Thread.current[:lograge_sql_queries]
+      return {} unless sql_queries
+
+      Thread.current[:lograge_sql_queries] = nil
+      { sql_queries: %('#{sql_queries.join("\n")}') }
+    end
+  end
+end
+
+ActiveSupport::LogSubscriber.log_subscribers.each do |subscriber|
+  Lograge.unsubscribe(:active_record, subscriber) if subscriber.is_a?(ActiveRecord::LogSubscriber)
+end
+
+Lograge::RequestLogSubscriber.attach_to :active_record

lib/lograge/sql/version.rb

@@ -0,0 +1,5 @@
+module Lograge
+  module Sql
+    VERSION = '0.1.0'
+  end
+end

lograge-sql.gemspec

@@ -0,0 +1,27 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'lograge/sql/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = 'lograge-sql'
+  spec.version       = Lograge::Sql::VERSION
+  spec.authors       = ['Mattia Giuffrida']
+  spec.email         = ['[email protected]']
+
+  spec.summary       = %q{An extension for Lograge to log SQL queries}
+  spec.description   = %q{An extension for Lograge to log SQL queries}
+  spec.homepage      = 'https://github.com/iMacTia/lograge-sql'
+
+  spec.files         = `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  spec.bindir        = 'exe'
+  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.require_paths = ['lib']
+
+  spec.add_runtime_dependency 'lograge', '~> 0.4'
+  spec.add_runtime_dependency 'activerecord', '>= 4'
+
+  spec.add_development_dependency 'bundler', '~> 1.11'
+  spec.add_development_dependency 'rake', '~> 10.0'
+  spec.add_development_dependency 'rspec', '~> 3.0'
+end

spec/lograge/sql_spec.rb

@@ -0,0 +1,7 @@
+require 'spec_helper'
+
+describe Lograge::Sql do
+  it 'has a version number' do
+    expect(Lograge::Sql::VERSION).not_to be nil
+  end
+end

spec/spec_helper.rb

@@ -0,0 +1,2 @@
+$LOAD_PATH.unshift File.expand_path('../../lib', __FILE__)
+require 'lograge/sql'