1 files changed, 34 insertions, 0 deletions
diff --git a/lib/gitlab/counters/legacy_counter.rb b/lib/gitlab/counters/legacy_counter.rb
new file mode 100644
index 00000000000..06951514ec3
--- /dev/null
+++ b/lib/gitlab/counters/legacy_counter.rb
@@ -0,0 +1,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