Class: Gitlab::BackgroundMigration::BackfillOccurrenceIdToVulnerabilityAssociations

Inherits:
BatchedMigrationJob
  • Object
show all
Defined in:
lib/gitlab/background_migration/backfill_occurrence_id_to_vulnerability_associations.rb

Constant Summary

Constants inherited from BatchedMigrationJob

Gitlab::BackgroundMigration::BatchedMigrationJob::DEFAULT_FEATURE_CATEGORY, Gitlab::BackgroundMigration::BatchedMigrationJob::MINIMUM_PAUSE_MS

Constants included from Database::DynamicModelHelpers

Database::DynamicModelHelpers::BATCH_SIZE

Instance Method Summary collapse

Methods inherited from BatchedMigrationJob

#batch_metrics, cursor, cursor?, cursor_columns, feature_category, #filter_batch, generic_instance, health_context_tables, #initialize, job_arguments, job_arguments_count, operation_name, scope_to, tables_to_check_for_vacuum

Methods included from Database::DynamicModelHelpers

define_batchable_model, #each_batch, #each_batch_range

Constructor Details

This class inherits a constructor from Gitlab::BackgroundMigration::BatchedMigrationJob

Instance Method Details

#perform ⇒ Object 



8
9
10
11
12
 
# File 'lib/gitlab/background_migration/backfill_occurrence_id_to_vulnerability_associations.rb', line 8

def perform
  each_sub_batch do |sub_batch|
    connection.exec_update(update_sql(sub_batch))
  end
end

#update_sql(sub_batch) ⇒ Object 



14
15
16
17
18
19
20
21
22
23
24
25
26
27
 
# File 'lib/gitlab/background_migration/backfill_occurrence_id_to_vulnerability_associations.rb', line 14

def update_sql(sub_batch)
  <<~SQL
  UPDATE
    #{batch_table}
  SET
    vulnerability_occurrence_id = vulnerabilities.finding_id
  FROM
    vulnerabilities
  WHERE
    vulnerabilities.id IN (#{sub_batch.select(:vulnerability_id).to_sql}) AND
    vulnerabilities.id = #{batch_table}.vulnerability_id AND
    #{batch_table}.vulnerability_occurrence_id IS NULL
  SQL
end