gitlab-org--gitlab-foss/app/models/packages/event.rb

55 lines
1.5 KiB
Ruby

# frozen_string_literal: true
class Packages::Event < ApplicationRecord
belongs_to :package, optional: true
UNIQUE_EVENTS_ALLOWED = %i[push_package delete_package pull_package].freeze
EVENT_SCOPES = ::Packages::Package.package_types.merge(container: 1000, tag: 1001).freeze
EVENT_PREFIX = "i_package"
enum event_scope: EVENT_SCOPES
enum event_type: {
push_package: 0,
delete_package: 1,
pull_package: 2,
search_package: 3,
list_package: 4,
list_repositories: 5,
delete_repository: 6,
delete_tag: 7,
delete_tag_bulk: 8,
list_tags: 9,
cli_metadata: 10
}
enum originator_type: { user: 0, deploy_token: 1, guest: 2 }
# Remove some of the events, for now, so we don't hammer Redis too hard.
# See: https://gitlab.com/gitlab-org/gitlab/-/issues/280770
def self.event_allowed?(event_type)
return true if UNIQUE_EVENTS_ALLOWED.include?(event_type.to_sym)
false
end
# counter names for unique user tracking (for MAU)
def self.unique_counters_for(event_scope, event_type, originator_type)
return [] unless event_allowed?(event_type)
return [] if originator_type.to_s == 'guest'
["#{EVENT_PREFIX}_#{event_scope}_#{originator_type}"]
end
# total counter names for tracking number of events
def self.counters_for(event_scope, event_type, originator_type)
return [] unless event_allowed?(event_type)
[
"#{EVENT_PREFIX}_#{event_type}",
"#{EVENT_PREFIX}_#{event_type}_by_#{originator_type}",
"#{EVENT_PREFIX}_#{event_scope}_#{event_type}"
]
end
end