pools 0.0.3 → 0.0.7

data/lib/cassandra/pooled.rb

@@ -0,0 +1,32 @@
+require 'pools'
+require 'cassandra/0.7'
+
+class Cassandra
+  class Pooled
+    include ::Pools::Pooled
+
+    def initialize(*args)
+      @cassandra_args = args.dup
+      super
+    end
+
+    def __connection
+      Cassandra.new(*@cassandra_args)
+    end
+
+    def __disconnect(client)
+      client.disconnect! if client
+    end
+
+    preparation_methods :login!, :disable_node_auto_discovery!
+
+    connection_methods :keyspace, :keyspaces, :servers, :schema,
+      :auth_request, :thrift_client_options, :thrift_client_class,
+      :insert, :remove, :count_columns, :multi_count_columns,
+      :get_columns, :multi_get_columns, :get, :multi_get, :exists?,
+      :get_range, :count_range, :batch, :schema_agreement?, :version,
+      :cluster_name, :ring, :partitioner, :truncate!, :clear_keyspace!,
+      :add_column_family, :drop_column_family, :rename_column_family,
+      :add_keyspace, :drop_keyspace, :rename_keyspace
+  end
+end

data/lib/pools/connection_pool.rb

@@ -219,6 +219,7 @@ module Pools
 
     def checkout_new_connection
       c = @pooled.__connection
+      @pooled.__prepare(c)
       @connections << c
       checkout_connection(c)
     end

data/lib/pools/pooled.rb

@@ -4,10 +4,11 @@ require 'active_support/concern'
 module Pools
   module Pooled
     extend ActiveSupport::Concern
-    attr_reader :connection_pool
+    attr_reader :connection_pool, :preparation_chain
 
     def initialize(*args)
       options = args.extract_options!
+      @preparation_chain = []
       @connection_pool = ConnectionPool.new(self, options)
       Pools.handler.add(@connection_pool, options[:pool_name])
     end
@@ -24,16 +25,28 @@ module Pools
       # Override in parent
     end
 
+    def __prepare(connection)
+      @preparation_chain.each { |args| connection.send(*args) }
+    end
+
     module ClassMethods
       def connection_methods(*methods)
         methods.each do |method|
-          define_method(method) do |*params|
+          define_method(method) do |*params, &block|
             with_connection do |client|
-              client.send(method, *params)
+              client.send(method, *params, &block)
             end
           end
         end
       end
+
+      def preparation_methods(*methods)
+        methods.each do |method|
+          define_method(method) do |*params|
+            @preparation_chain << ([method] + params)
+          end
+        end
+      end
     end
   end
 end

data/lib/redis/pooled.rb

@@ -1,3 +1,4 @@
+require 'pools'
 require 'redis'
 
 class Redis

metadata

@@ -1,29 +1,39 @@
 --- !ruby/object:Gem::Specification 
 name: pools
 version: !ruby/object:Gem::Version 
+  hash: 17
   prerelease: 
-  version: 0.0.3
+  segments: 
+  - 0
+  - 0
+  - 7
+  version: 0.0.7
 platform: ruby
 authors: 
-  - Michael Rykov
+- Michael Rykov
 autorequire: 
 bindir: bin
 cert_chain: []
 
-date: 2011-03-13 00:00:00 -08:00
+date: 2011-05-22 00:00:00 -07:00
 default_executable: 
 dependencies: 
-  - !ruby/object:Gem::Dependency 
-    name: activesupport
-    prerelease: false
-    requirement: &id001 !ruby/object:Gem::Requirement 
-      none: false
-      requirements: 
-        - - ~>
-          - !ruby/object:Gem::Version 
-            version: 3.0.5
-    type: :runtime
-    version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: activesupport
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 13
+        segments: 
+        - 3
+        - 0
+        - 5
+        version: 3.0.5
+  type: :runtime
+  version_requirements: *id001
 description: |
   Generalized connection pooling
 
@@ -35,17 +45,18 @@ extensions: []
 extra_rdoc_files: []
 
 files: 
-  - README.md
-  - Rakefile
-  - LICENSE
-  - lib/pools.rb
-  - lib/pools/connection_pool.rb
-  - lib/pools/handler.rb
-  - lib/pools/middleware.rb
-  - lib/pools/pooled.rb
-  - lib/redis/pooled.rb
-  - lib/redis/pooled_store.rb
-has_rdoc: true
+- README.md
+- Rakefile
+- LICENSE
+- lib/cassandra/pooled.rb
+- lib/pools/connection_pool.rb
+- lib/pools/handler.rb
+- lib/pools/middleware.rb
+- lib/pools/pooled.rb
+- lib/pools.rb
+- lib/redis/pooled.rb
+- lib/redis/pooled_store.rb
+has_rdoc: false
 homepage: http://github.com/rykov/pools
 licenses: []
 
@@ -53,23 +64,29 @@ post_install_message:
 rdoc_options: []
 
 require_paths: 
-  - lib
+- lib
 required_ruby_version: !ruby/object:Gem::Requirement 
   none: false
   requirements: 
-    - - ">="
-      - !ruby/object:Gem::Version 
-        version: "0"
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
 required_rubygems_version: !ruby/object:Gem::Requirement 
   none: false
   requirements: 
-    - - ">="
-      - !ruby/object:Gem::Version 
-        version: "0"
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
 requirements: []
 
 rubyforge_project: 
-rubygems_version: 1.5.1
+rubygems_version: 1.6.2
 signing_key: 
 specification_version: 3
 summary: Generalized connection pooling