mirror of
https://gh.wpcy.net/https://github.com/discourse/discourse.git
synced 2026-06-17 06:29:01 +08:00
30990006a9
This reduces chances of errors where consumers of strings mutate inputs and reduces memory usage of the app. Test suite passes now, but there may be some stuff left, so we will run a few sites on a branch prior to merging
43 lines
1 KiB
Ruby
Vendored
43 lines
1 KiB
Ruby
Vendored
# frozen_string_literal: true
|
|
|
|
class Jobs::Onceoff < Jobs::Base
|
|
sidekiq_options retry: false
|
|
|
|
def self.name_for(klass)
|
|
klass.name.sub(/^Jobs\:\:/, '')
|
|
end
|
|
|
|
def running_key_name
|
|
"#{self.class.name}:running"
|
|
end
|
|
|
|
# Pass `force: true` to force it happen again
|
|
def execute(args)
|
|
job_name = self.class.name_for(self.class)
|
|
has_lock = $redis.setnx(running_key_name, Time.now.to_i)
|
|
|
|
# If we can't get a lock, just noop
|
|
if args[:force] || has_lock
|
|
begin
|
|
return if OnceoffLog.where(job_name: job_name).exists? && !args[:force]
|
|
execute_onceoff(args)
|
|
OnceoffLog.create!(job_name: job_name)
|
|
ensure
|
|
$redis.del(running_key_name) if has_lock
|
|
end
|
|
end
|
|
|
|
end
|
|
|
|
def self.enqueue_all
|
|
previously_ran = OnceoffLog.pluck(:job_name).uniq
|
|
|
|
ObjectSpace.each_object(Class).select { |klass| klass < self }.each do |klass|
|
|
job_name = name_for(klass)
|
|
unless previously_ran.include?(job_name)
|
|
Jobs.enqueue(job_name.underscore.to_sym)
|
|
end
|
|
end
|
|
end
|
|
|
|
end
|
