50 lines
1.5 KiB
Ruby
50 lines
1.5 KiB
Ruby
class FixProjectRecordsWithInvalidVisibility < ActiveRecord::Migration
|
|
include Gitlab::Database::MigrationHelpers
|
|
|
|
BATCH_SIZE = 500
|
|
DOWNTIME = false
|
|
|
|
# This migration is idempotent and there's no sense in throwing away the
|
|
# partial result if it's interrupted
|
|
disable_ddl_transaction!
|
|
|
|
def up
|
|
projects = Arel::Table.new(:projects)
|
|
namespaces = Arel::Table.new(:namespaces)
|
|
|
|
finder_sql =
|
|
projects.
|
|
join(namespaces, Arel::Nodes::InnerJoin).
|
|
on(projects[:namespace_id].eq(namespaces[:id])).
|
|
where(projects[:visibility_level].gt(namespaces[:visibility_level])).
|
|
project(projects[:id], namespaces[:visibility_level]).
|
|
take(BATCH_SIZE).
|
|
to_sql
|
|
|
|
# Update matching rows in batches. Each batch can cause up to 3 UPDATE
|
|
# statements, in addition to the SELECT: one per visibility_level
|
|
loop do
|
|
to_update = connection.exec_query(finder_sql)
|
|
break if to_update.rows.count == 0
|
|
|
|
# row[0] is projects.id, row[1] is namespaces.visibility_level
|
|
updates = to_update.rows.each_with_object(Hash.new {|h, k| h[k] = [] }) do |row, obj|
|
|
obj[row[1]] << row[0]
|
|
end
|
|
|
|
updates.each do |visibility_level, project_ids|
|
|
updater = Arel::UpdateManager.new(ActiveRecord::Base).
|
|
table(projects).
|
|
set(projects[:visibility_level] => visibility_level).
|
|
where(projects[:id].in(project_ids))
|
|
|
|
ActiveRecord::Base.connection.exec_update(updater.to_sql, self.class.name, [])
|
|
end
|
|
end
|
|
end
|
|
|
|
def down
|
|
# no-op
|
|
end
|
|
end
|