Contents

module Replidog
  class ProxyHandler
    def initialize
      @proxies = {}
      @class_to_proxy = {}
    end

    def establish_connection(configuration, klass)
      @proxies[configuration] ||= Proxy.new(self, configuration)
      @class_to_proxy[klass.name] ||= @proxies[configuration]
    end

    def retrieve_proxy(klass)
      proxy = @class_to_proxy[klass.name]
      return proxy if proxy
      return nil if ActiveRecord::Base == klass
      retrieve_proxy(klass.superclass)
    end

    def remove_connection(klass)
      proxy = @class_to_proxy.delete(klass.name)
      return nil unless proxy

      @proxies.delete(proxy.configuration)
      proxy.clear_all_slave_connections!
      proxy.configuration.config
    end

    def clear_active_slave_connections!
      @proxies.each_value do |proxy|
        proxy.clear_active_slave_connections!
      end
    end

    def clear_reloadable_slave_connections!
      @proxies.each_value do |proxy|
        proxy.clear_reloadable_slave_connections!
      end
    end

    def clear_all_slave_connections!
      @proxies.each_value do |proxy|
        proxy.clear_all_slave_connections!
      end
    end

    def enable_query_cache!
      ActiveRecord::Base.connection_handler.connection_pools.each_value do |connection_pool|
        connection_pool.connection.enable_query_cache!
      end
      @proxies.each_value(&:enable_query_cache_for_slaves!)
    end

    def disable_query_cache!
      ActiveRecord::Base.connection_handler.connection_pools.each_value do |connection_pool|
        connection_pool.connection.disable_query_cache!
      end
      @proxies.each_value(&:disable_query_cache_for_slaves!)
    end

    def clear_query_cache
      ActiveRecord::Base.connection_handler.connection_pools.each_value do |connection_pool|
        connection_pool.connection.clear_query_cache
      end
      @proxies.each_value(&:clear_query_cache_for_slaves)
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
replidog-0.1.0	lib/replidog/proxy_handler.rb