blob: 6520cd63711cbb1386284e0c03c5de48d4338f77 (
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
|
# frozen_string_literal: true
module Gitlab
module BackgroundMigration
# Backfill projectfeatures.package_registry_access_level depending on projects.packages_enabled
class BackfillProjectFeaturePackageRegistryAccessLevel < ::Gitlab::BackgroundMigration::BatchedMigrationJob
FEATURE_DISABLED = 0 # ProjectFeature::DISABLED
FEATURE_PRIVATE = 10 # ProjectFeature::PRIVATE
FEATURE_ENABLED = 20 # ProjectFeature::ENABLED
FEATURE_PUBLIC = 30 # ProjectFeature::PUBLIC
PROJECT_PRIVATE = 0 # Gitlab::VisibilityLevel::PRIVATE
PROJECT_INTERNAL = 10 # Gitlab::VisibilityLevel::INTERNAL
PROJECT_PUBLIC = 20 # Gitlab::VisibilityLevel::PUBLIC
# Migration only version of ProjectFeature table
class ProjectFeature < ::ApplicationRecord
self.table_name = 'project_features'
end
operation_name :update_all
def perform
each_sub_batch do |sub_batch|
ProjectFeature.connection.execute(
<<~SQL
UPDATE project_features pf
SET package_registry_access_level = (CASE p.packages_enabled
WHEN true THEN (CASE p.visibility_level
WHEN #{PROJECT_PUBLIC} THEN #{FEATURE_PUBLIC}
WHEN #{PROJECT_INTERNAL} THEN #{FEATURE_ENABLED}
WHEN #{PROJECT_PRIVATE} THEN #{FEATURE_PRIVATE}
END)
WHEN false THEN #{FEATURE_DISABLED}
ELSE #{FEATURE_DISABLED}
END)
FROM projects p
WHERE pf.project_id = p.id AND
pf.project_id BETWEEN #{start_id} AND #{end_id}
SQL
)
end
end
end
end
end
|
