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