debian-mirror-gitlab/lib/gitlab/background_migration/backfill_artifact_expiry_date.rb
2021-03-11 19:13:27 +05:30

58 lines
1.8 KiB
Ruby

# 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