Sha256: f9cdb42b2e906fc5f0e07a90b5ca7d8dd737da6f3e2d11d14e9dec469b7e6972

Contents?: true

Size: 1.76 KB

Versions: 8

Compression:

Stored size: 1.76 KB

Contents

module Shoryuken
  module Middleware
    module Server
      class AutoExtendVisibility
        include Util

        EXTEND_UPFRONT_SECONDS = 5

        def call(worker, queue, sqs_msg, body)
          if sqs_msg.is_a?(Array)
            logger.warn { "Auto extend visibility isn't supported for batch workers" }
            return yield
          end

          timer = auto_visibility_timer(worker, queue, sqs_msg, body)
          yield
        ensure
          timer.kill if timer
        end

        private

        class MessageVisibilityExtender
          include Util

          def auto_extend(worker, queue, sqs_msg, body)
            queue_visibility_timeout = Shoryuken::Client.queues(queue).visibility_timeout

            Concurrent::TimerTask.new(execution_interval: queue_visibility_timeout - EXTEND_UPFRONT_SECONDS) do
              begin
                logger.debug do
                  "Extending message #{worker_name(worker.class, sqs_msg, body)}/#{queue}/#{sqs_msg.message_id} " \
                  "visibility timeout by #{queue_visibility_timeout}s."
                end

                sqs_msg.change_visibility(visibility_timeout: queue_visibility_timeout)
              rescue => e
                logger.error do
                  'Could not auto extend the message ' \
                  "#{worker_name(worker.class, sqs_msg, body)}/#{queue}/#{sqs_msg.message_id} " \
                  "visibility timeout. Error: #{e.message}"
                end
              end
            end
          end
        end

        def auto_visibility_timer(worker, queue, sqs_msg, body)
          return unless worker.class.auto_visibility_timeout?

          MessageVisibilityExtender.new.auto_extend(worker, queue, sqs_msg, body).tap(&:execute)
        end
      end
    end
  end
end

Version data entries

8 entries across 8 versions & 1 rubygems

Version Path
shoryuken-3.0.7 lib/shoryuken/middleware/server/auto_extend_visibility.rb
shoryuken-3.0.6 lib/shoryuken/middleware/server/auto_extend_visibility.rb
shoryuken-3.0.5 lib/shoryuken/middleware/server/auto_extend_visibility.rb
shoryuken-3.0.4 lib/shoryuken/middleware/server/auto_extend_visibility.rb
shoryuken-3.0.3 lib/shoryuken/middleware/server/auto_extend_visibility.rb
shoryuken-3.0.2 lib/shoryuken/middleware/server/auto_extend_visibility.rb
shoryuken-3.0.1 lib/shoryuken/middleware/server/auto_extend_visibility.rb
shoryuken-3.0.0 lib/shoryuken/middleware/server/auto_extend_visibility.rb