1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
require "pg"
require "active_record"
module RspecProfiling
module Collectors
class PSQL
def self.install
new.install
end
def self.uninstall
new.uninstall
end
def self.reset
new.results.destroy_all
end
def initialize
RspecProfiling.config.db_path ||= 'rspec_profiling'
establish_connection
end
def install
return if prepared?
connection.create_table(table) do |t|
t.string :branch, index: true
t.string :commit_hash, index: true
t.datetime :date, index: true
t.text :file, index: true
t.integer :line_number, index: true
t.text :description
t.decimal :time, index: true
t.string :status, index: true
t.text :exception
t.integer :query_count, index: true
t.decimal :query_time, index:true
t.integer :request_count, index: true
t.decimal :request_time, index: true
t.timestamps null: true
end
end
def uninstall
connection.drop_table(table)
end
def insert(attributes)
results.create!(attributes.except(:created_at, :events, :event_counts, :event_times, :event_events))
end
def results
@results ||= begin
establish_connection
Result.table_name = table
Result.attr_protected if Result.respond_to?(:attr_protected)
Result
end
end
private
def prepared?
if active_record5_or_up?
connection.data_source_exists?(table)
else
connection.table_exists?(table)
end
end
def active_record5_or_up?
ActiveRecord::VERSION::STRING[0].to_i >= 5
end
def connection
@connection ||= results.connection
end
def establish_connection
begin
PG.connect(dbname: 'postgres').exec("CREATE DATABASE #{database}")
rescue PG::DuplicateDatabase
# no op
end
Result.establish_connection(
:adapter => 'postgresql',
:database => database
)
end
def table
RspecProfiling.config.table_name
end
def database
RspecProfiling.config.db_path
end
class Result < ActiveRecord::Base
def to_s
[description, location].join(" - ")
end
private
def location
[file, line_number].join(":")
end
end
end
end
end
|
