Module: SidekiqAlive
- Defined in:
- lib/sidekiq_alive.rb,
lib/sidekiq_alive/config.rb,
lib/sidekiq_alive/server.rb,
lib/sidekiq_alive/worker.rb,
lib/sidekiq_alive/version.rb
Defined Under Namespace
Classes: Config, Server, Worker
Constant Summary
collapse
- VERSION =
'2.0.3'
Class Method Summary
collapse
Class Method Details
.alive? ⇒ Boolean
84
85
86
|
# File 'lib/sidekiq_alive.rb', line 84
def self.alive?
redis.ttl(current_lifeness_key) != -2
end
|
.banner ⇒ Object
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
|
# File 'lib/sidekiq_alive.rb', line 122
def self.banner
<<~BANNER
=================== SidekiqAlive =================
Hostname: #{hostname}
Liveness key: #{current_lifeness_key}
Port: #{config.port}
Time to live: #{config.time_to_live}s
Current instance register key: #{current_instance_register_key}
Worker running on queue: #{@queue}
starting ...
BANNER
end
|
.config ⇒ Object
98
99
100
|
# File 'lib/sidekiq_alive.rb', line 98
def self.config
@config ||= SidekiqAlive::Config.instance
end
|
.current_instance_register_key ⇒ Object
70
71
72
|
# File 'lib/sidekiq_alive.rb', line 70
def self.current_instance_register_key
"#{config.registered_instance_key}::#{hostname}"
end
|
.current_lifeness_key ⇒ Object
102
103
104
|
# File 'lib/sidekiq_alive.rb', line 102
def self.current_lifeness_key
"#{config.liveness_key}::#{hostname}"
end
|
.current_queue ⇒ Object
39
40
41
|
# File 'lib/sidekiq_alive.rb', line 39
def self.current_queue
"#{config.queue_prefix}-#{hostname}"
end
|
.hostname ⇒ Object
106
107
108
|
# File 'lib/sidekiq_alive.rb', line 106
def self.hostname
ENV['HOSTNAME'] || 'HOSTNAME_NOT_SET'
end
|
.logger ⇒ Object
94
95
96
|
# File 'lib/sidekiq_alive.rb', line 94
def self.logger
Sidekiq.logger
end
|
.purge_pending_jobs ⇒ Object
58
59
60
61
62
63
64
65
66
67
68
|
# File 'lib/sidekiq_alive.rb', line 58
def self.purge_pending_jobs
scheduled_set = Sidekiq::ScheduledSet.new
jobs = scheduled_set.select { |job| job.klass == 'SidekiqAlive::Worker' && job.queue == current_queue }
logger.info("[SidekiqAlive] Purging #{jobs.count} pending for #{hostname}")
jobs.each(&:delete)
logger.info("[SidekiqAlive] Removing queue #{current_queue}")
Sidekiq::Queue.new(current_queue).clear
end
|
.redis ⇒ Object
80
81
82
|
# File 'lib/sidekiq_alive.rb', line 80
def self.redis
Sidekiq.redis { |r| r }
end
|
.register_current_instance ⇒ Object
43
44
45
|
# File 'lib/sidekiq_alive.rb', line 43
def self.register_current_instance
register_instance(current_instance_register_key)
end
|
.register_instance(instance_name) ⇒ Object
149
150
151
152
153
|
# File 'lib/sidekiq_alive.rb', line 149
def self.register_instance(instance_name)
redis.set(instance_name,
Time.now.to_i,
ex: config.registration_ttl.to_i)
end
|
.registered_instances ⇒ Object
54
55
56
|
# File 'lib/sidekiq_alive.rb', line 54
def self.registered_instances
redis.keys("#{config.registered_instance_key}::*")
end
|
.setup {|config| ... } ⇒ Object
90
91
92
|
# File 'lib/sidekiq_alive.rb', line 90
def self.setup
yield(config)
end
|
.shutdown_info ⇒ Object
110
111
112
113
114
115
116
117
118
119
120
|
# File 'lib/sidekiq_alive.rb', line 110
def self.shutdown_info
<<~BANNER
=================== Shutting down SidekiqAlive =================
Hostname: #{hostname}
Liveness key: #{current_lifeness_key}
Current instance register key: #{current_instance_register_key}
BANNER
end
|
.start ⇒ Object
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
|
# File 'lib/sidekiq_alive.rb', line 8
def self.start
SidekiqAlive::Worker.sidekiq_options queue: current_queue
Sidekiq.configure_server do |sq_config|
sq_config.options[:queues] << current_queue
sq_config.on(:startup) do
SidekiqAlive.tap do |sa|
sa.logger.info(banner)
sa.register_current_instance
sa.store_alive_key
sa::Worker.perform_async(hostname)
@server_pid = fork do
sa::Server.run!
end
sa.logger.info(successful_startup_text)
end
end
sq_config.on(:quiet) do
SidekiqAlive.unregister_current_instance
end
sq_config.on(:shutdown) do
Process.kill('TERM', @server_pid) unless @server_pid.nil?
Process.wait(@server_pid) unless @server_pid.nil?
SidekiqAlive.unregister_current_instance
end
end
end
|
.store_alive_key ⇒ Object
74
75
76
77
78
|
# File 'lib/sidekiq_alive.rb', line 74
def self.store_alive_key
redis.set(current_lifeness_key,
Time.now.to_i,
ex: config.time_to_live.to_i)
end
|
.successful_startup_text ⇒ Object
139
140
141
142
143
144
145
146
147
|
# File 'lib/sidekiq_alive.rb', line 139
def self.successful_startup_text
<<~BANNER
Registered instances:
- #{registered_instances.join("\n\s\s- ")}
=================== SidekiqAlive Ready! =================
BANNER
end
|
.unregister_current_instance ⇒ Object
47
48
49
50
51
52
|
# File 'lib/sidekiq_alive.rb', line 47
def self.unregister_current_instance
logger.info(shutdown_info)
purge_pending_jobs
redis.del(current_instance_register_key)
end
|