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
end

def write(data)
  rescue_writable do
    socket.write_nonblock(data)
  end
end