debian-mirror-gitlab/app/services/ci/job_artifacts/destroy_associations_service.rb

# frozen_string_literal: true

module Ci
  module JobArtifacts
    # This class is used by Ci::JobArtifact's FastDestroyAll implementation.
    # Ci::JobArtifact.begin_fast_destroy instantiates this service and calls #destroy_records.
    # This will set @statistics_updates instance variables.
    # The same instance is passed to Ci::JobArtifact.finalize_fast_destroy, which then calls
    # #update_statistics, using @statistics_updates set by #destroy_records.
    class DestroyAssociationsService
      BATCH_SIZE = 100

      def initialize(job_artifacts_relation)
        @job_artifacts_relation = job_artifacts_relation
        @statistics_updates = {}
      end

      def destroy_records
        @job_artifacts_relation.each_batch(of: BATCH_SIZE) do |relation|
          service = Ci::JobArtifacts::DestroyBatchService.new(relation, pick_up_at: Time.current)
          result  = service.execute(update_stats: false)
          @statistics_updates.merge!(result[:statistics_updates]) do |_project, existing_updates, new_updates|
            existing_updates.concat(new_updates)
          end
        end
      end

      def update_statistics
        @statistics_updates.each do |project, increments|
          ProjectStatistics.bulk_increment_statistic(project, Ci::JobArtifact.project_statistics_name, increments)
        end
      end
    end
  end
end