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
  socket.connect_nonblock
rescue IO::WaitReadable
  if IO.select([socket], nil, nil, connect_timeout)
    retry
  else
    raise TimeoutError, "Connection timed out after #{connect_timeout} seconds"
  end
rescue IO::WaitWritable
  if IO.select(nil, [socket], nil, connect_timeout)
    retry
  else
    raise TimeoutError, "Connection timed out after #{connect_timeout} seconds"
  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)
  socket.read_nonblock(size)
rescue IO::WaitReadable
  if IO.select([socket], nil, nil, read_timeout)
    retry
  else
    raise TimeoutError, "Read timed out after #{read_timeout} seconds"
  end
end

def write(data)
  socket.write_nonblock(data)
rescue IO::WaitWritable
  if IO.select(nil, [socket], nil, write_timeout)
    retry
  else
    raise TimeoutError, "Read timed out after #{write_timeout} seconds"
  end
end