comparison mercurial/streamclone.py @ 33258:761ccfeff8b1

streamclone: stop using 'vfs.mustaudit = False'

Now that each call disable the auditing on its own, we can safely drop this the
mustaudit usage. No other code is modified.

33257:15e9cbe6ae49

    repo.ui.debug('%d files, %d bytes to transfer\n' %
                  (len(entries), total_bytes))

    svfs = repo.svfs
    oldaudit = svfs.mustaudit
    debugflag = repo.ui.debugflag
    svfs.mustaudit = False

    def emitrevlogdata():
        try:
            for name, size in entries:
                if debugflag:
                    repo.ui.debug('sending %s (%d bytes)\n' % (name, size))
                # partially encode name over the wire for backwards compat
                yield '%s\0%d\n' % (store.encodedir(name), size)
                if size <= 65536:
                    with svfs(name, 'rb', auditpath=False) as fp:
                        yield fp.read(size)
                else:
                    data = svfs(name, auditpath=False)
                    for chunk in util.filechunkiter(data, limit=size):
                        yield chunk
        finally:
            svfs.mustaudit = oldaudit

    return len(entries), total_bytes, emitrevlogdata()

def generatev1wireproto(repo):
    """Emit content for version 1 of streaming clone suitable for the wire.

33258:761ccfeff8b1

    repo.ui.debug('%d files, %d bytes to transfer\n' %
                  (len(entries), total_bytes))

    svfs = repo.svfs
    debugflag = repo.ui.debugflag

    def emitrevlogdata():
        for name, size in entries:
            if debugflag:
                repo.ui.debug('sending %s (%d bytes)\n' % (name, size))
            # partially encode name over the wire for backwards compat
            yield '%s\0%d\n' % (store.encodedir(name), size)
            if size <= 65536:
                with svfs(name, 'rb', auditpath=False) as fp:
                    yield fp.read(size)
            else:
                data = svfs(name, auditpath=False)
                for chunk in util.filechunkiter(data, limit=size):
                    yield chunk

    return len(entries), total_bytes, emitrevlogdata()

def generatev1wireproto(repo):
    """Emit content for version 1 of streaming clone suitable for the wire.