Mercurial > hg
changeset 49413:b380583ad2f5
perf-bundle: accept --type argument
| author | Pierre-Yves David <pierre-yves.david@octobus.net> |
|---|---|
| date | Mon, 11 Jul 2022 23:30:24 +0200 |
| parents | 3635aae8b2e9 |
| children | 3c5d0f879404 |
| files | contrib/perf.py mercurial/commands.py |
| diffstat | 2 files changed, 34 insertions(+), 8 deletions(-) [+] |
line wrap: on
line diff
--- a/contrib/perf.py Mon Jul 11 23:10:55 2022 +0200 +++ b/contrib/perf.py Mon Jul 11 23:30:24 2022 +0200 @@ -984,6 +984,13 @@ b'changesets to bundle', b'REV', ), + ( + b't', + b'type', + b'none', + b'bundlespec to use (see `hg help bundlespec`)', + b'TYPE', + ), ] + formatteropts, b'REVS', @@ -991,7 +998,7 @@ def perfbundle(ui, repo, *revs, **opts): """benchmark the creation of a bundle from a repository - For now, this create a `none-v1` bundle. + For now, this only supports "none" compression. """ from mercurial import bundlecaches from mercurial import discovery @@ -1018,11 +1025,31 @@ bases = [cl.node(r) for r in repo.revs(b"heads(::%ld - %ld)", revs, revs)] outgoing = discovery.outgoing(repo, bases, targets) - bundlespec = bundlecaches.parsebundlespec( - repo, b"none", strict=False - ) - - bversion = b'HG10' + bundlespec.wirecompression + bundle_spec = opts.get(b'type') + + bundle_spec = bundlecaches.parsebundlespec(repo, bundle_spec, strict=False) + + cgversion = bundle_spec.params[b"cg.version"] + if cgversion not in changegroup.supportedoutgoingversions(repo): + err = b"repository does not support bundle version %s" + raise error.Abort(err % cgversion) + + if cgversion == b'01': # bundle1 + bversion = b'HG10' + bundle_spec.wirecompression + bcompression = None + elif cgversion in (b'02', b'03'): + bversion = b'HG20' + bcompression = bundle_spec.wirecompression + else: + err = b'perf::bundle: unexpected changegroup version %s' + raise error.ProgrammingError(err % cgversion) + + if bcompression is None: + bcompression = b'UN' + + if bcompression != b'UN': + err = b'perf::bundle: compression currently unsupported: %s' + raise error.ProgrammingError(err % bcompression) def do_bundle(): bundle2.writenewbundle( @@ -1032,7 +1059,7 @@ os.devnull, bversion, outgoing, - {}, + bundle_spec.params, ) timer(do_bundle)