module Redis::Connection::SocketMixin

def _read_from_socket(nbytes)
  begin
    read_nonblock(nbytes)
  rescue Errno::EWOULDBLOCK, Errno::EAGAIN
    if IO.select([self], nil, nil, @timeout)
      retry
    else
      raise Redis::TimeoutError
    end
  end
rescue EOFError
  raise Errno::ECONNRESET
end

def gets
  crlf = nil
  while (crlf = @buffer.index(CRLF)) == nil
    @buffer << _read_from_socket(1024)
  end
  @buffer.slice!(0, crlf + CRLF.bytesize)
end

def initialize(*args)
  super(*args)
  @timeout = nil
  @buffer = ""
end

def read(nbytes)
  result = @buffer.slice!(0, nbytes)
  while result.bytesize < nbytes
    result << _read_from_socket(nbytes - result.bytesize)
  end
  result
end

def timeout=(timeout)
  if timeout && timeout > 0
    @timeout = timeout
  else
    @timeout = nil
  end
end