84 lines
2.0 KiB
Ruby
84 lines
2.0 KiB
Ruby
|
module Jobs
|
||
|
|
|
||
|
|
class Base
|
||
|
|
include Sidekiq::Worker
|
||
|
|
|
||
|
|
def self.delayed_perform(opts={})
|
||
|
|
self.new.perform(opts)
|
||
|
|
end
|
||
|
|
|
||
|
|
def self.mutex
|
||
|
|
@mutex ||= Mutex.new
|
||
|
|
end
|
||
|
|
|
||
|
|
def execute(opts={})
|
||
|
|
raise "Overwrite me!"
|
||
|
|
end
|
||
|
|
|
||
|
|
def perform(opts={})
|
||
|
|
opts = opts.with_indifferent_access
|
||
|
|
|
||
|
|
if opts.delete(:sync_exec)
|
||
|
|
if opts.has_key?(:current_site_id) and opts[:current_site_id] != RailsMultisite::ConnectionManagement.current_db
|
||
|
|
raise ArgumentError.new("You can't connect to another database when executing a job synchronously.")
|
||
|
|
else
|
||
|
|
return execute(opts)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
|
||
|
|
dbs =
|
||
|
|
if opts[:current_site_id]
|
||
|
|
[opts[:current_site_id]]
|
||
|
|
else
|
||
|
|
RailsMultisite::ConnectionManagement.all_dbs
|
||
|
|
end
|
||
|
|
|
||
|
|
dbs.each do |db|
|
||
|
|
begin
|
||
|
|
Jobs::Base.mutex.synchronize do
|
||
|
|
RailsMultisite::ConnectionManagement.establish_connection(:db => db)
|
||
|
|
execute(opts)
|
||
|
|
end
|
||
|
|
ensure
|
||
|
|
ActiveRecord::Base.connection_handler.clear_active_connections!
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
def self.enqueue(job_name, opts={})
|
||
|
|
|
||
|
|
klass_name = "Jobs::#{job_name.to_s.camelcase}"
|
||
|
|
klass = klass_name.constantize
|
||
|
|
|
||
|
|
# Unless we want to work on all sites
|
||
|
|
unless opts.delete(:all_sites)
|
||
|
|
opts[:current_site_id] ||= RailsMultisite::ConnectionManagement.current_db
|
||
|
|
end
|
||
|
|
|
||
|
|
# If we are able to queue a job, do it
|
||
|
|
if SiteSetting.queue_jobs?
|
||
|
|
if opts[:delay_for].present?
|
||
|
|
klass.delay_for(opts.delete(:delay_for)).delayed_perform(opts)
|
||
|
|
else
|
||
|
|
Sidekiq::Client.enqueue(klass_name.constantize, opts)
|
||
|
|
end
|
||
|
|
else
|
||
|
|
# Otherwise execute the job right away
|
||
|
|
opts.delete(:delay_for)
|
||
|
|
opts[:sync_exec] = true
|
||
|
|
klass.new.perform(opts)
|
||
|
|
end
|
||
|
|
|
||
|
|
end
|
||
|
|
|
||
|
|
def self.enqueue_in(secs, job_name, opts={})
|
||
|
|
enqueue(job_name, opts.merge!(delay_for: secs))
|
||
|
|
end
|
||
|
|
|
||
|
|
end
|
||
|
|
|
||
|
|
# Require all jobs
|
||
|
|
Dir["#{Rails.root}/lib/jobs/*"].each {|file| require_dependency file }
|