diff options
Diffstat (limited to 'lib/gitlab/usage_data_counters/jetbrains_plugin_activity_unique_counter.rb')
-rw-r--r-- | lib/gitlab/usage_data_counters/jetbrains_plugin_activity_unique_counter.rb | 28 |
1 files changed, 28 insertions, 0 deletions
diff --git a/lib/gitlab/usage_data_counters/jetbrains_plugin_activity_unique_counter.rb b/lib/gitlab/usage_data_counters/jetbrains_plugin_activity_unique_counter.rb new file mode 100644 index 00000000000..f3d5be5e28f --- /dev/null +++ b/lib/gitlab/usage_data_counters/jetbrains_plugin_activity_unique_counter.rb @@ -0,0 +1,28 @@ +# frozen_string_literal: true + +module Gitlab + module UsageDataCounters + module JetBrainsPluginActivityUniqueCounter + JETBRAINS_API_REQUEST_ACTION = 'i_code_review_user_jetbrains_api_request' + JETBRAINS_USER_AGENT_REGEX = /\Agitlab-jetbrains-plugin/.freeze + + class << self + def track_api_request_when_trackable(user_agent:, user:) + user_agent&.match?(JETBRAINS_USER_AGENT_REGEX) && track_unique_action_by_user(JETBRAINS_API_REQUEST_ACTION, user) + end + + private + + def track_unique_action_by_user(action, user) + return unless user + + track_unique_action(action, user.id) + end + + def track_unique_action(action, value) + Gitlab::UsageDataCounters::HLLRedisCounter.track_usage_event(action, value) + end + end + end + end +end |