Class: Dexter::Collector

Inherits:
Object
  • Object
show all
Defined in:
lib/dexter/collector.rb

Instance Method Summary collapse

Constructor Details

#initialize(options = {}) ⇒ Collector

Returns a new instance of Collector.



3
4
5
6
7
8
9
 
# File 'lib/dexter/collector.rb', line 3

def initialize(options = {})
  @top_queries = {}
  @new_queries = Set.new
  @mutex = Mutex.new
  @min_time = options[:min_time] * 60000 # convert minutes to ms
  @min_calls = options[:min_calls]
end

Instance Method Details

#add(query, total_time, calls = 1) ⇒ Object 



11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
 
# File 'lib/dexter/collector.rb', line 11

def add(query, total_time, calls = 1)
  fingerprint =
    begin
      PgQuery.fingerprint(query)
    rescue PgQuery::ParseError
      # do nothing
    end

  return unless fingerprint

  @top_queries[fingerprint] ||= {calls: 0, total_time: 0}
  @top_queries[fingerprint][:calls] += calls
  @top_queries[fingerprint][:total_time] += total_time
  @top_queries[fingerprint][:query] = query
  @mutex.synchronize do
    @new_queries << fingerprint
  end
end

#fetch_queries ⇒ Object 



30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
 
# File 'lib/dexter/collector.rb', line 30

def fetch_queries
  new_queries = nil

  @mutex.synchronize do
    new_queries = @new_queries.dup
    @new_queries.clear
  end

  queries = []
  @top_queries.each do |k, v|
    if new_queries.include?(k) && v[:total_time] >= @min_time && v[:calls] >= @min_calls
      query = Query.new(v[:query], k)
      query.total_time = v[:total_time]
      query.calls = v[:calls]
      queries << query
    end
  end

  queries.sort_by(&:fingerprint)
end