Class: Twingly::Metrics::ExponentiallyDecayingSample

Inherits:
Object
  • Object
show all
Defined in:
lib/twingly/metrics/exponentially_decaying_sample.rb

Constant Summary collapse

RESCALE_THRESHOLD =

1 hour

60 * 60

Instance Method Summary collapse

Constructor Details

#initialize(reservoir_size, alpha, values = nil) ⇒ ExponentiallyDecayingSample

Returns a new instance of ExponentiallyDecayingSample.



12
13
14
15
16
17
18
19
20
 
# File 'lib/twingly/metrics/exponentially_decaying_sample.rb', line 12

def initialize(reservoir_size, alpha, values = nil)
  @values = values || ConcurrentRedBlackTree.new
  @count = Atomic.new(0)
  @next_scale_time = Atomic.new(0)
  @alpha = alpha
  @reservoir_size = reservoir_size
  @mutex = Mutex.new
  clear
end

Instance Method Details

#clear ⇒ Object 



22
23
24
25
26
27
28
29
 
# File 'lib/twingly/metrics/exponentially_decaying_sample.rb', line 22

def clear
  @mutex.synchronize do
    @values.clear
    @count.value = 0
    @next_scale_time.value = Time.now + RESCALE_THRESHOLD
    @start_time = Time.now
  end
end

#rescale(now, next_time) ⇒ Object

rubocop:disable Metrics/AbcSize,Metrics/MethodLength



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
 
# File 'lib/twingly/metrics/exponentially_decaying_sample.rb', line 79

def rescale(now, next_time) # rubocop:disable Metrics/AbcSize,Metrics/MethodLength
  return unless @next_scale_time.compare_and_swap(next_time, now + RESCALE_THRESHOLD)

  @mutex.synchronize do
    old_start_time = @start_time
    @start_time = Time.now
    @values.keys.each do |key|
      value = @values.delete(key)
      new_key = key * Math.exp(-@alpha * (@start_time - old_start_time))

      if key.nan?
        warn "ExponentiallyDecayingSample found a key of NaN. " \
             "old_start_time: #{old_start_time.to_f} start_time: #{@start_time.to_f}"
        next
      end

      if new_key.nan?
        warn "ExponentiallyDecayingSample found a new_key of NaN. " \
             "key: #{key} old_start_time: #{old_start_time.to_f} start_time: #{@start_time.to_f}"
        next
      end

      @values[new_key] = value
    end
  end
end

#size ⇒ Object 



31
32
33
34
 
# File 'lib/twingly/metrics/exponentially_decaying_sample.rb', line 31

def size
  count = @count.value
  [count, @reservoir_size].min
end

#snapshot ⇒ Object 



36
37
38
39
40
 
# File 'lib/twingly/metrics/exponentially_decaying_sample.rb', line 36

def snapshot
  @mutex.synchronize do
    Snapshot.new(@values.values)
  end
end

#update(value, timestamp = Time.now) ⇒ Object

rubocop:disable Metrics/AbcSize,Metrics/CyclomaticComplexity,Metrics/MethodLength,Metrics/PerceivedComplexity



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
 
# File 'lib/twingly/metrics/exponentially_decaying_sample.rb', line 42

def update(value, timestamp = Time.now) # rubocop:disable Metrics/AbcSize,Metrics/CyclomaticComplexity,Metrics/MethodLength,Metrics/PerceivedComplexity
  @mutex.synchronize do
    priority = weight(timestamp - @start_time) / rand
    priority = Float::MAX if priority.infinite?
    new_count = @count.update { |v| v + 1 }

    if priority.nan?
      warn "ExponentiallyDecayingSample found priority of NaN. " \
           "timestamp: #{timestamp.to_f} start_time: #{@start_time.to_f}"
      return
    end

    if new_count <= @reservoir_size
      @values[priority] = value
    else
      first_priority = @values.first[0]
      if first_priority < priority
        unless @values[priority] # rubocop:disable Style/SoleNestedConditional
          @values[priority] = value

          first_priority = @values.first[0] until @values.delete(first_priority) # rubocop:disable Metrics/BlockNesting
        end
      end
    end
  end

  now = Time.new
  next_time = @next_scale_time.value
  return unless now >= next_time

  rescale(now, next_time)
end

#weight(time) ⇒ Object 



75
76
77
 
# File 'lib/twingly/metrics/exponentially_decaying_sample.rb', line 75

def weight(time)
  Math.exp(@alpha * time)
end