60 lines
2.2 KiB
Ruby
60 lines
2.2 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Ci
|
|
class ProcessPipelineService
|
|
attr_reader :pipeline
|
|
|
|
def initialize(pipeline)
|
|
@pipeline = pipeline
|
|
end
|
|
|
|
def execute(trigger_build_ids = nil, initial_process: false)
|
|
update_retried
|
|
ensure_scheduling_type_for_processables
|
|
|
|
if Feature.enabled?(:ci_atomic_processing, pipeline.project)
|
|
Ci::PipelineProcessing::AtomicProcessingService
|
|
.new(pipeline)
|
|
.execute
|
|
else
|
|
Ci::PipelineProcessing::LegacyProcessingService
|
|
.new(pipeline)
|
|
.execute(trigger_build_ids, initial_process: initial_process)
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
# This method is for compatibility and data consistency and should be removed with 9.3 version of GitLab
|
|
# This replicates what is db/post_migrate/20170416103934_upate_retried_for_ci_build.rb
|
|
# and ensures that functionality will not be broken before migration is run
|
|
# this updates only when there are data that needs to be updated, there are two groups with no retried flag
|
|
# rubocop: disable CodeReuse/ActiveRecord
|
|
def update_retried
|
|
# find the latest builds for each name
|
|
latest_statuses = pipeline.statuses.latest
|
|
.group(:name)
|
|
.having('count(*) > 1')
|
|
.pluck(Arel.sql('MAX(id)'), 'name')
|
|
|
|
# mark builds that are retried
|
|
pipeline.statuses.latest
|
|
.where(name: latest_statuses.map(&:second))
|
|
.where.not(id: latest_statuses.map(&:first))
|
|
.update_all(retried: true) if latest_statuses.any?
|
|
end
|
|
# rubocop: enable CodeReuse/ActiveRecord
|
|
|
|
# Set scheduling type of processables if they were created before scheduling_type
|
|
# data was deployed (https://gitlab.com/gitlab-org/gitlab/-/merge_requests/22246).
|
|
# Given that this service runs multiple times during the pipeline
|
|
# life cycle we need to ensure we populate the data once.
|
|
# See more: https://gitlab.com/gitlab-org/gitlab/issues/205426
|
|
def ensure_scheduling_type_for_processables
|
|
lease = Gitlab::ExclusiveLease.new("set-scheduling-types:#{pipeline.id}", timeout: 1.hour.to_i)
|
|
return unless lease.try_obtain
|
|
|
|
pipeline.processables.populate_scheduling_type!
|
|
end
|
|
end
|
|
end
|