ensure failed tasks run again

Author: patodevilla

lib/sidekiq_bouncer/bouncer.rb

@@ -57,11 +57,14 @@ def debounce(*params, key_or_args_indices:)
     # @param [NilClass|String] key
     # @return [False|*] true if should be excecuted
     def run(key)
-      return false unless let_in?(key)
-
-      yield.tap do
-        redis.call('DEL', key)
-      end
+      timestamp = nil
+      return false unless let_in?(key) { |t| timestamp = t }
+
+      redis.call('DEL', key)
+      yield
+    rescue StandardError => e
+      redis.call('SET', key, timestamp)
+      raise e
     end
 
     # @param [NilClass|String] key
@@ -75,7 +78,10 @@ def let_in?(key)
       # the span of DELAY_BUFFER. The first one will clear the timestamp, and the rest
       # will skip when they see that the timestamp is gone.
       timestamp = redis.call('GET', key)
-      return false if timestamp.nil? || now_i < timestamp.to_i
+      return false if timestamp.nil?
+
+      yield timestamp if block_given?
+      return false if now_i < timestamp.to_i
 
       true
     end

lib/sidekiq_bouncer/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module SidekiqBouncer
-  VERSION = '0.3.3'
+  VERSION = '0.3.4'
 end