comparison mercurial/localrepo.py @ 18121:f8a13f061a8a

branchmap: extract updatebranchcache from repo

18120:88990d3e3d75

        better alternative. But we are using this approach until it is ready.
        """
        cl = self.changelog
        return cl.rev(cl.tip())

    @unfilteredmethod # Until we get a smarter cache management
    def updatebranchcache(self):
        cl = self.changelog
        tip = cl.tip()
        if self._branchcache is not None and self._branchcachetip == tip:
            return

        oldtip = self._branchcachetip
        if oldtip is None or oldtip not in cl.nodemap:
            partial, last, lrev = branchmap.read(self)
        else:
            lrev = cl.rev(oldtip)
            partial = self._branchcache

        catip = self._cacheabletip()
        # if lrev == catip: cache is already up to date
        # if lrev >  catip: we have uncachable element in `partial` can't write
        #                   on disk
        if lrev < catip:
            ctxgen = (self[r] for r in cl.revs(lrev + 1, catip))
            branchmap.update(self, partial, ctxgen)
            branchmap.write(self, partial, cl.node(catip), catip)
            lrev = catip
        # If cacheable tip were lower than actual tip, we need to update the
        # cache up to tip. This update (from cacheable to actual tip) is not
        # written to disk since it's not cacheable.
        tiprev = len(self) - 1
        if lrev < tiprev:
            ctxgen = (self[r] for r in cl.revs(lrev + 1, tiprev))
            branchmap.update(self, partial, ctxgen)
        self._branchcache = partial
        self._branchcachetip = tip

    def branchmap(self):
        '''returns a dictionary {branch: [branchheads]}'''
        if self.changelog.filteredrevs:
            # some changeset are excluded we can't use the cache
            bmap = {}
            branchmap.update(self, bmap, (self[r] for r in self))
            return bmap
        else:
            self.updatebranchcache()
            return self._branchcache


    def _branchtip(self, heads):
        '''return the tipmost branch head in heads'''

                # be compliant anyway
                #
                # if minimal phase was 0 we don't need to retract anything
                phases.retractboundary(self, targetphase, [n])
            tr.close()
            self.updatebranchcache()
            return n
        finally:
            if tr:
                tr.release()
            lock.release()

            cl.finalize(trp)

            tr.close()

            if changesets > 0:
                self.updatebranchcache()
                def runhooks():
                    # forcefully update the on-disk branch cache
                    self.ui.debug("updating the branch cache\n")
                    self.hook("changegroup", node=hex(cl.node(clstart)),
                              source=srctype, url=url)

18121:f8a13f061a8a

        better alternative. But we are using this approach until it is ready.
        """
        cl = self.changelog
        return cl.rev(cl.tip())

    def branchmap(self):
        '''returns a dictionary {branch: [branchheads]}'''
        if self.changelog.filteredrevs:
            # some changeset are excluded we can't use the cache
            bmap = {}
            branchmap.update(self, bmap, (self[r] for r in self))
            return bmap
        else:
            branchmap.updatecache(self)
            return self._branchcache


    def _branchtip(self, heads):
        '''return the tipmost branch head in heads'''

                # be compliant anyway
                #
                # if minimal phase was 0 we don't need to retract anything
                phases.retractboundary(self, targetphase, [n])
            tr.close()
            branchmap.updatecache(self)
            return n
        finally:
            if tr:
                tr.release()
            lock.release()

            cl.finalize(trp)

            tr.close()

            if changesets > 0:
                branchmap.updatecache(self)
                def runhooks():
                    # forcefully update the on-disk branch cache
                    self.ui.debug("updating the branch cache\n")
                    self.hook("changegroup", node=hex(cl.node(clstart)),
                              source=srctype, url=url)