diff options
author | GitLab Bot <gitlab-bot@gitlab.com> | 2021-07-20 09:55:51 +0000 |
---|---|---|
committer | GitLab Bot <gitlab-bot@gitlab.com> | 2021-07-20 09:55:51 +0000 |
commit | e8d2c2579383897a1dd7f9debd359abe8ae8373d (patch) | |
tree | c42be41678c2586d49a75cabce89322082698334 /app/services/ci/queue/builds_table_strategy.rb | |
parent | fc845b37ec3a90aaa719975f607740c22ba6a113 (diff) | |
download | gitlab-ce-e8d2c2579383897a1dd7f9debd359abe8ae8373d.tar.gz |
Add latest changes from gitlab-org/gitlab@14-1-stable-eev14.1.0-rc42
Diffstat (limited to 'app/services/ci/queue/builds_table_strategy.rb')
-rw-r--r-- | app/services/ci/queue/builds_table_strategy.rb | 67 |
1 files changed, 67 insertions, 0 deletions
diff --git a/app/services/ci/queue/builds_table_strategy.rb b/app/services/ci/queue/builds_table_strategy.rb new file mode 100644 index 00000000000..c941734ac40 --- /dev/null +++ b/app/services/ci/queue/builds_table_strategy.rb @@ -0,0 +1,67 @@ +# frozen_string_literal: true + +module Ci + module Queue + class BuildsTableStrategy + attr_reader :runner + + def initialize(runner) + @runner = runner + end + + # rubocop:disable CodeReuse/ActiveRecord + def builds_for_shared_runner + relation = new_builds + # don't run projects which have not enabled shared runners and builds + .joins('INNER JOIN projects ON ci_builds.project_id = projects.id') + .where(projects: { shared_runners_enabled: true, pending_delete: false }) + .joins('LEFT JOIN project_features ON ci_builds.project_id = project_features.project_id') + .where('project_features.builds_access_level IS NULL or project_features.builds_access_level > 0') + + if Feature.enabled?(:ci_queueing_disaster_recovery_disable_fair_scheduling, runner, type: :ops, default_enabled: :yaml) + # if disaster recovery is enabled, we fallback to FIFO scheduling + relation.order('ci_builds.id ASC') + else + # Implement fair scheduling + # this returns builds that are ordered by number of running builds + # we prefer projects that don't use shared runners at all + relation + .joins("LEFT JOIN (#{running_builds_for_shared_runners.to_sql}) AS project_builds ON ci_builds.project_id = project_builds.project_id") + .order(Arel.sql('COALESCE(project_builds.running_builds, 0) ASC'), 'ci_builds.id ASC') + end + end + + def builds_matching_tag_ids(relation, ids) + # pick builds that does not have other tags than runner's one + relation.matches_tag_ids(ids) + end + + def builds_with_any_tags(relation) + # pick builds that have at least one tag + relation.with_any_tags + end + + def order(relation) + relation.order('id ASC') + end + + def new_builds + ::Ci::Build.pending.unstarted + end + + def build_ids(relation) + relation.pluck(:id) + end + + private + + def running_builds_for_shared_runners + ::Ci::Build.running + .where(runner: ::Ci::Runner.instance_type) + .group(:project_id) + .select(:project_id, 'COUNT(*) AS running_builds') + end + # rubocop:enable CodeReuse/ActiveRecord + end + end +end |