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
|
require 'redis'
require 'atomic'
# Instrument Redis time
module Peek
module RedisInstrumented
def call(*args, &block)
start = Time.now
super(*args, &block)
ensure
duration = (Time.now - start)
::Redis::Client.query_time.update { |value| value + duration }
::Redis::Client.query_count.update { |value| value + 1 }
end
end
end
class Redis::Client
prepend Peek::RedisInstrumented
class << self
attr_accessor :query_time, :query_count
end
self.query_count = Atomic.new(0)
self.query_time = Atomic.new(0)
end
module Peek
module Views
class Redis < View
def duration
::Redis::Client.query_time.value
end
def formatted_duration
ms = duration * 1000
if ms >= 1000
"%.2fms" % ms
else
"%.0fms" % ms
end
end
def calls
::Redis::Client.query_count.value
end
def results
{ :duration => formatted_duration, :calls => calls }
end
private
def setup_subscribers
# Reset each counter when a new request starts
subscribe 'start_processing.action_controller' do
::Redis::Client.query_time.value = 0
::Redis::Client.query_count.value = 0
end
end
end
end
end
|
