38 lines
1.1 KiB
Ruby
38 lines
1.1 KiB
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
# rubocop: disable Style/Documentation
|
||
|
|
class Gitlab::BackgroundMigration::DropInvalidVulnerabilities
|
||
|
|
# rubocop: disable Gitlab/NamespacedClass
|
||
|
|
class Vulnerability < ActiveRecord::Base
|
||
|
|
self.table_name = "vulnerabilities"
|
||
|
|
has_many :findings, class_name: 'VulnerabilitiesFinding', inverse_of: :vulnerability
|
||
|
|
end
|
||
|
|
|
||
|
|
class VulnerabilitiesFinding < ActiveRecord::Base
|
||
|
|
self.table_name = "vulnerability_occurrences"
|
||
|
|
belongs_to :vulnerability, class_name: 'Vulnerability', inverse_of: :findings, foreign_key: 'vulnerability_id'
|
||
|
|
end
|
||
|
|
# rubocop: enable Gitlab/NamespacedClass
|
||
|
|
|
||
|
|
# rubocop: disable CodeReuse/ActiveRecord
|
||
|
|
def perform(start_id, end_id)
|
||
|
|
Vulnerability
|
||
|
|
.where(id: start_id..end_id)
|
||
|
|
.left_joins(:findings)
|
||
|
|
.where(vulnerability_occurrences: { vulnerability_id: nil })
|
||
|
|
.delete_all
|
||
|
|
|
||
|
|
mark_job_as_succeeded(start_id, end_id)
|
||
|
|
end
|
||
|
|
# rubocop: enable CodeReuse/ActiveRecord
|
||
|
|
|
||
|
|
private
|
||
|
|
|
||
|
|
def mark_job_as_succeeded(*arguments)
|
||
|
|
Gitlab::Database::BackgroundMigrationJob.mark_all_as_succeeded(
|
||
|
|
'DropInvalidVulnerabilities',
|
||
|
|
arguments
|
||
|
|
)
|
||
|
|
end
|
||
|
|
end
|