blob: 0a5dfd72392bf5d08028bb7cca774b5c40d1d9a6 (
plain)
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
|
# frozen_string_literal: true
class MigrateLicenseManagementArtifactsToLicenseScanning < ActiveRecord::Migration[6.0]
DOWNTIME = false
LICENSE_MANAGEMENT_FILE_TYPE = 10
LICENSE_SCANNING_FILE_TYPE = 101
disable_ddl_transaction!
class JobArtifact < ActiveRecord::Base
include EachBatch
self.table_name = 'ci_job_artifacts'
end
# We're updating file_type of ci artifacts from license_management to license_scanning
# But before that we need to delete "rogue" artifacts for CI builds that have associated with them
# both license_scanning and license_management artifacts. It's an edge case and usually, we don't have
# such builds in the database.
def up
return unless Gitlab.ee?
JobArtifact
.where("file_type = 10 OR file_type = 101")
.each_batch(column: :job_id, of: 1000) do |relation|
min, max = relation.pluck('MIN(job_id)', 'MAX(job_id)').flatten
ActiveRecord::Base.connection.execute <<~SQL
WITH ci_job_artifacts_with_row_number as (
SELECT job_id, id, ROW_NUMBER() OVER (PARTITION BY job_id ORDER BY id ASC) as row_number
FROM ci_job_artifacts
WHERE (file_type = #{LICENSE_SCANNING_FILE_TYPE} OR file_type = #{LICENSE_MANAGEMENT_FILE_TYPE})
AND job_id >= #{Integer(min)} AND job_id < #{Integer(max)}
)
DELETE FROM ci_job_artifacts
WHERE ci_job_artifacts.id IN (SELECT id from ci_job_artifacts_with_row_number WHERE ci_job_artifacts_with_row_number.row_number > 1)
SQL
end
JobArtifact.where(file_type: LICENSE_MANAGEMENT_FILE_TYPE).each_batch(column: :job_id, of: 1000) do |relation|
relation.update_all(file_type: LICENSE_SCANNING_FILE_TYPE)
end
end
def down
# no-op
# we're deleting duplicating artifacts and updating file_type for license_management artifacts
end
end
|
