payrollhero / dispatch-rider / #3015

module DispatchRider

  module Handlers

    class Base

      include NamedProcess

      extend InheritanceTracking

      class << self

        def set_default_retry( amount ) 

          define_method(:retry_timeout) do

            amount

          end

        end

      end

      attr_reader :raw_message

      def do_process(raw_message)

        with_named_process(raw_message) do

          @raw_message = raw_message

          process(raw_message.body)

        end

      rescue Exception => e

        self.retry if self.retry_on_failure?

        raise e

      end

      def process(message)

        raise NotImplementedError, "Method 'process' not overridden in subclass!"

      end

      protected

      def extend_timeout(timeout)

        raw_message.extend_timeout(timeout)

      end

      def return_to_queue

        raw_message.return_to_queue

      end

      def retry

        timeout = retry_timeout

        case timeout

        when :immediate

          return_to_queue

        else

          extend_timeout(timeout)

        end

      end

      def retry_on_failure?

        self.respond_to? :retry_timeout

      end

    end

  end

end