35 lines
854 B
Ruby
35 lines
854 B
Ruby
|
# 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
|