BigW Consortium Gitlab

20170502101023_cleanup_namespaceless_pending_delete_projects.rb 988 Bytes
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
# This is the counterpart of RequeuePendingDeleteProjects and cleans all
# projects with `pending_delete = true` and that do not have a namespace.
class CleanupNamespacelessPendingDeleteProjects < ActiveRecord::Migration
  include Gitlab::Database::MigrationHelpers

  DOWNTIME = false

  disable_ddl_transaction!

  def up
    @offset = 0

    loop do
      ids = pending_delete_batch

16
      break if ids.empty?
17

18
      args = ids.map { |id| Array(id) }
19 20 21 22 23 24 25 26 27 28 29 30 31 32

      NamespacelessProjectDestroyWorker.bulk_perform_async(args)

      @offset += 1
    end
  end

  def down
    # noop
  end

  private

  def pending_delete_batch
Douwe Maan committed
33
    connection.exec_query(find_batch).map{ |row| row['id'].to_i }
34 35 36 37 38 39
  end

  BATCH_SIZE = 5000

  def find_batch
    projects = Arel::Table.new(:projects)
40 41 42 43 44 45
    projects.project(projects[:id])
      .where(projects[:pending_delete].eq(true))
      .where(projects[:namespace_id].eq(nil))
      .skip(@offset * BATCH_SIZE)
      .take(BATCH_SIZE)
      .to_sql
46 47
  end
end