spring_onion 0.1.1 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9029f4da9a3fa2875d738d9bd505fef3784ac40987453f30974eb3ce1d226bf8
-  data.tar.gz: 53eaebd2f28fa3257a9477e2fc498f08be2e944375a9316a1d746c58f6b37f4e
+  metadata.gz: f81f2b07f2a586b8e02444106908d6d8eb8631d772db5bff80dd0fa6ee72b550
+  data.tar.gz: 0e23da7945c119c5c5dcf90f2ead4985647536d0c7f3cb091b47e8f2c94ae168
 SHA512:
-  metadata.gz: 8013cf7f3086f2f0ee8abfb6261327f98f0983adc3adb1dd2945b5cc6039e68e979181f77ed95c4d87825cd32ab54cb6baf497e47c1c7424211cc27628d93d4f
-  data.tar.gz: 57cf5c24c5bf52b543c64a281b2474612ba0861f85180748d92ee26fcf420675477004d8dd73a1f5e59fa8d1297e0092746767292073a03bbd888a916a153c1a
+  metadata.gz: 115c03712806b8b97d9ff1e65f512db06e8ebc9f4e555b743d97c57b9159c58dc4f43410b7fb73db178a584425e22b403f6005ed3be6cb9a01c57b3295c686e3
+  data.tar.gz: 4aa1951677c97b2fa2b45e2eeb3d49b51543e1632537c316a9d73d775b73b95472ced9074e06adc32d4c523bef5f069b9c4826c1f4bdf0a1e638e61d458dc2e1

data/Gemfile.lock

@@ -1,8 +1,9 @@
 PATH
   remote: .
   specs:
-    spring_onion (0.1.1)
+    spring_onion (0.2.1)
       activerecord
+      coderay
       mysql2
 
 GEM
@@ -24,6 +25,7 @@ GEM
       rake
       thor (>= 0.14.0)
     ast (2.4.1)
+    coderay (1.1.3)
     concurrent-ruby (1.1.6)
     diff-lcs (1.4.4)
     i18n (1.8.3)

data/README.md

@@ -39,8 +39,8 @@ SpringOnion.source_filter_re = //
 
 class Employee < ActiveRecord::Base; end
 
-Employee.all.to_a.count
-#=> SpringOnion	INFO	2020-07-18 01:53:27 +0900	{"sql":"SELECT `employees`.* FROM `employees`","explain":[{"line":1,"select_type":"SIMPLE","table":"employees","partitions":null,"type":"ALL","possible_keys":null,"key":null,"key_len":null,"ref":null,"rows":298936,"filtered":100.0,"Extra":null}],"violations":{"line 1":["slow_type"]},"backtrace":["/foo/bar/zoo/baz.rb:18:in `\u003ctop (required)\u003e'"]}
+p Employee.all.to_a.count
+#=> SpringOnion	INFO	2020-07-18 01:53:27 +0900	{"sql":"SELECT `employees`.* FROM `employees`","explain":[{"line":1,"select_type":"SIMPLE","table":"employees","partitions":null,"type":"ALL","possible_keys":null,"key":null,"key_len":null,"ref":null,"rows":298936,"filtered":100.0,"Extra":null}],"warnings":{"line 1":["slow_type"]},"backtrace":["/foo/bar/zoo/baz.rb:18:in `\u003ctop (required)\u003e'"]}
 #=> 300024
 ```
 
@@ -65,7 +65,7 @@ Employee.all.to_a.count
             "Extra": null
         }
     ],
-    "violations": {
+    "warnings": {
         "line 1": [
             "slow_type"
         ]

data/lib/spring_onion.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 
 require 'logger'
+require 'coderay'
 require 'active_support'
 
 require 'spring_onion/config'

data/lib/spring_onion/config.rb

@@ -17,7 +17,7 @@ module SpringOnion
     'Using temporary'
   )
 
-  @violations = {
+  @warnings = {
     slow_select_type: ->(exp) { SLOW_SELECT_TYPE_RE =~ exp['select_type'] },
     slow_type: ->(exp) { SLOW_TYPE_RE =~ exp['type'] },
     slow_possible_keys: ->(exp) { SLOW_POSSIBLE_KEYS_RE =~ exp['possible_keys'] },
@@ -42,22 +42,21 @@ module SpringOnion
     @ignore_sql_filter_re !~ sql && @sql_filter_re =~ sql
   end
 
-  @source_filter_re = Regexp.union(
-    [%r{/app/}].tap do |ary|
-      re = ENV['SPRING_ONION_SOURCE_FILTER_RE']
-      ary << Regexp.new(re) if re
-    end
-  )
+  @source_filter_re = ENV['SPRING_ONION_SOURCE_FILTER_RE'].yield_self do |re|
+    re ? Regexp.new(re) : %r{/app/}
+  end
 
   @ignore_source_filter_re = Regexp.union(
-    [RbConfig::TOPDIR, *Gem.path].tap do |ary|
+    [RbConfig::TOPDIR, *Gem.path, '/.rbenv/versions/'].tap do |ary|
       re = ENV['SPRING_ONION_IGNORE_SOURCE_FILTER_RE']
       ary << Regexp.new(re) if re
     end
   )
 
   @source_filter = lambda do |backtrace_lines|
-    backtrace_lines.grep_v(@ignore_source_filter_re).grep(@source_filter_re)
+    backtrace_lines = backtrace_lines.grep_v(@ignore_source_filter_re)
+    idx = backtrace_lines.index { |l| @source_filter_re =~ l }
+    idx ? backtrace_lines.slice(idx..-1) : []
   end
 
   @logger = Logger.new($stdout).tap do |logger|
@@ -67,14 +66,18 @@ module SpringOnion
   end
 
   @trace_len = 3
+  @json_pretty = (/\A(1|true)\z/i =~ ENV['SPRING_ONION_JSON_PRETTY'])
+  @color = /\A(1|true)\z/i =~ ENV.fetch('SPRING_ONION_COLOR', $stdout.tty?.to_s)
 
   class << self
     attr_accessor :enabled,
                   :connection,
-                  :violations,
+                  :warnings,
                   :sql_filter_re, :ignore_sql_filter_re, :sql_filter,
                   :source_filter_re, :ignore_source_filter_re, :source_filter,
                   :logger,
-                  :trace_len
+                  :trace_len,
+                  :json_pretty,
+                  :color
   end
 end

data/lib/spring_onion/explainer.rb

@@ -30,25 +30,34 @@ module SpringOnion
     end
 
     def _validate(exp, sql, trace)
-      violations = SpringOnion.violations
-      violation_names_by_line = {}
+      warnings = SpringOnion.warnings
+      warning_names_by_line = {}
 
       exp.each_with_index do |row, i|
-        violation_names = violations.select do |_name, validator|
+        warning_names = warnings.select do |_name, validator|
           validator.call(row)
         end.keys
 
-        violation_names_by_line["line #{i + 1}"] = violation_names unless violation_names.empty?
+        warning_names_by_line["line #{i + 1}"] = warning_names unless warning_names.empty?
       end
 
-      return if violation_names_by_line.empty?
+      return if warning_names_by_line.empty?
 
-      SpringOnion.logger.info({
+      h = {
         sql: sql,
         explain: exp.each_with_index.map { |r, i| { line: i + 1 }.merge(r) },
-        violations: violation_names_by_line,
+        warnings: warning_names_by_line,
         backtrace: trace.slice(0, SpringOnion.trace_len),
-      }.to_json)
+      }
+
+      line = if SpringOnion.json_pretty
+               JSON.pretty_generate(h)
+             else
+               JSON.dump(h)
+             end
+
+      line = CodeRay.scan(line, :json).terminal if SpringOnion.color
+      SpringOnion.logger.info(line)
     end
   end
 end

data/lib/spring_onion/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module SpringOnion
-  VERSION = '0.1.1'
+  VERSION = '0.2.1'
 end

data/spring_onion.gemspec

@@ -12,7 +12,7 @@ Gem::Specification.new do |spec|
   spec.description   = 'Log MySQL queries with EXPLAIN that may be slow.'
   spec.homepage      = 'https://github.com/winebarrel/spring_onion'
   spec.license       = 'MIT'
-  spec.required_ruby_version = Gem::Requirement.new('>= 2.3.0')
+  spec.required_ruby_version = Gem::Requirement.new('>= 2.5.0')
 
   # Specify which files should be added to the gem when it is released.
   # The `git ls-files -z` loads the files in the RubyGem that have been added into git.
@@ -24,6 +24,7 @@ Gem::Specification.new do |spec|
   spec.require_paths = ['lib']
 
   spec.add_dependency 'activerecord'
+  spec.add_dependency 'coderay'
   spec.add_dependency 'mysql2'
   spec.add_development_dependency 'appraisal'
   spec.add_development_dependency 'bundler'

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: spring_onion
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.2.1
 platform: ruby
 authors:
 - winebarrel
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2020-07-17 00:00:00.000000000 Z
+date: 2020-07-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord
@@ -24,6 +24,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: coderay
+  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: mysql2
   requirement: !ruby/object:Gem::Requirement
@@ -142,7 +156,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: 2.3.0
+      version: 2.5.0
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="