blob: dd24b3abf2e7fb9c777910ff1cf49fcf295854d3 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
require 'sidekiq/api'
module SidekiqJobsManager
def setup
ActiveJob::Base.queue_adapter = :sidekiq
unless can_run?
puts "Cannot run integration tests for sidekiq. To be able to run integration tests for sidekiq you need to install and start redis.\n"
exit
end
end
def clear_jobs
Sidekiq::ScheduledSet.new.clear
Sidekiq::Queue.new("integration_tests").clear
end
def start_workers
fork do
logfile = Rails.root.join("log/sidekiq.log").to_s
pidfile = Rails.root.join("tmp/sidekiq.pid").to_s
::Process.daemon(true, true)
[$stdout, $stderr].each do |io|
File.open(logfile, 'ab') do |f|
io.reopen(f)
end
io.sync = true
end
$stdin.reopen('/dev/null')
Sidekiq::Logging.initialize_logger(logfile)
File.open(File.expand_path(pidfile), 'w') do |f|
f.puts ::Process.pid
end
self_read, self_write = IO.pipe
trap "TERM" do
self_write.puts("TERM")
end
require 'celluloid'
require 'sidekiq/launcher'
sidekiq = Sidekiq::Launcher.new({queues: ["integration_tests"],
environment: "test",
concurrency: 1,
timeout: 1,
})
Sidekiq.poll_interval = 0.5
Sidekiq::Scheduled.const_set :INITIAL_WAIT, 1
begin
sidekiq.run
while readable_io = IO.select([self_read])
signal = readable_io.first[0].gets.strip
raise Interrupt if signal == "TERM"
end
rescue Interrupt
sidekiq.stop
exit(0)
end
end
sleep 1
end
def stop_workers
pidfile = Rails.root.join("tmp/sidekiq.pid").to_s
Process.kill 'TERM', File.open(pidfile).read.to_i
FileUtils.rm_f pidfile
rescue
end
def can_run?
begin
Sidekiq.redis(&:info)
Sidekiq.logger = nil
rescue
return false
end
true
end
end
|