comparison mercurial/localrepo.py @ 10432:8a8030fc57d6

localrepo: provide indeterminate progress information while bundling

10431:ba5e508b5e92

            collect = changegroup.collector(cl, msng_mnfst_set, changedfiles)

            # Create a changenode group generator that will call our functions
            # back to lookup the owning changenode and collect information.
            group = cl.group(msng_cl_lst, identity, collect)
            for chnk in group:
                yield chnk

            # Figure out which manifest nodes (of the ones we think might be
            # part of the changegroup) the recipient must know about and
            # remove them from the changegroup.
            has_mnfst_set = set()

            msng_mnfst_lst.sort(key=mnfst.rev)
            # Create a generator for the manifestnodes that calls our lookup
            # and data collection functions back.
            group = mnfst.group(msng_mnfst_lst, lookup_manifest_link,
                                filenode_collector(changedfiles))
            for chnk in group:
                yield chnk

            # These are no longer needed, dereference and toss the memory for
            # them.
            msng_mnfst_lst = None
            msng_mnfst_set.clear()

                    if isinstance(fname, int):
                        continue
                    msng_filenode_set.setdefault(fname, {})
                    changedfiles[fname] = 1
            # Go through all our files in order sorted by name.
            for fname in sorted(changedfiles):
                filerevlog = self.file(fname)
                if not len(filerevlog):
                    raise util.Abort(_("empty or missing revlog for %s") % fname)
                # Toss out the filenodes that the recipient isn't really

                    # lookup function as we need to collect no information
                    # from filenodes.
                    group = filerevlog.group(msng_filenode_lst,
                                             lookup_filenode_link_func(fname))
                    for chnk in group:
                        yield chnk
                if fname in msng_filenode_set:
                    # Don't need this anymore, toss it to free memory.
                    del msng_filenode_set[fname]
            # Signal that no more groups are left.
            yield changegroup.closechunk()

            if msng_cl_lst:
                self.hook('outgoing', node=hex(msng_cl_lst[0]), source=source)

        return util.chunkbuffer(gengroup())

            # construct a list of all changed files
            changedfiles = {}
            mmfs = {}
            collect = changegroup.collector(cl, mmfs, changedfiles)

            for chnk in cl.group(nodes, identity, collect):
                yield chnk

            mnfst = self.manifest
            nodeiter = gennodelst(mnfst)
            for chnk in mnfst.group(nodeiter, lookuprevlink_func(mnfst)):
                yield chnk

            for fname in sorted(changedfiles):
                filerevlog = self.file(fname)
                if not len(filerevlog):
                    raise util.Abort(_("empty or missing revlog for %s") % fname)
                nodeiter = gennodelst(filerevlog)

                if nodeiter:
                    yield changegroup.chunkheader(len(fname))
                    yield fname
                    lookup = lookuprevlink_func(filerevlog)
                    for chnk in filerevlog.group(nodeiter, lookup):
                        yield chnk

            yield changegroup.closechunk()

            if nodes:
                self.hook('outgoing', node=hex(nodes[0]), source=source)

10432:8a8030fc57d6

            collect = changegroup.collector(cl, msng_mnfst_set, changedfiles)

            # Create a changenode group generator that will call our functions
            # back to lookup the owning changenode and collect information.
            group = cl.group(msng_cl_lst, identity, collect)
            cnt = 0
            for chnk in group:
                yield chnk
                self.ui.progress('bundle changes', cnt, unit='chunks')
                cnt += 1
            self.ui.progress('bundle changes', None, unit='chunks')


            # Figure out which manifest nodes (of the ones we think might be
            # part of the changegroup) the recipient must know about and
            # remove them from the changegroup.
            has_mnfst_set = set()

            msng_mnfst_lst.sort(key=mnfst.rev)
            # Create a generator for the manifestnodes that calls our lookup
            # and data collection functions back.
            group = mnfst.group(msng_mnfst_lst, lookup_manifest_link,
                                filenode_collector(changedfiles))
            cnt = 0
            for chnk in group:
                yield chnk
                self.ui.progress('bundle manifests', cnt, unit='chunks')
                cnt += 1
            self.ui.progress('bundle manifests', None, unit='chunks')

            # These are no longer needed, dereference and toss the memory for
            # them.
            msng_mnfst_lst = None
            msng_mnfst_set.clear()

                    if isinstance(fname, int):
                        continue
                    msng_filenode_set.setdefault(fname, {})
                    changedfiles[fname] = 1
            # Go through all our files in order sorted by name.
            cnt = 0
            for fname in sorted(changedfiles):
                filerevlog = self.file(fname)
                if not len(filerevlog):
                    raise util.Abort(_("empty or missing revlog for %s") % fname)
                # Toss out the filenodes that the recipient isn't really

                    # lookup function as we need to collect no information
                    # from filenodes.
                    group = filerevlog.group(msng_filenode_lst,
                                             lookup_filenode_link_func(fname))
                    for chnk in group:
                        self.ui.progress(
                            'bundle files', cnt, item=fname, unit='chunks')
                        cnt += 1
                        yield chnk
                if fname in msng_filenode_set:
                    # Don't need this anymore, toss it to free memory.
                    del msng_filenode_set[fname]
            # Signal that no more groups are left.
            yield changegroup.closechunk()
            self.ui.progress('bundle files', None, unit='chunks')

            if msng_cl_lst:
                self.hook('outgoing', node=hex(msng_cl_lst[0]), source=source)

        return util.chunkbuffer(gengroup())

            # construct a list of all changed files
            changedfiles = {}
            mmfs = {}
            collect = changegroup.collector(cl, mmfs, changedfiles)

            cnt = 0
            for chnk in cl.group(nodes, identity, collect):
                self.ui.progress('bundle changes', cnt, unit='chunks')
                cnt += 1
                yield chnk
            self.ui.progress('bundle changes', None, unit='chunks')

            mnfst = self.manifest
            nodeiter = gennodelst(mnfst)
            cnt = 0
            for chnk in mnfst.group(nodeiter, lookuprevlink_func(mnfst)):
                self.ui.progress('bundle manifests', cnt, unit='chunks')
                cnt += 1
                yield chnk
            self.ui.progress('bundle manifests', None, unit='chunks')

            cnt = 0
            for fname in sorted(changedfiles):
                filerevlog = self.file(fname)
                if not len(filerevlog):
                    raise util.Abort(_("empty or missing revlog for %s") % fname)
                nodeiter = gennodelst(filerevlog)

                if nodeiter:
                    yield changegroup.chunkheader(len(fname))
                    yield fname
                    lookup = lookuprevlink_func(filerevlog)
                    for chnk in filerevlog.group(nodeiter, lookup):
                        self.ui.progress(
                            'bundle files', cnt, item=fname, unit='chunks')
                        cnt += 1
                        yield chnk
            self.ui.progress('bundle files', None, unit='chunks')

            yield changegroup.closechunk()

            if nodes:
                self.hook('outgoing', node=hex(nodes[0]), source=source)