Merge pull request #44 from Shopify/reset

add support for reset of RateQueue

lib/limiter/rate_queue.rb

@@ -8,10 +8,19 @@ def initialize(size, interval: 60, balanced: false, &blk)
       @size = size
       @interval = interval
 
-      @ring = balanced ? balanced_ring : unbalanced_ring
-      @head = 0
+      @balanced = balanced
+
       @mutex = Mutex.new
       @blk = blk
+
+      reset
+    end
+
+    def reset
+      @mutex.synchronize do
+        @ring = @balanced ? balanced_ring : unbalanced_ring
+        @head = 0
+      end
     end
 
     def shift

test/limiter/rate_queue_test.rb

@@ -36,6 +36,15 @@ def test_shift_is_rate_limited_across_multiple_threads
       end
     end
 
+    def test_shift_is_not_rate_limited
+      assert_elapsed(0) do
+        COUNT.times do
+          @queue.shift
+          @queue.reset
+        end
+      end
+    end
+
     def test_block_was_called_on_rate_limit
       @block_hit = false
       @queue = RateQueue.new(RATE, interval: INTERVAL) { @block_hit = true }