comparison mercurial/localrepo.py @ 20927:24a443948627

localrepo: move the changegroupsubset method in changegroup module

This is a gratuitous code move aimed at reducing the localrepo bloatness.

The method had few callers, not enough to be kept in local repo.

The peer API remains unchanged.

20926:7c1ed40e3325

    def changegroup(self, basenodes, source):
        return self._repo.changegroup(basenodes, source)

    def changegroupsubset(self, bases, heads, source):
        return self._repo.changegroupsubset(bases, heads, source)

class localrepository(object):

    supportedformats = set(('revlogv1', 'generaldelta'))
    _basesupported = supportedformats | set(('store', 'fncache', 'shared',

        pass

    def push(self, remote, force=False, revs=None, newbranch=False):
        return exchange.push(self, remote, force, revs, newbranch)

    def changegroupsubset(self, roots, heads, source):
        """Compute a changegroup consisting of all the nodes that are
        descendants of any of the roots and ancestors of any of the heads.
        Return a chunkbuffer object whose read() method will return
        successive changegroup chunks.

        It is fairly complex as determining which filenodes and which
        manifest nodes need to be included for the changeset to be complete
        is non-trivial.

        Another wrinkle is doing the reverse, figuring out which changeset in
        the changegroup a particular filenode or manifestnode belongs to.
        """
        cl = self.changelog
        if not roots:
            roots = [nullid]
        # TODO: remove call to nodesbetween.
        csets, roots, heads = cl.nodesbetween(roots, heads)
        discbases = []
        for n in roots:
            discbases.extend([p for p in cl.parents(n) if p != nullid])
        outgoing = discovery.outgoing(cl, discbases, heads)
        bundler = changegroup.bundle10(self)
        return changegroup.getsubset(self, outgoing, bundler, source)

    def getlocalbundle(self, source, outgoing, bundlecaps=None):
        """Like getbundle, but taking a discovery.outgoing as an argument.

        This is only implemented for local repos and reuses potentially
        precomputed sets in outgoing."""

                                   discovery.outgoing(cl, common, heads),
                                   bundlecaps=bundlecaps)

    def changegroup(self, basenodes, source):
        # to avoid a race we use changegroupsubset() (issue1320)
        return self.changegroupsubset(basenodes, self.heads(), source)

    @unfilteredmethod
    def addchangegroup(self, source, srctype, url, emptyok=False):
        """Add the changegroup returned by source.read() to this repo.
        srctype is a string like 'push', 'pull', or 'unbundle'.  url is

20927:24a443948627

    def changegroup(self, basenodes, source):
        return self._repo.changegroup(basenodes, source)

    def changegroupsubset(self, bases, heads, source):
        return changegroup.changegroupsubset(self._repo, bases, heads, source)

class localrepository(object):

    supportedformats = set(('revlogv1', 'generaldelta'))
    _basesupported = supportedformats | set(('store', 'fncache', 'shared',

        pass

    def push(self, remote, force=False, revs=None, newbranch=False):
        return exchange.push(self, remote, force, revs, newbranch)

    def getlocalbundle(self, source, outgoing, bundlecaps=None):
        """Like getbundle, but taking a discovery.outgoing as an argument.

        This is only implemented for local repos and reuses potentially
        precomputed sets in outgoing."""

                                   discovery.outgoing(cl, common, heads),
                                   bundlecaps=bundlecaps)

    def changegroup(self, basenodes, source):
        # to avoid a race we use changegroupsubset() (issue1320)
        return changegroup.changegroupsubset(self, basenodes, self.heads(),
                                             source)

    @unfilteredmethod
    def addchangegroup(self, source, srctype, url, emptyok=False):
        """Add the changegroup returned by source.read() to this repo.
        srctype is a string like 'push', 'pull', or 'unbundle'.  url is