Show a warn message when batch isn't supported

ruby-shoryuken · Jan 10, 2017 · 025fc83 · 025fc83
1 parent 3dce8d3
commit 025fc83
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/lib/shoryuken/middleware/server/auto_extend_visibility.rb b/lib/shoryuken/middleware/server/auto_extend_visibility.rb
@@ -4,10 +4,18 @@ 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)
+
           begin
             yield
           ensure

diff --git a/lib/shoryuken/middleware/server/exponential_backoff_retry.rb b/lib/shoryuken/middleware/server/exponential_backoff_retry.rb
@@ -5,6 +5,11 @@ class ExponentialBackoffRetry
         include Util
 
         def call(worker, queue, sqs_msg, body)
+          if sqs_msg.is_a?(Array)
+            logger.warn { "Exponential backoff isn't supported for batch workers"  }
+            return yield
+          end
+
           started_at = Time.now
           yield
         rescue