blob: 06951514ec3f429feb233742fc984a4c8f9d84c9 (
plain)
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
|
# frozen_string_literal: true
module Gitlab
module Counters
# This class is a wrapper over ActiveRecord counter
# for attributes that have not adopted Redis-backed BufferedCounter.
class LegacyCounter
def initialize(counter_record, attribute)
@counter_record = counter_record
@attribute = attribute
@current_value = counter_record.method(attribute).call
end
def increment(amount)
updated = counter_record.class.update_counters(counter_record.id, { attribute => amount })
if updated == 1
counter_record.execute_after_commit_callbacks
@current_value += amount
end
@current_value
end
def reset!
counter_record.update!(attribute => 0)
end
private
attr_reader :counter_record, :attribute
end
end
end
|