comparison contrib/perf.py @ 32731:6f791ca70640

perf: have a generic "clearstorecache" function

There are multiple places where we will want to purge some store cache. So we
promote the existing _clearobsstore function to a something reusable.

32730:b8ff7d0ff361

    # - repo.tags of such Mercurial isn't "callable", and repo.tags()
    #   in perftags() causes failure soon
    # - perf.py itself has been available since 1.1 (or eb240755386d)
    raise error.Abort(("tags API of this hg command is unknown"))

# perf commands

@command('perfwalk', formatteropts)
def perfwalk(ui, repo, *pats, **opts):
    timer, fm = gettimer(ui, opts)

        else:
            for r in repo.revs(expr): pass
    timer(d)
    fm.end()

def _clearobsstore(repo):
    unfi = repo.unfiltered()
    if 'obsstore' in vars(unfi):
        del unfi.obsstore
        del unfi._filecache['obsstore']

@command('perfvolatilesets',
         [('', 'clear-obsstore', False, 'drop obsstore between each call.'),
         ] + formatteropts)
def perfvolatilesets(ui, repo, *names, **opts):
    """benchmark the computation of various volatile set

    def getobs(name):
        def d():
            repo.invalidatevolatilesets()
            if opts['clear_obsstore']:
                _clearobsstore(repo)
            obsolete.getrevs(repo, name)
        return d

    allobs = sorted(obsolete.cachefuncs)
    if names:

    def getfiltered(name):
        def d():
            repo.invalidatevolatilesets()
            if opts['clear_obsstore']:
                _clearobsstore(repo)
            repoview.filterrevs(repo, name)
        return d

    allfilter = sorted(repoview.filtertable)
    if names:

32731:6f791ca70640

    # - repo.tags of such Mercurial isn't "callable", and repo.tags()
    #   in perftags() causes failure soon
    # - perf.py itself has been available since 1.1 (or eb240755386d)
    raise error.Abort(("tags API of this hg command is unknown"))

# utilities to clear cache

def clearfilecache(repo, attrname):
    unfi = repo.unfiltered()
    if attrname in vars(unfi):
        delattr(unfi, attrname)
    unfi._filecache.pop(attrname, None)

# perf commands

@command('perfwalk', formatteropts)
def perfwalk(ui, repo, *pats, **opts):
    timer, fm = gettimer(ui, opts)

        else:
            for r in repo.revs(expr): pass
    timer(d)
    fm.end()

@command('perfvolatilesets',
         [('', 'clear-obsstore', False, 'drop obsstore between each call.'),
         ] + formatteropts)
def perfvolatilesets(ui, repo, *names, **opts):
    """benchmark the computation of various volatile set

    def getobs(name):
        def d():
            repo.invalidatevolatilesets()
            if opts['clear_obsstore']:
                clearfilecache(repo, 'obsstore')
            obsolete.getrevs(repo, name)
        return d

    allobs = sorted(obsolete.cachefuncs)
    if names:

    def getfiltered(name):
        def d():
            repo.invalidatevolatilesets()
            if opts['clear_obsstore']:
                clearfilecache(repo, 'obsstore')
            repoview.filterrevs(repo, name)
        return d

    allfilter = sorted(repoview.filtertable)
    if names: