Contents

require 'thrift_client/connection'

module ThriftHelpers
  class Server
    class ServerMarkedDown < StandardError; end

    def initialize(connection_string, cached = true)
      @connection_string = connection_string
      @connection = nil
      @cached = cached
      @marked_down_til = nil
    end

    def mark_down!(til)
      close(true)
      @marked_down_til = Time.now + til
    end

    def up?
      !down?
    end

    def down?
      @marked_down_til && @marked_down_til > Time.now
    end

    def to_s
      @connection_string
    end

    def open(trans, wrap, conn_timeout, trans_timeout)
      if down?
        raise ServerMarkedDown, "marked down until #{@marked_down_til}"
      end

      if @connection.nil? || (@cached && !@connection.open?)
        @connection = Connection::Factory.create(trans, wrap, @connection_string, conn_timeout)
        @connection.connect!
      end

      if wrap || trans.respond_to?(:timeout=)
        timeout = trans_timeout
      end

      self
    end

    def open?
      @connection && @connection.open?
    end

    def close(teardown = false)
      if teardown || !@cached
        @connection.close rescue nil #TODO
        @connection = nil
      end
    end

    def transport
      return nil unless @connection
      @connection.transport
    end

    module TransportInterface
      def read(sz)
        transport.read(sz)
      end

      def read_byte
        transport.read_byte
      end

      def read_into_buffer(buffer, size)
        transport.read_into_buffer(buffer, size)
      end

      def read_all(sz)
        transport.read_all(sz)
      end

      def write(buf)
        transport.write(buf)
      end
      alias_method :<<, :write

      def flush
        transport.flush
      end

      def timeout=(timeout)
        transport.timeout = timeout
      end

      def timeout
        transport.timeout
      end
    end
    include TransportInterface
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
thrift_client-0.8.2	lib/thrift_client/server.rb