activerecord_partitioning 0.0.1

data/README.md

@@ -0,0 +1,6 @@
+ActiveRecord Partitioning
+======================
+
+An ActiveRecord ConnectionPools class supports switching connection pool by database config instead of ActiveRecord model class name.
+
+It is only support ActiveRecord 2.3.18 currently.

data/lib/activerecord_partitioning/connection_pools.rb

@@ -0,0 +1,39 @@
+module ActiveRecordPartitioning
+  class NoActiveConnectionPoolError < StandardError
+  end
+
+  class ConnectionPools
+    attr_reader :store
+
+    def initialize(store={})
+      @store = store
+    end
+
+    def [](key)
+      config = ActiveRecordPartitioning.current_connection_pool_config
+      raise NoActiveConnectionPoolError if config.nil?
+      @store[connection_pool_key(config)]
+    end
+
+    def []=(key, pool)
+      @store[connection_pool_key(pool.spec.config)] = pool
+    end
+
+    def delete_if(&block)
+      @store.delete_if(&block)
+    end
+
+    def each_value(&block)
+      @store.each_value(&block)
+    end
+
+    def size
+      @store.size
+    end
+
+    private
+    def connection_pool_key(config)
+      config[:url]
+    end
+  end
+end

data/lib/activerecord_partitioning.rb

@@ -0,0 +1,28 @@
+require 'active_record'
+require 'activerecord_partitioning/connection_pools'
+
+module ActiveRecordPartitioning
+  module_function
+  def setup(base_config = {}, pools = {})
+    @base_config = base_config.symbolize_keys
+    ActiveRecord::Base.connection_handler = ActiveRecord::ConnectionAdapters::ConnectionHandler.new(ConnectionPools.new(pools))
+  end
+
+  def with_connection_pool(config, &block)
+    self.current_connection_pool_config = config = config.symbolize_keys
+    if ActiveRecord::Base.connection_pool.nil?
+      ActiveRecord::Base.establish_connection(@base_config.merge(config))
+    end
+    yield if block_given?
+  ensure
+    self.current_connection_pool_config = nil
+  end
+
+  def current_connection_pool_config
+    Thread.current[:current_connection_pool_config]
+  end
+
+  def current_connection_pool_config=(config)
+    Thread.current[:current_connection_pool_config] = config
+  end
+end

metadata

@@ -0,0 +1,113 @@
+--- !ruby/object:Gem::Specification 
+name: activerecord_partitioning
+version: !ruby/object:Gem::Version 
+  hash: 29
+  prerelease: 
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Xiao Li
+- sdqali
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2013-10-10 00:00:00 Z
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: activerecord
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - "="
+      - !ruby/object:Gem::Version 
+        hash: 39
+        segments: 
+        - 2
+        - 3
+        - 18
+        version: 2.3.18
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: rake
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id002
+- !ruby/object:Gem::Dependency 
+  name: sqlite3
+  prerelease: false
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id003
+description: An ActiveRecord ConnectionPools class supports switching connection pool by database config instead of ActiveRecord model class name
+email: 
+- swing1979@gmail.com
+- sadiqalikm@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- README.md
+- lib/activerecord_partitioning/connection_pools.rb
+- lib/activerecord_partitioning.rb
+homepage: https://github.com/ThoughtWorksStudios/activerecord_partitioning
+licenses: 
+- MIT
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: An ActiveRecord ConnectionPools class supports switching connection pool by database config instead of ActiveRecord model class name
+test_files: []
+