aboutsummaryrefslogblamecommitdiffstats
path: root/activejob/lib/active_job/queue_adapter.rb
blob: dd05800bafb3e70de60e6f1c4cf40224e8b9c440 (plain) (tree)
1
2
3
4
5
6
7
8
9
                             
 
                                                    

                
                                                                   
                                                                     
                              
                                 
 
               
                                                                                               
                                                                                          
                                 

       
                                                                       
                       
                                                                     
                                                                      


                       
 



                            
                                                                     
                                                         
                    
                                                  
                                                   



             


                                                       

                                                                                         
              
                                                       


                                                                                                             


                                 
             
           
 



                                                       
 
                                                              
 


                                                                        
       
     
   
# frozen_string_literal: true

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_name, instance_accessor: false, instance_predicate: false
      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

      def queue_adapter_name
        _queue_adapter_name
      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)
        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
            assign_adapter(name_or_adapter_or_class.to_s,
                           ActiveJob::QueueAdapters.lookup(name_or_adapter_or_class).new)
          else
            if queue_adapter?(name_or_adapter_or_class)
              adapter_name = "#{name_or_adapter_or_class.class.name.demodulize.remove('Adapter').underscore}"
              assign_adapter(adapter_name,
                             name_or_adapter_or_class)
            else
              raise ArgumentError
            end
          end
        end

        def assign_adapter(adapter_name, queue_adapter)
          self._queue_adapter_name = adapter_name
          self._queue_adapter = queue_adapter
        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