use sidekiq

This commit is contained in:
Pete Matsyburka
2024-06-18 16:51:42 +03:00
parent 9539c476c0
commit 3034c00bb3
41 changed files with 332 additions and 132 deletions
+70
View File
@@ -0,0 +1,70 @@
# frozen_string_literal: true
require 'puma/plugin'
# rubocop:disable Metrics
Puma::Plugin.create do
def start(launcher)
return if ENV['LOCAL_REDIS_URL'].to_s.empty?
@puma_pid = $PROCESS_ID
launcher.events.on_booted do
@redis_server_pid = fork_redis
end
in_background { monitor_redis }
at_exit do
stop_redis_server if Process.pid == @puma_pid
end
launcher.events.on_stopped { stop_redis_server }
launcher.events.on_restart { stop_redis_server }
end
private
def monitor_redis
loop do
if redis_dead?
Process.kill(:INT, @puma_pid)
break
end
sleep 5
end
end
def redis_dead?
return false unless @redis_server_pid
Process.waitpid(@redis_server_pid, Process::WNOHANG)
false
rescue Errno::ECHILD, Errno::ESRCH
true
end
def fork_redis
fork do
Process.setsid
Dir.chdir(ENV.fetch('WORKDIR', nil)) unless ENV['WORKDIR'].to_s.empty?
exec('redis-server', '--requirepass', Digest::SHA1.hexdigest("redis#{ENV.fetch('SECRET_KEY_BASE', '')}"),
out: '/dev/null')
end
end
def stop_redis_server
if @redis_server_pid
Process.kill(:INT, @redis_server_pid)
Process.wait(@redis_server_pid)
end
rescue Errno::ECHILD, Errno::ESRCH
nil
end
end
# rubocop:enable Metrics
+74
View File
@@ -0,0 +1,74 @@
# frozen_string_literal: true
require 'puma/plugin'
# rubocop:disable Metrics
Puma::Plugin.create do
def config(cfg)
return if cfg.instance_variable_get(:@options)[:workers] <= 0
cfg.on_worker_boot { start_sidekiq! }
cfg.on_worker_shutdown { @sidekiq&.stop }
cfg.on_refork { @sidekiq&.stop }
end
def start(launcher)
launcher.events.on_booted do
next if Puma.stats_hash[:workers].to_i != 0
start_sidekiq!
end
launcher.events.on_stopped { Thread.new { @sidekiq&.stop }.join }
launcher.events.on_restart { Thread.new { @sidekiq&.stop }.join }
end
def fire_event(config, event)
arr = config[:lifecycle_events][event]
arr.each(&:call)
arr.clear
end
def start_sidekiq!
Thread.new do
wait_for_redis!
configs = Sidekiq.configure_embed do |config|
config.logger.level = Logger::INFO
sidekiq_config = YAML.load_file('config/sidekiq.yml')
config.queues = sidekiq_config['queues']
config.concurrency = ENV.fetch('SIDEKIQ_THREADS', 5).to_i
config.merge!(sidekiq_config)
config[:max_retries] = 13
ActiveSupport.run_load_hooks(:sidekiq_config, config)
end.instance_variable_get(:@config)
@sidekiq = Sidekiq::Launcher.new(configs, embedded: true)
@sidekiq.run
fire_event(configs, :startup)
end
end
def wait_for_redis!
attempt = 0
loop do
attempt += 1
sleep (attempt - 1) / 10.0
RedisClient.new(url: ENV.fetch('REDIS_URL', nil)).call('GET', '1')
break
rescue RedisClient::CannotConnectError
raise('Unable to connect to redis') if attempt > 10
end
end
end
# rubocop:enable Metrics
+1 -1
View File
@@ -73,7 +73,7 @@ module Submitters
next if submitter.email.blank?
next if submitter.preferences['send_email'] == false
SendSubmitterInvitationEmailJob.perform_later('submitter_id' => submitter.id)
SendSubmitterInvitationEmailJob.perform_async('submitter_id' => submitter.id)
end
end
end
+2 -2
View File
@@ -14,14 +14,14 @@ module Submitters
unless submitter.submission_events.exists?(event_type: 'start_form')
SubmissionEvents.create_with_tracking_data(submitter, 'start_form', request)
SendFormStartedWebhookRequestJob.perform_later({ 'submitter_id' => submitter.id })
SendFormStartedWebhookRequestJob.perform_async({ 'submitter_id' => submitter.id })
end
update_submitter!(submitter, params, request)
submitter.submission.save!
ProcessSubmitterCompletionJob.perform_later({ 'submitter_id' => submitter.id }) if submitter.completed_at?
ProcessSubmitterCompletionJob.perform_async({ 'submitter_id' => submitter.id }) if submitter.completed_at?
submitter
end