Add latest changes from gitlab-org/gitlab@13-2-stable-ee

1 files changed, 86 insertions, 0 deletions

diff --git a/lib/gitlab/usage_data_counters/track_unique_actions.rb b/lib/gitlab/usage_data_counters/track_unique_actions.rb
new file mode 100644
index 00000000000..9fb5a29748e
--- /dev/null
+++ b/lib/gitlab/usage_data_counters/track_unique_actions.rb
@@ -0,0 +1,86 @@
+# frozen_string_literal: true
+
+module Gitlab
+  module UsageDataCounters
+    module TrackUniqueActions
+      KEY_EXPIRY_LENGTH = 29.days
+      FEATURE_FLAG = :track_unique_actions
+
+      WIKI_ACTION = :wiki_action
+      DESIGN_ACTION = :design_action
+      PUSH_ACTION = :project_action
+
+      ACTION_TRANSFORMATIONS = HashWithIndifferentAccess.new({
+        wiki: {
+          created: WIKI_ACTION,
+          updated: WIKI_ACTION,
+          destroyed: WIKI_ACTION
+        },
+        design: {
+          created: DESIGN_ACTION,
+          updated: DESIGN_ACTION,
+          destroyed: DESIGN_ACTION
+        },
+        project: {
+          pushed: PUSH_ACTION
+        }
+      }).freeze
+
+      class << self
+        def track_action(event_action:, event_target:, author_id:, time: Time.zone.now)
+          return unless Gitlab::CurrentSettings.usage_ping_enabled
+          return unless Feature.enabled?(FEATURE_FLAG)
+          return unless valid_target?(event_target)
+          return unless valid_action?(event_action)
+
+          transformed_target = transform_target(event_target)
+          transformed_action = transform_action(event_action, transformed_target)
+
+          add_event(transformed_action, author_id, time)
+        end
+
+        def count_unique_events(event_action:, date_from:, date_to:)
+          keys = (date_from.to_date..date_to.to_date).map { |date| key(event_action, date) }
+
+          Gitlab::Redis::SharedState.with do |redis|
+            redis.pfcount(*keys)
+          end
+        end
+
+        private
+
+        def transform_action(event_action, event_target)
+          ACTION_TRANSFORMATIONS.dig(event_target, event_action) || event_action
+        end
+
+        def transform_target(event_target)
+          Event::TARGET_TYPES.key(event_target)
+        end
+
+        def valid_target?(target)
+          Event::TARGET_TYPES.value?(target)
+        end
+
+        def valid_action?(action)
+          Event.actions.key?(action)
+        end
+
+        def key(event_action, date)
+          year_day = date.strftime('%G-%j')
+          "#{year_day}-{#{event_action}}"
+        end
+
+        def add_event(event_action, author_id, date)
+          target_key = key(event_action, date)
+
+          Gitlab::Redis::SharedState.with do |redis|
+            redis.multi do |multi|
+              multi.pfadd(target_key, author_id)
+              multi.expire(target_key, KEY_EXPIRY_LENGTH)
+            end
+          end
+        end
+      end
+    end
+  end
+end