41 lines
1.1 KiB
Ruby
41 lines
1.1 KiB
Ruby
|
module Ci
|
||
|
|
# This class responsible for assigning
|
||
|
|
# proper pending build to runner on runner API request
|
||
|
|
class RegisterBuildService
|
||
|
|
def execute(current_runner)
|
||
|
|
builds = Ci::Build.pending.unstarted
|
||
|
|
|
||
|
|
builds =
|
||
|
|
if current_runner.shared?
|
||
|
|
# don't run projects which have not enables shared runners
|
||
|
|
builds.includes(:project).where(ci_projects: { shared_runners_enabled: true })
|
||
|
|
else
|
||
|
|
# do run projects which are only assigned to this runner
|
||
|
|
builds.where(project_id: current_runner.projects)
|
||
|
|
end
|
||
|
|
|
||
|
|
builds = builds.order('created_at ASC')
|
||
|
|
|
||
|
|
build = builds.find do |build|
|
||
|
|
(build.tag_list - current_runner.tag_list).empty?
|
||
|
|
end
|
||
|
|
|
||
|
|
|
||
|
|
if build
|
||
|
|
# In case when 2 runners try to assign the same build, second runner will be declined
|
||
|
|
# with StateMachine::InvalidTransition in run! method.
|
||
|
|
build.with_lock do
|
||
|
|
build.runner_id = current_runner.id
|
||
|
|
build.save!
|
||
|
|
build.run!
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
build
|
||
|
|
|
||
|
|
rescue StateMachine::InvalidTransition
|
||
|
|
nil
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|