52 lines
1.6 KiB
Ruby
52 lines
1.6 KiB
Ruby
|
require "active_support/core_ext/string/inflections"
|
||
|
|
|
||
|
|
module ActiveJob
|
||
|
|
# The <tt>ActiveJob::QueueAdapter</tt> module is used to load the
|
||
|
|
# correct adapter. The default queue adapter is the +:async+ queue.
|
||
|
|
module QueueAdapter #:nodoc:
|
||
|
|
extend ActiveSupport::Concern
|
||
|
|
|
||
|
|
included do
|
||
|
|
class_attribute :_queue_adapter, instance_accessor: false, instance_predicate: false
|
||
|
|
self.queue_adapter = :async
|
||
|
|
end
|
||
|
|
|
||
|
|
# Includes the setter method for changing the active queue adapter.
|
||
|
|
module ClassMethods
|
||
|
|
# Returns the backend queue provider. The default queue adapter
|
||
|
|
# is the +:async+ queue. See QueueAdapters for more information.
|
||
|
|
def queue_adapter
|
||
|
|
_queue_adapter
|
||
|
|
end
|
||
|
|
|
||
|
|
# Specify the backend queue provider. The default queue adapter
|
||
|
|
# is the +:async+ queue. See QueueAdapters for more
|
||
|
|
# information.
|
||
|
|
def queue_adapter=(name_or_adapter_or_class)
|
||
|
|
self._queue_adapter = interpret_adapter(name_or_adapter_or_class)
|
||
|
|
end
|
||
|
|
|
||
|
|
private
|
||
|
|
|
||
|
|
def interpret_adapter(name_or_adapter_or_class)
|
||
|
|
case name_or_adapter_or_class
|
||
|
|
when Symbol, String
|
||
|
|
ActiveJob::QueueAdapters.lookup(name_or_adapter_or_class).new
|
||
|
|
else
|
||
|
|
if queue_adapter?(name_or_adapter_or_class)
|
||
|
|
name_or_adapter_or_class
|
||
|
|
else
|
||
|
|
raise ArgumentError
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
QUEUE_ADAPTER_METHODS = [:enqueue, :enqueue_at].freeze
|
||
|
|
|
||
|
|
def queue_adapter?(object)
|
||
|
|
QUEUE_ADAPTER_METHODS.all? { |meth| object.respond_to?(meth) }
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|