module Redis::Connection::SocketMixin

def _write_to_socket(data)
  begin
    write_nonblock(data)
  rescue IO::WaitWritable
    if IO.select(nil, [self], nil, @write_timeout)
      retry
    else
      raise Redis::TimeoutError
    end
  rescue IO::WaitReadable
    if IO.select([self], nil, nil, @write_timeout)
      retry
    else
      raise Redis::TimeoutError
    end
  end
rescue EOFError
  raise Errno::ECONNRESET
end