Class: SqlOptimizerGenerator

Inherits:
Rails::Generators::Base
  • Object
show all
Defined in:
lib/generators/sql_optimizer_generator.rb

Instance Method Summary collapse

Instance Method Details

#controller_content ⇒ Object 



58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
 
# File 'lib/generators/sql_optimizer_generator.rb', line 58

def controller_content
  <<~RUBY
    class SqlOptimizerController < ApplicationController

      layout false
    
      # GET /sql_optimizer
      def index
        @query_logs = QueryLog.where.not(source: nil)
        @popular_queries = @query_logs.group_by(&:query).sort_by { |_, val| val.size }.last(3)
        @max_query = @query_logs.order(duration: :desc).first
        @min_query = @query_logs.order(:duration).first
      end
    
      # GET /graph
      def graph
        @query_logs = QueryLog.where.not(source: nil)
        render json: collect_graph.to_json
      end
    
      private
    
      def collect_graph
        @query_logs.group_by(&:query).first(10).map.with_index do |query, i|
          durations = query[1].map(&:duration)
          avg_time  = durations.reduce(:+) / durations.size.to_f
          {
            index: i,
            avg_time: avg_time.round(2),
            data: {
              query: query[1].first.query,
              name:  query[1].first.source,
              size:  query[1].size
            }
          }
        end
      end
    
    end
  RUBY
end

#initializer_content ⇒ Object 



50
51
52
53
54
55
56
 
# File 'lib/generators/sql_optimizer_generator.rb', line 50

def initializer_content
  <<~RUBY
    # This is for tracking sql queries
    QueryTrace.enable!

  RUBY
end

#migrate_query_logs ⇒ Object 



4
5
6
7
8
9
10
11
12
13
 
# File 'lib/generators/sql_optimizer_generator.rb', line 4

def migrate_query_logs
  migration_number = Dir.glob("#{Rails.root}/db/migrate/*").max_by { |name| name[/\d+/].to_i }[/\d+/].to_i + 1
  create_file "db/migrate/#{migration_number}_create_query_logs.rb", migration_content
  create_file 'app/models/query_log.rb', model_content
  create_file 'app/controllers/sql_optimizer.rb', controller_content
  application(nil, env: :development) { initializer_content }
  route route_content
  rake 'db:migrate'
  puts 'Done'
end

#migration_content ⇒ Object 



15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
 
# File 'lib/generators/sql_optimizer_generator.rb', line 15

def migration_content
  <<~RUBY
    class CreateQueryLogs < ActiveRecord::Migration[6.0]
      def change
        create_table :query_logs do |t|
          t.string :query
          t.float :duration
          t.string :source
          t.bigint :follow_id
          t.integer :n_plus_one_size, default: 1

          t.timestamps
        end
      end
    end
  RUBY
end

#model_content ⇒ Object 



33
34
35
36
37
38
 
# File 'lib/generators/sql_optimizer_generator.rb', line 33

def model_content
  <<~RUBY
    class QueryLog < ApplicationRecord
    end
  RUBY
end

#route_content ⇒ Object 



40
41
42
43
44
45
46
47
48
 
# File 'lib/generators/sql_optimizer_generator.rb', line 40

def route_content
  <<~RUBY
    if Rails.env.development?
      get 'sql_optimizer', to: 'sql_optimizer#index'
      get 'graph',         to: 'sql_optimizer#graph'
    end

  RUBY
end