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