Call error block on retries

lib/relish.rb

@@ -99,23 +99,27 @@ def delete(id, version)
     end
   end
 
-  def set_error_handler(&block)
-    @error_handler = block
+  def set_error_handler(&blk)
+    @error_handler = blk
   end
 
   def rescue_dynamodb_error
     tries = @tries
     begin
       yield
     rescue => e
-      if (tries -= 1).zero?
-        raise
-      else
-        unless @error_handler.nil?
-          @error_handler.call(e)
-        end
+      retries = ((tries -= 1) > 0)
+
+      unless @error_handler.nil?
+        @error_handler.call(e, retries)
+      end
+
+      if retries
         retry
+      else
+        raise
       end
+
     end
   end
 end

spec/relish_spec.rb

@@ -32,15 +32,19 @@
         assert_requested(:post, @dynamo_url, times: 3)
       end
 
-      it "calls a custom proc so consumers can log/measure Dynamo errors" do
+      it "calls a custom proc so consumers can log/measure Dynamo errors with retries" do
         @error = nil
-        @relish.set_error_handler { |e| @error = e }
+        @retries = []
+        @relish.set_error_handler do |e, r|
+          @error = e
+          @retries << r
+        end
         assert_raise Excon::Errors::ServiceUnavailable do
           @relish.copy("1234", "1", { name: "foobar" })
         end
         assert_equal Excon::Errors::ServiceUnavailable, @error.class
+        assert_equal [true, true, false], @retries
       end
     end
-
   end
 end