discourse/config/initializers/100-sidekiq.rb

82 lines
2.2 KiB
Ruby
Raw Normal View History

require "sidekiq/pausable"
Sidekiq.configure_client do |config|
2014-04-23 09:01:17 +08:00
config.redis = Discourse.sidekiq_redis_config
end
2013-08-08 01:25:05 +08:00
Sidekiq.configure_server do |config|
2014-04-23 09:01:17 +08:00
config.redis = Discourse.sidekiq_redis_config
2015-09-03 10:00:19 +08:00
config.server_middleware do |chain|
chain.add Sidekiq::Pausable
end
2013-08-08 01:25:05 +08:00
end
if Sidekiq.server?
# defer queue should simply run in sidekiq
Scheduler::Defer.async = false
# warm up AR
RailsMultisite::ConnectionManagement.each_connection do
(ActiveRecord::Base.connection.tables - %w[schema_migrations]).each do |table|
table.classify.constantize.first rescue nil
end
end
Rails.application.config.after_initialize do
scheduler_hostname = ENV["UNICORN_SCHEDULER_HOSTNAME"]
if !scheduler_hostname || scheduler_hostname.split(',').include?(`hostname`.strip)
require 'scheduler/scheduler'
manager = Scheduler::Manager.new($redis.without_namespace)
Scheduler::Manager.discover_schedules.each do |schedule|
manager.ensure_schedule!(schedule)
end
Thread.new do
while true
begin
manager.tick
rescue => e
# the show must go on
2017-07-28 09:20:09 +08:00
Discourse.handle_job_exception(e, message: "While ticking scheduling manager")
end
sleep 1
end
end
end
end
end
2013-05-30 00:36:15 +08:00
Sidekiq.logger.level = Logger::WARN
class SidekiqLogsterReporter < Sidekiq::ExceptionHandler::Logger
def call(ex, context = {})
return if Jobs::HandledExceptionWrapper === ex
Discourse.reset_active_record_cache_if_needed(ex)
# Pass context to Logster
fake_env = {}
context.each do |key, value|
Logster.add_to_env(fake_env, key, value)
end
text = "Job exception: #{ex}\n"
if ex.backtrace
Logster.add_to_env(fake_env, :backtrace, ex.backtrace)
end
2015-05-06 10:28:32 +08:00
Logster.add_to_env(fake_env, :current_hostname, Discourse.current_hostname)
Thread.current[Logster::Logger::LOGSTER_ENV] = fake_env
Logster.logger.error(text)
rescue => e
Logster.logger.fatal("Failed to log exception #{ex} #{hash}\nReason: #{e.class} #{e}\n#{e.backtrace.join("\n")}")
ensure
Thread.current[Logster::Logger::LOGSTER_ENV] = nil
end
end
Sidekiq.error_handlers.clear
Sidekiq.error_handlers << SidekiqLogsterReporter.new