comparison mercurial/changegroup.py @ 28232:829d369fc5a8

changegroup: write root manifests and subdir manifests in a single loop

This is another step towards making the manifest generation recurse
along the directory trees. The loop over 'tmfnodes' now takes the form
of a queue. At this point, we only add to the queue twice: we add the
root manifests, and, while visiting the root manifest revisions, we
add all subdirectory revisions (for treemanifest repos). Thus, any
iterations over 'tmfnodes' after the first will not add any items and
the "queue" will just keep shrinking.

28231:3faba927dd93

    def generatemanifests(self, commonrevs, clrevorder, fastpathlinkrev, mfs,
                          mfchangedfiles, fnodes):
        repo = self._repo
        ml = repo.manifest
        tmfnodes = {}

        # Callback for the manifest, used to collect linkrevs for filelog
        # revisions.
        # Returns the linkrev node (collected in lookupcl).
        def makelookupmflinknode(dir):

                        if clrevorder[clnode] < clrevorder[tmfclnode]:
                            tmfclnodes[n] = clnode
                return clnode
            return lookupmflinknode

        mfnodes = self.prune(ml, mfs, commonrevs)
        size = 0
        for x in self._packmanifests('', mfnodes, makelookupmflinknode('')):
            size += len(x)
            yield x
        for dir, nodes in tmfnodes.iteritems():
            prunednodes = self.prune(ml.dirlog(dir), nodes, commonrevs)
            for x in self._packmanifests(dir, prunednodes,
                                         makelookupmflinknode(dir)):
                size += len(x)
                yield x
        self._verbosenote(_('%8.i (manifests)\n') % size)
        yield self._manifestsdone()

    # The 'source' parameter is useful for extensions
    def generatefiles(self, changedfiles, linknodes, commonrevs, source):

28232:829d369fc5a8

    def generatemanifests(self, commonrevs, clrevorder, fastpathlinkrev, mfs,
                          mfchangedfiles, fnodes):
        repo = self._repo
        ml = repo.manifest
        tmfnodes = {'': mfs}

        # Callback for the manifest, used to collect linkrevs for filelog
        # revisions.
        # Returns the linkrev node (collected in lookupcl).
        def makelookupmflinknode(dir):

                        if clrevorder[clnode] < clrevorder[tmfclnode]:
                            tmfclnodes[n] = clnode
                return clnode
            return lookupmflinknode

        size = 0
        while tmfnodes:
            dir = min(tmfnodes)
            nodes = tmfnodes[dir]
            prunednodes = self.prune(ml.dirlog(dir), nodes, commonrevs)
            for x in self._packmanifests(dir, prunednodes,
                                         makelookupmflinknode(dir)):
                size += len(x)
                yield x
            del tmfnodes[dir]
        self._verbosenote(_('%8.i (manifests)\n') % size)
        yield self._manifestsdone()

    # The 'source' parameter is useful for extensions
    def generatefiles(self, changedfiles, linknodes, commonrevs, source):