Make sure a pool can be re-used

jarib · May 12, 2012 · 9d962db · 9d962db
1 parent 304507a
commit 9d962db
Showing 1 changed file with 13 additions and 4 deletions.
diff --git a/lib/vnctools/server_pool.rb b/lib/vnctools/server_pool.rb
@@ -3,16 +3,21 @@ class ServerPool
     include Observable
 
     def initialize(capacity, klass = Server)
-      @capacity = capacity
+      @capacity     = capacity
-      @servers  = Array.new(capacity) { klass.new }
+      @running      = []
-      @running  = []
+      @server_class = klass
+
+      create_servers
     end
 
     def stop
-      running.each do |s|
+      running.dup.each do |s|
         fire :on_display_stopping, s
         s.stop
+        running.delete s
       end
+
+      create_servers
     end
 
     def size
@@ -58,6 +63,10 @@ def next_server
       server
     end
 
+    def create_servers
+      @servers = Array.new(@capacity) { @server_class.new }
+    end
+
     class TooManyDisplaysError < StandardError
     end