Add latest changes from gitlab-org/gitlab@13-7-stable-eev13.7.0-rc42

1 files changed, 40 insertions, 0 deletions

diff --git a/lib/gitlab/usage_data_counters.rb b/lib/gitlab/usage_data_counters.rb
new file mode 100644
index 00000000000..ca7699e64e1
--- /dev/null
+++ b/lib/gitlab/usage_data_counters.rb
@@ -0,0 +1,40 @@
+# frozen_string_literal: true
+
+module Gitlab
+  module UsageDataCounters
+    COUNTERS = [
+      GuestPackageEventCounter,
+      WikiPageCounter,
+      WebIdeCounter,
+      NoteCounter,
+      SnippetCounter,
+      SearchCounter,
+      CycleAnalyticsCounter,
+      ProductivityAnalyticsCounter,
+      SourceCodeCounter,
+      MergeRequestCounter,
+      DesignsCounter,
+      KubernetesAgentCounter,
+      StaticSiteEditorCounter
+    ].freeze
+
+    UsageDataCounterError = Class.new(StandardError)
+    UnknownEvent = Class.new(UsageDataCounterError)
+
+    class << self
+      def counters
+        self::COUNTERS
+      end
+
+      def count(event_name)
+        counters.each do |counter|
+          event = counter.fetch_supported_event(event_name)
+
+          return counter.count(event) if event
+        end
+
+        raise UnknownEvent, "Cannot find counter for event #{event_name}"
+      end
+    end
+  end
+end