blob: 250e0cb2369b036426b4c532f2f9ed5e7ca4fc99 (
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
|
# frozen_string_literal: true
class EnsureUniqueDebianPackages < Gitlab::Database::Migration[2.1]
BATCH_SIZE = 1_000
disable_ddl_transaction!
restrict_gitlab_migration gitlab_schema: :gitlab_main
class Package < MigrationRecord
include EachBatch
self.table_name = 'packages_packages'
enum package_type: { debian: 9 }
enum status: { pending_destruction: 4 }
end
def up
Package.distinct_each_batch(column: :project_id) do |package_projects|
project_ids = package_projects.pluck(:project_id)
duplicates = Package.debian
.not_pending_destruction
.where(project_id: project_ids)
.select('project_id, name, version, MAX(id) as last_id')
.group(:project_id, :name, :version)
.having('count(id) > 1')
loop do
duplicates.limit(BATCH_SIZE).each do |duplicate|
Package.debian
.not_pending_destruction
.where(
project_id: duplicate.project_id,
name: duplicate.name,
version: duplicate.version,
id: ..duplicate.last_id - 1
).update_all status: :pending_destruction
end
break unless duplicates.exists?
end
end
end
def down
# nothing to do
end
end
|
