0fc9f9d3e7
DB schema generated by a migration may look different in rails 4 and 5 (because rails 5 may use different default values). For this reason it's important to explicitly set for which rails version a migration was written for. See https://stackoverflow.com/questions/35929869/activerecordmigration-deprecation-warning-asks-for-rails-version-but-im-no/35930912#35930912
49 lines
1,020 B
Ruby
49 lines
1,020 B
Ruby
# See http://doc.gitlab.com/ce/development/migration_style_guide.html
|
|
# for more information on how to write migrations for GitLab.
|
|
|
|
class RequeuePendingDeleteProjects < ActiveRecord::Migration[4.2]
|
|
include Gitlab::Database::MigrationHelpers
|
|
|
|
DOWNTIME = false
|
|
|
|
def up
|
|
admin = User.find_by(admin: true)
|
|
return unless admin
|
|
|
|
@offset = 0
|
|
|
|
loop do
|
|
ids = pending_delete_batch
|
|
|
|
break if ids.rows.count.zero?
|
|
|
|
args = ids.map { |id| [id['id'], admin.id, {}] }
|
|
|
|
Sidekiq::Client.push_bulk('class' => "ProjectDestroyWorker", 'args' => args)
|
|
|
|
@offset += 1
|
|
end
|
|
end
|
|
|
|
def down
|
|
# noop
|
|
end
|
|
|
|
private
|
|
|
|
def pending_delete_batch
|
|
connection.exec_query(find_batch)
|
|
end
|
|
|
|
BATCH_SIZE = 5000
|
|
|
|
def find_batch
|
|
projects = Arel::Table.new(:projects)
|
|
projects.project(projects[:id])
|
|
.where(projects[:pending_delete].eq(true))
|
|
.where(projects[:namespace_id].not_eq(nil))
|
|
.skip(@offset * BATCH_SIZE)
|
|
.take(BATCH_SIZE)
|
|
.to_sql
|
|
end
|
|
end
|