summaryrefslogtreecommitdiff
path: root/lib/tasks/gitlab/traces.rake
blob: 5e1ec481ece9e83263946f78a244270ee5cbf498 (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
36
37
38
require 'logger'
require 'resolv-replace'

desc "GitLab | Archive legacy traces to trace artifacts"
namespace :gitlab do
  namespace :traces do
    task archive: :environment do
      logger = Logger.new(STDOUT)
      logger.info('Archiving legacy traces')

      Ci::Build.finished.without_archived_trace
        .order(id: :asc)
        .find_in_batches(batch_size: 1000) do |jobs|
        job_ids = jobs.map { |job| [job.id] }

        ArchiveTraceWorker.bulk_perform_async(job_ids)

        logger.info("Scheduled #{job_ids.count} jobs. From #{job_ids.min} to #{job_ids.max}")
      end
    end

    task migrate: :environment do
      logger = Logger.new(STDOUT)
      logger.info('Starting transfer of job traces')

      Ci::Build.joins(:project)
        .with_archived_trace_stored_locally
        .find_each(batch_size: 10) do |build|

        build.job_artifacts_trace.file.migrate!(ObjectStorage::Store::REMOTE)

        logger.info("Transferred job trace of #{build.id} to object storage")
      rescue => e
        logger.error("Failed to transfer artifacts of #{build.id} with error: #{e.message}")
      end
    end
  end
end