2014-02-12 22:30:33 -06:00
|
|
|
require "sidekiq/pausable"
|
|
|
|
|
|
|
|
Sidekiq.configure_client do |config|
|
2014-04-22 20:01:17 -05:00
|
|
|
config.redis = Discourse.sidekiq_redis_config
|
2014-02-12 22:30:33 -06:00
|
|
|
end
|
|
|
|
|
2013-08-07 12:25:05 -05:00
|
|
|
Sidekiq.configure_server do |config|
|
2014-04-22 20:01:17 -05:00
|
|
|
config.redis = Discourse.sidekiq_redis_config
|
2015-09-02 21:00:19 -05:00
|
|
|
|
2014-02-12 22:30:33 -06:00
|
|
|
config.server_middleware do |chain|
|
|
|
|
chain.add Sidekiq::Pausable
|
|
|
|
end
|
2013-08-07 12:25:05 -05:00
|
|
|
end
|
|
|
|
|
2014-02-05 17:14:41 -06:00
|
|
|
if Sidekiq.server?
|
2018-05-22 21:05:17 -05:00
|
|
|
# defer queue should simply run in sidekiq
|
|
|
|
Scheduler::Defer.async = false
|
|
|
|
|
2014-02-07 22:18:54 -06:00
|
|
|
# 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
|
|
|
|
|
2014-04-10 11:40:49 -05:00
|
|
|
Rails.application.config.after_initialize do
|
2017-04-26 01:13:26 -05:00
|
|
|
scheduler_hostname = ENV["UNICORN_SCHEDULER_HOSTNAME"]
|
|
|
|
|
2017-06-19 00:45:51 -05:00
|
|
|
if !scheduler_hostname || scheduler_hostname.split(',').include?(`hostname`.strip)
|
2017-04-26 01:13:26 -05:00
|
|
|
require 'scheduler/scheduler'
|
2017-10-11 02:46:53 -05:00
|
|
|
manager = Scheduler::Manager.new($redis.without_namespace)
|
2017-04-26 01:13:26 -05:00
|
|
|
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-27 20:20:09 -05:00
|
|
|
Discourse.handle_job_exception(e, message: "While ticking scheduling manager")
|
2017-04-26 01:13:26 -05:00
|
|
|
end
|
|
|
|
sleep 1
|
2014-04-10 11:40:49 -05:00
|
|
|
end
|
2014-02-05 17:14:41 -06:00
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
2013-09-02 02:14:41 -05:00
|
|
|
end
|
2013-05-29 11:36:15 -05:00
|
|
|
|
|
|
|
Sidekiq.logger.level = Logger::WARN
|
2014-05-06 17:23:52 -05:00
|
|
|
|
2014-07-17 15:22:46 -05:00
|
|
|
class SidekiqLogsterReporter < Sidekiq::ExceptionHandler::Logger
|
2014-07-11 17:03:07 -05:00
|
|
|
def call(ex, context = {})
|
2015-08-13 21:51:23 -05:00
|
|
|
|
|
|
|
return if Jobs::HandledExceptionWrapper === ex
|
2017-02-17 11:09:53 -06:00
|
|
|
Discourse.reset_active_record_cache_if_needed(ex)
|
2015-08-13 21:51:23 -05:00
|
|
|
|
2014-07-11 17:03:07 -05:00
|
|
|
# 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"
|
2014-05-06 17:23:52 -05:00
|
|
|
if ex.backtrace
|
2014-07-11 17:03:07 -05:00
|
|
|
Logster.add_to_env(fake_env, :backtrace, ex.backtrace)
|
2014-05-06 17:23:52 -05:00
|
|
|
end
|
2014-07-11 17:03:07 -05:00
|
|
|
|
2015-05-05 21:28:32 -05:00
|
|
|
Logster.add_to_env(fake_env, :current_hostname, Discourse.current_hostname)
|
|
|
|
|
2014-07-11 17:03:07 -05:00
|
|
|
Thread.current[Logster::Logger::LOGSTER_ENV] = fake_env
|
|
|
|
Logster.logger.error(text)
|
2014-05-06 17:23:52 -05:00
|
|
|
rescue => e
|
2014-07-11 17:03:07 -05:00
|
|
|
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
|
2014-05-06 17:23:52 -05:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2014-07-17 15:22:46 -05:00
|
|
|
Sidekiq.error_handlers.clear
|
2014-07-11 17:03:07 -05:00
|
|
|
Sidekiq.error_handlers << SidekiqLogsterReporter.new
|