wye 0.0.1

data/Gemfile

@@ -0,0 +1,3 @@
+source "http://rubygems.org"
+
+gemspec

data/MIT-LICENSE

@@ -0,0 +1,20 @@
+Copyright 2012 Daniel Duvall
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.rdoc

@@ -0,0 +1,14 @@
+= Wye
+
+Wye provides users of ActiveRecord 3.2 with scopes and block methods for
+contextual execution of database queries on alternative database connections.
+Wye's patterns are most useful to applications making use of either replicated
+or federated databases.
+
+== Basic Usage Patterns
+
+(Documentation is forthcoming.)
+
+== Configuration
+
+(Documentation is forthcoming.)

data/lib/wye/active_record/connection_handler.rb

@@ -0,0 +1,81 @@
+require 'active_record'
+
+module Wye
+  module ActiveRecord
+    # Implements an identical interface to the standard AR connection handler,
+    # but with support for alternate connection pools for each spec.
+    #
+    class ConnectionHandler < ::ActiveRecord::ConnectionAdapters::ConnectionHandler
+      attr_reader :connection_pools
+      attr_reader :switch
+
+      def initialize(*arguments)
+        super
+        @class_to_alternate_pools = {}
+        @switch = Switch.new(::ActiveRecord::Base)
+      end
+
+      def establish_connection(name, spec)
+        super.tap do
+          @class_to_alternate_pools[name] ||= {}
+
+          alternate_specs(spec).each do |alternate,spec|
+            @connection_pools[spec] ||= ::ActiveRecord::ConnectionAdapters::ConnectionPool.new(spec)
+            @class_to_alternate_pools[name][alternate] = @connection_pools[spec]
+          end
+        end
+      end
+
+      def remove_connection(klass)
+        super.tap do
+          if pools = @class_to_alternate_pools.delete(klass.name)
+            pools.each do |(alternate,pool)|
+              @connection_pools.delete pool.spec
+              pool.automatic_reconnect = false
+              pool.disconnect!
+            end
+          end
+        end
+      end
+
+      def retrieve_alternate_connection_pool(klass, alternate)
+        pools = @class_to_alternate_pools[klass.name]
+        return pools[alternate] if pools && pools.include?(alternate)
+        return nil if ::ActiveRecord::Base == klass
+        retrieve_alternate_connection_pool(klass.superclass, alternate)
+      end
+
+      def retrieve_main_connection_pool(klass)
+        pool = @class_to_pool[klass.name]
+        return pool if pool
+        return nil if ::ActiveRecord::Base == klass
+        retrieve_main_connection_pool(klass.superclass)
+      end
+
+      def retrieve_connection_pool(klass)
+        if alternate = switch.current_alternate(klass)
+          retrieve_alternate_connection_pool(klass, alternate)
+        else
+          retrieve_main_connection_pool(klass)
+        end
+      end
+
+      private
+
+      # Returns the alternate specs for the given one.
+      #
+      def alternate_specs(spec)
+        (spec.config[:alternates] || {}).inject({}) do |specs,(alternate,alternate_config)|
+          specs.tap { |specs| specs[alternate] = make_alternate_spec(spec, alternate_config) }
+        end
+      end
+
+      def make_alternate_spec(spec, alternate_config)
+        config = spec.config.clone
+        config.delete(:alternates)
+        alternate_config.each { |key,value| config[key.to_sym] = value }
+        ::ActiveRecord::Base::ConnectionSpecification.new(config, "#{config[:adapter]}_connection")
+      end
+    end
+  end
+end

data/lib/wye/active_record.rb

@@ -0,0 +1,5 @@
+module Wye
+  module ActiveRecord
+    autoload :ConnectionHandler, 'wye/active_record/connection_handler'
+  end
+end

data/lib/wye/railtie.rb

@@ -0,0 +1,12 @@
+require 'wye'
+require 'rails'
+
+module Wye
+  class Railtie < Rails::Railtie
+    initializer 'wye.set_connection_handler', :before => 'active_record.initialize_database' do
+      ActiveSupport.on_load(:active_record) do
+        ::ActiveRecord::Base.connection_handler = Wye::ActiveRecord::ConnectionHandler.new
+      end
+    end
+  end
+end

data/lib/wye/switch.rb

@@ -0,0 +1,43 @@
+module Wye
+  class Switch
+    attr_reader :base_class
+
+    def initialize(base_class)
+      @base_class = base_class
+      @thread_to_class_to_alternate = {}
+      @mutex = Mutex.new
+    end
+
+    def current_alternate(klass, cta = nil)
+      return nil unless klass
+
+      cta ||= class_to_alternate
+      alternate = cta[klass]
+
+      return alternate.first unless alternate.nil? || alternate.empty?
+      return nil if klass == base_class
+      current_alternate(klass.superclass, cta)
+    end
+
+    def on(klass, alternate)
+      class_to_alternate[klass] ||= []
+      class_to_alternate[klass].unshift(alternate)
+      yield
+    ensure
+      class_to_alternate[klass].shift
+      clean_up_after_dead_threads
+    end
+
+    private
+
+    def class_to_alternate(thread = Thread.current)
+      cta = @thread_to_class_to_alternate[thread] ||= {}
+      cta.empty? ? (thread == Thread.main ? cta : class_to_alternate(Thread.main)) : cta
+    end
+
+    def clean_up_after_dead_threads
+      @mutex.synchronize { @thread_to_class_to_alternate.delete_if { |thread| !thread.alive? } }
+    end
+
+  end
+end

data/lib/wye/version.rb

@@ -0,0 +1,3 @@
+module Wye
+  VERSION = '0.0.1'
+end

data/lib/wye.rb

@@ -0,0 +1,6 @@
+module Wye
+  autoload :ActiveRecord, 'wye/active_record'
+  autoload :Switch, 'wye/switch'
+end
+
+require 'wye/railtie' if defined?(Rails)

metadata

@@ -0,0 +1,89 @@
+--- !ruby/object:Gem::Specification
+name: wye
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Daniel Duvall
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-11-07 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: activerecord
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 3.2.8
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 3.2.8
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '2.5'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '2.5'
+description: Wye provides users of ActiveRecord 3.2 with scopes and block methods
+  for contextual execution of database queries on alternative database connections.
+  Wye's patterns are most useful to applications making use of either replicated or
+  federated databases.
+email:
+- dan@mutual.io
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/wye/active_record/connection_handler.rb
+- lib/wye/active_record.rb
+- lib/wye/railtie.rb
+- lib/wye/switch.rb
+- lib/wye/version.rb
+- lib/wye.rb
+- MIT-LICENSE
+- Gemfile
+- README.rdoc
+homepage: https://github.com/lettersandlight/wye
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: Simple contextual switching of ActiveRecord connection pools.
+test_files: []