Class: ActiveSupport::Testing::Parallelization::Server

Inherits:
Object
  • Object
show all
Includes:
DRb::DRbUndumped
Defined in:
activesupport/lib/active_support/testing/parallelization/server.rb

Instance Method Summary collapse

Constructor Details

#initialize(distributor:) ⇒ Server

Returns a new instance of Server.



13
14
15
16
17
18
 
# File 'activesupport/lib/active_support/testing/parallelization/server.rb', line 13

def initialize(distributor:)
  @distributor = distributor
  @active_workers = Concurrent::Map.new
  @worker_pids = Concurrent::Map.new
  @in_flight = Concurrent::Map.new
end

Instance Method Details

#<<(o) ⇒ Object 



31
32
33
34
 
# File 'activesupport/lib/active_support/testing/parallelization/server.rb', line 31

def <<(o)
  o[2] = DRbObject.new(o[2]) if o
  @distributor.add_test(o)
end

#active_workers? ⇒ Boolean

Returns:

  • (Boolean) 


63
64
65
 
# File 'activesupport/lib/active_support/testing/parallelization/server.rb', line 63

def active_workers?
  @active_workers.size > 0
end

#interrupt ⇒ Object 



67
68
69
 
# File 'activesupport/lib/active_support/testing/parallelization/server.rb', line 67

def interrupt
  @distributor.interrupt
end

#pop(worker_id) ⇒ Object 



36
37
38
39
40
41
42
 
# File 'activesupport/lib/active_support/testing/parallelization/server.rb', line 36

def pop(worker_id)
  if test = @distributor.take(worker_id: worker_id)
    @in_flight[[test[0].to_s, test[1]]] = test
  end

  test
end

#record(reporter, result) ⇒ Object

Raises:

  • (DRb::DRbConnError) 


20
21
22
23
24
25
26
27
28
29
 
# File 'activesupport/lib/active_support/testing/parallelization/server.rb', line 20

def record(reporter, result)
  raise DRb::DRbConnError if result.is_a?(DRb::DRbUnknown)

  @in_flight.delete([result.klass, result.name])

  reporter.synchronize do
    reporter.prerecord(PrerecordResultClass.new(result.klass), result.name)
    reporter.record(result)
  end
end

#remove_dead_workers(dead_pids) ⇒ Object 



54
55
56
57
58
59
60
61
 
# File 'activesupport/lib/active_support/testing/parallelization/server.rb', line 54

def remove_dead_workers(dead_pids)
  dead_pids.each do |dead_pid|
    if worker_id = @worker_pids.key(dead_pid)
      @active_workers.delete(worker_id)
      @worker_pids.delete(worker_id)
    end
  end
end

#shutdown ⇒ Object 



71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
 
# File 'activesupport/lib/active_support/testing/parallelization/server.rb', line 71

def shutdown
  # Wait for initial queue to drain
  while @distributor.pending?
    sleep 0.1
  end

  @distributor.close

  wait_for_active_workers

  @in_flight.values.each do |(klass, name, reporter)|
    result = Minitest::Result.from(klass.new(name))
    error = RuntimeError.new("result not reported")
    error.set_backtrace([""])
    result.failures << Minitest::UnexpectedError.new(error)
    reporter.synchronize do
      reporter.record(result)
    end
  end
rescue Interrupt
  warn "Interrupted. Exiting..."

  @distributor.close

  wait_for_active_workers
end

#start_worker(worker_id, worker_pid) ⇒ Object 



44
45
46
47
 
# File 'activesupport/lib/active_support/testing/parallelization/server.rb', line 44

def start_worker(worker_id, worker_pid)
  @active_workers[worker_id] = true
  @worker_pids[worker_id] = worker_pid
end

#stop_worker(worker_id, worker_pid) ⇒ Object 



49
50
51
52
 
# File 'activesupport/lib/active_support/testing/parallelization/server.rb', line 49

def stop_worker(worker_id, worker_pid)
  @active_workers.delete(worker_id)
  @worker_pids.delete(worker_id)
end