Class: Buildkite::TestCollector::Session

Inherits:
Object
  • Object
show all
Defined in:
lib/buildkite/test_collector/session.rb

Constant Summary collapse

UPLOAD_SESSION_TIMEOUT = 
60
UPLOAD_API_MAX_RESULTS = 
5000

Instance Method Summary collapse

Constructor Details

#initialize ⇒ Session

Returns a new instance of Session.



10
11
12
13
 
# File 'lib/buildkite/test_collector/session.rb', line 10

def initialize
  @send_queue_ids = []
  @upload_futures = []
end

Instance Method Details

#add_example_to_send_queue(id) ⇒ Object 



15
16
17
18
19
20
21
22
 
# File 'lib/buildkite/test_collector/session.rb', line 15

def add_example_to_send_queue(id)
  @send_queue_ids << id

  if @send_queue_ids.size >= Buildkite::TestCollector.batch_size
    send_ids = @send_queue_ids.shift(Buildkite::TestCollector.batch_size)
    upload_data(send_ids)
  end
end

#close ⇒ Object 



31
32
33
34
35
36
37
38
39
40
41
42
 
# File 'lib/buildkite/test_collector/session.rb', line 31

def close
  return if @upload_futures.empty?

  begin
    Concurrent::Promises.zip(*@upload_futures).value!(UPLOAD_SESSION_TIMEOUT)
  rescue StandardError
    # Timeout or other error - futures will continue running in the thread pool
    # with their own per-upload timeouts, we just stop waiting for them
  ensure
    @upload_futures.clear
  end
end

#send_remaining_data ⇒ Object 



24
25
26
27
28
29
 
# File 'lib/buildkite/test_collector/session.rb', line 24

def send_remaining_data
  return if @send_queue_ids.empty?

  upload_data(@send_queue_ids)
  @send_queue_ids.clear
end