Contents

require 'yaml'

module EbisuConnection
  class ConnectionManager
    CHECK_INTERVAL = 1.minute

    class << self
      attr_writer :slaves_file
      attr_accessor :slave_type

      def slaves_file
        @slaves_file || File.join(Rails.root, "config/slave.yaml")
      end
    end

    def initialize
      @mutex = Mutex.new
    end

    def slave_connection
      slaves.sample.connection
    end

    def put_aside!
      return unless @file_mtime

      now = Time.now
      @check_time ||= now
      return if now - @check_time < CHECK_INTERVAL
      @check_time = now

      mtime = File.mtime(self.class.slaves_file)
      return if @file_mtime == mtime

      clear_all_connection!
    end

    def clear_all_connection!
      @mutex.synchronize do
        @slaves.values.each do |s|
          s.all_disconnect!
        end

        @slaves = nil
        @slave_conf = nil
        @spec = nil
      end
    end

    private

    def slaves
      @mutex.synchronize do
        @slaves ||= {}
        @slaves[current_thread_id] ||= get_slaves
      end
    end

    def get_slaves
      EbisuConnection::Slaves.new(slaves_conf, spec)
    end

    def slaves_conf
      @slaves_conf ||= get_slaves_conf
    end

    def spec
      @spec ||= get_spec
    end

    def get_slaves_conf
      @file_mtime = File.mtime(self.class.slaves_file)
      conf = YAML.load_file(self.class.slaves_file)
      self.class.slave_type ? conf[self.class.slave_type] : conf
    end

    def get_spec
      ret = ActiveRecord::Base.configurations[Rails.env]
      ret.merge(ret["slave"] || {})
    end

    def current_thread_id
      Thread.current.object_id
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
ebisu_connection-0.0.1	lib/ebisu_connection/connection_manager.rb