43 lines
2 KiB
Ruby
43 lines
2 KiB
Ruby
# 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
|
|
|
|
def perform
|
|
each_sub_batch(operation_name: :update_all) 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
|