gitlab-org--gitlab-foss/lib/gitlab/background_migration/backfill_artifact_expiry_date.rb

# frozen_string_literal: true

module Gitlab
  module BackgroundMigration
    # Backfill expire_at for a range of Ci::JobArtifact
    class BackfillArtifactExpiryDate
      include Gitlab::Utils::StrongMemoize

      SWITCH_DATE = Date.new(2020, 06, 22).freeze
      OLD_ARTIFACT_AGE = 15.months
      BATCH_SIZE = 1_000
      OLD_ARTIFACT_EXPIRY_OFFSET = 3.months
      RECENT_ARTIFACT_EXPIRY_OFFSET = 1.year

      # Ci::JobArtifact model
      class Ci::JobArtifact < ActiveRecord::Base
        include ::EachBatch

        self.table_name = 'ci_job_artifacts'

        scope :without_expiry_date, -> { where(expire_at: nil) }
        scope :before_switch, -> { where("date(created_at AT TIME ZONE 'UTC') < ?::date", SWITCH_DATE) }
        scope :between, -> (start_id, end_id) { where(id: start_id..end_id) }
        scope :old, -> { where(self.arel_table[:created_at].lt(OLD_ARTIFACT_AGE.ago)) }
        scope :recent, -> { where(self.arel_table[:created_at].gt(OLD_ARTIFACT_AGE.ago)) }
      end

      def perform(start_id, end_id)
        Ci::JobArtifact
          .without_expiry_date.before_switch
          .between(start_id, end_id)
          .each_batch(of: BATCH_SIZE) do |batch|
          batch.old.update_all(expire_at: old_artifact_expiry_date)
          batch.recent.update_all(expire_at: recent_artifact_expiry_date)
        end
      end

      private

      def offset_date
        strong_memoize(:offset_date) do
          current_date = Time.current
          target_date = Time.zone.local(current_date.year, current_date.month, 22, 0, 0, 0)

          current_date.day < 22 ? target_date : target_date.next_month
        end
      end

      def old_artifact_expiry_date
        offset_date + OLD_ARTIFACT_EXPIRY_OFFSET
      end

      def recent_artifact_expiry_date
        offset_date + RECENT_ARTIFACT_EXPIRY_OFFSET
      end
    end
  end
end