BigW Consortium Gitlab

runner.rb 4.87 KB
Newer Older
1 2
module Ci
  class Runner < ActiveRecord::Base
3
    extend Gitlab::Ci::Model
4
    include Gitlab::SQL::Pattern
5

6
    RUNNER_QUEUE_EXPIRY_TIME = 60.minutes
7
    ONLINE_CONTACT_TIMEOUT = 1.hour
8
    AVAILABLE_SCOPES = %w[specific shared active paused online].freeze
9
    FORM_EDITABLE = %i[description tag_list active run_untagged locked access_level].freeze
10

11
    has_many :builds
12
    has_many :runner_projects, dependent: :destroy # rubocop:disable Cop/ActiveRecordDependent
13
    has_many :projects, through: :runner_projects
14 15 16 17 18 19 20 21 22

    has_one :last_build, ->() { order('id DESC') }, class_name: 'Ci::Build'

    before_validation :set_default_values

    scope :specific, ->() { where(is_shared: false) }
    scope :shared, ->() { where(is_shared: true) }
    scope :active, ->() { where(active: true) }
    scope :paused, ->() { where(active: false) }
23
    scope :online, ->() { where('contacted_at > ?', contact_time_deadline) }
24
    scope :ordered, ->() { order(id: :desc) }
25

26
    scope :owned_or_shared, ->(project_id) do
27
      joins('LEFT JOIN ci_runner_projects ON ci_runner_projects.runner_id = ci_runners.id')
28
        .where("ci_runner_projects.project_id = :project_id OR ci_runners.is_shared = true", project_id: project_id)
29 30
    end

Lin Jen-Shin committed
31
    scope :assignable_for, ->(project) do
32 33
      # FIXME: That `to_sql` is needed to workaround a weird Rails bug.
      #        Without that, placeholders would miss one and couldn't match.
34 35
      where(locked: false)
        .where.not("id IN (#{project.runners.select(:id).to_sql})").specific
36 37
    end

38
    validate :tag_constraints
39
    validates :access_level, presence: true
40

41 42
    acts_as_taggable

43 44
    after_destroy :cleanup_runner_queue

45
    enum access_level: {
46 47
      not_protected: 0,
      ref_protected: 1
48 49
    }

50 51 52 53 54 55 56 57 58 59 60 61
    # Searches for runners matching the given query.
    #
    # This method uses ILIKE on PostgreSQL and LIKE on MySQL.
    #
    # This method performs a *partial* match on tokens, thus a query for "a"
    # will match any runner where the token contains the letter "a". As a result
    # you should *not* use this method for non-admin purposes as otherwise users
    # might be able to query a list of all runners.
    #
    # query - The search query as a String
    #
    # Returns an ActiveRecord::Relation.
62
    def self.search(query)
63
      fuzzy_search(query, [:token, :description])
64 65
    end

66 67 68 69
    def self.contact_time_deadline
      ONLINE_CONTACT_TIMEOUT.ago
    end

70 71 72 73 74 75 76
    def set_default_values
      self.token = SecureRandom.hex(15) if self.token.blank?
    end

    def assign_to(project, current_user = nil)
      self.is_shared = false if shared?
      self.save
77
      project.runner_projects.create(runner_id: self.id)
78 79 80
    end

    def display_name
81
      return short_sha if description.blank?
82 83 84 85 86 87 88 89

      description
    end

    def shared?
      is_shared
    end

90
    def online?
91
      contacted_at && contacted_at > self.class.contact_time_deadline
92 93 94 95 96 97 98 99 100 101 102 103
    end

    def status
      if contacted_at.nil?
        :not_connected
      elsif active?
        online? ? :online : :offline
      else
        :paused
      end
    end

104 105 106 107 108 109 110 111
    def belongs_to_one_project?
      runner_projects.count == 1
    end

    def specific?
      !shared?
    end

112
    def can_pick?(build)
113
      return false if self.ref_protected? && !build.protected?
Shinya Maeda committed
114

115
      assignable_for?(build.project_id) && accepting_tags?(build)
116 117
    end

118 119 120 121 122
    def only_for?(project)
      projects == [project]
    end

    def short_sha
Kamil Trzcinski committed
123
      token[0...8] if token
124
    end
125 126 127 128

    def has_tags?
      tag_list.any?
    end
129

130 131 132 133 134 135 136 137
    def predefined_variables
      [
        { key: 'CI_RUNNER_ID', value: id.to_s, public: true },
        { key: 'CI_RUNNER_DESCRIPTION', value: description, public: true },
        { key: 'CI_RUNNER_TAGS', value: tag_list.to_s, public: true }
      ]
    end

Kim "BKC" Carlbäcker committed
138
    def tick_runner_queue
139
      SecureRandom.hex.tap do |new_update|
140
        ::Gitlab::Workhorse.set_key_and_notify(runner_queue_key, new_update,
141
          expire: RUNNER_QUEUE_EXPIRY_TIME, overwrite: true)
142
      end
143 144
    end

Kim "BKC" Carlbäcker committed
145
    def ensure_runner_queue_value
146 147 148
      new_value = SecureRandom.hex
      ::Gitlab::Workhorse.set_key_and_notify(runner_queue_key, new_value,
        expire: RUNNER_QUEUE_EXPIRY_TIME, overwrite: false)
Kim "BKC" Carlbäcker committed
149 150
    end

151
    def runner_queue_value_latest?(value)
Kim "BKC" Carlbäcker committed
152
      ensure_runner_queue_value == value if value.present?
153 154
    end

155 156
    private

157
    def cleanup_runner_queue
158
      Gitlab::Redis::Queues.with do |redis|
159 160 161 162
        redis.del(runner_queue_key)
      end
    end

Kim "BKC" Carlbäcker committed
163
    def runner_queue_key
164
      "runner:build_queue:#{self.token}"
165 166
    end

167
    def tag_constraints
168 169 170 171 172
      unless has_tags? || run_untagged?
        errors.add(:tags_list,
          'can not be empty when runner is not allowed to pick untagged jobs')
      end
    end
173

174 175
    def assignable_for?(project_id)
      is_shared? || projects.exists?(id: project_id)
176 177
    end

178 179
    def accepting_tags?(build)
      (run_untagged? || build.has_tags?) && (build.tag_list - tag_list).empty?
180
    end
181 182
  end
end