comparison contrib/perf.py @ 51008:2f1967ffefb1

perf: change the way we approach revlog reading

If the `reading` context manager is available, we should use it over explicit
file handle management. This will help us to make file handle management a
matter more internal to the revlog.

51007:ccddd2f54013

            except NotImplementedError:
                pass

    revs = list(rl.revs(startrev, len(rl) - 1))

    def rlfh(rl):
        if rl._inline:
            indexfile = getattr(rl, '_indexfile', None)
            if indexfile is None:
                # compatibility with <= hg-5.8
                indexfile = getattr(rl, 'indexfile')
            return getsvfs(repo)(indexfile)
        else:
            datafile = getattr(rl, 'datafile', getattr(rl, 'datafile'))
            return getsvfs(repo)(datafile)

    def doread():
        rl.clearcaches()
        for rev in revs:
            segmentforrevs(rev, rev)

    def doreadcachedfh():
        rl.clearcaches()
        fh = rlfh(rl)
        for rev in revs:
            segmentforrevs(rev, rev, df=fh)

    def doreadbatch():
        rl.clearcaches()
        segmentforrevs(revs[0], revs[-1])

    def doreadbatchcachedfh():
        rl.clearcaches()
        fh = rlfh(rl)
        segmentforrevs(revs[0], revs[-1], df=fh)

    def dochunk():
        rl.clearcaches()
        fh = rlfh(rl)
        for rev in revs:
            rl._chunk(rev, df=fh)

    chunks = [None]

    def dochunkbatch():
        rl.clearcaches()
        fh = rlfh(rl)
        # Save chunks as a side-effect.
        chunks[0] = rl._chunks(revs, df=fh)

    def docompress(compressor):
        rl.clearcaches()

        try:

51008:2f1967ffefb1

            except NotImplementedError:
                pass

    revs = list(rl.revs(startrev, len(rl) - 1))

    @contextlib.contextmanager
    def reading(rl):
        if getattr(rl, 'reading', None) is not None:
            with rl.reading():
                yield None
        elif rl._inline:
            indexfile = getattr(rl, '_indexfile', None)
            if indexfile is None:
                # compatibility with <= hg-5.8
                indexfile = getattr(rl, 'indexfile')
            yield getsvfs(repo)(indexfile)
        else:
            datafile = getattr(rl, 'datafile', getattr(rl, 'datafile'))
            yield getsvfs(repo)(datafile)

    def doread():
        rl.clearcaches()
        for rev in revs:
            segmentforrevs(rev, rev)

    def doreadcachedfh():
        rl.clearcaches()
        with reading(rl) as fh:
            if fh is not None:
                for rev in revs:
                    segmentforrevs(rev, rev, df=fh)
            else:
                for rev in revs:
                    segmentforrevs(rev, rev)

    def doreadbatch():
        rl.clearcaches()
        segmentforrevs(revs[0], revs[-1])

    def doreadbatchcachedfh():
        rl.clearcaches()
        with reading(rl) as fh:
            if fh is not None:
                segmentforrevs(revs[0], revs[-1], df=fh)
            else:
                segmentforrevs(revs[0], revs[-1])

    def dochunk():
        rl.clearcaches()
        with reading(rl) as fh:
            if fh is not None:
                for rev in revs:
                    rl._chunk(rev, df=fh)
            else:
                for rev in revs:
                    rl._chunk(rev)

    chunks = [None]

    def dochunkbatch():
        rl.clearcaches()
        with reading(rl) as fh:
            if fh is not None:
                # Save chunks as a side-effect.
                chunks[0] = rl._chunks(revs, df=fh)
            else:
                # Save chunks as a side-effect.
                chunks[0] = rl._chunks(revs)

    def docompress(compressor):
        rl.clearcaches()

        try: