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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
# frozen_string_literal: true
module Projects
class UpdateRepositoryStorageService
include UpdateRepositoryStorageMethods
delegate :project, to: :repository_storage_move
private
def track_repository(_destination_storage_name)
# Connect project to pool repository from the new shard
project.swap_pool_repository!
# Connect project to the repository from the new shard
project.track_project_repository
# Link repository from the new shard to pool repository from the new shard
project.link_pool_repository if replicate_object_pool_on_move_ff_enabled?
end
def mirror_repositories
if project.repository_exists?
mirror_repository(type: Gitlab::GlRepository::PROJECT)
end
if project.wiki.repository_exists?
mirror_repository(type: Gitlab::GlRepository::WIKI)
end
if project.design_repository.exists?
mirror_repository(type: ::Gitlab::GlRepository::DESIGN)
end
end
def mirror_object_pool(destination_storage_name)
return unless replicate_object_pool_on_move_ff_enabled?
return unless project.repository_exists?
pool_repository = project.pool_repository
return unless pool_repository
# If pool repository already exists, then we will link the moved project repository to it
return if pool_repository_exists_for?(shard_name: destination_storage_name, pool_repository: pool_repository)
target_pool_repository = create_pool_repository_for!(
shard_name: destination_storage_name,
pool_repository: pool_repository
)
checksum, new_checksum = replicate_object_pool_repository(from: pool_repository, to: target_pool_repository)
if checksum != new_checksum
raise Error,
format(s_('UpdateRepositoryStorage|Failed to verify %{type} repository checksum from %{old} to %{new}'),
type: 'object_pool', old: checksum, new: new_checksum)
end
end
def remove_old_paths
super
if project.wiki.repository_exists?
Gitlab::Git::Repository.new(
source_storage_name,
"#{project.wiki.disk_path}.git",
nil,
nil
).remove
end
if project.design_repository.exists?
Gitlab::Git::Repository.new(
source_storage_name,
"#{project.design_repository.disk_path}.git",
nil,
nil
).remove
end
end
def pool_repository_exists_for?(shard_name:, pool_repository:)
PoolRepository.by_source_project_and_shard_name(
pool_repository.source_project,
shard_name
).exists?
end
def create_pool_repository_for!(shard_name:, pool_repository:)
# Set state `ready` because we manually replicate object pool
PoolRepository.create!(
shard: Shard.by_name(shard_name),
source_project: pool_repository.source_project,
disk_path: pool_repository.disk_path,
state: 'ready'
)
end
def replicate_object_pool_repository(from:, to:)
old_object_pool = from.object_pool
new_object_pool = to.object_pool
checksum = old_object_pool.repository.checksum
new_object_pool.repository.replicate(old_object_pool.repository)
new_checksum = new_object_pool.repository.checksum
[checksum, new_checksum]
end
def replicate_object_pool_on_move_ff_enabled?
Feature.enabled?(:replicate_object_pool_on_move, project)
end
end
end
|
