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

#initializer_content ⇒ Object 



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

def initializer_content
  "    # This is for tracking sql queries\n    QueryTrace.enable!\n\n  RUBY\nend\n"

#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
  "    class CreateQueryLogs < ActiveRecord::Migration[6.0]\n      def change\n        create_table :query_logs do |t|\n          t.string :query\n          t.float :duration\n          t.string :source\n          t.bigint :follow_id\n          t.integer :n_plus_one_size, default: 1\n\n          t.timestamps\n        end\n      end\n    end\n  RUBY\nend\n"

#model_content ⇒ Object 



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

def model_content
  "    class QueryLog < ApplicationRecord\n    end\n  RUBY\nend\n"

#route_content ⇒ Object 



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

def route_content
  "    if Rails.env.development?\n      get 'sql_optimizer', to: 'sql_optimizer#index'\n      get 'graph',         to: 'sql_optimizer#graph'\n    end\n\n  RUBY\nend\n"