mirror of
https://github.com/facebook/sapling.git
synced 2024-10-11 09:17:30 +03:00
9dc21f8d0b
Summary: D13853115 adds `edenscm/` to `sys.path` and code still uses `import mercurial`. That has nasty problems if both `import mercurial` and `import edenscm.mercurial` are used, because Python would think `mercurial.foo` and `edenscm.mercurial.foo` are different modules so code like `try: ... except mercurial.error.Foo: ...`, or `isinstance(x, mercurial.foo.Bar)` would fail to handle the `edenscm.mercurial` version. There are also some module-level states (ex. `extensions._extensions`) that would cause trouble if they have multiple versions in a single process. Change imports to use the `edenscm` so ideally the `mercurial` is no longer imported at all. Add checks in extensions.py to catch unexpected extensions importing modules from the old (wrong) locations when running tests. Reviewed By: phillco Differential Revision: D13868981 fbshipit-source-id: f4e2513766957fd81d85407994f7521a08e4de48
119 lines
3.3 KiB
Python
119 lines
3.3 KiB
Python
# Copyright 2018 Facebook, Inc.
|
|
#
|
|
# This software may be used and distributed according to the terms of the
|
|
# GNU General Public License version 2 or any later version.
|
|
|
|
from __future__ import absolute_import
|
|
|
|
import random
|
|
import time
|
|
import traceback
|
|
|
|
from edenscm.mercurial import context, error, registrar, scmutil
|
|
from edenscm.mercurial.i18n import _
|
|
|
|
from ..perfsuite import editsgenerator
|
|
|
|
|
|
cmdtable = {}
|
|
command = registrar.command(cmdtable)
|
|
|
|
testedwith = "ships-with-fb-hgext"
|
|
|
|
|
|
def _mainloop(repo, ui, tr, count, batchsize):
|
|
i = 0
|
|
batchcount = 0
|
|
start = time.time()
|
|
base = scmutil.revsingle(repo, ui.config("repogenerator", "startcommit", "tip"))
|
|
goalrev = ui.configint("repogenerator", "numcommits", 10000) - 1
|
|
ui.write(_("starting commit is: %d (goal is %d)\n") % (base.rev(), goalrev))
|
|
generator = editsgenerator.randomeditsgenerator(base)
|
|
|
|
while base.rev() < goalrev:
|
|
# Make a commit:
|
|
wctx = context.overlayworkingctx(repo)
|
|
wctx.setbase(base)
|
|
generator.makerandomedits(wctx)
|
|
memctx = wctx.tomemctx("memory commit", parents=(base.rev(), None))
|
|
newctx = repo[repo.commitctx(memctx)]
|
|
|
|
# Log production rate:
|
|
elapsed = time.time() - start
|
|
if i % 5 == 0:
|
|
ui.write(
|
|
_(
|
|
"created %s, %0.2f sec elapsed "
|
|
"(%0.2f commits/sec, %s per hour, %s per day)\n"
|
|
)
|
|
% (
|
|
i,
|
|
elapsed,
|
|
i / elapsed,
|
|
"{:,}".format(int(i / elapsed * 3600)),
|
|
"{:,}".format(int(i / elapsed * 86400)),
|
|
)
|
|
)
|
|
base = newctx
|
|
i += 1
|
|
batchcount += 1
|
|
if batchcount > batchsize:
|
|
ui.status(_("committing txn...\n"))
|
|
tr.close()
|
|
tr = repo.transaction("newtxn_")
|
|
batchcount = 0
|
|
if i >= count:
|
|
ui.status(_("generated %d commits; quitting\n") % count)
|
|
return
|
|
|
|
|
|
@command(
|
|
"repogenerator",
|
|
[
|
|
("", "batch-size", 50000, _("size of transactiions to commit")),
|
|
("n", "count", 50000, _("number of commits to generate")),
|
|
("", "seed", 0, _("random seed to use")),
|
|
],
|
|
_("hg repogenerator [OPTION] [REV]"),
|
|
)
|
|
def repogenerator(ui, repo, *revs, **opts):
|
|
"""Generates random commits for large-scale repo generation
|
|
|
|
The starting revision is configurable::
|
|
|
|
[repogenerator]
|
|
startcommit = tip
|
|
|
|
The number of commits is configurable::
|
|
|
|
[repogenerator]
|
|
numcommits = 10000
|
|
|
|
The shape of generated paths can be tweaked::
|
|
|
|
[repogenerator]
|
|
filenamedircount = 3
|
|
filenameleaflength = 3
|
|
"""
|
|
if opts["seed"]:
|
|
random.seed(opts["seed"])
|
|
|
|
with repo.wlock(), repo.lock():
|
|
try:
|
|
tr = repo.transaction("")
|
|
except error.AbandonedTransactionFoundError:
|
|
ui.status(_("recovering abandoned transaction...\n"))
|
|
repo.recover()
|
|
tr = repo.transaction("")
|
|
|
|
try:
|
|
_mainloop(repo, ui, tr, opts["count"], opts["batch_size"])
|
|
tr.close()
|
|
except KeyboardInterrupt:
|
|
ui.status(_("interrupted...\n"))
|
|
tr.abort()
|
|
except Exception:
|
|
traceback.print_exc()
|
|
tr.abort()
|
|
raise
|