Contents

#
# ActiveRecord read from a slave
#
module ActiveRecordSlave

  # Install ActiveRecord::Slave into ActiveRecord to redirect reads to the slave
  # By default, only the default Database adapter (ActiveRecord::Base.connection.class)
  # is extended with slave read capabilities
  def self.install!(adapter_class = nil)
    if slave_config = ActiveRecord::Base.connection.config[:slave]
      Rails.logger.info "ActiveRecordSlave.install! v#{ActiveRecordSlave::VERSION} Establishing connection to slave database"
      Slave.establish_connection(slave_config)

      # Inject a new #select method into the ActiveRecord Database adapter
      base = adapter_class || ActiveRecord::Base.connection.class
      base.send(:include, InstanceMethods)
      base.alias_method_chain(:select, :slave_reader)
    end
  end

  # Force reads for the supplied block to read from the master database
  # Only applies to calls made within the current thread
  def self.read_from_master
    # Set :master indicator in thread local storage so that it is visible
    # during the select call
    current = Thread.current[:active_record_slave]
    Thread.current[:active_record_slave] = :master
    yield
  ensure
    Thread.current[:active_record_slave] = current
  end

end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
active_record_slave-0.2.0	lib/active_record_slave/active_record_slave.rb