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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
# frozen_string_literal: true
module Gitlab
module BackgroundMigration
# Migrates internal_ids records for `usage: issues` from project to namespace scope.
# For project issues it will be project namespace, for group issues it will be group namespace.
class IssuesInternalIdScopeUpdater < ::Gitlab::BackgroundMigration::BatchedMigrationJob
operation_name :issues_internal_id_scope_updater
feature_category :database
ISSUES_USAGE = 0 # see Enums::InternalId#usage_resources[:issues]
scope_to ->(relation) do
relation.where(usage: ISSUES_USAGE).where.not(project_id: nil)
end
def perform
each_sub_batch do |sub_batch|
create_namespace_scoped_records(sub_batch)
delete_project_scoped_records(sub_batch)
end
end
private
def delete_project_scoped_records(sub_batch)
# There is no need to keep the project scoped issues usage as we move to scoping issues to namespace.
# Also in case we do decide to move back to scoping issues usage to project, we are better off if the
# project record is not present as that would result in overlapping IIDs because project scoped issues
# usage will have outdated IIDs left in the DB
log_info("Deleted internal_ids records", ids: sub_batch.pluck(:id))
connection.execute(
<<~SQL
DELETE FROM internal_ids WHERE id IN (#{sub_batch.select(:id).to_sql})
SQL
)
end
def create_namespace_scoped_records(sub_batch)
# Creates a corresponding namespace scoped record for every `issues` usage scoped to a project.
# On conflict it means the record was already created when a new issue is created with the
# newly namespace scoped Issue model, see Issue#has_internal_id definition. In which case to
# make sure we have the namespace_id scoped record set to the greatest of the two last_values.
created_records_ids = connection.execute(
<<~SQL
INSERT INTO internal_ids (usage, last_value, namespace_id)
SELECT #{ISSUES_USAGE}, last_value, project_namespace_id
FROM internal_ids
INNER JOIN projects ON projects.id = internal_ids.project_id
WHERE internal_ids.id IN(#{sub_batch.select(:id).to_sql})
ON CONFLICT (usage, namespace_id) WHERE namespace_id IS NOT NULL
DO UPDATE SET last_value = GREATEST(EXCLUDED.last_value, internal_ids.last_value)
RETURNING id;
SQL
)
log_info("Created/updated internal_ids records", ids: created_records_ids.field_values('id'))
end
def log_info(message, **extra)
::Gitlab::BackgroundMigration::Logger.info(migrator: self.class.to_s, message: message, **extra)
end
end
end
end
|