1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55
|
# frozen_string_literal: true
module Ml
class DestroyModelService
def initialize(model, user)
@model = model
@user = user
end
def execute
package_deletion_result = ::Packages::MarkPackagesForDestructionService.new(
packages: @model.all_packages,
current_user: @user
).execute
return packages_not_deleted(package_deletion_result.message) if package_deletion_result.error?
return error unless @model.destroy
audit_destroy_event(@model)
success
end
private
def success
ServiceResponse.success(payload: payload)
end
def error
ServiceResponse.error(message: @model.errors.full_messages, payload: payload)
end
def packages_not_deleted(error_message)
ServiceResponse.error(message: error_message, payload: payload)
end
def payload
{ model: @model }
end
def audit_destroy_event(model)
audit_context = {
name: 'ml_model_destroyed',
author: @user,
scope: model.project,
target: model,
message: "MlModel #{model.name} destroyed"
}
::Gitlab::Audit::Auditor.audit(audit_context)
end
end
end
|