diff options
Diffstat (limited to 'app/models/packages/event.rb')
-rw-r--r-- | app/models/packages/event.rb | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/app/models/packages/event.rb b/app/models/packages/event.rb index f1d0af64ccd..959c94931ec 100644 --- a/app/models/packages/event.rb +++ b/app/models/packages/event.rb @@ -3,6 +3,7 @@ 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 enum event_scope: EVENT_SCOPES @@ -22,4 +23,20 @@ class Packages::Event < ApplicationRecord } enum originator_type: { user: 0, deploy_token: 1, guest: 2 } + + def self.allowed_event_name(event_scope, event_type, originator) + return unless event_allowed?(event_scope, event_type, originator) + + # remove `package` from the event name to avoid issues with HLLRedisCounter class parsing + "i_package_#{event_scope}_#{originator}_#{event_type.gsub(/_packages?/, "")}" + end + + # 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_scope, event_type, originator) + return false if originator.to_sym == :guest + return true if UNIQUE_EVENTS_ALLOWED.include?(event_type.to_sym) + + false + end end |