debian-mirror-gitlab/app/services/ci/runners/stale_managers_cleanup_service.rb

# frozen_string_literal: true

module Ci
  module Runners
    class StaleManagersCleanupService
      MAX_DELETIONS = 1000

      def execute
        ServiceResponse.success(payload: {
          # the `stale` relationship can return duplicates, so we don't try to return a precise count here
          deleted_managers: delete_stale_runner_managers > 0
        })
      end

      private

      def delete_stale_runner_managers
        total_deleted_count = 0
        loop do
          sub_batch_limit = [100, MAX_DELETIONS].min

          # delete_all discards part of the `stale` scope query, so we expliclitly wrap it with a SELECT as a workaround
          deleted_count = Ci::RunnerManager.id_in(Ci::RunnerManager.stale.limit(sub_batch_limit)).delete_all
          total_deleted_count += deleted_count

          break if deleted_count == 0 || total_deleted_count >= MAX_DELETIONS
        end

        total_deleted_count
      end
    end
  end
end