discourse/app/jobs/onceoff/onceoff.rb

# frozen_string_literal: true

class Jobs::Onceoff < ::Jobs::Base
  sidekiq_options retry: false

  def self.name_for(klass)
    klass.name.sub(/\AJobs\:\:/, "")
  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 = Discourse.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
        Discourse.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)
        Jobs.enqueue(job_name.underscore.to_sym) if previously_ran.exclude?(job_name)
      end
  end
end