class HTTP::Timeout::PerOperation

def connect(socket_class, host, port)
  ::Timeout.timeout(connect_timeout, TimeoutError) do
    @socket = socket_class.open(host, port)
  end
end

def connect_ssl
  rescue_readable do
    rescue_writable do
      socket.connect_nonblock
    end
  end
end

def initialize(*args)
  super
  @read_timeout = options.fetch(:read_timeout, READ_TIMEOUT)
  @write_timeout = options.fetch(:write_timeout, WRITE_TIMEOUT)
  @connect_timeout = options.fetch(:connect_timeout, CONNECT_TIMEOUT)
end

def readpartial(size)
  rescue_readable do
    socket.read_nonblock(size)
  end
rescue EOFError
  :eof
end

def readpartial(size)
  loop do
    result = socket.read_nonblock(size, :exception => false)
    if result.nil?
      return :eof
    elsif result != :wait_readable
      return result
    end
    unless IO.select([socket], nil, nil, read_timeout)
      fail TimeoutError, "Read timed out after #{read_timeout} seconds"
    end
  end
end

def write(data)
  rescue_writable do
    socket.write_nonblock(data)
  end
rescue EOFError
  :eof
end

def write(data)
  loop do
    result = socket.write_nonblock(data, :exception => false)
    return result unless result == :wait_writable
    unless IO.select(nil, [socket], nil, write_timeout)
      fail TimeoutError, "Read timed out after #{write_timeout} seconds"
    end
  end
end