comparison mercurial/bundle2.py @ 34151:550343626bb2

bundle2: move part counter to partiterator

As part of moving the part iterator logic to a separate class, let's move the
part counting logic and the output for it.

Differential Revision: https://phab.mercurial-scm.org/D704

34150:e9e0e1143fc5

        op = bundleoperation(repo, lambda: tr)
        _processchangegroup(op, unbundler, tr, source, url, **kwargs)
        return op

class partiterator(object):
    def __init__(self, unbundler):
        self.unbundler = unbundler

    def __enter__(self):
        return enumerate(self.unbundler.iterparts())

    def __exit__(self, type, value, tb):
        pass

def processbundle(repo, unbundler, transactiongetter=None, op=None):
    """This function process a bundle, apply effect to/from a repo

    It iterates over each part then searches for and uses the proper handling

        else:
            msg.append(' with-transaction')
        msg.append('\n')
        repo.ui.debug(''.join(msg))

    with partiterator(unbundler) as parts:
        part = None
        nbpart = 0
        try:
            for nbpart, part in parts:
                _processpart(op, part)
        except Exception as exc:
            # Any exceptions seeking to the end of the bundle at this point are
            # almost certainly related to the underlying stream being bad.
            # And, chances are that the exception we're handling is related to
            # getting in that bad state. So, we swallow the seeking error and
            # re-raise the original error.
            seekerror = False
            try:
                for nbpart, part in parts:
                    # consume the bundle content
                    part.seek(0, 2)
            except Exception:
                seekerror = True

            # that form if we need to.
            if seekerror:
                raise exc
            else:
                raise
        finally:
            repo.ui.debug('bundle2-input-bundle: %i parts total\n' % nbpart)

    return op

def _processchangegroup(op, cg, tr, source, url, **kwargs):
    ret = cg.apply(op.repo, tr, source, url, **kwargs)

34151:550343626bb2

        op = bundleoperation(repo, lambda: tr)
        _processchangegroup(op, unbundler, tr, source, url, **kwargs)
        return op

class partiterator(object):
    def __init__(self, repo, unbundler):
        self.repo = repo
        self.unbundler = unbundler
        self.iterator = None
        self.count = 0

    def __enter__(self):
        def func():
            itr = enumerate(self.unbundler.iterparts())
            for count, p in itr:
                self.count = count
                yield p
        self.iterator = func()
        return self.iterator

    def __exit__(self, type, value, tb):
        if not self.iterator:
            return

        self.repo.ui.debug('bundle2-input-bundle: %i parts total\n' %
                           self.count)

def processbundle(repo, unbundler, transactiongetter=None, op=None):
    """This function process a bundle, apply effect to/from a repo

    It iterates over each part then searches for and uses the proper handling

        else:
            msg.append(' with-transaction')
        msg.append('\n')
        repo.ui.debug(''.join(msg))

    with partiterator(repo, unbundler) as parts:
        part = None
        try:
            for part in parts:
                _processpart(op, part)
        except Exception as exc:
            # Any exceptions seeking to the end of the bundle at this point are
            # almost certainly related to the underlying stream being bad.
            # And, chances are that the exception we're handling is related to
            # getting in that bad state. So, we swallow the seeking error and
            # re-raise the original error.
            seekerror = False
            try:
                for part in parts:
                    # consume the bundle content
                    part.seek(0, 2)
            except Exception:
                seekerror = True

            # that form if we need to.
            if seekerror:
                raise exc
            else:
                raise

    return op

def _processchangegroup(op, cg, tr, source, url, **kwargs):
    ret = cg.apply(op.repo, tr, source, url, **kwargs)