runner.rb 4.61 KB
Newer Older
Douwe Maan's avatar
Douwe Maan committed
1
2
3
module Ci
  class Runner < ActiveRecord::Base
    extend Ci::Model
4

5
    RUNNER_QUEUE_EXPIRY_TIME = 60.minutes
6
    LAST_CONTACT_TIME = 1.hour.ago
Douwe Maan's avatar
Douwe Maan committed
7
8
    AVAILABLE_SCOPES = %w[specific shared active paused online].freeze
    FORM_EDITABLE = %i[description tag_list active run_untagged locked].freeze
9

10
11
12
    has_many :builds
    has_many :runner_projects, dependent: :destroy
    has_many :projects, through: :runner_projects, foreign_key: :gl_project_id
Douwe Maan's avatar
Douwe Maan committed
13
14
15
16
17
18
19
20
21

    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) }
22
    scope :online, ->() { where('contacted_at > ?', LAST_CONTACT_TIME) }
23
    scope :ordered, ->() { order(id: :desc) }
Douwe Maan's avatar
Douwe Maan committed
24

25
    scope :owned_or_shared, ->(project_id) do
Douwe Maan's avatar
Douwe Maan committed
26
27
      joins('LEFT JOIN ci_runner_projects ON ci_runner_projects.runner_id = ci_runners.id').
        where("ci_runner_projects.gl_project_id = :project_id OR ci_runners.is_shared = true", project_id: project_id)
28
29
    end

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

37
    validate :tag_constraints
38

Douwe Maan's avatar
Douwe Maan committed
39
40
    acts_as_taggable

41
42
    after_destroy :cleanup_runner_queue

43
44
45
46
47
48
49
50
51
52
53
54
    # 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.
Douwe Maan's avatar
Douwe Maan committed
55
    def self.search(query)
56
      t = arel_table
57
58
59
      pattern = "%#{query}%"

      where(t[:token].matches(pattern).or(t[:description].matches(pattern)))
Douwe Maan's avatar
Douwe Maan committed
60
61
62
63
64
65
66
67
68
    end

    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
69
      project.runner_projects.create(runner_id: self.id)
Douwe Maan's avatar
Douwe Maan committed
70
71
72
    end

    def display_name
73
      return short_sha if description.blank?
Douwe Maan's avatar
Douwe Maan committed
74
75
76
77
78
79
80
81

      description
    end

    def shared?
      is_shared
    end

82
83
84
85
86
87
88
89
90
91
92
93
94
95
    def online?
      contacted_at && contacted_at > LAST_CONTACT_TIME
    end

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

Douwe Maan's avatar
Douwe Maan committed
96
97
98
99
100
101
102
103
    def belongs_to_one_project?
      runner_projects.count == 1
    end

    def specific?
      !shared?
    end

104
    def can_pick?(build)
Lin Jen-Shin's avatar
Lin Jen-Shin committed
105
      assignable_for?(build.project) && accepting_tags?(build)
106
107
    end

Douwe Maan's avatar
Douwe Maan committed
108
109
110
111
112
    def only_for?(project)
      projects == [project]
    end

    def short_sha
Kamil Trzcinski's avatar
Kamil Trzcinski committed
113
      token[0...8] if token
Douwe Maan's avatar
Douwe Maan committed
114
    end
115
116
117
118

    def has_tags?
      tag_list.any?
    end
119

120
121
122
123
124
125
126
127
    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's avatar
Kim "BKC" Carlbäcker committed
128
    def tick_runner_queue
129
130
131
132
133
      SecureRandom.hex.tap do |new_update|
        Gitlab::Redis.with do |redis|
          redis.set(runner_queue_key, new_update, ex: RUNNER_QUEUE_EXPIRY_TIME)
        end
      end
134
135
    end

Kim "BKC" Carlbäcker's avatar
Kim "BKC" Carlbäcker committed
136
137
    def ensure_runner_queue_value
      Gitlab::Redis.with do |redis|
Kamil Trzcinski's avatar
Kamil Trzcinski committed
138
        value = SecureRandom.hex
139
        redis.set(runner_queue_key, value, ex: RUNNER_QUEUE_EXPIRY_TIME, nx: true)
Kim "BKC" Carlbäcker's avatar
Kim "BKC" Carlbäcker committed
140
        redis.get(runner_queue_key)
Kim "BKC" Carlbäcker's avatar
Kim "BKC" Carlbäcker committed
141
142
143
144
145
      end
    end

    def is_runner_queue_value_latest?(value)
      ensure_runner_queue_value == value if value.present?
146
147
    end

148
149
    private

150
151
152
153
154
155
    def cleanup_runner_queue
      Gitlab::Redis.with do |redis|
        redis.del(runner_queue_key)
      end
    end

Kim "BKC" Carlbäcker's avatar
Kim "BKC" Carlbäcker committed
156
    def runner_queue_key
157
      "runner:build_queue:#{self.token}"
158
159
    end

160
    def tag_constraints
161
162
163
164
165
      unless has_tags? || run_untagged?
        errors.add(:tags_list,
          'can not be empty when runner is not allowed to pick untagged jobs')
      end
    end
166

Lin Jen-Shin's avatar
Lin Jen-Shin committed
167
    def assignable_for?(project)
168
169
170
      !locked? || projects.exists?(id: project.id)
    end

171
172
    def accepting_tags?(build)
      (run_untagged? || build.has_tags?) && (build.tag_list - tag_list).empty?
173
    end
Douwe Maan's avatar
Douwe Maan committed
174
175
  end
end