create a SizedStack for the connection pool

Author: Aaron Patterson and Terence Lee

lib/resque.rb

@@ -19,6 +19,7 @@
 require 'resque/consumer'
 require 'resque/threaded_pool'
 require 'resque/connection_pool'
+require 'resque/sized_stack'
 
 require 'resque/vendor/utf8_util'
 

lib/resque/sized_stack.rb

@@ -0,0 +1,14 @@
+require 'thread'
+
+module Resque
+  ###
+  # Sized LIFO queue.
+  class SizedStack < SizedQueue
+    def initialize(size)
+      super
+      class << @que
+        alias :shift :pop
+      end
+    end
+  end
+end

test/sized_stack_test.rb

@@ -0,0 +1,25 @@
+require "test_helper"
+
+module Resque
+  describe SizedStack do
+    it "can be constructed" do
+      assert Resque::SizedStack.new(10)
+    end
+
+    it "pops off what we push" do
+      stack = Resque::SizedStack.new 10
+      thing = Object.new
+      stack.push thing
+      assert_equal thing, stack.pop
+    end
+
+    it "is a stack" do
+      stack = Resque::SizedStack.new 10
+      items = 10.times.map { Object.new }
+      items.each { |i| stack << i }
+      items.reverse_each do |i|
+        assert_equal i, stack.pop
+      end
+    end
+  end
+end