ryandotsmith/queue_classic

# frozen_string_literal: true

module QC
  # QC Configuration
  module Config
    # You can use the APP_NAME to query for
    # postgres related process information in the
    # pg_stat_activity table.
    def app_name
      @app_name ||= ENV['QC_APP_NAME'] || 'queue_classic'
    end

    # Number of seconds to block on the listen chanel for new jobs.
    def wait_time
      @wait_time ||= (ENV['QC_LISTEN_TIME'] || 5).to_i
    end

    # Why do you want to change the table name?
    # Just deal with the default OK?
    # If you do want to change this, you will
    # need to update the PL/pgSQL lock_head() function.
    # Come on. Don't do it.... Just stick with the default.
    def table_name
      @table_name ||= 'queue_classic_jobs'
    end

    def queue
      @queue = ENV['QUEUE'] || 'default'
    end

    # The default queue used by `QC.enqueue`.
    def default_queue
      @default_queue ||= Queue.new(QC.queue)
    end

    def default_queue=(queue)
      @default_queue = queue
    end

    # Each row in the table will have a column that
    # notes the queue. You can point your workers
    # at different queues.
    def queues
      @queues ||= (ENV.fetch('QUEUES', nil) && ENV['QUEUES'].split(',').map(&:strip)) || []
    end

    # Set this to 1 for strict FIFO.
    # There is nothing special about 9....
    def top_bound
      @top_bound ||= (ENV['QC_TOP_BOUND'] || 9).to_i
    end

    # Set this variable if you wish for
    # the worker to fork a UNIX process for
    # each locked job. Remember to re-establish
    # any database connections. See the worker
    # for more details.
    def fork_worker?
      @fork_worker ||= !ENV['QC_FORK_WORKER'].nil?
    end

    # The worker class instantiated by QC's rake tasks.
    def default_worker_class
      @default_worker_class ||= begin
        class_name = ENV.fetch('QC_DEFAULT_WORKER_CLASS', nil)
        class_name ? Kernel.const_get(class_name) : QC::Worker
      rescue NameError
        QC::Worker
      end
    end

    def default_worker_class=(worker_class)
      @default_worker_class = worker_class
    end

    # reset memoized configuration
    def reset_config
      # TODO: we might want to think about storing these in a Hash.
      @app_name = nil
      @wait_time = nil
      @table_name = nil
      @queue = nil
      @default_queue = nil
      @queues = nil
      @top_bound = nil
      @fork_worker = nil
      @worker_class = nil
    end
  end
end