db-charmer 1.6.19 → 1.7.0.pre1

data/lib/db_charmer/multi_db_migrations.rb

@@ -1,67 +0,0 @@
-module DbCharmer
-  module MultiDbMigrations
-    def self.extended(base)
-      class << base
-        alias_method_chain :migrate, :db_wrapper
-      end
-    end
-
-    @@multi_db_names = {}
-    def multi_db_names
-      puts "Retrieving connections for #{self.name}"
-      @@multi_db_names[self.name] || @@multi_db_names['ActiveRecord::Migration']
-    end
-
-    def multi_db_names=(names)
-      puts "Setting connections for #{self.name}"
-      @@multi_db_names[self.name] = names
-    end
-
-    def migrate_with_db_wrapper(direction)
-      if names = multi_db_names
-        names.each do |multi_db_name|
-          on_db(multi_db_name) do
-            migrate_without_db_wrapper(direction)
-          end
-        end
-      else
-        migrate_without_db_wrapper(direction)
-      end
-    end
-
-    def on_db(db_name)
-      name = db_name.is_a?(Hash) ? db_name[:connection_name] : db_name.inspect
-      announce "Switching connection to #{name}"
-      # Switch connection
-      old_proxy = ActiveRecord::Base.db_charmer_connection_proxy
-      db_name = nil if db_name == :default
-      ActiveRecord::Base.switch_connection_to(db_name, DbCharmer.connections_should_exist?)
-      # Yield the block
-      yield
-    ensure
-      # Switch it back
-      ActiveRecord::Base.verify_active_connections!
-      announce "Switching connection back"
-      ActiveRecord::Base.switch_connection_to(old_proxy)
-    end
-
-    def db_magic(opts = {})
-      # Collect connections from all possible options
-      conns = [ opts[:connection], opts[:connections] ]
-      conns << shard_connections(opts[:sharded_connection]) if opts[:sharded_connection]
-
-      # Get a unique set of connections
-      conns = conns.flatten.compact.uniq
-      raise ArgumentError, "No connection name - no magic!" unless conns.any?
-
-      # Save connections
-      self.multi_db_names = conns
-    end
-
-    # Return a list of connections to shards in a sharded connection
-    def shard_connections(conn_name)
-      conn = DbCharmer::Sharding.sharded_connection(conn_name)
-      conn.shard_connections
-    end
-  end
-end

data/lib/db_charmer/multi_db_proxy.rb

@@ -1,63 +0,0 @@
-module DbCharmer
-  module MultiDbProxy
-    # Simple proxy class that switches connections and then proxies all the calls
-    # This class is used to implement chained on_db calls
-    class OnDbProxy < BlankSlate
-      def initialize(proxy_target, slave)
-        @proxy_target = proxy_target
-        @slave = slave
-      end
-
-    private
-
-      def method_missing(meth, *args, &block)
-        # Switch connection and proxy the method call
-        @proxy_target.on_db(@slave) do |proxy_target|
-          res = proxy_target.__send__(meth, *args, &block)
-
-          # If result is a scope/association, return a new proxy for it, otherwise return the result itself
-          (res.proxy?) ? OnDbProxy.new(res, @slave) : res
-        end
-      end
-    end
-
-    module ClassMethods
-      def on_db(con, proxy_target = nil)
-        proxy_target ||= self
-
-        # Chain call
-        return OnDbProxy.new(proxy_target, con) unless block_given?
-
-        # Block call
-        begin
-          self.db_charmer_connection_level += 1
-          old_proxy = db_charmer_connection_proxy
-          switch_connection_to(con, DbCharmer.connections_should_exist?)
-          yield(proxy_target)
-        ensure
-          switch_connection_to(old_proxy)
-          self.db_charmer_connection_level -= 1
-        end
-      end
-    end
-
-    module InstanceMethods
-      def on_db(con, proxy_target = nil, &block)
-        proxy_target ||= self
-        self.class.on_db(con, proxy_target, &block)
-      end
-    end
-
-    module MasterSlaveClassMethods
-      def on_slave(con = nil, proxy_target = nil, &block)
-        con ||= db_charmer_random_slave
-        raise ArgumentError, "No slaves found in the class and no slave connection given" unless con
-        on_db(con, proxy_target, &block)
-      end
-
-      def on_master(proxy_target = nil, &block)
-        on_db(db_charmer_default_connection, proxy_target, &block)
-      end
-    end
-  end
-end

data/lib/db_charmer/scope_proxy.rb

@@ -1,22 +0,0 @@
-module DbCharmer
-  module ScopeProxy
-    module InstanceMethods
-      def proxy?
-        true
-      end
-
-      def on_db(con, proxy_target = nil, &block)
-        proxy_target ||= self
-        proxy_scope.on_db(con, proxy_target, &block)
-      end
-
-      def on_slave(con = nil, &block)
-        proxy_scope.on_slave(con, self, &block)
-      end
-
-      def on_master(&block)
-        proxy_scope.on_master(self, &block)
-      end
-    end
-  end
-end