view contrib/fuzz/dirstate_corpus.py @ 50942:75d3306fbc9a stable

tags: avoid expensive access to repo.changelog in a loop

repo.changelog needs some cache invalidation when run on filtered repository.
Accessing it in that loop can be expensive when there is many heads (e.g.
mozilla try and it 25 000 heads).

Note that the loop itself seems useless, but after this patch it no longer take
about ⅛  of the time we spend computing cache for mozilla try.

before :
! wall 0.350994 comb 0.350000 user 0.330000 sys 0.020000 (median of 28)

after :
! wall 0.319520 comb 0.310000 user 0.290000 sys 0.020000 (median of 30)

import argparse
import os
import zipfile

ap = argparse.ArgumentParser()
ap.add_argument("out", metavar="some.zip", type=str, nargs=1)
args = ap.parse_args()

reporoot = os.path.normpath(os.path.join(os.path.dirname(__file__), '..', '..'))
dirstate = os.path.join(reporoot, '.hg', 'dirstate')

with zipfile.ZipFile(args.out[0], "w", zipfile.ZIP_STORED) as zf:
    if os.path.exists(dirstate):
        with open(dirstate, 'rb') as f:
            zf.writestr("dirstate", f.read())