Class: Ci::UpdateBuildQueueService

Inherits:
Object
  • Object
show all
Defined in:
app/services/ci/update_build_queue_service.rb

Constant Summary collapse

InvalidQueueTransition = 
Class.new(StandardError)

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(metrics = ::Gitlab::Ci::Queue::Metrics) ⇒ UpdateBuildQueueService

Returns a new instance of UpdateBuildQueueService.



9
10
11
 
# File 'app/services/ci/update_build_queue_service.rb', line 9

def initialize(metrics = ::Gitlab::Ci::Queue::Metrics)
  @metrics = metrics
end

Instance Attribute Details

#metrics ⇒ Object (readonly)

Returns the value of attribute metrics.



7
8
9
 
# File 'app/services/ci/update_build_queue_service.rb', line 7

def metrics
  @metrics
end

Instance Method Details

#pop(build, transition) ⇒ Object

Remove a build from the pending builds queue

Raises:



33
34
35
36
37
 
# File 'app/services/ci/update_build_queue_service.rb', line 33

def pop(build, transition)
  raise InvalidQueueTransition unless transition.from == 'pending'

  transition.within_transaction { remove!(build) }
end

#push(build, transition) ⇒ Object

Add a build to the pending builds queue

Raises:



16
17
18
19
20
21
22
23
24
25
26
27
28
 
# File 'app/services/ci/update_build_queue_service.rb', line 16

def push(build, transition)
  raise InvalidQueueTransition unless transition.to == 'pending'

  transition.within_transaction do
    result = build.create_queuing_entry!

    unless result.empty?
      metrics.increment_queue_operation(:build_queue_push)

      result.rows.dig(0, 0)
    end
  end
end

#remove!(build) ⇒ Object

Force remove build from the queue, without checking a transition state



42
43
44
45
46
47
48
49
50
 
# File 'app/services/ci/update_build_queue_service.rb', line 42

def remove!(build)
  removed = build.all_queuing_entries.delete_all

  if removed > 0
    metrics.increment_queue_operation(:build_queue_pop)

    build.id
  end
end

#tick(build) ⇒ Object

Unblock runner associated with given project / build



94
95
96
 
# File 'app/services/ci/update_build_queue_service.rb', line 94

def tick(build)
  tick_for(build, build.project.all_available_runners)
end

#track(build, transition) ⇒ Object

Add shared runner build tracking entry (used for queuing).

Raises:



55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
 
# File 'app/services/ci/update_build_queue_service.rb', line 55

def track(build, transition)
  return unless build.shared_runner_build?

  raise InvalidQueueTransition unless transition.to == 'running'

  transition.within_transaction do
    result = ::Ci::RunningBuild.upsert_shared_runner_build!(build)

    unless result.empty?
      metrics.increment_queue_operation(:shared_runner_build_new)

      result.rows.dig(0, 0)
    end
  end
end

#untrack(build, transition) ⇒ Object

Remove a runtime build tracking entry for a shared runner build (used for queuing).

Raises:



75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
 
# File 'app/services/ci/update_build_queue_service.rb', line 75

def untrack(build, transition)
  return unless build.shared_runner_build?

  raise InvalidQueueTransition unless transition.from == 'running'

  transition.within_transaction do
    removed = build.all_runtime_metadata.delete_all

    if removed > 0
      metrics.increment_queue_operation(:shared_runner_build_done)

      build.id
    end
  end
end