83 lines
1.8 KiB
Ruby
83 lines
1.8 KiB
Ruby
require 'resque'
|
|
|
|
ENV['FORK_PER_JOB'] = 'false'
|
|
|
|
def shutdown
|
|
puts "Cleaning up resources..."
|
|
Stats.destroy!
|
|
EventMachine.stop_event_loop
|
|
puts "Terminated!"
|
|
exit!
|
|
end
|
|
|
|
Resque.before_first_fork do
|
|
JamWebEventMachine.start
|
|
|
|
#ActiveRecord::Base.establish_connection
|
|
config = {
|
|
influxdb_database: APP_CONFIG.influxdb_database,
|
|
influxdb_username: APP_CONFIG.influxdb_username,
|
|
influxdb_password: APP_CONFIG.influxdb_password,
|
|
influxdb_hosts: APP_CONFIG.influxdb_hosts,
|
|
influxdb_port: APP_CONFIG.influxdb_port,
|
|
influxdb_async: true # if we use async=true, the forked job will die before the stat is sent
|
|
}
|
|
|
|
# handle these events and force a shutdown. this is required I think due to influxdb-client.
|
|
Signal.trap("TERM") do
|
|
shutdown
|
|
end
|
|
Signal.trap("INT") do
|
|
shutdown
|
|
end
|
|
|
|
JamRuby::Stats.init(config)
|
|
|
|
end
|
|
# https://devcenter.heroku.com/articles/forked-pg-connections
|
|
Resque.before_fork do
|
|
|
|
#defined?(ActiveRecord::Base) and
|
|
# ActiveRecord::Base.connection.disconnect!
|
|
|
|
#JamRuby::Stats.destroy!
|
|
end
|
|
|
|
Resque.after_fork do
|
|
#defined?(ActiveRecord::Base) and
|
|
# ActiveRecord::Base.establish_connection
|
|
|
|
end
|
|
|
|
# for jobs that do not extend lonely job, just extend this module and get stats
|
|
module JamRuby
|
|
module ResqueStats
|
|
def around_perform(*args)
|
|
Stats.timer('job.stats') do
|
|
begin
|
|
yield
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
|
|
require 'resque-lonely_job'
|
|
|
|
# for jobs that extend lonely job, we override around_perform already implemented in LonelyJob, and call into it
|
|
module Resque
|
|
module Plugins
|
|
module JamLonelyJob
|
|
def around_perform(*args)
|
|
Stats.timer('job.stats') do
|
|
super
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
Resque::Plugins::JamLonelyJob.module_eval { include Resque::Plugins::LonelyJob }
|
|
|