summaryrefslogtreecommitdiff
path: root/app/services/ci/job_artifacts/destroy_associations_service.rb
blob: fd3e69a5913783b4fb48da66235df94813c3f0c4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
# 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