gitlab-org--gitlab-foss/lib/gitlab/background_migration/drop_invalid_vulnerabilities.rb

37 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