2018-11-18 11:00:15 +05:30
|
|
|
# frozen_string_literal: true
|
|
|
|
|
|
|
|
module ResourceEvents
|
|
|
|
class ChangeLabelsService
|
|
|
|
attr_reader :resource, :user
|
|
|
|
|
|
|
|
def initialize(resource, user)
|
|
|
|
@resource, @user = resource, user
|
|
|
|
end
|
|
|
|
|
|
|
|
def execute(added_labels: [], removed_labels: [])
|
|
|
|
label_hash = {
|
|
|
|
resource_column(resource) => resource.id,
|
|
|
|
user_id: user.id,
|
2019-07-31 22:56:46 +05:30
|
|
|
created_at: resource.system_note_timestamp
|
2018-11-18 11:00:15 +05:30
|
|
|
}
|
|
|
|
|
|
|
|
labels = added_labels.map do |label|
|
|
|
|
label_hash.merge(label_id: label.id, action: ResourceLabelEvent.actions['add'])
|
|
|
|
end
|
|
|
|
labels += removed_labels.map do |label|
|
|
|
|
label_hash.merge(label_id: label.id, action: ResourceLabelEvent.actions['remove'])
|
|
|
|
end
|
|
|
|
|
|
|
|
Gitlab::Database.bulk_insert(ResourceLabelEvent.table_name, labels)
|
2018-11-20 20:47:30 +05:30
|
|
|
resource.expire_note_etag_cache
|
2018-11-18 11:00:15 +05:30
|
|
|
end
|
|
|
|
|
|
|
|
private
|
|
|
|
|
|
|
|
def resource_column(resource)
|
|
|
|
case resource
|
|
|
|
when Issue
|
|
|
|
:issue_id
|
|
|
|
when MergeRequest
|
|
|
|
:merge_request_id
|
|
|
|
else
|
|
|
|
raise ArgumentError, "Unknown resource type #{resource.class.name}"
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
2019-12-04 20:38:33 +05:30
|
|
|
|
|
|
|
ResourceEvents::ChangeLabelsService.prepend_if_ee('EE::ResourceEvents::ChangeLabelsService')
|