Move wrapper into separate file

lib/connection_pool.rb

@@ -1,6 +1,7 @@
 require 'connection_pool/version'
 require 'connection_pool/errors'
 require 'connection_pool/timed_stack'
+require 'connection_pool/wrapper'
 
 
 # Generic connection pool class for sharing a limited number of objects or network connections

lib/connection_pool/wrapper.rb

@@ -0,0 +1,39 @@
+class ConnectionPool
+  class Wrapper < ::BasicObject
+    METHODS = [:with, :pool_shutdown, :wrapped_pool]
+
+    def initialize(options = {}, &block)
+      @pool = options.fetch(:pool) { ::ConnectionPool.new(options, &block) }
+    end
+
+    def wrapped_pool
+      @pool
+    end
+
+    def with(&block)
+      @pool.with(&block)
+    end
+
+    def pool_shutdown(&block)
+      @pool.shutdown(&block)
+    end
+
+    def pool_size
+      @pool.size
+    end
+
+    def pool_available
+      @pool.available
+    end
+
+    def respond_to?(id, *args)
+      METHODS.include?(id) || with { |c| c.respond_to?(id, *args) }
+    end
+
+    def method_missing(name, *args, &block)
+      with do |connection|
+        connection.send(name, *args, &block)
+      end
+    end
+  end
+end