40 lines
1.3 KiB
Ruby
40 lines
1.3 KiB
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
class Gitlab::Seeder::ProjectFeatures
|
||
|
|
include ActionView::Helpers::NumberHelper
|
||
|
|
|
||
|
|
BATCH_SIZE = 100_000
|
||
|
|
|
||
|
|
def seed!
|
||
|
|
create_project_features!
|
||
|
|
end
|
||
|
|
|
||
|
|
def create_project_features!
|
||
|
|
Gitlab::Seeder.with_mass_insert(Project.count, "Project features") do
|
||
|
|
Project.each_batch(of: BATCH_SIZE) do |batch, index|
|
||
|
|
range = batch.pluck(Arel.sql('MIN(id)'), Arel.sql('MAX(id)')).first
|
||
|
|
count = index * batch.size
|
||
|
|
|
||
|
|
Gitlab::Seeder.log_message("Creating project features: #{count}.")
|
||
|
|
ActiveRecord::Base.connection.execute <<~SQL
|
||
|
|
INSERT INTO project_features (project_id, merge_requests_access_level, issues_access_level, wiki_access_level, pages_access_level)
|
||
|
|
SELECT
|
||
|
|
id,
|
||
|
|
#{ProjectFeature::ENABLED} AS merge_requests_access_level,
|
||
|
|
#{ProjectFeature::ENABLED} AS issues_access_level,
|
||
|
|
#{ProjectFeature::ENABLED} AS wiki_access_level,
|
||
|
|
#{ProjectFeature::ENABLED} AS pages_access_level
|
||
|
|
FROM projects
|
||
|
|
WHERE projects.id BETWEEN #{range.first} AND #{range.last}
|
||
|
|
ON CONFLICT DO NOTHING;
|
||
|
|
SQL
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
Gitlab::Seeder.quiet do
|
||
|
|
projects = Gitlab::Seeder::ProjectFeatures.new
|
||
|
|
projects.seed!
|
||
|
|
end
|