##// END OF EJS Templates
perf: introduce a `--iteration` to `perfdirstate`...
marmoute -
r43466:5f9b1250 default
parent child Browse files
Show More
@@ -1,3778 +1,3789 b''
1 1 # perf.py - performance test routines
2 2 '''helper extension to measure performance
3 3
4 4 Configurations
5 5 ==============
6 6
7 7 ``perf``
8 8 --------
9 9
10 10 ``all-timing``
11 11 When set, additional statistics will be reported for each benchmark: best,
12 12 worst, median average. If not set only the best timing is reported
13 13 (default: off).
14 14
15 15 ``presleep``
16 16 number of second to wait before any group of runs (default: 1)
17 17
18 18 ``pre-run``
19 19 number of run to perform before starting measurement.
20 20
21 21 ``profile-benchmark``
22 22 Enable profiling for the benchmarked section.
23 23 (The first iteration is benchmarked)
24 24
25 25 ``run-limits``
26 26 Control the number of runs each benchmark will perform. The option value
27 27 should be a list of `<time>-<numberofrun>` pairs. After each run the
28 28 conditions are considered in order with the following logic:
29 29
30 30 If benchmark has been running for <time> seconds, and we have performed
31 31 <numberofrun> iterations, stop the benchmark,
32 32
33 33 The default value is: `3.0-100, 10.0-3`
34 34
35 35 ``stub``
36 36 When set, benchmarks will only be run once, useful for testing
37 37 (default: off)
38 38 '''
39 39
40 40 # "historical portability" policy of perf.py:
41 41 #
42 42 # We have to do:
43 43 # - make perf.py "loadable" with as wide Mercurial version as possible
44 44 # This doesn't mean that perf commands work correctly with that Mercurial.
45 45 # BTW, perf.py itself has been available since 1.1 (or eb240755386d).
46 46 # - make historical perf command work correctly with as wide Mercurial
47 47 # version as possible
48 48 #
49 49 # We have to do, if possible with reasonable cost:
50 50 # - make recent perf command for historical feature work correctly
51 51 # with early Mercurial
52 52 #
53 53 # We don't have to do:
54 54 # - make perf command for recent feature work correctly with early
55 55 # Mercurial
56 56
57 57 from __future__ import absolute_import
58 58 import contextlib
59 59 import functools
60 60 import gc
61 61 import os
62 62 import random
63 63 import shutil
64 64 import struct
65 65 import sys
66 66 import tempfile
67 67 import threading
68 68 import time
69 69 from mercurial import (
70 70 changegroup,
71 71 cmdutil,
72 72 commands,
73 73 copies,
74 74 error,
75 75 extensions,
76 76 hg,
77 77 mdiff,
78 78 merge,
79 79 revlog,
80 80 util,
81 81 )
82 82
83 83 # for "historical portability":
84 84 # try to import modules separately (in dict order), and ignore
85 85 # failure, because these aren't available with early Mercurial
86 86 try:
87 87 from mercurial import branchmap # since 2.5 (or bcee63733aad)
88 88 except ImportError:
89 89 pass
90 90 try:
91 91 from mercurial import obsolete # since 2.3 (or ad0d6c2b3279)
92 92 except ImportError:
93 93 pass
94 94 try:
95 95 from mercurial import registrar # since 3.7 (or 37d50250b696)
96 96
97 97 dir(registrar) # forcibly load it
98 98 except ImportError:
99 99 registrar = None
100 100 try:
101 101 from mercurial import repoview # since 2.5 (or 3a6ddacb7198)
102 102 except ImportError:
103 103 pass
104 104 try:
105 105 from mercurial.utils import repoviewutil # since 5.0
106 106 except ImportError:
107 107 repoviewutil = None
108 108 try:
109 109 from mercurial import scmutil # since 1.9 (or 8b252e826c68)
110 110 except ImportError:
111 111 pass
112 112 try:
113 113 from mercurial import setdiscovery # since 1.9 (or cb98fed52495)
114 114 except ImportError:
115 115 pass
116 116
117 117 try:
118 118 from mercurial import profiling
119 119 except ImportError:
120 120 profiling = None
121 121
122 122
123 123 def identity(a):
124 124 return a
125 125
126 126
127 127 try:
128 128 from mercurial import pycompat
129 129
130 130 getargspec = pycompat.getargspec # added to module after 4.5
131 131 _byteskwargs = pycompat.byteskwargs # since 4.1 (or fbc3f73dc802)
132 132 _sysstr = pycompat.sysstr # since 4.0 (or 2219f4f82ede)
133 133 _bytestr = pycompat.bytestr # since 4.2 (or b70407bd84d5)
134 134 _xrange = pycompat.xrange # since 4.8 (or 7eba8f83129b)
135 135 fsencode = pycompat.fsencode # since 3.9 (or f4a5e0e86a7e)
136 136 if pycompat.ispy3:
137 137 _maxint = sys.maxsize # per py3 docs for replacing maxint
138 138 else:
139 139 _maxint = sys.maxint
140 140 except (NameError, ImportError, AttributeError):
141 141 import inspect
142 142
143 143 getargspec = inspect.getargspec
144 144 _byteskwargs = identity
145 145 _bytestr = str
146 146 fsencode = identity # no py3 support
147 147 _maxint = sys.maxint # no py3 support
148 148 _sysstr = lambda x: x # no py3 support
149 149 _xrange = xrange
150 150
151 151 try:
152 152 # 4.7+
153 153 queue = pycompat.queue.Queue
154 154 except (NameError, AttributeError, ImportError):
155 155 # <4.7.
156 156 try:
157 157 queue = pycompat.queue
158 158 except (NameError, AttributeError, ImportError):
159 159 import Queue as queue
160 160
161 161 try:
162 162 from mercurial import logcmdutil
163 163
164 164 makelogtemplater = logcmdutil.maketemplater
165 165 except (AttributeError, ImportError):
166 166 try:
167 167 makelogtemplater = cmdutil.makelogtemplater
168 168 except (AttributeError, ImportError):
169 169 makelogtemplater = None
170 170
171 171 # for "historical portability":
172 172 # define util.safehasattr forcibly, because util.safehasattr has been
173 173 # available since 1.9.3 (or 94b200a11cf7)
174 174 _undefined = object()
175 175
176 176
177 177 def safehasattr(thing, attr):
178 178 return getattr(thing, _sysstr(attr), _undefined) is not _undefined
179 179
180 180
181 181 setattr(util, 'safehasattr', safehasattr)
182 182
183 183 # for "historical portability":
184 184 # define util.timer forcibly, because util.timer has been available
185 185 # since ae5d60bb70c9
186 186 if safehasattr(time, 'perf_counter'):
187 187 util.timer = time.perf_counter
188 188 elif os.name == b'nt':
189 189 util.timer = time.clock
190 190 else:
191 191 util.timer = time.time
192 192
193 193 # for "historical portability":
194 194 # use locally defined empty option list, if formatteropts isn't
195 195 # available, because commands.formatteropts has been available since
196 196 # 3.2 (or 7a7eed5176a4), even though formatting itself has been
197 197 # available since 2.2 (or ae5f92e154d3)
198 198 formatteropts = getattr(
199 199 cmdutil, "formatteropts", getattr(commands, "formatteropts", [])
200 200 )
201 201
202 202 # for "historical portability":
203 203 # use locally defined option list, if debugrevlogopts isn't available,
204 204 # because commands.debugrevlogopts has been available since 3.7 (or
205 205 # 5606f7d0d063), even though cmdutil.openrevlog() has been available
206 206 # since 1.9 (or a79fea6b3e77).
207 207 revlogopts = getattr(
208 208 cmdutil,
209 209 "debugrevlogopts",
210 210 getattr(
211 211 commands,
212 212 "debugrevlogopts",
213 213 [
214 214 (b'c', b'changelog', False, b'open changelog'),
215 215 (b'm', b'manifest', False, b'open manifest'),
216 216 (b'', b'dir', False, b'open directory manifest'),
217 217 ],
218 218 ),
219 219 )
220 220
221 221 cmdtable = {}
222 222
223 223 # for "historical portability":
224 224 # define parsealiases locally, because cmdutil.parsealiases has been
225 225 # available since 1.5 (or 6252852b4332)
226 226 def parsealiases(cmd):
227 227 return cmd.split(b"|")
228 228
229 229
230 230 if safehasattr(registrar, 'command'):
231 231 command = registrar.command(cmdtable)
232 232 elif safehasattr(cmdutil, 'command'):
233 233 command = cmdutil.command(cmdtable)
234 234 if b'norepo' not in getargspec(command).args:
235 235 # for "historical portability":
236 236 # wrap original cmdutil.command, because "norepo" option has
237 237 # been available since 3.1 (or 75a96326cecb)
238 238 _command = command
239 239
240 240 def command(name, options=(), synopsis=None, norepo=False):
241 241 if norepo:
242 242 commands.norepo += b' %s' % b' '.join(parsealiases(name))
243 243 return _command(name, list(options), synopsis)
244 244
245 245
246 246 else:
247 247 # for "historical portability":
248 248 # define "@command" annotation locally, because cmdutil.command
249 249 # has been available since 1.9 (or 2daa5179e73f)
250 250 def command(name, options=(), synopsis=None, norepo=False):
251 251 def decorator(func):
252 252 if synopsis:
253 253 cmdtable[name] = func, list(options), synopsis
254 254 else:
255 255 cmdtable[name] = func, list(options)
256 256 if norepo:
257 257 commands.norepo += b' %s' % b' '.join(parsealiases(name))
258 258 return func
259 259
260 260 return decorator
261 261
262 262
263 263 try:
264 264 import mercurial.registrar
265 265 import mercurial.configitems
266 266
267 267 configtable = {}
268 268 configitem = mercurial.registrar.configitem(configtable)
269 269 configitem(
270 270 b'perf',
271 271 b'presleep',
272 272 default=mercurial.configitems.dynamicdefault,
273 273 experimental=True,
274 274 )
275 275 configitem(
276 276 b'perf',
277 277 b'stub',
278 278 default=mercurial.configitems.dynamicdefault,
279 279 experimental=True,
280 280 )
281 281 configitem(
282 282 b'perf',
283 283 b'parentscount',
284 284 default=mercurial.configitems.dynamicdefault,
285 285 experimental=True,
286 286 )
287 287 configitem(
288 288 b'perf',
289 289 b'all-timing',
290 290 default=mercurial.configitems.dynamicdefault,
291 291 experimental=True,
292 292 )
293 293 configitem(
294 294 b'perf', b'pre-run', default=mercurial.configitems.dynamicdefault,
295 295 )
296 296 configitem(
297 297 b'perf',
298 298 b'profile-benchmark',
299 299 default=mercurial.configitems.dynamicdefault,
300 300 )
301 301 configitem(
302 302 b'perf',
303 303 b'run-limits',
304 304 default=mercurial.configitems.dynamicdefault,
305 305 experimental=True,
306 306 )
307 307 except (ImportError, AttributeError):
308 308 pass
309 309 except TypeError:
310 310 # compatibility fix for a11fd395e83f
311 311 # hg version: 5.2
312 312 configitem(
313 313 b'perf', b'presleep', default=mercurial.configitems.dynamicdefault,
314 314 )
315 315 configitem(
316 316 b'perf', b'stub', default=mercurial.configitems.dynamicdefault,
317 317 )
318 318 configitem(
319 319 b'perf', b'parentscount', default=mercurial.configitems.dynamicdefault,
320 320 )
321 321 configitem(
322 322 b'perf', b'all-timing', default=mercurial.configitems.dynamicdefault,
323 323 )
324 324 configitem(
325 325 b'perf', b'pre-run', default=mercurial.configitems.dynamicdefault,
326 326 )
327 327 configitem(
328 328 b'perf',
329 329 b'profile-benchmark',
330 330 default=mercurial.configitems.dynamicdefault,
331 331 )
332 332 configitem(
333 333 b'perf', b'run-limits', default=mercurial.configitems.dynamicdefault,
334 334 )
335 335
336 336
337 337 def getlen(ui):
338 338 if ui.configbool(b"perf", b"stub", False):
339 339 return lambda x: 1
340 340 return len
341 341
342 342
343 343 class noop(object):
344 344 """dummy context manager"""
345 345
346 346 def __enter__(self):
347 347 pass
348 348
349 349 def __exit__(self, *args):
350 350 pass
351 351
352 352
353 353 NOOPCTX = noop()
354 354
355 355
356 356 def gettimer(ui, opts=None):
357 357 """return a timer function and formatter: (timer, formatter)
358 358
359 359 This function exists to gather the creation of formatter in a single
360 360 place instead of duplicating it in all performance commands."""
361 361
362 362 # enforce an idle period before execution to counteract power management
363 363 # experimental config: perf.presleep
364 364 time.sleep(getint(ui, b"perf", b"presleep", 1))
365 365
366 366 if opts is None:
367 367 opts = {}
368 368 # redirect all to stderr unless buffer api is in use
369 369 if not ui._buffers:
370 370 ui = ui.copy()
371 371 uifout = safeattrsetter(ui, b'fout', ignoremissing=True)
372 372 if uifout:
373 373 # for "historical portability":
374 374 # ui.fout/ferr have been available since 1.9 (or 4e1ccd4c2b6d)
375 375 uifout.set(ui.ferr)
376 376
377 377 # get a formatter
378 378 uiformatter = getattr(ui, 'formatter', None)
379 379 if uiformatter:
380 380 fm = uiformatter(b'perf', opts)
381 381 else:
382 382 # for "historical portability":
383 383 # define formatter locally, because ui.formatter has been
384 384 # available since 2.2 (or ae5f92e154d3)
385 385 from mercurial import node
386 386
387 387 class defaultformatter(object):
388 388 """Minimized composition of baseformatter and plainformatter
389 389 """
390 390
391 391 def __init__(self, ui, topic, opts):
392 392 self._ui = ui
393 393 if ui.debugflag:
394 394 self.hexfunc = node.hex
395 395 else:
396 396 self.hexfunc = node.short
397 397
398 398 def __nonzero__(self):
399 399 return False
400 400
401 401 __bool__ = __nonzero__
402 402
403 403 def startitem(self):
404 404 pass
405 405
406 406 def data(self, **data):
407 407 pass
408 408
409 409 def write(self, fields, deftext, *fielddata, **opts):
410 410 self._ui.write(deftext % fielddata, **opts)
411 411
412 412 def condwrite(self, cond, fields, deftext, *fielddata, **opts):
413 413 if cond:
414 414 self._ui.write(deftext % fielddata, **opts)
415 415
416 416 def plain(self, text, **opts):
417 417 self._ui.write(text, **opts)
418 418
419 419 def end(self):
420 420 pass
421 421
422 422 fm = defaultformatter(ui, b'perf', opts)
423 423
424 424 # stub function, runs code only once instead of in a loop
425 425 # experimental config: perf.stub
426 426 if ui.configbool(b"perf", b"stub", False):
427 427 return functools.partial(stub_timer, fm), fm
428 428
429 429 # experimental config: perf.all-timing
430 430 displayall = ui.configbool(b"perf", b"all-timing", False)
431 431
432 432 # experimental config: perf.run-limits
433 433 limitspec = ui.configlist(b"perf", b"run-limits", [])
434 434 limits = []
435 435 for item in limitspec:
436 436 parts = item.split(b'-', 1)
437 437 if len(parts) < 2:
438 438 ui.warn((b'malformatted run limit entry, missing "-": %s\n' % item))
439 439 continue
440 440 try:
441 441 time_limit = float(_sysstr(parts[0]))
442 442 except ValueError as e:
443 443 ui.warn(
444 444 (
445 445 b'malformatted run limit entry, %s: %s\n'
446 446 % (_bytestr(e), item)
447 447 )
448 448 )
449 449 continue
450 450 try:
451 451 run_limit = int(_sysstr(parts[1]))
452 452 except ValueError as e:
453 453 ui.warn(
454 454 (
455 455 b'malformatted run limit entry, %s: %s\n'
456 456 % (_bytestr(e), item)
457 457 )
458 458 )
459 459 continue
460 460 limits.append((time_limit, run_limit))
461 461 if not limits:
462 462 limits = DEFAULTLIMITS
463 463
464 464 profiler = None
465 465 if profiling is not None:
466 466 if ui.configbool(b"perf", b"profile-benchmark", False):
467 467 profiler = profiling.profile(ui)
468 468
469 469 prerun = getint(ui, b"perf", b"pre-run", 0)
470 470 t = functools.partial(
471 471 _timer,
472 472 fm,
473 473 displayall=displayall,
474 474 limits=limits,
475 475 prerun=prerun,
476 476 profiler=profiler,
477 477 )
478 478 return t, fm
479 479
480 480
481 481 def stub_timer(fm, func, setup=None, title=None):
482 482 if setup is not None:
483 483 setup()
484 484 func()
485 485
486 486
487 487 @contextlib.contextmanager
488 488 def timeone():
489 489 r = []
490 490 ostart = os.times()
491 491 cstart = util.timer()
492 492 yield r
493 493 cstop = util.timer()
494 494 ostop = os.times()
495 495 a, b = ostart, ostop
496 496 r.append((cstop - cstart, b[0] - a[0], b[1] - a[1]))
497 497
498 498
499 499 # list of stop condition (elapsed time, minimal run count)
500 500 DEFAULTLIMITS = (
501 501 (3.0, 100),
502 502 (10.0, 3),
503 503 )
504 504
505 505
506 506 def _timer(
507 507 fm,
508 508 func,
509 509 setup=None,
510 510 title=None,
511 511 displayall=False,
512 512 limits=DEFAULTLIMITS,
513 513 prerun=0,
514 514 profiler=None,
515 515 ):
516 516 gc.collect()
517 517 results = []
518 518 begin = util.timer()
519 519 count = 0
520 520 if profiler is None:
521 521 profiler = NOOPCTX
522 522 for i in range(prerun):
523 523 if setup is not None:
524 524 setup()
525 525 func()
526 526 keepgoing = True
527 527 while keepgoing:
528 528 if setup is not None:
529 529 setup()
530 530 with profiler:
531 531 with timeone() as item:
532 532 r = func()
533 533 profiler = NOOPCTX
534 534 count += 1
535 535 results.append(item[0])
536 536 cstop = util.timer()
537 537 # Look for a stop condition.
538 538 elapsed = cstop - begin
539 539 for t, mincount in limits:
540 540 if elapsed >= t and count >= mincount:
541 541 keepgoing = False
542 542 break
543 543
544 544 formatone(fm, results, title=title, result=r, displayall=displayall)
545 545
546 546
547 547 def formatone(fm, timings, title=None, result=None, displayall=False):
548 548
549 549 count = len(timings)
550 550
551 551 fm.startitem()
552 552
553 553 if title:
554 554 fm.write(b'title', b'! %s\n', title)
555 555 if result:
556 556 fm.write(b'result', b'! result: %s\n', result)
557 557
558 558 def display(role, entry):
559 559 prefix = b''
560 560 if role != b'best':
561 561 prefix = b'%s.' % role
562 562 fm.plain(b'!')
563 563 fm.write(prefix + b'wall', b' wall %f', entry[0])
564 564 fm.write(prefix + b'comb', b' comb %f', entry[1] + entry[2])
565 565 fm.write(prefix + b'user', b' user %f', entry[1])
566 566 fm.write(prefix + b'sys', b' sys %f', entry[2])
567 567 fm.write(prefix + b'count', b' (%s of %%d)' % role, count)
568 568 fm.plain(b'\n')
569 569
570 570 timings.sort()
571 571 min_val = timings[0]
572 572 display(b'best', min_val)
573 573 if displayall:
574 574 max_val = timings[-1]
575 575 display(b'max', max_val)
576 576 avg = tuple([sum(x) / count for x in zip(*timings)])
577 577 display(b'avg', avg)
578 578 median = timings[len(timings) // 2]
579 579 display(b'median', median)
580 580
581 581
582 582 # utilities for historical portability
583 583
584 584
585 585 def getint(ui, section, name, default):
586 586 # for "historical portability":
587 587 # ui.configint has been available since 1.9 (or fa2b596db182)
588 588 v = ui.config(section, name, None)
589 589 if v is None:
590 590 return default
591 591 try:
592 592 return int(v)
593 593 except ValueError:
594 594 raise error.ConfigError(
595 595 b"%s.%s is not an integer ('%s')" % (section, name, v)
596 596 )
597 597
598 598
599 599 def safeattrsetter(obj, name, ignoremissing=False):
600 600 """Ensure that 'obj' has 'name' attribute before subsequent setattr
601 601
602 602 This function is aborted, if 'obj' doesn't have 'name' attribute
603 603 at runtime. This avoids overlooking removal of an attribute, which
604 604 breaks assumption of performance measurement, in the future.
605 605
606 606 This function returns the object to (1) assign a new value, and
607 607 (2) restore an original value to the attribute.
608 608
609 609 If 'ignoremissing' is true, missing 'name' attribute doesn't cause
610 610 abortion, and this function returns None. This is useful to
611 611 examine an attribute, which isn't ensured in all Mercurial
612 612 versions.
613 613 """
614 614 if not util.safehasattr(obj, name):
615 615 if ignoremissing:
616 616 return None
617 617 raise error.Abort(
618 618 (
619 619 b"missing attribute %s of %s might break assumption"
620 620 b" of performance measurement"
621 621 )
622 622 % (name, obj)
623 623 )
624 624
625 625 origvalue = getattr(obj, _sysstr(name))
626 626
627 627 class attrutil(object):
628 628 def set(self, newvalue):
629 629 setattr(obj, _sysstr(name), newvalue)
630 630
631 631 def restore(self):
632 632 setattr(obj, _sysstr(name), origvalue)
633 633
634 634 return attrutil()
635 635
636 636
637 637 # utilities to examine each internal API changes
638 638
639 639
640 640 def getbranchmapsubsettable():
641 641 # for "historical portability":
642 642 # subsettable is defined in:
643 643 # - branchmap since 2.9 (or 175c6fd8cacc)
644 644 # - repoview since 2.5 (or 59a9f18d4587)
645 645 # - repoviewutil since 5.0
646 646 for mod in (branchmap, repoview, repoviewutil):
647 647 subsettable = getattr(mod, 'subsettable', None)
648 648 if subsettable:
649 649 return subsettable
650 650
651 651 # bisecting in bcee63733aad::59a9f18d4587 can reach here (both
652 652 # branchmap and repoview modules exist, but subsettable attribute
653 653 # doesn't)
654 654 raise error.Abort(
655 655 b"perfbranchmap not available with this Mercurial",
656 656 hint=b"use 2.5 or later",
657 657 )
658 658
659 659
660 660 def getsvfs(repo):
661 661 """Return appropriate object to access files under .hg/store
662 662 """
663 663 # for "historical portability":
664 664 # repo.svfs has been available since 2.3 (or 7034365089bf)
665 665 svfs = getattr(repo, 'svfs', None)
666 666 if svfs:
667 667 return svfs
668 668 else:
669 669 return getattr(repo, 'sopener')
670 670
671 671
672 672 def getvfs(repo):
673 673 """Return appropriate object to access files under .hg
674 674 """
675 675 # for "historical portability":
676 676 # repo.vfs has been available since 2.3 (or 7034365089bf)
677 677 vfs = getattr(repo, 'vfs', None)
678 678 if vfs:
679 679 return vfs
680 680 else:
681 681 return getattr(repo, 'opener')
682 682
683 683
684 684 def repocleartagscachefunc(repo):
685 685 """Return the function to clear tags cache according to repo internal API
686 686 """
687 687 if util.safehasattr(repo, b'_tagscache'): # since 2.0 (or 9dca7653b525)
688 688 # in this case, setattr(repo, '_tagscache', None) or so isn't
689 689 # correct way to clear tags cache, because existing code paths
690 690 # expect _tagscache to be a structured object.
691 691 def clearcache():
692 692 # _tagscache has been filteredpropertycache since 2.5 (or
693 693 # 98c867ac1330), and delattr() can't work in such case
694 694 if b'_tagscache' in vars(repo):
695 695 del repo.__dict__[b'_tagscache']
696 696
697 697 return clearcache
698 698
699 699 repotags = safeattrsetter(repo, b'_tags', ignoremissing=True)
700 700 if repotags: # since 1.4 (or 5614a628d173)
701 701 return lambda: repotags.set(None)
702 702
703 703 repotagscache = safeattrsetter(repo, b'tagscache', ignoremissing=True)
704 704 if repotagscache: # since 0.6 (or d7df759d0e97)
705 705 return lambda: repotagscache.set(None)
706 706
707 707 # Mercurial earlier than 0.6 (or d7df759d0e97) logically reaches
708 708 # this point, but it isn't so problematic, because:
709 709 # - repo.tags of such Mercurial isn't "callable", and repo.tags()
710 710 # in perftags() causes failure soon
711 711 # - perf.py itself has been available since 1.1 (or eb240755386d)
712 712 raise error.Abort(b"tags API of this hg command is unknown")
713 713
714 714
715 715 # utilities to clear cache
716 716
717 717
718 718 def clearfilecache(obj, attrname):
719 719 unfiltered = getattr(obj, 'unfiltered', None)
720 720 if unfiltered is not None:
721 721 obj = obj.unfiltered()
722 722 if attrname in vars(obj):
723 723 delattr(obj, attrname)
724 724 obj._filecache.pop(attrname, None)
725 725
726 726
727 727 def clearchangelog(repo):
728 728 if repo is not repo.unfiltered():
729 729 object.__setattr__(repo, r'_clcachekey', None)
730 730 object.__setattr__(repo, r'_clcache', None)
731 731 clearfilecache(repo.unfiltered(), 'changelog')
732 732
733 733
734 734 # perf commands
735 735
736 736
737 737 @command(b'perfwalk', formatteropts)
738 738 def perfwalk(ui, repo, *pats, **opts):
739 739 opts = _byteskwargs(opts)
740 740 timer, fm = gettimer(ui, opts)
741 741 m = scmutil.match(repo[None], pats, {})
742 742 timer(
743 743 lambda: len(
744 744 list(
745 745 repo.dirstate.walk(m, subrepos=[], unknown=True, ignored=False)
746 746 )
747 747 )
748 748 )
749 749 fm.end()
750 750
751 751
752 752 @command(b'perfannotate', formatteropts)
753 753 def perfannotate(ui, repo, f, **opts):
754 754 opts = _byteskwargs(opts)
755 755 timer, fm = gettimer(ui, opts)
756 756 fc = repo[b'.'][f]
757 757 timer(lambda: len(fc.annotate(True)))
758 758 fm.end()
759 759
760 760
761 761 @command(
762 762 b'perfstatus',
763 763 [(b'u', b'unknown', False, b'ask status to look for unknown files')]
764 764 + formatteropts,
765 765 )
766 766 def perfstatus(ui, repo, **opts):
767 767 """benchmark the performance of a single status call
768 768
769 769 The repository data are preserved between each call.
770 770
771 771 By default, only the status of the tracked file are requested. If
772 772 `--unknown` is passed, the "unknown" files are also tracked.
773 773 """
774 774 opts = _byteskwargs(opts)
775 775 # m = match.always(repo.root, repo.getcwd())
776 776 # timer(lambda: sum(map(len, repo.dirstate.status(m, [], False, False,
777 777 # False))))
778 778 timer, fm = gettimer(ui, opts)
779 779 timer(lambda: sum(map(len, repo.status(unknown=opts[b'unknown']))))
780 780 fm.end()
781 781
782 782
783 783 @command(b'perfaddremove', formatteropts)
784 784 def perfaddremove(ui, repo, **opts):
785 785 opts = _byteskwargs(opts)
786 786 timer, fm = gettimer(ui, opts)
787 787 try:
788 788 oldquiet = repo.ui.quiet
789 789 repo.ui.quiet = True
790 790 matcher = scmutil.match(repo[None])
791 791 opts[b'dry_run'] = True
792 792 if b'uipathfn' in getargspec(scmutil.addremove).args:
793 793 uipathfn = scmutil.getuipathfn(repo)
794 794 timer(lambda: scmutil.addremove(repo, matcher, b"", uipathfn, opts))
795 795 else:
796 796 timer(lambda: scmutil.addremove(repo, matcher, b"", opts))
797 797 finally:
798 798 repo.ui.quiet = oldquiet
799 799 fm.end()
800 800
801 801
802 802 def clearcaches(cl):
803 803 # behave somewhat consistently across internal API changes
804 804 if util.safehasattr(cl, b'clearcaches'):
805 805 cl.clearcaches()
806 806 elif util.safehasattr(cl, b'_nodecache'):
807 807 from mercurial.node import nullid, nullrev
808 808
809 809 cl._nodecache = {nullid: nullrev}
810 810 cl._nodepos = None
811 811
812 812
813 813 @command(b'perfheads', formatteropts)
814 814 def perfheads(ui, repo, **opts):
815 815 """benchmark the computation of a changelog heads"""
816 816 opts = _byteskwargs(opts)
817 817 timer, fm = gettimer(ui, opts)
818 818 cl = repo.changelog
819 819
820 820 def s():
821 821 clearcaches(cl)
822 822
823 823 def d():
824 824 len(cl.headrevs())
825 825
826 826 timer(d, setup=s)
827 827 fm.end()
828 828
829 829
830 830 @command(
831 831 b'perftags',
832 832 formatteropts
833 833 + [(b'', b'clear-revlogs', False, b'refresh changelog and manifest'),],
834 834 )
835 835 def perftags(ui, repo, **opts):
836 836 opts = _byteskwargs(opts)
837 837 timer, fm = gettimer(ui, opts)
838 838 repocleartagscache = repocleartagscachefunc(repo)
839 839 clearrevlogs = opts[b'clear_revlogs']
840 840
841 841 def s():
842 842 if clearrevlogs:
843 843 clearchangelog(repo)
844 844 clearfilecache(repo.unfiltered(), 'manifest')
845 845 repocleartagscache()
846 846
847 847 def t():
848 848 return len(repo.tags())
849 849
850 850 timer(t, setup=s)
851 851 fm.end()
852 852
853 853
854 854 @command(b'perfancestors', formatteropts)
855 855 def perfancestors(ui, repo, **opts):
856 856 opts = _byteskwargs(opts)
857 857 timer, fm = gettimer(ui, opts)
858 858 heads = repo.changelog.headrevs()
859 859
860 860 def d():
861 861 for a in repo.changelog.ancestors(heads):
862 862 pass
863 863
864 864 timer(d)
865 865 fm.end()
866 866
867 867
868 868 @command(b'perfancestorset', formatteropts)
869 869 def perfancestorset(ui, repo, revset, **opts):
870 870 opts = _byteskwargs(opts)
871 871 timer, fm = gettimer(ui, opts)
872 872 revs = repo.revs(revset)
873 873 heads = repo.changelog.headrevs()
874 874
875 875 def d():
876 876 s = repo.changelog.ancestors(heads)
877 877 for rev in revs:
878 878 rev in s
879 879
880 880 timer(d)
881 881 fm.end()
882 882
883 883
884 884 @command(b'perfdiscovery', formatteropts, b'PATH')
885 885 def perfdiscovery(ui, repo, path, **opts):
886 886 """benchmark discovery between local repo and the peer at given path
887 887 """
888 888 repos = [repo, None]
889 889 timer, fm = gettimer(ui, opts)
890 890 path = ui.expandpath(path)
891 891
892 892 def s():
893 893 repos[1] = hg.peer(ui, opts, path)
894 894
895 895 def d():
896 896 setdiscovery.findcommonheads(ui, *repos)
897 897
898 898 timer(d, setup=s)
899 899 fm.end()
900 900
901 901
902 902 @command(
903 903 b'perfbookmarks',
904 904 formatteropts
905 905 + [(b'', b'clear-revlogs', False, b'refresh changelog and manifest'),],
906 906 )
907 907 def perfbookmarks(ui, repo, **opts):
908 908 """benchmark parsing bookmarks from disk to memory"""
909 909 opts = _byteskwargs(opts)
910 910 timer, fm = gettimer(ui, opts)
911 911
912 912 clearrevlogs = opts[b'clear_revlogs']
913 913
914 914 def s():
915 915 if clearrevlogs:
916 916 clearchangelog(repo)
917 917 clearfilecache(repo, b'_bookmarks')
918 918
919 919 def d():
920 920 repo._bookmarks
921 921
922 922 timer(d, setup=s)
923 923 fm.end()
924 924
925 925
926 926 @command(b'perfbundleread', formatteropts, b'BUNDLE')
927 927 def perfbundleread(ui, repo, bundlepath, **opts):
928 928 """Benchmark reading of bundle files.
929 929
930 930 This command is meant to isolate the I/O part of bundle reading as
931 931 much as possible.
932 932 """
933 933 from mercurial import (
934 934 bundle2,
935 935 exchange,
936 936 streamclone,
937 937 )
938 938
939 939 opts = _byteskwargs(opts)
940 940
941 941 def makebench(fn):
942 942 def run():
943 943 with open(bundlepath, b'rb') as fh:
944 944 bundle = exchange.readbundle(ui, fh, bundlepath)
945 945 fn(bundle)
946 946
947 947 return run
948 948
949 949 def makereadnbytes(size):
950 950 def run():
951 951 with open(bundlepath, b'rb') as fh:
952 952 bundle = exchange.readbundle(ui, fh, bundlepath)
953 953 while bundle.read(size):
954 954 pass
955 955
956 956 return run
957 957
958 958 def makestdioread(size):
959 959 def run():
960 960 with open(bundlepath, b'rb') as fh:
961 961 while fh.read(size):
962 962 pass
963 963
964 964 return run
965 965
966 966 # bundle1
967 967
968 968 def deltaiter(bundle):
969 969 for delta in bundle.deltaiter():
970 970 pass
971 971
972 972 def iterchunks(bundle):
973 973 for chunk in bundle.getchunks():
974 974 pass
975 975
976 976 # bundle2
977 977
978 978 def forwardchunks(bundle):
979 979 for chunk in bundle._forwardchunks():
980 980 pass
981 981
982 982 def iterparts(bundle):
983 983 for part in bundle.iterparts():
984 984 pass
985 985
986 986 def iterpartsseekable(bundle):
987 987 for part in bundle.iterparts(seekable=True):
988 988 pass
989 989
990 990 def seek(bundle):
991 991 for part in bundle.iterparts(seekable=True):
992 992 part.seek(0, os.SEEK_END)
993 993
994 994 def makepartreadnbytes(size):
995 995 def run():
996 996 with open(bundlepath, b'rb') as fh:
997 997 bundle = exchange.readbundle(ui, fh, bundlepath)
998 998 for part in bundle.iterparts():
999 999 while part.read(size):
1000 1000 pass
1001 1001
1002 1002 return run
1003 1003
1004 1004 benches = [
1005 1005 (makestdioread(8192), b'read(8k)'),
1006 1006 (makestdioread(16384), b'read(16k)'),
1007 1007 (makestdioread(32768), b'read(32k)'),
1008 1008 (makestdioread(131072), b'read(128k)'),
1009 1009 ]
1010 1010
1011 1011 with open(bundlepath, b'rb') as fh:
1012 1012 bundle = exchange.readbundle(ui, fh, bundlepath)
1013 1013
1014 1014 if isinstance(bundle, changegroup.cg1unpacker):
1015 1015 benches.extend(
1016 1016 [
1017 1017 (makebench(deltaiter), b'cg1 deltaiter()'),
1018 1018 (makebench(iterchunks), b'cg1 getchunks()'),
1019 1019 (makereadnbytes(8192), b'cg1 read(8k)'),
1020 1020 (makereadnbytes(16384), b'cg1 read(16k)'),
1021 1021 (makereadnbytes(32768), b'cg1 read(32k)'),
1022 1022 (makereadnbytes(131072), b'cg1 read(128k)'),
1023 1023 ]
1024 1024 )
1025 1025 elif isinstance(bundle, bundle2.unbundle20):
1026 1026 benches.extend(
1027 1027 [
1028 1028 (makebench(forwardchunks), b'bundle2 forwardchunks()'),
1029 1029 (makebench(iterparts), b'bundle2 iterparts()'),
1030 1030 (
1031 1031 makebench(iterpartsseekable),
1032 1032 b'bundle2 iterparts() seekable',
1033 1033 ),
1034 1034 (makebench(seek), b'bundle2 part seek()'),
1035 1035 (makepartreadnbytes(8192), b'bundle2 part read(8k)'),
1036 1036 (makepartreadnbytes(16384), b'bundle2 part read(16k)'),
1037 1037 (makepartreadnbytes(32768), b'bundle2 part read(32k)'),
1038 1038 (makepartreadnbytes(131072), b'bundle2 part read(128k)'),
1039 1039 ]
1040 1040 )
1041 1041 elif isinstance(bundle, streamclone.streamcloneapplier):
1042 1042 raise error.Abort(b'stream clone bundles not supported')
1043 1043 else:
1044 1044 raise error.Abort(b'unhandled bundle type: %s' % type(bundle))
1045 1045
1046 1046 for fn, title in benches:
1047 1047 timer, fm = gettimer(ui, opts)
1048 1048 timer(fn, title=title)
1049 1049 fm.end()
1050 1050
1051 1051
1052 1052 @command(
1053 1053 b'perfchangegroupchangelog',
1054 1054 formatteropts
1055 1055 + [
1056 1056 (b'', b'cgversion', b'02', b'changegroup version'),
1057 1057 (b'r', b'rev', b'', b'revisions to add to changegroup'),
1058 1058 ],
1059 1059 )
1060 1060 def perfchangegroupchangelog(ui, repo, cgversion=b'02', rev=None, **opts):
1061 1061 """Benchmark producing a changelog group for a changegroup.
1062 1062
1063 1063 This measures the time spent processing the changelog during a
1064 1064 bundle operation. This occurs during `hg bundle` and on a server
1065 1065 processing a `getbundle` wire protocol request (handles clones
1066 1066 and pull requests).
1067 1067
1068 1068 By default, all revisions are added to the changegroup.
1069 1069 """
1070 1070 opts = _byteskwargs(opts)
1071 1071 cl = repo.changelog
1072 1072 nodes = [cl.lookup(r) for r in repo.revs(rev or b'all()')]
1073 1073 bundler = changegroup.getbundler(cgversion, repo)
1074 1074
1075 1075 def d():
1076 1076 state, chunks = bundler._generatechangelog(cl, nodes)
1077 1077 for chunk in chunks:
1078 1078 pass
1079 1079
1080 1080 timer, fm = gettimer(ui, opts)
1081 1081
1082 1082 # Terminal printing can interfere with timing. So disable it.
1083 1083 with ui.configoverride({(b'progress', b'disable'): True}):
1084 1084 timer(d)
1085 1085
1086 1086 fm.end()
1087 1087
1088 1088
1089 1089 @command(b'perfdirs', formatteropts)
1090 1090 def perfdirs(ui, repo, **opts):
1091 1091 opts = _byteskwargs(opts)
1092 1092 timer, fm = gettimer(ui, opts)
1093 1093 dirstate = repo.dirstate
1094 1094 b'a' in dirstate
1095 1095
1096 1096 def d():
1097 1097 dirstate.hasdir(b'a')
1098 1098 del dirstate._map._dirs
1099 1099
1100 1100 timer(d)
1101 1101 fm.end()
1102 1102
1103 1103
1104 @command(b'perfdirstate', formatteropts)
1104 @command(b'perfdirstate', [
1105 (b'', b'iteration', None,
1106 b'benchmark a full iteration for the dirstate'),
1107 ] + formatteropts)
1105 1108 def perfdirstate(ui, repo, **opts):
1106 """benchmap the time necessary to load a dirstate from scratch
1107
1109 """benchmap the time of various distate operations
1110
1111 By default benchmark the time necessary to load a dirstate from scratch.
1108 1112 The dirstate is loaded to the point were a "contains" request can be
1109 1113 answered.
1110 1114 """
1111 1115 opts = _byteskwargs(opts)
1112 1116 timer, fm = gettimer(ui, opts)
1113 1117 b"a" in repo.dirstate
1114 1118
1115 def setup():
1116 repo.dirstate.invalidate()
1117
1118 def d():
1119 b"a" in repo.dirstate
1119 if opts[b'iteration']:
1120 setup = None
1121 dirstate = repo.dirstate
1122 def d():
1123 for f in dirstate:
1124 pass
1125 else:
1126 def setup():
1127 repo.dirstate.invalidate()
1128
1129 def d():
1130 b"a" in repo.dirstate
1120 1131
1121 1132 timer(d, setup=setup)
1122 1133 fm.end()
1123 1134
1124 1135
1125 1136 @command(b'perfdirstatedirs', formatteropts)
1126 1137 def perfdirstatedirs(ui, repo, **opts):
1127 1138 """benchmap a 'dirstate.hasdir' call from an empty `dirs` cache
1128 1139 """
1129 1140 opts = _byteskwargs(opts)
1130 1141 timer, fm = gettimer(ui, opts)
1131 1142 repo.dirstate.hasdir(b"a")
1132 1143
1133 1144 def setup():
1134 1145 del repo.dirstate._map._dirs
1135 1146
1136 1147 def d():
1137 1148 repo.dirstate.hasdir(b"a")
1138 1149
1139 1150 timer(d, setup=setup)
1140 1151 fm.end()
1141 1152
1142 1153
1143 1154 @command(b'perfdirstatefoldmap', formatteropts)
1144 1155 def perfdirstatefoldmap(ui, repo, **opts):
1145 1156 """benchmap a `dirstate._map.filefoldmap.get()` request
1146 1157
1147 1158 The dirstate filefoldmap cache is dropped between every request.
1148 1159 """
1149 1160 opts = _byteskwargs(opts)
1150 1161 timer, fm = gettimer(ui, opts)
1151 1162 dirstate = repo.dirstate
1152 1163 dirstate._map.filefoldmap.get(b'a')
1153 1164
1154 1165 def setup():
1155 1166 del dirstate._map.filefoldmap
1156 1167
1157 1168 def d():
1158 1169 dirstate._map.filefoldmap.get(b'a')
1159 1170
1160 1171 timer(d, setup=setup)
1161 1172 fm.end()
1162 1173
1163 1174
1164 1175 @command(b'perfdirfoldmap', formatteropts)
1165 1176 def perfdirfoldmap(ui, repo, **opts):
1166 1177 """benchmap a `dirstate._map.dirfoldmap.get()` request
1167 1178
1168 1179 The dirstate dirfoldmap cache is dropped between every request.
1169 1180 """
1170 1181 opts = _byteskwargs(opts)
1171 1182 timer, fm = gettimer(ui, opts)
1172 1183 dirstate = repo.dirstate
1173 1184 dirstate._map.dirfoldmap.get(b'a')
1174 1185
1175 1186 def setup():
1176 1187 del dirstate._map.dirfoldmap
1177 1188 del dirstate._map._dirs
1178 1189
1179 1190 def d():
1180 1191 dirstate._map.dirfoldmap.get(b'a')
1181 1192
1182 1193 timer(d, setup=setup)
1183 1194 fm.end()
1184 1195
1185 1196
1186 1197 @command(b'perfdirstatewrite', formatteropts)
1187 1198 def perfdirstatewrite(ui, repo, **opts):
1188 1199 """benchmap the time it take to write a dirstate on disk
1189 1200 """
1190 1201 opts = _byteskwargs(opts)
1191 1202 timer, fm = gettimer(ui, opts)
1192 1203 ds = repo.dirstate
1193 1204 b"a" in ds
1194 1205
1195 1206 def setup():
1196 1207 ds._dirty = True
1197 1208
1198 1209 def d():
1199 1210 ds.write(repo.currenttransaction())
1200 1211
1201 1212 timer(d, setup=setup)
1202 1213 fm.end()
1203 1214
1204 1215
1205 1216 def _getmergerevs(repo, opts):
1206 1217 """parse command argument to return rev involved in merge
1207 1218
1208 1219 input: options dictionnary with `rev`, `from` and `bse`
1209 1220 output: (localctx, otherctx, basectx)
1210 1221 """
1211 1222 if opts[b'from']:
1212 1223 fromrev = scmutil.revsingle(repo, opts[b'from'])
1213 1224 wctx = repo[fromrev]
1214 1225 else:
1215 1226 wctx = repo[None]
1216 1227 # we don't want working dir files to be stat'd in the benchmark, so
1217 1228 # prime that cache
1218 1229 wctx.dirty()
1219 1230 rctx = scmutil.revsingle(repo, opts[b'rev'], opts[b'rev'])
1220 1231 if opts[b'base']:
1221 1232 fromrev = scmutil.revsingle(repo, opts[b'base'])
1222 1233 ancestor = repo[fromrev]
1223 1234 else:
1224 1235 ancestor = wctx.ancestor(rctx)
1225 1236 return (wctx, rctx, ancestor)
1226 1237
1227 1238
1228 1239 @command(
1229 1240 b'perfmergecalculate',
1230 1241 [
1231 1242 (b'r', b'rev', b'.', b'rev to merge against'),
1232 1243 (b'', b'from', b'', b'rev to merge from'),
1233 1244 (b'', b'base', b'', b'the revision to use as base'),
1234 1245 ]
1235 1246 + formatteropts,
1236 1247 )
1237 1248 def perfmergecalculate(ui, repo, **opts):
1238 1249 opts = _byteskwargs(opts)
1239 1250 timer, fm = gettimer(ui, opts)
1240 1251
1241 1252 wctx, rctx, ancestor = _getmergerevs(repo, opts)
1242 1253
1243 1254 def d():
1244 1255 # acceptremote is True because we don't want prompts in the middle of
1245 1256 # our benchmark
1246 1257 merge.calculateupdates(
1247 1258 repo,
1248 1259 wctx,
1249 1260 rctx,
1250 1261 [ancestor],
1251 1262 branchmerge=False,
1252 1263 force=False,
1253 1264 acceptremote=True,
1254 1265 followcopies=True,
1255 1266 )
1256 1267
1257 1268 timer(d)
1258 1269 fm.end()
1259 1270
1260 1271
1261 1272 @command(
1262 1273 b'perfmergecopies',
1263 1274 [
1264 1275 (b'r', b'rev', b'.', b'rev to merge against'),
1265 1276 (b'', b'from', b'', b'rev to merge from'),
1266 1277 (b'', b'base', b'', b'the revision to use as base'),
1267 1278 ]
1268 1279 + formatteropts,
1269 1280 )
1270 1281 def perfmergecopies(ui, repo, **opts):
1271 1282 """measure runtime of `copies.mergecopies`"""
1272 1283 opts = _byteskwargs(opts)
1273 1284 timer, fm = gettimer(ui, opts)
1274 1285 wctx, rctx, ancestor = _getmergerevs(repo, opts)
1275 1286
1276 1287 def d():
1277 1288 # acceptremote is True because we don't want prompts in the middle of
1278 1289 # our benchmark
1279 1290 copies.mergecopies(repo, wctx, rctx, ancestor)
1280 1291
1281 1292 timer(d)
1282 1293 fm.end()
1283 1294
1284 1295
1285 1296 @command(b'perfpathcopies', [], b"REV REV")
1286 1297 def perfpathcopies(ui, repo, rev1, rev2, **opts):
1287 1298 """benchmark the copy tracing logic"""
1288 1299 opts = _byteskwargs(opts)
1289 1300 timer, fm = gettimer(ui, opts)
1290 1301 ctx1 = scmutil.revsingle(repo, rev1, rev1)
1291 1302 ctx2 = scmutil.revsingle(repo, rev2, rev2)
1292 1303
1293 1304 def d():
1294 1305 copies.pathcopies(ctx1, ctx2)
1295 1306
1296 1307 timer(d)
1297 1308 fm.end()
1298 1309
1299 1310
1300 1311 @command(
1301 1312 b'perfphases',
1302 1313 [(b'', b'full', False, b'include file reading time too'),],
1303 1314 b"",
1304 1315 )
1305 1316 def perfphases(ui, repo, **opts):
1306 1317 """benchmark phasesets computation"""
1307 1318 opts = _byteskwargs(opts)
1308 1319 timer, fm = gettimer(ui, opts)
1309 1320 _phases = repo._phasecache
1310 1321 full = opts.get(b'full')
1311 1322
1312 1323 def d():
1313 1324 phases = _phases
1314 1325 if full:
1315 1326 clearfilecache(repo, b'_phasecache')
1316 1327 phases = repo._phasecache
1317 1328 phases.invalidate()
1318 1329 phases.loadphaserevs(repo)
1319 1330
1320 1331 timer(d)
1321 1332 fm.end()
1322 1333
1323 1334
1324 1335 @command(b'perfphasesremote', [], b"[DEST]")
1325 1336 def perfphasesremote(ui, repo, dest=None, **opts):
1326 1337 """benchmark time needed to analyse phases of the remote server"""
1327 1338 from mercurial.node import bin
1328 1339 from mercurial import (
1329 1340 exchange,
1330 1341 hg,
1331 1342 phases,
1332 1343 )
1333 1344
1334 1345 opts = _byteskwargs(opts)
1335 1346 timer, fm = gettimer(ui, opts)
1336 1347
1337 1348 path = ui.paths.getpath(dest, default=(b'default-push', b'default'))
1338 1349 if not path:
1339 1350 raise error.Abort(
1340 1351 b'default repository not configured!',
1341 1352 hint=b"see 'hg help config.paths'",
1342 1353 )
1343 1354 dest = path.pushloc or path.loc
1344 1355 ui.statusnoi18n(b'analysing phase of %s\n' % util.hidepassword(dest))
1345 1356 other = hg.peer(repo, opts, dest)
1346 1357
1347 1358 # easier to perform discovery through the operation
1348 1359 op = exchange.pushoperation(repo, other)
1349 1360 exchange._pushdiscoverychangeset(op)
1350 1361
1351 1362 remotesubset = op.fallbackheads
1352 1363
1353 1364 with other.commandexecutor() as e:
1354 1365 remotephases = e.callcommand(
1355 1366 b'listkeys', {b'namespace': b'phases'}
1356 1367 ).result()
1357 1368 del other
1358 1369 publishing = remotephases.get(b'publishing', False)
1359 1370 if publishing:
1360 1371 ui.statusnoi18n(b'publishing: yes\n')
1361 1372 else:
1362 1373 ui.statusnoi18n(b'publishing: no\n')
1363 1374
1364 1375 nodemap = repo.changelog.nodemap
1365 1376 nonpublishroots = 0
1366 1377 for nhex, phase in remotephases.iteritems():
1367 1378 if nhex == b'publishing': # ignore data related to publish option
1368 1379 continue
1369 1380 node = bin(nhex)
1370 1381 if node in nodemap and int(phase):
1371 1382 nonpublishroots += 1
1372 1383 ui.statusnoi18n(b'number of roots: %d\n' % len(remotephases))
1373 1384 ui.statusnoi18n(b'number of known non public roots: %d\n' % nonpublishroots)
1374 1385
1375 1386 def d():
1376 1387 phases.remotephasessummary(repo, remotesubset, remotephases)
1377 1388
1378 1389 timer(d)
1379 1390 fm.end()
1380 1391
1381 1392
1382 1393 @command(
1383 1394 b'perfmanifest',
1384 1395 [
1385 1396 (b'm', b'manifest-rev', False, b'Look up a manifest node revision'),
1386 1397 (b'', b'clear-disk', False, b'clear on-disk caches too'),
1387 1398 ]
1388 1399 + formatteropts,
1389 1400 b'REV|NODE',
1390 1401 )
1391 1402 def perfmanifest(ui, repo, rev, manifest_rev=False, clear_disk=False, **opts):
1392 1403 """benchmark the time to read a manifest from disk and return a usable
1393 1404 dict-like object
1394 1405
1395 1406 Manifest caches are cleared before retrieval."""
1396 1407 opts = _byteskwargs(opts)
1397 1408 timer, fm = gettimer(ui, opts)
1398 1409 if not manifest_rev:
1399 1410 ctx = scmutil.revsingle(repo, rev, rev)
1400 1411 t = ctx.manifestnode()
1401 1412 else:
1402 1413 from mercurial.node import bin
1403 1414
1404 1415 if len(rev) == 40:
1405 1416 t = bin(rev)
1406 1417 else:
1407 1418 try:
1408 1419 rev = int(rev)
1409 1420
1410 1421 if util.safehasattr(repo.manifestlog, b'getstorage'):
1411 1422 t = repo.manifestlog.getstorage(b'').node(rev)
1412 1423 else:
1413 1424 t = repo.manifestlog._revlog.lookup(rev)
1414 1425 except ValueError:
1415 1426 raise error.Abort(
1416 1427 b'manifest revision must be integer or full node'
1417 1428 )
1418 1429
1419 1430 def d():
1420 1431 repo.manifestlog.clearcaches(clear_persisted_data=clear_disk)
1421 1432 repo.manifestlog[t].read()
1422 1433
1423 1434 timer(d)
1424 1435 fm.end()
1425 1436
1426 1437
1427 1438 @command(b'perfchangeset', formatteropts)
1428 1439 def perfchangeset(ui, repo, rev, **opts):
1429 1440 opts = _byteskwargs(opts)
1430 1441 timer, fm = gettimer(ui, opts)
1431 1442 n = scmutil.revsingle(repo, rev).node()
1432 1443
1433 1444 def d():
1434 1445 repo.changelog.read(n)
1435 1446 # repo.changelog._cache = None
1436 1447
1437 1448 timer(d)
1438 1449 fm.end()
1439 1450
1440 1451
1441 1452 @command(b'perfignore', formatteropts)
1442 1453 def perfignore(ui, repo, **opts):
1443 1454 """benchmark operation related to computing ignore"""
1444 1455 opts = _byteskwargs(opts)
1445 1456 timer, fm = gettimer(ui, opts)
1446 1457 dirstate = repo.dirstate
1447 1458
1448 1459 def setupone():
1449 1460 dirstate.invalidate()
1450 1461 clearfilecache(dirstate, b'_ignore')
1451 1462
1452 1463 def runone():
1453 1464 dirstate._ignore
1454 1465
1455 1466 timer(runone, setup=setupone, title=b"load")
1456 1467 fm.end()
1457 1468
1458 1469
1459 1470 @command(
1460 1471 b'perfindex',
1461 1472 [
1462 1473 (b'', b'rev', [], b'revision to be looked up (default tip)'),
1463 1474 (b'', b'no-lookup', None, b'do not revision lookup post creation'),
1464 1475 ]
1465 1476 + formatteropts,
1466 1477 )
1467 1478 def perfindex(ui, repo, **opts):
1468 1479 """benchmark index creation time followed by a lookup
1469 1480
1470 1481 The default is to look `tip` up. Depending on the index implementation,
1471 1482 the revision looked up can matters. For example, an implementation
1472 1483 scanning the index will have a faster lookup time for `--rev tip` than for
1473 1484 `--rev 0`. The number of looked up revisions and their order can also
1474 1485 matters.
1475 1486
1476 1487 Example of useful set to test:
1477 1488 * tip
1478 1489 * 0
1479 1490 * -10:
1480 1491 * :10
1481 1492 * -10: + :10
1482 1493 * :10: + -10:
1483 1494 * -10000:
1484 1495 * -10000: + 0
1485 1496
1486 1497 It is not currently possible to check for lookup of a missing node. For
1487 1498 deeper lookup benchmarking, checkout the `perfnodemap` command."""
1488 1499 import mercurial.revlog
1489 1500
1490 1501 opts = _byteskwargs(opts)
1491 1502 timer, fm = gettimer(ui, opts)
1492 1503 mercurial.revlog._prereadsize = 2 ** 24 # disable lazy parser in old hg
1493 1504 if opts[b'no_lookup']:
1494 1505 if opts['rev']:
1495 1506 raise error.Abort('--no-lookup and --rev are mutually exclusive')
1496 1507 nodes = []
1497 1508 elif not opts[b'rev']:
1498 1509 nodes = [repo[b"tip"].node()]
1499 1510 else:
1500 1511 revs = scmutil.revrange(repo, opts[b'rev'])
1501 1512 cl = repo.changelog
1502 1513 nodes = [cl.node(r) for r in revs]
1503 1514
1504 1515 unfi = repo.unfiltered()
1505 1516 # find the filecache func directly
1506 1517 # This avoid polluting the benchmark with the filecache logic
1507 1518 makecl = unfi.__class__.changelog.func
1508 1519
1509 1520 def setup():
1510 1521 # probably not necessary, but for good measure
1511 1522 clearchangelog(unfi)
1512 1523
1513 1524 def d():
1514 1525 cl = makecl(unfi)
1515 1526 for n in nodes:
1516 1527 cl.rev(n)
1517 1528
1518 1529 timer(d, setup=setup)
1519 1530 fm.end()
1520 1531
1521 1532
1522 1533 @command(
1523 1534 b'perfnodemap',
1524 1535 [
1525 1536 (b'', b'rev', [], b'revision to be looked up (default tip)'),
1526 1537 (b'', b'clear-caches', True, b'clear revlog cache between calls'),
1527 1538 ]
1528 1539 + formatteropts,
1529 1540 )
1530 1541 def perfnodemap(ui, repo, **opts):
1531 1542 """benchmark the time necessary to look up revision from a cold nodemap
1532 1543
1533 1544 Depending on the implementation, the amount and order of revision we look
1534 1545 up can varies. Example of useful set to test:
1535 1546 * tip
1536 1547 * 0
1537 1548 * -10:
1538 1549 * :10
1539 1550 * -10: + :10
1540 1551 * :10: + -10:
1541 1552 * -10000:
1542 1553 * -10000: + 0
1543 1554
1544 1555 The command currently focus on valid binary lookup. Benchmarking for
1545 1556 hexlookup, prefix lookup and missing lookup would also be valuable.
1546 1557 """
1547 1558 import mercurial.revlog
1548 1559
1549 1560 opts = _byteskwargs(opts)
1550 1561 timer, fm = gettimer(ui, opts)
1551 1562 mercurial.revlog._prereadsize = 2 ** 24 # disable lazy parser in old hg
1552 1563
1553 1564 unfi = repo.unfiltered()
1554 1565 clearcaches = opts['clear_caches']
1555 1566 # find the filecache func directly
1556 1567 # This avoid polluting the benchmark with the filecache logic
1557 1568 makecl = unfi.__class__.changelog.func
1558 1569 if not opts[b'rev']:
1559 1570 raise error.Abort('use --rev to specify revisions to look up')
1560 1571 revs = scmutil.revrange(repo, opts[b'rev'])
1561 1572 cl = repo.changelog
1562 1573 nodes = [cl.node(r) for r in revs]
1563 1574
1564 1575 # use a list to pass reference to a nodemap from one closure to the next
1565 1576 nodeget = [None]
1566 1577
1567 1578 def setnodeget():
1568 1579 # probably not necessary, but for good measure
1569 1580 clearchangelog(unfi)
1570 1581 nodeget[0] = makecl(unfi).nodemap.get
1571 1582
1572 1583 def d():
1573 1584 get = nodeget[0]
1574 1585 for n in nodes:
1575 1586 get(n)
1576 1587
1577 1588 setup = None
1578 1589 if clearcaches:
1579 1590
1580 1591 def setup():
1581 1592 setnodeget()
1582 1593
1583 1594 else:
1584 1595 setnodeget()
1585 1596 d() # prewarm the data structure
1586 1597 timer(d, setup=setup)
1587 1598 fm.end()
1588 1599
1589 1600
1590 1601 @command(b'perfstartup', formatteropts)
1591 1602 def perfstartup(ui, repo, **opts):
1592 1603 opts = _byteskwargs(opts)
1593 1604 timer, fm = gettimer(ui, opts)
1594 1605
1595 1606 def d():
1596 1607 if os.name != r'nt':
1597 1608 os.system(
1598 1609 b"HGRCPATH= %s version -q > /dev/null" % fsencode(sys.argv[0])
1599 1610 )
1600 1611 else:
1601 1612 os.environ[r'HGRCPATH'] = r' '
1602 1613 os.system(r"%s version -q > NUL" % sys.argv[0])
1603 1614
1604 1615 timer(d)
1605 1616 fm.end()
1606 1617
1607 1618
1608 1619 @command(b'perfparents', formatteropts)
1609 1620 def perfparents(ui, repo, **opts):
1610 1621 """benchmark the time necessary to fetch one changeset's parents.
1611 1622
1612 1623 The fetch is done using the `node identifier`, traversing all object layers
1613 1624 from the repository object. The first N revisions will be used for this
1614 1625 benchmark. N is controlled by the ``perf.parentscount`` config option
1615 1626 (default: 1000).
1616 1627 """
1617 1628 opts = _byteskwargs(opts)
1618 1629 timer, fm = gettimer(ui, opts)
1619 1630 # control the number of commits perfparents iterates over
1620 1631 # experimental config: perf.parentscount
1621 1632 count = getint(ui, b"perf", b"parentscount", 1000)
1622 1633 if len(repo.changelog) < count:
1623 1634 raise error.Abort(b"repo needs %d commits for this test" % count)
1624 1635 repo = repo.unfiltered()
1625 1636 nl = [repo.changelog.node(i) for i in _xrange(count)]
1626 1637
1627 1638 def d():
1628 1639 for n in nl:
1629 1640 repo.changelog.parents(n)
1630 1641
1631 1642 timer(d)
1632 1643 fm.end()
1633 1644
1634 1645
1635 1646 @command(b'perfctxfiles', formatteropts)
1636 1647 def perfctxfiles(ui, repo, x, **opts):
1637 1648 opts = _byteskwargs(opts)
1638 1649 x = int(x)
1639 1650 timer, fm = gettimer(ui, opts)
1640 1651
1641 1652 def d():
1642 1653 len(repo[x].files())
1643 1654
1644 1655 timer(d)
1645 1656 fm.end()
1646 1657
1647 1658
1648 1659 @command(b'perfrawfiles', formatteropts)
1649 1660 def perfrawfiles(ui, repo, x, **opts):
1650 1661 opts = _byteskwargs(opts)
1651 1662 x = int(x)
1652 1663 timer, fm = gettimer(ui, opts)
1653 1664 cl = repo.changelog
1654 1665
1655 1666 def d():
1656 1667 len(cl.read(x)[3])
1657 1668
1658 1669 timer(d)
1659 1670 fm.end()
1660 1671
1661 1672
1662 1673 @command(b'perflookup', formatteropts)
1663 1674 def perflookup(ui, repo, rev, **opts):
1664 1675 opts = _byteskwargs(opts)
1665 1676 timer, fm = gettimer(ui, opts)
1666 1677 timer(lambda: len(repo.lookup(rev)))
1667 1678 fm.end()
1668 1679
1669 1680
1670 1681 @command(
1671 1682 b'perflinelogedits',
1672 1683 [
1673 1684 (b'n', b'edits', 10000, b'number of edits'),
1674 1685 (b'', b'max-hunk-lines', 10, b'max lines in a hunk'),
1675 1686 ],
1676 1687 norepo=True,
1677 1688 )
1678 1689 def perflinelogedits(ui, **opts):
1679 1690 from mercurial import linelog
1680 1691
1681 1692 opts = _byteskwargs(opts)
1682 1693
1683 1694 edits = opts[b'edits']
1684 1695 maxhunklines = opts[b'max_hunk_lines']
1685 1696
1686 1697 maxb1 = 100000
1687 1698 random.seed(0)
1688 1699 randint = random.randint
1689 1700 currentlines = 0
1690 1701 arglist = []
1691 1702 for rev in _xrange(edits):
1692 1703 a1 = randint(0, currentlines)
1693 1704 a2 = randint(a1, min(currentlines, a1 + maxhunklines))
1694 1705 b1 = randint(0, maxb1)
1695 1706 b2 = randint(b1, b1 + maxhunklines)
1696 1707 currentlines += (b2 - b1) - (a2 - a1)
1697 1708 arglist.append((rev, a1, a2, b1, b2))
1698 1709
1699 1710 def d():
1700 1711 ll = linelog.linelog()
1701 1712 for args in arglist:
1702 1713 ll.replacelines(*args)
1703 1714
1704 1715 timer, fm = gettimer(ui, opts)
1705 1716 timer(d)
1706 1717 fm.end()
1707 1718
1708 1719
1709 1720 @command(b'perfrevrange', formatteropts)
1710 1721 def perfrevrange(ui, repo, *specs, **opts):
1711 1722 opts = _byteskwargs(opts)
1712 1723 timer, fm = gettimer(ui, opts)
1713 1724 revrange = scmutil.revrange
1714 1725 timer(lambda: len(revrange(repo, specs)))
1715 1726 fm.end()
1716 1727
1717 1728
1718 1729 @command(b'perfnodelookup', formatteropts)
1719 1730 def perfnodelookup(ui, repo, rev, **opts):
1720 1731 opts = _byteskwargs(opts)
1721 1732 timer, fm = gettimer(ui, opts)
1722 1733 import mercurial.revlog
1723 1734
1724 1735 mercurial.revlog._prereadsize = 2 ** 24 # disable lazy parser in old hg
1725 1736 n = scmutil.revsingle(repo, rev).node()
1726 1737 cl = mercurial.revlog.revlog(getsvfs(repo), b"00changelog.i")
1727 1738
1728 1739 def d():
1729 1740 cl.rev(n)
1730 1741 clearcaches(cl)
1731 1742
1732 1743 timer(d)
1733 1744 fm.end()
1734 1745
1735 1746
1736 1747 @command(
1737 1748 b'perflog',
1738 1749 [(b'', b'rename', False, b'ask log to follow renames')] + formatteropts,
1739 1750 )
1740 1751 def perflog(ui, repo, rev=None, **opts):
1741 1752 opts = _byteskwargs(opts)
1742 1753 if rev is None:
1743 1754 rev = []
1744 1755 timer, fm = gettimer(ui, opts)
1745 1756 ui.pushbuffer()
1746 1757 timer(
1747 1758 lambda: commands.log(
1748 1759 ui, repo, rev=rev, date=b'', user=b'', copies=opts.get(b'rename')
1749 1760 )
1750 1761 )
1751 1762 ui.popbuffer()
1752 1763 fm.end()
1753 1764
1754 1765
1755 1766 @command(b'perfmoonwalk', formatteropts)
1756 1767 def perfmoonwalk(ui, repo, **opts):
1757 1768 """benchmark walking the changelog backwards
1758 1769
1759 1770 This also loads the changelog data for each revision in the changelog.
1760 1771 """
1761 1772 opts = _byteskwargs(opts)
1762 1773 timer, fm = gettimer(ui, opts)
1763 1774
1764 1775 def moonwalk():
1765 1776 for i in repo.changelog.revs(start=(len(repo) - 1), stop=-1):
1766 1777 ctx = repo[i]
1767 1778 ctx.branch() # read changelog data (in addition to the index)
1768 1779
1769 1780 timer(moonwalk)
1770 1781 fm.end()
1771 1782
1772 1783
1773 1784 @command(
1774 1785 b'perftemplating',
1775 1786 [(b'r', b'rev', [], b'revisions to run the template on'),] + formatteropts,
1776 1787 )
1777 1788 def perftemplating(ui, repo, testedtemplate=None, **opts):
1778 1789 """test the rendering time of a given template"""
1779 1790 if makelogtemplater is None:
1780 1791 raise error.Abort(
1781 1792 b"perftemplating not available with this Mercurial",
1782 1793 hint=b"use 4.3 or later",
1783 1794 )
1784 1795
1785 1796 opts = _byteskwargs(opts)
1786 1797
1787 1798 nullui = ui.copy()
1788 1799 nullui.fout = open(os.devnull, r'wb')
1789 1800 nullui.disablepager()
1790 1801 revs = opts.get(b'rev')
1791 1802 if not revs:
1792 1803 revs = [b'all()']
1793 1804 revs = list(scmutil.revrange(repo, revs))
1794 1805
1795 1806 defaulttemplate = (
1796 1807 b'{date|shortdate} [{rev}:{node|short}]'
1797 1808 b' {author|person}: {desc|firstline}\n'
1798 1809 )
1799 1810 if testedtemplate is None:
1800 1811 testedtemplate = defaulttemplate
1801 1812 displayer = makelogtemplater(nullui, repo, testedtemplate)
1802 1813
1803 1814 def format():
1804 1815 for r in revs:
1805 1816 ctx = repo[r]
1806 1817 displayer.show(ctx)
1807 1818 displayer.flush(ctx)
1808 1819
1809 1820 timer, fm = gettimer(ui, opts)
1810 1821 timer(format)
1811 1822 fm.end()
1812 1823
1813 1824
1814 1825 def _displaystats(ui, opts, entries, data):
1815 1826 pass
1816 1827 # use a second formatter because the data are quite different, not sure
1817 1828 # how it flies with the templater.
1818 1829 fm = ui.formatter(b'perf-stats', opts)
1819 1830 for key, title in entries:
1820 1831 values = data[key]
1821 1832 nbvalues = len(data)
1822 1833 values.sort()
1823 1834 stats = {
1824 1835 'key': key,
1825 1836 'title': title,
1826 1837 'nbitems': len(values),
1827 1838 'min': values[0][0],
1828 1839 '10%': values[(nbvalues * 10) // 100][0],
1829 1840 '25%': values[(nbvalues * 25) // 100][0],
1830 1841 '50%': values[(nbvalues * 50) // 100][0],
1831 1842 '75%': values[(nbvalues * 75) // 100][0],
1832 1843 '80%': values[(nbvalues * 80) // 100][0],
1833 1844 '85%': values[(nbvalues * 85) // 100][0],
1834 1845 '90%': values[(nbvalues * 90) // 100][0],
1835 1846 '95%': values[(nbvalues * 95) // 100][0],
1836 1847 '99%': values[(nbvalues * 99) // 100][0],
1837 1848 'max': values[-1][0],
1838 1849 }
1839 1850 fm.startitem()
1840 1851 fm.data(**stats)
1841 1852 # make node pretty for the human output
1842 1853 fm.plain('### %s (%d items)\n' % (title, len(values)))
1843 1854 lines = [
1844 1855 'min',
1845 1856 '10%',
1846 1857 '25%',
1847 1858 '50%',
1848 1859 '75%',
1849 1860 '80%',
1850 1861 '85%',
1851 1862 '90%',
1852 1863 '95%',
1853 1864 '99%',
1854 1865 'max',
1855 1866 ]
1856 1867 for l in lines:
1857 1868 fm.plain('%s: %s\n' % (l, stats[l]))
1858 1869 fm.end()
1859 1870
1860 1871
1861 1872 @command(
1862 1873 b'perfhelper-mergecopies',
1863 1874 formatteropts
1864 1875 + [
1865 1876 (b'r', b'revs', [], b'restrict search to these revisions'),
1866 1877 (b'', b'timing', False, b'provides extra data (costly)'),
1867 1878 (b'', b'stats', False, b'provides statistic about the measured data'),
1868 1879 ],
1869 1880 )
1870 1881 def perfhelpermergecopies(ui, repo, revs=[], **opts):
1871 1882 """find statistics about potential parameters for `perfmergecopies`
1872 1883
1873 1884 This command find (base, p1, p2) triplet relevant for copytracing
1874 1885 benchmarking in the context of a merge. It reports values for some of the
1875 1886 parameters that impact merge copy tracing time during merge.
1876 1887
1877 1888 If `--timing` is set, rename detection is run and the associated timing
1878 1889 will be reported. The extra details come at the cost of slower command
1879 1890 execution.
1880 1891
1881 1892 Since rename detection is only run once, other factors might easily
1882 1893 affect the precision of the timing. However it should give a good
1883 1894 approximation of which revision triplets are very costly.
1884 1895 """
1885 1896 opts = _byteskwargs(opts)
1886 1897 fm = ui.formatter(b'perf', opts)
1887 1898 dotiming = opts[b'timing']
1888 1899 dostats = opts[b'stats']
1889 1900
1890 1901 output_template = [
1891 1902 ("base", "%(base)12s"),
1892 1903 ("p1", "%(p1.node)12s"),
1893 1904 ("p2", "%(p2.node)12s"),
1894 1905 ("p1.nb-revs", "%(p1.nbrevs)12d"),
1895 1906 ("p1.nb-files", "%(p1.nbmissingfiles)12d"),
1896 1907 ("p1.renames", "%(p1.renamedfiles)12d"),
1897 1908 ("p1.time", "%(p1.time)12.3f"),
1898 1909 ("p2.nb-revs", "%(p2.nbrevs)12d"),
1899 1910 ("p2.nb-files", "%(p2.nbmissingfiles)12d"),
1900 1911 ("p2.renames", "%(p2.renamedfiles)12d"),
1901 1912 ("p2.time", "%(p2.time)12.3f"),
1902 1913 ("renames", "%(nbrenamedfiles)12d"),
1903 1914 ("total.time", "%(time)12.3f"),
1904 1915 ]
1905 1916 if not dotiming:
1906 1917 output_template = [
1907 1918 i
1908 1919 for i in output_template
1909 1920 if not ('time' in i[0] or 'renames' in i[0])
1910 1921 ]
1911 1922 header_names = [h for (h, v) in output_template]
1912 1923 output = ' '.join([v for (h, v) in output_template]) + '\n'
1913 1924 header = ' '.join(['%12s'] * len(header_names)) + '\n'
1914 1925 fm.plain(header % tuple(header_names))
1915 1926
1916 1927 if not revs:
1917 1928 revs = ['all()']
1918 1929 revs = scmutil.revrange(repo, revs)
1919 1930
1920 1931 if dostats:
1921 1932 alldata = {
1922 1933 'nbrevs': [],
1923 1934 'nbmissingfiles': [],
1924 1935 }
1925 1936 if dotiming:
1926 1937 alldata['parentnbrenames'] = []
1927 1938 alldata['totalnbrenames'] = []
1928 1939 alldata['parenttime'] = []
1929 1940 alldata['totaltime'] = []
1930 1941
1931 1942 roi = repo.revs('merge() and %ld', revs)
1932 1943 for r in roi:
1933 1944 ctx = repo[r]
1934 1945 p1 = ctx.p1()
1935 1946 p2 = ctx.p2()
1936 1947 bases = repo.changelog._commonancestorsheads(p1.rev(), p2.rev())
1937 1948 for b in bases:
1938 1949 b = repo[b]
1939 1950 p1missing = copies._computeforwardmissing(b, p1)
1940 1951 p2missing = copies._computeforwardmissing(b, p2)
1941 1952 data = {
1942 1953 b'base': b.hex(),
1943 1954 b'p1.node': p1.hex(),
1944 1955 b'p1.nbrevs': len(repo.revs('only(%d, %d)', p1.rev(), b.rev())),
1945 1956 b'p1.nbmissingfiles': len(p1missing),
1946 1957 b'p2.node': p2.hex(),
1947 1958 b'p2.nbrevs': len(repo.revs('only(%d, %d)', p2.rev(), b.rev())),
1948 1959 b'p2.nbmissingfiles': len(p2missing),
1949 1960 }
1950 1961 if dostats:
1951 1962 if p1missing:
1952 1963 alldata['nbrevs'].append(
1953 1964 (data['p1.nbrevs'], b.hex(), p1.hex())
1954 1965 )
1955 1966 alldata['nbmissingfiles'].append(
1956 1967 (data['p1.nbmissingfiles'], b.hex(), p1.hex())
1957 1968 )
1958 1969 if p2missing:
1959 1970 alldata['nbrevs'].append(
1960 1971 (data['p2.nbrevs'], b.hex(), p2.hex())
1961 1972 )
1962 1973 alldata['nbmissingfiles'].append(
1963 1974 (data['p2.nbmissingfiles'], b.hex(), p2.hex())
1964 1975 )
1965 1976 if dotiming:
1966 1977 begin = util.timer()
1967 1978 mergedata = copies.mergecopies(repo, p1, p2, b)
1968 1979 end = util.timer()
1969 1980 # not very stable timing since we did only one run
1970 1981 data['time'] = end - begin
1971 1982 # mergedata contains five dicts: "copy", "movewithdir",
1972 1983 # "diverge", "renamedelete" and "dirmove".
1973 1984 # The first 4 are about renamed file so lets count that.
1974 1985 renames = len(mergedata[0])
1975 1986 renames += len(mergedata[1])
1976 1987 renames += len(mergedata[2])
1977 1988 renames += len(mergedata[3])
1978 1989 data['nbrenamedfiles'] = renames
1979 1990 begin = util.timer()
1980 1991 p1renames = copies.pathcopies(b, p1)
1981 1992 end = util.timer()
1982 1993 data['p1.time'] = end - begin
1983 1994 begin = util.timer()
1984 1995 p2renames = copies.pathcopies(b, p2)
1985 1996 data['p2.time'] = end - begin
1986 1997 end = util.timer()
1987 1998 data['p1.renamedfiles'] = len(p1renames)
1988 1999 data['p2.renamedfiles'] = len(p2renames)
1989 2000
1990 2001 if dostats:
1991 2002 if p1missing:
1992 2003 alldata['parentnbrenames'].append(
1993 2004 (data['p1.renamedfiles'], b.hex(), p1.hex())
1994 2005 )
1995 2006 alldata['parenttime'].append(
1996 2007 (data['p1.time'], b.hex(), p1.hex())
1997 2008 )
1998 2009 if p2missing:
1999 2010 alldata['parentnbrenames'].append(
2000 2011 (data['p2.renamedfiles'], b.hex(), p2.hex())
2001 2012 )
2002 2013 alldata['parenttime'].append(
2003 2014 (data['p2.time'], b.hex(), p2.hex())
2004 2015 )
2005 2016 if p1missing or p2missing:
2006 2017 alldata['totalnbrenames'].append(
2007 2018 (
2008 2019 data['nbrenamedfiles'],
2009 2020 b.hex(),
2010 2021 p1.hex(),
2011 2022 p2.hex(),
2012 2023 )
2013 2024 )
2014 2025 alldata['totaltime'].append(
2015 2026 (data['time'], b.hex(), p1.hex(), p2.hex())
2016 2027 )
2017 2028 fm.startitem()
2018 2029 fm.data(**data)
2019 2030 # make node pretty for the human output
2020 2031 out = data.copy()
2021 2032 out['base'] = fm.hexfunc(b.node())
2022 2033 out['p1.node'] = fm.hexfunc(p1.node())
2023 2034 out['p2.node'] = fm.hexfunc(p2.node())
2024 2035 fm.plain(output % out)
2025 2036
2026 2037 fm.end()
2027 2038 if dostats:
2028 2039 # use a second formatter because the data are quite different, not sure
2029 2040 # how it flies with the templater.
2030 2041 entries = [
2031 2042 ('nbrevs', 'number of revision covered'),
2032 2043 ('nbmissingfiles', 'number of missing files at head'),
2033 2044 ]
2034 2045 if dotiming:
2035 2046 entries.append(
2036 2047 ('parentnbrenames', 'rename from one parent to base')
2037 2048 )
2038 2049 entries.append(('totalnbrenames', 'total number of renames'))
2039 2050 entries.append(('parenttime', 'time for one parent'))
2040 2051 entries.append(('totaltime', 'time for both parents'))
2041 2052 _displaystats(ui, opts, entries, alldata)
2042 2053
2043 2054
2044 2055 @command(
2045 2056 b'perfhelper-pathcopies',
2046 2057 formatteropts
2047 2058 + [
2048 2059 (b'r', b'revs', [], b'restrict search to these revisions'),
2049 2060 (b'', b'timing', False, b'provides extra data (costly)'),
2050 2061 (b'', b'stats', False, b'provides statistic about the measured data'),
2051 2062 ],
2052 2063 )
2053 2064 def perfhelperpathcopies(ui, repo, revs=[], **opts):
2054 2065 """find statistic about potential parameters for the `perftracecopies`
2055 2066
2056 2067 This command find source-destination pair relevant for copytracing testing.
2057 2068 It report value for some of the parameters that impact copy tracing time.
2058 2069
2059 2070 If `--timing` is set, rename detection is run and the associated timing
2060 2071 will be reported. The extra details comes at the cost of a slower command
2061 2072 execution.
2062 2073
2063 2074 Since the rename detection is only run once, other factors might easily
2064 2075 affect the precision of the timing. However it should give a good
2065 2076 approximation of which revision pairs are very costly.
2066 2077 """
2067 2078 opts = _byteskwargs(opts)
2068 2079 fm = ui.formatter(b'perf', opts)
2069 2080 dotiming = opts[b'timing']
2070 2081 dostats = opts[b'stats']
2071 2082
2072 2083 if dotiming:
2073 2084 header = '%12s %12s %12s %12s %12s %12s\n'
2074 2085 output = (
2075 2086 "%(source)12s %(destination)12s "
2076 2087 "%(nbrevs)12d %(nbmissingfiles)12d "
2077 2088 "%(nbrenamedfiles)12d %(time)18.5f\n"
2078 2089 )
2079 2090 header_names = (
2080 2091 "source",
2081 2092 "destination",
2082 2093 "nb-revs",
2083 2094 "nb-files",
2084 2095 "nb-renames",
2085 2096 "time",
2086 2097 )
2087 2098 fm.plain(header % header_names)
2088 2099 else:
2089 2100 header = '%12s %12s %12s %12s\n'
2090 2101 output = (
2091 2102 "%(source)12s %(destination)12s "
2092 2103 "%(nbrevs)12d %(nbmissingfiles)12d\n"
2093 2104 )
2094 2105 fm.plain(header % ("source", "destination", "nb-revs", "nb-files"))
2095 2106
2096 2107 if not revs:
2097 2108 revs = ['all()']
2098 2109 revs = scmutil.revrange(repo, revs)
2099 2110
2100 2111 if dostats:
2101 2112 alldata = {
2102 2113 'nbrevs': [],
2103 2114 'nbmissingfiles': [],
2104 2115 }
2105 2116 if dotiming:
2106 2117 alldata['nbrenames'] = []
2107 2118 alldata['time'] = []
2108 2119
2109 2120 roi = repo.revs('merge() and %ld', revs)
2110 2121 for r in roi:
2111 2122 ctx = repo[r]
2112 2123 p1 = ctx.p1().rev()
2113 2124 p2 = ctx.p2().rev()
2114 2125 bases = repo.changelog._commonancestorsheads(p1, p2)
2115 2126 for p in (p1, p2):
2116 2127 for b in bases:
2117 2128 base = repo[b]
2118 2129 parent = repo[p]
2119 2130 missing = copies._computeforwardmissing(base, parent)
2120 2131 if not missing:
2121 2132 continue
2122 2133 data = {
2123 2134 b'source': base.hex(),
2124 2135 b'destination': parent.hex(),
2125 2136 b'nbrevs': len(repo.revs('only(%d, %d)', p, b)),
2126 2137 b'nbmissingfiles': len(missing),
2127 2138 }
2128 2139 if dostats:
2129 2140 alldata['nbrevs'].append(
2130 2141 (data['nbrevs'], base.hex(), parent.hex(),)
2131 2142 )
2132 2143 alldata['nbmissingfiles'].append(
2133 2144 (data['nbmissingfiles'], base.hex(), parent.hex(),)
2134 2145 )
2135 2146 if dotiming:
2136 2147 begin = util.timer()
2137 2148 renames = copies.pathcopies(base, parent)
2138 2149 end = util.timer()
2139 2150 # not very stable timing since we did only one run
2140 2151 data['time'] = end - begin
2141 2152 data['nbrenamedfiles'] = len(renames)
2142 2153 if dostats:
2143 2154 alldata['time'].append(
2144 2155 (data['time'], base.hex(), parent.hex(),)
2145 2156 )
2146 2157 alldata['nbrenames'].append(
2147 2158 (data['nbrenamedfiles'], base.hex(), parent.hex(),)
2148 2159 )
2149 2160 fm.startitem()
2150 2161 fm.data(**data)
2151 2162 out = data.copy()
2152 2163 out['source'] = fm.hexfunc(base.node())
2153 2164 out['destination'] = fm.hexfunc(parent.node())
2154 2165 fm.plain(output % out)
2155 2166
2156 2167 fm.end()
2157 2168 if dostats:
2158 2169 # use a second formatter because the data are quite different, not sure
2159 2170 # how it flies with the templater.
2160 2171 fm = ui.formatter(b'perf', opts)
2161 2172 entries = [
2162 2173 ('nbrevs', 'number of revision covered'),
2163 2174 ('nbmissingfiles', 'number of missing files at head'),
2164 2175 ]
2165 2176 if dotiming:
2166 2177 entries.append(('nbrenames', 'renamed files'))
2167 2178 entries.append(('time', 'time'))
2168 2179 _displaystats(ui, opts, entries, alldata)
2169 2180
2170 2181
2171 2182 @command(b'perfcca', formatteropts)
2172 2183 def perfcca(ui, repo, **opts):
2173 2184 opts = _byteskwargs(opts)
2174 2185 timer, fm = gettimer(ui, opts)
2175 2186 timer(lambda: scmutil.casecollisionauditor(ui, False, repo.dirstate))
2176 2187 fm.end()
2177 2188
2178 2189
2179 2190 @command(b'perffncacheload', formatteropts)
2180 2191 def perffncacheload(ui, repo, **opts):
2181 2192 opts = _byteskwargs(opts)
2182 2193 timer, fm = gettimer(ui, opts)
2183 2194 s = repo.store
2184 2195
2185 2196 def d():
2186 2197 s.fncache._load()
2187 2198
2188 2199 timer(d)
2189 2200 fm.end()
2190 2201
2191 2202
2192 2203 @command(b'perffncachewrite', formatteropts)
2193 2204 def perffncachewrite(ui, repo, **opts):
2194 2205 opts = _byteskwargs(opts)
2195 2206 timer, fm = gettimer(ui, opts)
2196 2207 s = repo.store
2197 2208 lock = repo.lock()
2198 2209 s.fncache._load()
2199 2210 tr = repo.transaction(b'perffncachewrite')
2200 2211 tr.addbackup(b'fncache')
2201 2212
2202 2213 def d():
2203 2214 s.fncache._dirty = True
2204 2215 s.fncache.write(tr)
2205 2216
2206 2217 timer(d)
2207 2218 tr.close()
2208 2219 lock.release()
2209 2220 fm.end()
2210 2221
2211 2222
2212 2223 @command(b'perffncacheencode', formatteropts)
2213 2224 def perffncacheencode(ui, repo, **opts):
2214 2225 opts = _byteskwargs(opts)
2215 2226 timer, fm = gettimer(ui, opts)
2216 2227 s = repo.store
2217 2228 s.fncache._load()
2218 2229
2219 2230 def d():
2220 2231 for p in s.fncache.entries:
2221 2232 s.encode(p)
2222 2233
2223 2234 timer(d)
2224 2235 fm.end()
2225 2236
2226 2237
2227 2238 def _bdiffworker(q, blocks, xdiff, ready, done):
2228 2239 while not done.is_set():
2229 2240 pair = q.get()
2230 2241 while pair is not None:
2231 2242 if xdiff:
2232 2243 mdiff.bdiff.xdiffblocks(*pair)
2233 2244 elif blocks:
2234 2245 mdiff.bdiff.blocks(*pair)
2235 2246 else:
2236 2247 mdiff.textdiff(*pair)
2237 2248 q.task_done()
2238 2249 pair = q.get()
2239 2250 q.task_done() # for the None one
2240 2251 with ready:
2241 2252 ready.wait()
2242 2253
2243 2254
2244 2255 def _manifestrevision(repo, mnode):
2245 2256 ml = repo.manifestlog
2246 2257
2247 2258 if util.safehasattr(ml, b'getstorage'):
2248 2259 store = ml.getstorage(b'')
2249 2260 else:
2250 2261 store = ml._revlog
2251 2262
2252 2263 return store.revision(mnode)
2253 2264
2254 2265
2255 2266 @command(
2256 2267 b'perfbdiff',
2257 2268 revlogopts
2258 2269 + formatteropts
2259 2270 + [
2260 2271 (
2261 2272 b'',
2262 2273 b'count',
2263 2274 1,
2264 2275 b'number of revisions to test (when using --startrev)',
2265 2276 ),
2266 2277 (b'', b'alldata', False, b'test bdiffs for all associated revisions'),
2267 2278 (b'', b'threads', 0, b'number of thread to use (disable with 0)'),
2268 2279 (b'', b'blocks', False, b'test computing diffs into blocks'),
2269 2280 (b'', b'xdiff', False, b'use xdiff algorithm'),
2270 2281 ],
2271 2282 b'-c|-m|FILE REV',
2272 2283 )
2273 2284 def perfbdiff(ui, repo, file_, rev=None, count=None, threads=0, **opts):
2274 2285 """benchmark a bdiff between revisions
2275 2286
2276 2287 By default, benchmark a bdiff between its delta parent and itself.
2277 2288
2278 2289 With ``--count``, benchmark bdiffs between delta parents and self for N
2279 2290 revisions starting at the specified revision.
2280 2291
2281 2292 With ``--alldata``, assume the requested revision is a changeset and
2282 2293 measure bdiffs for all changes related to that changeset (manifest
2283 2294 and filelogs).
2284 2295 """
2285 2296 opts = _byteskwargs(opts)
2286 2297
2287 2298 if opts[b'xdiff'] and not opts[b'blocks']:
2288 2299 raise error.CommandError(b'perfbdiff', b'--xdiff requires --blocks')
2289 2300
2290 2301 if opts[b'alldata']:
2291 2302 opts[b'changelog'] = True
2292 2303
2293 2304 if opts.get(b'changelog') or opts.get(b'manifest'):
2294 2305 file_, rev = None, file_
2295 2306 elif rev is None:
2296 2307 raise error.CommandError(b'perfbdiff', b'invalid arguments')
2297 2308
2298 2309 blocks = opts[b'blocks']
2299 2310 xdiff = opts[b'xdiff']
2300 2311 textpairs = []
2301 2312
2302 2313 r = cmdutil.openrevlog(repo, b'perfbdiff', file_, opts)
2303 2314
2304 2315 startrev = r.rev(r.lookup(rev))
2305 2316 for rev in range(startrev, min(startrev + count, len(r) - 1)):
2306 2317 if opts[b'alldata']:
2307 2318 # Load revisions associated with changeset.
2308 2319 ctx = repo[rev]
2309 2320 mtext = _manifestrevision(repo, ctx.manifestnode())
2310 2321 for pctx in ctx.parents():
2311 2322 pman = _manifestrevision(repo, pctx.manifestnode())
2312 2323 textpairs.append((pman, mtext))
2313 2324
2314 2325 # Load filelog revisions by iterating manifest delta.
2315 2326 man = ctx.manifest()
2316 2327 pman = ctx.p1().manifest()
2317 2328 for filename, change in pman.diff(man).items():
2318 2329 fctx = repo.file(filename)
2319 2330 f1 = fctx.revision(change[0][0] or -1)
2320 2331 f2 = fctx.revision(change[1][0] or -1)
2321 2332 textpairs.append((f1, f2))
2322 2333 else:
2323 2334 dp = r.deltaparent(rev)
2324 2335 textpairs.append((r.revision(dp), r.revision(rev)))
2325 2336
2326 2337 withthreads = threads > 0
2327 2338 if not withthreads:
2328 2339
2329 2340 def d():
2330 2341 for pair in textpairs:
2331 2342 if xdiff:
2332 2343 mdiff.bdiff.xdiffblocks(*pair)
2333 2344 elif blocks:
2334 2345 mdiff.bdiff.blocks(*pair)
2335 2346 else:
2336 2347 mdiff.textdiff(*pair)
2337 2348
2338 2349 else:
2339 2350 q = queue()
2340 2351 for i in _xrange(threads):
2341 2352 q.put(None)
2342 2353 ready = threading.Condition()
2343 2354 done = threading.Event()
2344 2355 for i in _xrange(threads):
2345 2356 threading.Thread(
2346 2357 target=_bdiffworker, args=(q, blocks, xdiff, ready, done)
2347 2358 ).start()
2348 2359 q.join()
2349 2360
2350 2361 def d():
2351 2362 for pair in textpairs:
2352 2363 q.put(pair)
2353 2364 for i in _xrange(threads):
2354 2365 q.put(None)
2355 2366 with ready:
2356 2367 ready.notify_all()
2357 2368 q.join()
2358 2369
2359 2370 timer, fm = gettimer(ui, opts)
2360 2371 timer(d)
2361 2372 fm.end()
2362 2373
2363 2374 if withthreads:
2364 2375 done.set()
2365 2376 for i in _xrange(threads):
2366 2377 q.put(None)
2367 2378 with ready:
2368 2379 ready.notify_all()
2369 2380
2370 2381
2371 2382 @command(
2372 2383 b'perfunidiff',
2373 2384 revlogopts
2374 2385 + formatteropts
2375 2386 + [
2376 2387 (
2377 2388 b'',
2378 2389 b'count',
2379 2390 1,
2380 2391 b'number of revisions to test (when using --startrev)',
2381 2392 ),
2382 2393 (b'', b'alldata', False, b'test unidiffs for all associated revisions'),
2383 2394 ],
2384 2395 b'-c|-m|FILE REV',
2385 2396 )
2386 2397 def perfunidiff(ui, repo, file_, rev=None, count=None, **opts):
2387 2398 """benchmark a unified diff between revisions
2388 2399
2389 2400 This doesn't include any copy tracing - it's just a unified diff
2390 2401 of the texts.
2391 2402
2392 2403 By default, benchmark a diff between its delta parent and itself.
2393 2404
2394 2405 With ``--count``, benchmark diffs between delta parents and self for N
2395 2406 revisions starting at the specified revision.
2396 2407
2397 2408 With ``--alldata``, assume the requested revision is a changeset and
2398 2409 measure diffs for all changes related to that changeset (manifest
2399 2410 and filelogs).
2400 2411 """
2401 2412 opts = _byteskwargs(opts)
2402 2413 if opts[b'alldata']:
2403 2414 opts[b'changelog'] = True
2404 2415
2405 2416 if opts.get(b'changelog') or opts.get(b'manifest'):
2406 2417 file_, rev = None, file_
2407 2418 elif rev is None:
2408 2419 raise error.CommandError(b'perfunidiff', b'invalid arguments')
2409 2420
2410 2421 textpairs = []
2411 2422
2412 2423 r = cmdutil.openrevlog(repo, b'perfunidiff', file_, opts)
2413 2424
2414 2425 startrev = r.rev(r.lookup(rev))
2415 2426 for rev in range(startrev, min(startrev + count, len(r) - 1)):
2416 2427 if opts[b'alldata']:
2417 2428 # Load revisions associated with changeset.
2418 2429 ctx = repo[rev]
2419 2430 mtext = _manifestrevision(repo, ctx.manifestnode())
2420 2431 for pctx in ctx.parents():
2421 2432 pman = _manifestrevision(repo, pctx.manifestnode())
2422 2433 textpairs.append((pman, mtext))
2423 2434
2424 2435 # Load filelog revisions by iterating manifest delta.
2425 2436 man = ctx.manifest()
2426 2437 pman = ctx.p1().manifest()
2427 2438 for filename, change in pman.diff(man).items():
2428 2439 fctx = repo.file(filename)
2429 2440 f1 = fctx.revision(change[0][0] or -1)
2430 2441 f2 = fctx.revision(change[1][0] or -1)
2431 2442 textpairs.append((f1, f2))
2432 2443 else:
2433 2444 dp = r.deltaparent(rev)
2434 2445 textpairs.append((r.revision(dp), r.revision(rev)))
2435 2446
2436 2447 def d():
2437 2448 for left, right in textpairs:
2438 2449 # The date strings don't matter, so we pass empty strings.
2439 2450 headerlines, hunks = mdiff.unidiff(
2440 2451 left, b'', right, b'', b'left', b'right', binary=False
2441 2452 )
2442 2453 # consume iterators in roughly the way patch.py does
2443 2454 b'\n'.join(headerlines)
2444 2455 b''.join(sum((list(hlines) for hrange, hlines in hunks), []))
2445 2456
2446 2457 timer, fm = gettimer(ui, opts)
2447 2458 timer(d)
2448 2459 fm.end()
2449 2460
2450 2461
2451 2462 @command(b'perfdiffwd', formatteropts)
2452 2463 def perfdiffwd(ui, repo, **opts):
2453 2464 """Profile diff of working directory changes"""
2454 2465 opts = _byteskwargs(opts)
2455 2466 timer, fm = gettimer(ui, opts)
2456 2467 options = {
2457 2468 'w': 'ignore_all_space',
2458 2469 'b': 'ignore_space_change',
2459 2470 'B': 'ignore_blank_lines',
2460 2471 }
2461 2472
2462 2473 for diffopt in ('', 'w', 'b', 'B', 'wB'):
2463 2474 opts = dict((options[c], b'1') for c in diffopt)
2464 2475
2465 2476 def d():
2466 2477 ui.pushbuffer()
2467 2478 commands.diff(ui, repo, **opts)
2468 2479 ui.popbuffer()
2469 2480
2470 2481 diffopt = diffopt.encode('ascii')
2471 2482 title = b'diffopts: %s' % (diffopt and (b'-' + diffopt) or b'none')
2472 2483 timer(d, title=title)
2473 2484 fm.end()
2474 2485
2475 2486
2476 2487 @command(b'perfrevlogindex', revlogopts + formatteropts, b'-c|-m|FILE')
2477 2488 def perfrevlogindex(ui, repo, file_=None, **opts):
2478 2489 """Benchmark operations against a revlog index.
2479 2490
2480 2491 This tests constructing a revlog instance, reading index data,
2481 2492 parsing index data, and performing various operations related to
2482 2493 index data.
2483 2494 """
2484 2495
2485 2496 opts = _byteskwargs(opts)
2486 2497
2487 2498 rl = cmdutil.openrevlog(repo, b'perfrevlogindex', file_, opts)
2488 2499
2489 2500 opener = getattr(rl, 'opener') # trick linter
2490 2501 indexfile = rl.indexfile
2491 2502 data = opener.read(indexfile)
2492 2503
2493 2504 header = struct.unpack(b'>I', data[0:4])[0]
2494 2505 version = header & 0xFFFF
2495 2506 if version == 1:
2496 2507 revlogio = revlog.revlogio()
2497 2508 inline = header & (1 << 16)
2498 2509 else:
2499 2510 raise error.Abort(b'unsupported revlog version: %d' % version)
2500 2511
2501 2512 rllen = len(rl)
2502 2513
2503 2514 node0 = rl.node(0)
2504 2515 node25 = rl.node(rllen // 4)
2505 2516 node50 = rl.node(rllen // 2)
2506 2517 node75 = rl.node(rllen // 4 * 3)
2507 2518 node100 = rl.node(rllen - 1)
2508 2519
2509 2520 allrevs = range(rllen)
2510 2521 allrevsrev = list(reversed(allrevs))
2511 2522 allnodes = [rl.node(rev) for rev in range(rllen)]
2512 2523 allnodesrev = list(reversed(allnodes))
2513 2524
2514 2525 def constructor():
2515 2526 revlog.revlog(opener, indexfile)
2516 2527
2517 2528 def read():
2518 2529 with opener(indexfile) as fh:
2519 2530 fh.read()
2520 2531
2521 2532 def parseindex():
2522 2533 revlogio.parseindex(data, inline)
2523 2534
2524 2535 def getentry(revornode):
2525 2536 index = revlogio.parseindex(data, inline)[0]
2526 2537 index[revornode]
2527 2538
2528 2539 def getentries(revs, count=1):
2529 2540 index = revlogio.parseindex(data, inline)[0]
2530 2541
2531 2542 for i in range(count):
2532 2543 for rev in revs:
2533 2544 index[rev]
2534 2545
2535 2546 def resolvenode(node):
2536 2547 nodemap = revlogio.parseindex(data, inline)[1]
2537 2548 # This only works for the C code.
2538 2549 if nodemap is None:
2539 2550 return
2540 2551
2541 2552 try:
2542 2553 nodemap[node]
2543 2554 except error.RevlogError:
2544 2555 pass
2545 2556
2546 2557 def resolvenodes(nodes, count=1):
2547 2558 nodemap = revlogio.parseindex(data, inline)[1]
2548 2559 if nodemap is None:
2549 2560 return
2550 2561
2551 2562 for i in range(count):
2552 2563 for node in nodes:
2553 2564 try:
2554 2565 nodemap[node]
2555 2566 except error.RevlogError:
2556 2567 pass
2557 2568
2558 2569 benches = [
2559 2570 (constructor, b'revlog constructor'),
2560 2571 (read, b'read'),
2561 2572 (parseindex, b'create index object'),
2562 2573 (lambda: getentry(0), b'retrieve index entry for rev 0'),
2563 2574 (lambda: resolvenode(b'a' * 20), b'look up missing node'),
2564 2575 (lambda: resolvenode(node0), b'look up node at rev 0'),
2565 2576 (lambda: resolvenode(node25), b'look up node at 1/4 len'),
2566 2577 (lambda: resolvenode(node50), b'look up node at 1/2 len'),
2567 2578 (lambda: resolvenode(node75), b'look up node at 3/4 len'),
2568 2579 (lambda: resolvenode(node100), b'look up node at tip'),
2569 2580 # 2x variation is to measure caching impact.
2570 2581 (lambda: resolvenodes(allnodes), b'look up all nodes (forward)'),
2571 2582 (lambda: resolvenodes(allnodes, 2), b'look up all nodes 2x (forward)'),
2572 2583 (lambda: resolvenodes(allnodesrev), b'look up all nodes (reverse)'),
2573 2584 (
2574 2585 lambda: resolvenodes(allnodesrev, 2),
2575 2586 b'look up all nodes 2x (reverse)',
2576 2587 ),
2577 2588 (lambda: getentries(allrevs), b'retrieve all index entries (forward)'),
2578 2589 (
2579 2590 lambda: getentries(allrevs, 2),
2580 2591 b'retrieve all index entries 2x (forward)',
2581 2592 ),
2582 2593 (
2583 2594 lambda: getentries(allrevsrev),
2584 2595 b'retrieve all index entries (reverse)',
2585 2596 ),
2586 2597 (
2587 2598 lambda: getentries(allrevsrev, 2),
2588 2599 b'retrieve all index entries 2x (reverse)',
2589 2600 ),
2590 2601 ]
2591 2602
2592 2603 for fn, title in benches:
2593 2604 timer, fm = gettimer(ui, opts)
2594 2605 timer(fn, title=title)
2595 2606 fm.end()
2596 2607
2597 2608
2598 2609 @command(
2599 2610 b'perfrevlogrevisions',
2600 2611 revlogopts
2601 2612 + formatteropts
2602 2613 + [
2603 2614 (b'd', b'dist', 100, b'distance between the revisions'),
2604 2615 (b's', b'startrev', 0, b'revision to start reading at'),
2605 2616 (b'', b'reverse', False, b'read in reverse'),
2606 2617 ],
2607 2618 b'-c|-m|FILE',
2608 2619 )
2609 2620 def perfrevlogrevisions(
2610 2621 ui, repo, file_=None, startrev=0, reverse=False, **opts
2611 2622 ):
2612 2623 """Benchmark reading a series of revisions from a revlog.
2613 2624
2614 2625 By default, we read every ``-d/--dist`` revision from 0 to tip of
2615 2626 the specified revlog.
2616 2627
2617 2628 The start revision can be defined via ``-s/--startrev``.
2618 2629 """
2619 2630 opts = _byteskwargs(opts)
2620 2631
2621 2632 rl = cmdutil.openrevlog(repo, b'perfrevlogrevisions', file_, opts)
2622 2633 rllen = getlen(ui)(rl)
2623 2634
2624 2635 if startrev < 0:
2625 2636 startrev = rllen + startrev
2626 2637
2627 2638 def d():
2628 2639 rl.clearcaches()
2629 2640
2630 2641 beginrev = startrev
2631 2642 endrev = rllen
2632 2643 dist = opts[b'dist']
2633 2644
2634 2645 if reverse:
2635 2646 beginrev, endrev = endrev - 1, beginrev - 1
2636 2647 dist = -1 * dist
2637 2648
2638 2649 for x in _xrange(beginrev, endrev, dist):
2639 2650 # Old revisions don't support passing int.
2640 2651 n = rl.node(x)
2641 2652 rl.revision(n)
2642 2653
2643 2654 timer, fm = gettimer(ui, opts)
2644 2655 timer(d)
2645 2656 fm.end()
2646 2657
2647 2658
2648 2659 @command(
2649 2660 b'perfrevlogwrite',
2650 2661 revlogopts
2651 2662 + formatteropts
2652 2663 + [
2653 2664 (b's', b'startrev', 1000, b'revision to start writing at'),
2654 2665 (b'', b'stoprev', -1, b'last revision to write'),
2655 2666 (b'', b'count', 3, b'number of passes to perform'),
2656 2667 (b'', b'details', False, b'print timing for every revisions tested'),
2657 2668 (b'', b'source', b'full', b'the kind of data feed in the revlog'),
2658 2669 (b'', b'lazydeltabase', True, b'try the provided delta first'),
2659 2670 (b'', b'clear-caches', True, b'clear revlog cache between calls'),
2660 2671 ],
2661 2672 b'-c|-m|FILE',
2662 2673 )
2663 2674 def perfrevlogwrite(ui, repo, file_=None, startrev=1000, stoprev=-1, **opts):
2664 2675 """Benchmark writing a series of revisions to a revlog.
2665 2676
2666 2677 Possible source values are:
2667 2678 * `full`: add from a full text (default).
2668 2679 * `parent-1`: add from a delta to the first parent
2669 2680 * `parent-2`: add from a delta to the second parent if it exists
2670 2681 (use a delta from the first parent otherwise)
2671 2682 * `parent-smallest`: add from the smallest delta (either p1 or p2)
2672 2683 * `storage`: add from the existing precomputed deltas
2673 2684
2674 2685 Note: This performance command measures performance in a custom way. As a
2675 2686 result some of the global configuration of the 'perf' command does not
2676 2687 apply to it:
2677 2688
2678 2689 * ``pre-run``: disabled
2679 2690
2680 2691 * ``profile-benchmark``: disabled
2681 2692
2682 2693 * ``run-limits``: disabled use --count instead
2683 2694 """
2684 2695 opts = _byteskwargs(opts)
2685 2696
2686 2697 rl = cmdutil.openrevlog(repo, b'perfrevlogwrite', file_, opts)
2687 2698 rllen = getlen(ui)(rl)
2688 2699 if startrev < 0:
2689 2700 startrev = rllen + startrev
2690 2701 if stoprev < 0:
2691 2702 stoprev = rllen + stoprev
2692 2703
2693 2704 lazydeltabase = opts['lazydeltabase']
2694 2705 source = opts['source']
2695 2706 clearcaches = opts['clear_caches']
2696 2707 validsource = (
2697 2708 b'full',
2698 2709 b'parent-1',
2699 2710 b'parent-2',
2700 2711 b'parent-smallest',
2701 2712 b'storage',
2702 2713 )
2703 2714 if source not in validsource:
2704 2715 raise error.Abort('invalid source type: %s' % source)
2705 2716
2706 2717 ### actually gather results
2707 2718 count = opts['count']
2708 2719 if count <= 0:
2709 2720 raise error.Abort('invalide run count: %d' % count)
2710 2721 allresults = []
2711 2722 for c in range(count):
2712 2723 timing = _timeonewrite(
2713 2724 ui,
2714 2725 rl,
2715 2726 source,
2716 2727 startrev,
2717 2728 stoprev,
2718 2729 c + 1,
2719 2730 lazydeltabase=lazydeltabase,
2720 2731 clearcaches=clearcaches,
2721 2732 )
2722 2733 allresults.append(timing)
2723 2734
2724 2735 ### consolidate the results in a single list
2725 2736 results = []
2726 2737 for idx, (rev, t) in enumerate(allresults[0]):
2727 2738 ts = [t]
2728 2739 for other in allresults[1:]:
2729 2740 orev, ot = other[idx]
2730 2741 assert orev == rev
2731 2742 ts.append(ot)
2732 2743 results.append((rev, ts))
2733 2744 resultcount = len(results)
2734 2745
2735 2746 ### Compute and display relevant statistics
2736 2747
2737 2748 # get a formatter
2738 2749 fm = ui.formatter(b'perf', opts)
2739 2750 displayall = ui.configbool(b"perf", b"all-timing", False)
2740 2751
2741 2752 # print individual details if requested
2742 2753 if opts['details']:
2743 2754 for idx, item in enumerate(results, 1):
2744 2755 rev, data = item
2745 2756 title = 'revisions #%d of %d, rev %d' % (idx, resultcount, rev)
2746 2757 formatone(fm, data, title=title, displayall=displayall)
2747 2758
2748 2759 # sorts results by median time
2749 2760 results.sort(key=lambda x: sorted(x[1])[len(x[1]) // 2])
2750 2761 # list of (name, index) to display)
2751 2762 relevants = [
2752 2763 ("min", 0),
2753 2764 ("10%", resultcount * 10 // 100),
2754 2765 ("25%", resultcount * 25 // 100),
2755 2766 ("50%", resultcount * 70 // 100),
2756 2767 ("75%", resultcount * 75 // 100),
2757 2768 ("90%", resultcount * 90 // 100),
2758 2769 ("95%", resultcount * 95 // 100),
2759 2770 ("99%", resultcount * 99 // 100),
2760 2771 ("99.9%", resultcount * 999 // 1000),
2761 2772 ("99.99%", resultcount * 9999 // 10000),
2762 2773 ("99.999%", resultcount * 99999 // 100000),
2763 2774 ("max", -1),
2764 2775 ]
2765 2776 if not ui.quiet:
2766 2777 for name, idx in relevants:
2767 2778 data = results[idx]
2768 2779 title = '%s of %d, rev %d' % (name, resultcount, data[0])
2769 2780 formatone(fm, data[1], title=title, displayall=displayall)
2770 2781
2771 2782 # XXX summing that many float will not be very precise, we ignore this fact
2772 2783 # for now
2773 2784 totaltime = []
2774 2785 for item in allresults:
2775 2786 totaltime.append(
2776 2787 (
2777 2788 sum(x[1][0] for x in item),
2778 2789 sum(x[1][1] for x in item),
2779 2790 sum(x[1][2] for x in item),
2780 2791 )
2781 2792 )
2782 2793 formatone(
2783 2794 fm,
2784 2795 totaltime,
2785 2796 title="total time (%d revs)" % resultcount,
2786 2797 displayall=displayall,
2787 2798 )
2788 2799 fm.end()
2789 2800
2790 2801
2791 2802 class _faketr(object):
2792 2803 def add(s, x, y, z=None):
2793 2804 return None
2794 2805
2795 2806
2796 2807 def _timeonewrite(
2797 2808 ui,
2798 2809 orig,
2799 2810 source,
2800 2811 startrev,
2801 2812 stoprev,
2802 2813 runidx=None,
2803 2814 lazydeltabase=True,
2804 2815 clearcaches=True,
2805 2816 ):
2806 2817 timings = []
2807 2818 tr = _faketr()
2808 2819 with _temprevlog(ui, orig, startrev) as dest:
2809 2820 dest._lazydeltabase = lazydeltabase
2810 2821 revs = list(orig.revs(startrev, stoprev))
2811 2822 total = len(revs)
2812 2823 topic = 'adding'
2813 2824 if runidx is not None:
2814 2825 topic += ' (run #%d)' % runidx
2815 2826 # Support both old and new progress API
2816 2827 if util.safehasattr(ui, 'makeprogress'):
2817 2828 progress = ui.makeprogress(topic, unit='revs', total=total)
2818 2829
2819 2830 def updateprogress(pos):
2820 2831 progress.update(pos)
2821 2832
2822 2833 def completeprogress():
2823 2834 progress.complete()
2824 2835
2825 2836 else:
2826 2837
2827 2838 def updateprogress(pos):
2828 2839 ui.progress(topic, pos, unit='revs', total=total)
2829 2840
2830 2841 def completeprogress():
2831 2842 ui.progress(topic, None, unit='revs', total=total)
2832 2843
2833 2844 for idx, rev in enumerate(revs):
2834 2845 updateprogress(idx)
2835 2846 addargs, addkwargs = _getrevisionseed(orig, rev, tr, source)
2836 2847 if clearcaches:
2837 2848 dest.index.clearcaches()
2838 2849 dest.clearcaches()
2839 2850 with timeone() as r:
2840 2851 dest.addrawrevision(*addargs, **addkwargs)
2841 2852 timings.append((rev, r[0]))
2842 2853 updateprogress(total)
2843 2854 completeprogress()
2844 2855 return timings
2845 2856
2846 2857
2847 2858 def _getrevisionseed(orig, rev, tr, source):
2848 2859 from mercurial.node import nullid
2849 2860
2850 2861 linkrev = orig.linkrev(rev)
2851 2862 node = orig.node(rev)
2852 2863 p1, p2 = orig.parents(node)
2853 2864 flags = orig.flags(rev)
2854 2865 cachedelta = None
2855 2866 text = None
2856 2867
2857 2868 if source == b'full':
2858 2869 text = orig.revision(rev)
2859 2870 elif source == b'parent-1':
2860 2871 baserev = orig.rev(p1)
2861 2872 cachedelta = (baserev, orig.revdiff(p1, rev))
2862 2873 elif source == b'parent-2':
2863 2874 parent = p2
2864 2875 if p2 == nullid:
2865 2876 parent = p1
2866 2877 baserev = orig.rev(parent)
2867 2878 cachedelta = (baserev, orig.revdiff(parent, rev))
2868 2879 elif source == b'parent-smallest':
2869 2880 p1diff = orig.revdiff(p1, rev)
2870 2881 parent = p1
2871 2882 diff = p1diff
2872 2883 if p2 != nullid:
2873 2884 p2diff = orig.revdiff(p2, rev)
2874 2885 if len(p1diff) > len(p2diff):
2875 2886 parent = p2
2876 2887 diff = p2diff
2877 2888 baserev = orig.rev(parent)
2878 2889 cachedelta = (baserev, diff)
2879 2890 elif source == b'storage':
2880 2891 baserev = orig.deltaparent(rev)
2881 2892 cachedelta = (baserev, orig.revdiff(orig.node(baserev), rev))
2882 2893
2883 2894 return (
2884 2895 (text, tr, linkrev, p1, p2),
2885 2896 {'node': node, 'flags': flags, 'cachedelta': cachedelta},
2886 2897 )
2887 2898
2888 2899
2889 2900 @contextlib.contextmanager
2890 2901 def _temprevlog(ui, orig, truncaterev):
2891 2902 from mercurial import vfs as vfsmod
2892 2903
2893 2904 if orig._inline:
2894 2905 raise error.Abort('not supporting inline revlog (yet)')
2895 2906 revlogkwargs = {}
2896 2907 k = 'upperboundcomp'
2897 2908 if util.safehasattr(orig, k):
2898 2909 revlogkwargs[k] = getattr(orig, k)
2899 2910
2900 2911 origindexpath = orig.opener.join(orig.indexfile)
2901 2912 origdatapath = orig.opener.join(orig.datafile)
2902 2913 indexname = 'revlog.i'
2903 2914 dataname = 'revlog.d'
2904 2915
2905 2916 tmpdir = tempfile.mkdtemp(prefix='tmp-hgperf-')
2906 2917 try:
2907 2918 # copy the data file in a temporary directory
2908 2919 ui.debug('copying data in %s\n' % tmpdir)
2909 2920 destindexpath = os.path.join(tmpdir, 'revlog.i')
2910 2921 destdatapath = os.path.join(tmpdir, 'revlog.d')
2911 2922 shutil.copyfile(origindexpath, destindexpath)
2912 2923 shutil.copyfile(origdatapath, destdatapath)
2913 2924
2914 2925 # remove the data we want to add again
2915 2926 ui.debug('truncating data to be rewritten\n')
2916 2927 with open(destindexpath, 'ab') as index:
2917 2928 index.seek(0)
2918 2929 index.truncate(truncaterev * orig._io.size)
2919 2930 with open(destdatapath, 'ab') as data:
2920 2931 data.seek(0)
2921 2932 data.truncate(orig.start(truncaterev))
2922 2933
2923 2934 # instantiate a new revlog from the temporary copy
2924 2935 ui.debug('truncating adding to be rewritten\n')
2925 2936 vfs = vfsmod.vfs(tmpdir)
2926 2937 vfs.options = getattr(orig.opener, 'options', None)
2927 2938
2928 2939 dest = revlog.revlog(
2929 2940 vfs, indexfile=indexname, datafile=dataname, **revlogkwargs
2930 2941 )
2931 2942 if dest._inline:
2932 2943 raise error.Abort('not supporting inline revlog (yet)')
2933 2944 # make sure internals are initialized
2934 2945 dest.revision(len(dest) - 1)
2935 2946 yield dest
2936 2947 del dest, vfs
2937 2948 finally:
2938 2949 shutil.rmtree(tmpdir, True)
2939 2950
2940 2951
2941 2952 @command(
2942 2953 b'perfrevlogchunks',
2943 2954 revlogopts
2944 2955 + formatteropts
2945 2956 + [
2946 2957 (b'e', b'engines', b'', b'compression engines to use'),
2947 2958 (b's', b'startrev', 0, b'revision to start at'),
2948 2959 ],
2949 2960 b'-c|-m|FILE',
2950 2961 )
2951 2962 def perfrevlogchunks(ui, repo, file_=None, engines=None, startrev=0, **opts):
2952 2963 """Benchmark operations on revlog chunks.
2953 2964
2954 2965 Logically, each revlog is a collection of fulltext revisions. However,
2955 2966 stored within each revlog are "chunks" of possibly compressed data. This
2956 2967 data needs to be read and decompressed or compressed and written.
2957 2968
2958 2969 This command measures the time it takes to read+decompress and recompress
2959 2970 chunks in a revlog. It effectively isolates I/O and compression performance.
2960 2971 For measurements of higher-level operations like resolving revisions,
2961 2972 see ``perfrevlogrevisions`` and ``perfrevlogrevision``.
2962 2973 """
2963 2974 opts = _byteskwargs(opts)
2964 2975
2965 2976 rl = cmdutil.openrevlog(repo, b'perfrevlogchunks', file_, opts)
2966 2977
2967 2978 # _chunkraw was renamed to _getsegmentforrevs.
2968 2979 try:
2969 2980 segmentforrevs = rl._getsegmentforrevs
2970 2981 except AttributeError:
2971 2982 segmentforrevs = rl._chunkraw
2972 2983
2973 2984 # Verify engines argument.
2974 2985 if engines:
2975 2986 engines = set(e.strip() for e in engines.split(b','))
2976 2987 for engine in engines:
2977 2988 try:
2978 2989 util.compressionengines[engine]
2979 2990 except KeyError:
2980 2991 raise error.Abort(b'unknown compression engine: %s' % engine)
2981 2992 else:
2982 2993 engines = []
2983 2994 for e in util.compengines:
2984 2995 engine = util.compengines[e]
2985 2996 try:
2986 2997 if engine.available():
2987 2998 engine.revlogcompressor().compress(b'dummy')
2988 2999 engines.append(e)
2989 3000 except NotImplementedError:
2990 3001 pass
2991 3002
2992 3003 revs = list(rl.revs(startrev, len(rl) - 1))
2993 3004
2994 3005 def rlfh(rl):
2995 3006 if rl._inline:
2996 3007 return getsvfs(repo)(rl.indexfile)
2997 3008 else:
2998 3009 return getsvfs(repo)(rl.datafile)
2999 3010
3000 3011 def doread():
3001 3012 rl.clearcaches()
3002 3013 for rev in revs:
3003 3014 segmentforrevs(rev, rev)
3004 3015
3005 3016 def doreadcachedfh():
3006 3017 rl.clearcaches()
3007 3018 fh = rlfh(rl)
3008 3019 for rev in revs:
3009 3020 segmentforrevs(rev, rev, df=fh)
3010 3021
3011 3022 def doreadbatch():
3012 3023 rl.clearcaches()
3013 3024 segmentforrevs(revs[0], revs[-1])
3014 3025
3015 3026 def doreadbatchcachedfh():
3016 3027 rl.clearcaches()
3017 3028 fh = rlfh(rl)
3018 3029 segmentforrevs(revs[0], revs[-1], df=fh)
3019 3030
3020 3031 def dochunk():
3021 3032 rl.clearcaches()
3022 3033 fh = rlfh(rl)
3023 3034 for rev in revs:
3024 3035 rl._chunk(rev, df=fh)
3025 3036
3026 3037 chunks = [None]
3027 3038
3028 3039 def dochunkbatch():
3029 3040 rl.clearcaches()
3030 3041 fh = rlfh(rl)
3031 3042 # Save chunks as a side-effect.
3032 3043 chunks[0] = rl._chunks(revs, df=fh)
3033 3044
3034 3045 def docompress(compressor):
3035 3046 rl.clearcaches()
3036 3047
3037 3048 try:
3038 3049 # Swap in the requested compression engine.
3039 3050 oldcompressor = rl._compressor
3040 3051 rl._compressor = compressor
3041 3052 for chunk in chunks[0]:
3042 3053 rl.compress(chunk)
3043 3054 finally:
3044 3055 rl._compressor = oldcompressor
3045 3056
3046 3057 benches = [
3047 3058 (lambda: doread(), b'read'),
3048 3059 (lambda: doreadcachedfh(), b'read w/ reused fd'),
3049 3060 (lambda: doreadbatch(), b'read batch'),
3050 3061 (lambda: doreadbatchcachedfh(), b'read batch w/ reused fd'),
3051 3062 (lambda: dochunk(), b'chunk'),
3052 3063 (lambda: dochunkbatch(), b'chunk batch'),
3053 3064 ]
3054 3065
3055 3066 for engine in sorted(engines):
3056 3067 compressor = util.compengines[engine].revlogcompressor()
3057 3068 benches.append(
3058 3069 (
3059 3070 functools.partial(docompress, compressor),
3060 3071 b'compress w/ %s' % engine,
3061 3072 )
3062 3073 )
3063 3074
3064 3075 for fn, title in benches:
3065 3076 timer, fm = gettimer(ui, opts)
3066 3077 timer(fn, title=title)
3067 3078 fm.end()
3068 3079
3069 3080
3070 3081 @command(
3071 3082 b'perfrevlogrevision',
3072 3083 revlogopts
3073 3084 + formatteropts
3074 3085 + [(b'', b'cache', False, b'use caches instead of clearing')],
3075 3086 b'-c|-m|FILE REV',
3076 3087 )
3077 3088 def perfrevlogrevision(ui, repo, file_, rev=None, cache=None, **opts):
3078 3089 """Benchmark obtaining a revlog revision.
3079 3090
3080 3091 Obtaining a revlog revision consists of roughly the following steps:
3081 3092
3082 3093 1. Compute the delta chain
3083 3094 2. Slice the delta chain if applicable
3084 3095 3. Obtain the raw chunks for that delta chain
3085 3096 4. Decompress each raw chunk
3086 3097 5. Apply binary patches to obtain fulltext
3087 3098 6. Verify hash of fulltext
3088 3099
3089 3100 This command measures the time spent in each of these phases.
3090 3101 """
3091 3102 opts = _byteskwargs(opts)
3092 3103
3093 3104 if opts.get(b'changelog') or opts.get(b'manifest'):
3094 3105 file_, rev = None, file_
3095 3106 elif rev is None:
3096 3107 raise error.CommandError(b'perfrevlogrevision', b'invalid arguments')
3097 3108
3098 3109 r = cmdutil.openrevlog(repo, b'perfrevlogrevision', file_, opts)
3099 3110
3100 3111 # _chunkraw was renamed to _getsegmentforrevs.
3101 3112 try:
3102 3113 segmentforrevs = r._getsegmentforrevs
3103 3114 except AttributeError:
3104 3115 segmentforrevs = r._chunkraw
3105 3116
3106 3117 node = r.lookup(rev)
3107 3118 rev = r.rev(node)
3108 3119
3109 3120 def getrawchunks(data, chain):
3110 3121 start = r.start
3111 3122 length = r.length
3112 3123 inline = r._inline
3113 3124 iosize = r._io.size
3114 3125 buffer = util.buffer
3115 3126
3116 3127 chunks = []
3117 3128 ladd = chunks.append
3118 3129 for idx, item in enumerate(chain):
3119 3130 offset = start(item[0])
3120 3131 bits = data[idx]
3121 3132 for rev in item:
3122 3133 chunkstart = start(rev)
3123 3134 if inline:
3124 3135 chunkstart += (rev + 1) * iosize
3125 3136 chunklength = length(rev)
3126 3137 ladd(buffer(bits, chunkstart - offset, chunklength))
3127 3138
3128 3139 return chunks
3129 3140
3130 3141 def dodeltachain(rev):
3131 3142 if not cache:
3132 3143 r.clearcaches()
3133 3144 r._deltachain(rev)
3134 3145
3135 3146 def doread(chain):
3136 3147 if not cache:
3137 3148 r.clearcaches()
3138 3149 for item in slicedchain:
3139 3150 segmentforrevs(item[0], item[-1])
3140 3151
3141 3152 def doslice(r, chain, size):
3142 3153 for s in slicechunk(r, chain, targetsize=size):
3143 3154 pass
3144 3155
3145 3156 def dorawchunks(data, chain):
3146 3157 if not cache:
3147 3158 r.clearcaches()
3148 3159 getrawchunks(data, chain)
3149 3160
3150 3161 def dodecompress(chunks):
3151 3162 decomp = r.decompress
3152 3163 for chunk in chunks:
3153 3164 decomp(chunk)
3154 3165
3155 3166 def dopatch(text, bins):
3156 3167 if not cache:
3157 3168 r.clearcaches()
3158 3169 mdiff.patches(text, bins)
3159 3170
3160 3171 def dohash(text):
3161 3172 if not cache:
3162 3173 r.clearcaches()
3163 3174 r.checkhash(text, node, rev=rev)
3164 3175
3165 3176 def dorevision():
3166 3177 if not cache:
3167 3178 r.clearcaches()
3168 3179 r.revision(node)
3169 3180
3170 3181 try:
3171 3182 from mercurial.revlogutils.deltas import slicechunk
3172 3183 except ImportError:
3173 3184 slicechunk = getattr(revlog, '_slicechunk', None)
3174 3185
3175 3186 size = r.length(rev)
3176 3187 chain = r._deltachain(rev)[0]
3177 3188 if not getattr(r, '_withsparseread', False):
3178 3189 slicedchain = (chain,)
3179 3190 else:
3180 3191 slicedchain = tuple(slicechunk(r, chain, targetsize=size))
3181 3192 data = [segmentforrevs(seg[0], seg[-1])[1] for seg in slicedchain]
3182 3193 rawchunks = getrawchunks(data, slicedchain)
3183 3194 bins = r._chunks(chain)
3184 3195 text = bytes(bins[0])
3185 3196 bins = bins[1:]
3186 3197 text = mdiff.patches(text, bins)
3187 3198
3188 3199 benches = [
3189 3200 (lambda: dorevision(), b'full'),
3190 3201 (lambda: dodeltachain(rev), b'deltachain'),
3191 3202 (lambda: doread(chain), b'read'),
3192 3203 ]
3193 3204
3194 3205 if getattr(r, '_withsparseread', False):
3195 3206 slicing = (lambda: doslice(r, chain, size), b'slice-sparse-chain')
3196 3207 benches.append(slicing)
3197 3208
3198 3209 benches.extend(
3199 3210 [
3200 3211 (lambda: dorawchunks(data, slicedchain), b'rawchunks'),
3201 3212 (lambda: dodecompress(rawchunks), b'decompress'),
3202 3213 (lambda: dopatch(text, bins), b'patch'),
3203 3214 (lambda: dohash(text), b'hash'),
3204 3215 ]
3205 3216 )
3206 3217
3207 3218 timer, fm = gettimer(ui, opts)
3208 3219 for fn, title in benches:
3209 3220 timer(fn, title=title)
3210 3221 fm.end()
3211 3222
3212 3223
3213 3224 @command(
3214 3225 b'perfrevset',
3215 3226 [
3216 3227 (b'C', b'clear', False, b'clear volatile cache between each call.'),
3217 3228 (b'', b'contexts', False, b'obtain changectx for each revision'),
3218 3229 ]
3219 3230 + formatteropts,
3220 3231 b"REVSET",
3221 3232 )
3222 3233 def perfrevset(ui, repo, expr, clear=False, contexts=False, **opts):
3223 3234 """benchmark the execution time of a revset
3224 3235
3225 3236 Use the --clean option if need to evaluate the impact of build volatile
3226 3237 revisions set cache on the revset execution. Volatile cache hold filtered
3227 3238 and obsolete related cache."""
3228 3239 opts = _byteskwargs(opts)
3229 3240
3230 3241 timer, fm = gettimer(ui, opts)
3231 3242
3232 3243 def d():
3233 3244 if clear:
3234 3245 repo.invalidatevolatilesets()
3235 3246 if contexts:
3236 3247 for ctx in repo.set(expr):
3237 3248 pass
3238 3249 else:
3239 3250 for r in repo.revs(expr):
3240 3251 pass
3241 3252
3242 3253 timer(d)
3243 3254 fm.end()
3244 3255
3245 3256
3246 3257 @command(
3247 3258 b'perfvolatilesets',
3248 3259 [(b'', b'clear-obsstore', False, b'drop obsstore between each call.'),]
3249 3260 + formatteropts,
3250 3261 )
3251 3262 def perfvolatilesets(ui, repo, *names, **opts):
3252 3263 """benchmark the computation of various volatile set
3253 3264
3254 3265 Volatile set computes element related to filtering and obsolescence."""
3255 3266 opts = _byteskwargs(opts)
3256 3267 timer, fm = gettimer(ui, opts)
3257 3268 repo = repo.unfiltered()
3258 3269
3259 3270 def getobs(name):
3260 3271 def d():
3261 3272 repo.invalidatevolatilesets()
3262 3273 if opts[b'clear_obsstore']:
3263 3274 clearfilecache(repo, b'obsstore')
3264 3275 obsolete.getrevs(repo, name)
3265 3276
3266 3277 return d
3267 3278
3268 3279 allobs = sorted(obsolete.cachefuncs)
3269 3280 if names:
3270 3281 allobs = [n for n in allobs if n in names]
3271 3282
3272 3283 for name in allobs:
3273 3284 timer(getobs(name), title=name)
3274 3285
3275 3286 def getfiltered(name):
3276 3287 def d():
3277 3288 repo.invalidatevolatilesets()
3278 3289 if opts[b'clear_obsstore']:
3279 3290 clearfilecache(repo, b'obsstore')
3280 3291 repoview.filterrevs(repo, name)
3281 3292
3282 3293 return d
3283 3294
3284 3295 allfilter = sorted(repoview.filtertable)
3285 3296 if names:
3286 3297 allfilter = [n for n in allfilter if n in names]
3287 3298
3288 3299 for name in allfilter:
3289 3300 timer(getfiltered(name), title=name)
3290 3301 fm.end()
3291 3302
3292 3303
3293 3304 @command(
3294 3305 b'perfbranchmap',
3295 3306 [
3296 3307 (b'f', b'full', False, b'Includes build time of subset'),
3297 3308 (
3298 3309 b'',
3299 3310 b'clear-revbranch',
3300 3311 False,
3301 3312 b'purge the revbranch cache between computation',
3302 3313 ),
3303 3314 ]
3304 3315 + formatteropts,
3305 3316 )
3306 3317 def perfbranchmap(ui, repo, *filternames, **opts):
3307 3318 """benchmark the update of a branchmap
3308 3319
3309 3320 This benchmarks the full repo.branchmap() call with read and write disabled
3310 3321 """
3311 3322 opts = _byteskwargs(opts)
3312 3323 full = opts.get(b"full", False)
3313 3324 clear_revbranch = opts.get(b"clear_revbranch", False)
3314 3325 timer, fm = gettimer(ui, opts)
3315 3326
3316 3327 def getbranchmap(filtername):
3317 3328 """generate a benchmark function for the filtername"""
3318 3329 if filtername is None:
3319 3330 view = repo
3320 3331 else:
3321 3332 view = repo.filtered(filtername)
3322 3333 if util.safehasattr(view._branchcaches, '_per_filter'):
3323 3334 filtered = view._branchcaches._per_filter
3324 3335 else:
3325 3336 # older versions
3326 3337 filtered = view._branchcaches
3327 3338
3328 3339 def d():
3329 3340 if clear_revbranch:
3330 3341 repo.revbranchcache()._clear()
3331 3342 if full:
3332 3343 view._branchcaches.clear()
3333 3344 else:
3334 3345 filtered.pop(filtername, None)
3335 3346 view.branchmap()
3336 3347
3337 3348 return d
3338 3349
3339 3350 # add filter in smaller subset to bigger subset
3340 3351 possiblefilters = set(repoview.filtertable)
3341 3352 if filternames:
3342 3353 possiblefilters &= set(filternames)
3343 3354 subsettable = getbranchmapsubsettable()
3344 3355 allfilters = []
3345 3356 while possiblefilters:
3346 3357 for name in possiblefilters:
3347 3358 subset = subsettable.get(name)
3348 3359 if subset not in possiblefilters:
3349 3360 break
3350 3361 else:
3351 3362 assert False, b'subset cycle %s!' % possiblefilters
3352 3363 allfilters.append(name)
3353 3364 possiblefilters.remove(name)
3354 3365
3355 3366 # warm the cache
3356 3367 if not full:
3357 3368 for name in allfilters:
3358 3369 repo.filtered(name).branchmap()
3359 3370 if not filternames or b'unfiltered' in filternames:
3360 3371 # add unfiltered
3361 3372 allfilters.append(None)
3362 3373
3363 3374 if util.safehasattr(branchmap.branchcache, 'fromfile'):
3364 3375 branchcacheread = safeattrsetter(branchmap.branchcache, b'fromfile')
3365 3376 branchcacheread.set(classmethod(lambda *args: None))
3366 3377 else:
3367 3378 # older versions
3368 3379 branchcacheread = safeattrsetter(branchmap, b'read')
3369 3380 branchcacheread.set(lambda *args: None)
3370 3381 branchcachewrite = safeattrsetter(branchmap.branchcache, b'write')
3371 3382 branchcachewrite.set(lambda *args: None)
3372 3383 try:
3373 3384 for name in allfilters:
3374 3385 printname = name
3375 3386 if name is None:
3376 3387 printname = b'unfiltered'
3377 3388 timer(getbranchmap(name), title=str(printname))
3378 3389 finally:
3379 3390 branchcacheread.restore()
3380 3391 branchcachewrite.restore()
3381 3392 fm.end()
3382 3393
3383 3394
3384 3395 @command(
3385 3396 b'perfbranchmapupdate',
3386 3397 [
3387 3398 (b'', b'base', [], b'subset of revision to start from'),
3388 3399 (b'', b'target', [], b'subset of revision to end with'),
3389 3400 (b'', b'clear-caches', False, b'clear cache between each runs'),
3390 3401 ]
3391 3402 + formatteropts,
3392 3403 )
3393 3404 def perfbranchmapupdate(ui, repo, base=(), target=(), **opts):
3394 3405 """benchmark branchmap update from for <base> revs to <target> revs
3395 3406
3396 3407 If `--clear-caches` is passed, the following items will be reset before
3397 3408 each update:
3398 3409 * the changelog instance and associated indexes
3399 3410 * the rev-branch-cache instance
3400 3411
3401 3412 Examples:
3402 3413
3403 3414 # update for the one last revision
3404 3415 $ hg perfbranchmapupdate --base 'not tip' --target 'tip'
3405 3416
3406 3417 $ update for change coming with a new branch
3407 3418 $ hg perfbranchmapupdate --base 'stable' --target 'default'
3408 3419 """
3409 3420 from mercurial import branchmap
3410 3421 from mercurial import repoview
3411 3422
3412 3423 opts = _byteskwargs(opts)
3413 3424 timer, fm = gettimer(ui, opts)
3414 3425 clearcaches = opts[b'clear_caches']
3415 3426 unfi = repo.unfiltered()
3416 3427 x = [None] # used to pass data between closure
3417 3428
3418 3429 # we use a `list` here to avoid possible side effect from smartset
3419 3430 baserevs = list(scmutil.revrange(repo, base))
3420 3431 targetrevs = list(scmutil.revrange(repo, target))
3421 3432 if not baserevs:
3422 3433 raise error.Abort(b'no revisions selected for --base')
3423 3434 if not targetrevs:
3424 3435 raise error.Abort(b'no revisions selected for --target')
3425 3436
3426 3437 # make sure the target branchmap also contains the one in the base
3427 3438 targetrevs = list(set(baserevs) | set(targetrevs))
3428 3439 targetrevs.sort()
3429 3440
3430 3441 cl = repo.changelog
3431 3442 allbaserevs = list(cl.ancestors(baserevs, inclusive=True))
3432 3443 allbaserevs.sort()
3433 3444 alltargetrevs = frozenset(cl.ancestors(targetrevs, inclusive=True))
3434 3445
3435 3446 newrevs = list(alltargetrevs.difference(allbaserevs))
3436 3447 newrevs.sort()
3437 3448
3438 3449 allrevs = frozenset(unfi.changelog.revs())
3439 3450 basefilterrevs = frozenset(allrevs.difference(allbaserevs))
3440 3451 targetfilterrevs = frozenset(allrevs.difference(alltargetrevs))
3441 3452
3442 3453 def basefilter(repo, visibilityexceptions=None):
3443 3454 return basefilterrevs
3444 3455
3445 3456 def targetfilter(repo, visibilityexceptions=None):
3446 3457 return targetfilterrevs
3447 3458
3448 3459 msg = b'benchmark of branchmap with %d revisions with %d new ones\n'
3449 3460 ui.status(msg % (len(allbaserevs), len(newrevs)))
3450 3461 if targetfilterrevs:
3451 3462 msg = b'(%d revisions still filtered)\n'
3452 3463 ui.status(msg % len(targetfilterrevs))
3453 3464
3454 3465 try:
3455 3466 repoview.filtertable[b'__perf_branchmap_update_base'] = basefilter
3456 3467 repoview.filtertable[b'__perf_branchmap_update_target'] = targetfilter
3457 3468
3458 3469 baserepo = repo.filtered(b'__perf_branchmap_update_base')
3459 3470 targetrepo = repo.filtered(b'__perf_branchmap_update_target')
3460 3471
3461 3472 # try to find an existing branchmap to reuse
3462 3473 subsettable = getbranchmapsubsettable()
3463 3474 candidatefilter = subsettable.get(None)
3464 3475 while candidatefilter is not None:
3465 3476 candidatebm = repo.filtered(candidatefilter).branchmap()
3466 3477 if candidatebm.validfor(baserepo):
3467 3478 filtered = repoview.filterrevs(repo, candidatefilter)
3468 3479 missing = [r for r in allbaserevs if r in filtered]
3469 3480 base = candidatebm.copy()
3470 3481 base.update(baserepo, missing)
3471 3482 break
3472 3483 candidatefilter = subsettable.get(candidatefilter)
3473 3484 else:
3474 3485 # no suitable subset where found
3475 3486 base = branchmap.branchcache()
3476 3487 base.update(baserepo, allbaserevs)
3477 3488
3478 3489 def setup():
3479 3490 x[0] = base.copy()
3480 3491 if clearcaches:
3481 3492 unfi._revbranchcache = None
3482 3493 clearchangelog(repo)
3483 3494
3484 3495 def bench():
3485 3496 x[0].update(targetrepo, newrevs)
3486 3497
3487 3498 timer(bench, setup=setup)
3488 3499 fm.end()
3489 3500 finally:
3490 3501 repoview.filtertable.pop(b'__perf_branchmap_update_base', None)
3491 3502 repoview.filtertable.pop(b'__perf_branchmap_update_target', None)
3492 3503
3493 3504
3494 3505 @command(
3495 3506 b'perfbranchmapload',
3496 3507 [
3497 3508 (b'f', b'filter', b'', b'Specify repoview filter'),
3498 3509 (b'', b'list', False, b'List brachmap filter caches'),
3499 3510 (b'', b'clear-revlogs', False, b'refresh changelog and manifest'),
3500 3511 ]
3501 3512 + formatteropts,
3502 3513 )
3503 3514 def perfbranchmapload(ui, repo, filter=b'', list=False, **opts):
3504 3515 """benchmark reading the branchmap"""
3505 3516 opts = _byteskwargs(opts)
3506 3517 clearrevlogs = opts[b'clear_revlogs']
3507 3518
3508 3519 if list:
3509 3520 for name, kind, st in repo.cachevfs.readdir(stat=True):
3510 3521 if name.startswith(b'branch2'):
3511 3522 filtername = name.partition(b'-')[2] or b'unfiltered'
3512 3523 ui.status(
3513 3524 b'%s - %s\n' % (filtername, util.bytecount(st.st_size))
3514 3525 )
3515 3526 return
3516 3527 if not filter:
3517 3528 filter = None
3518 3529 subsettable = getbranchmapsubsettable()
3519 3530 if filter is None:
3520 3531 repo = repo.unfiltered()
3521 3532 else:
3522 3533 repo = repoview.repoview(repo, filter)
3523 3534
3524 3535 repo.branchmap() # make sure we have a relevant, up to date branchmap
3525 3536
3526 3537 try:
3527 3538 fromfile = branchmap.branchcache.fromfile
3528 3539 except AttributeError:
3529 3540 # older versions
3530 3541 fromfile = branchmap.read
3531 3542
3532 3543 currentfilter = filter
3533 3544 # try once without timer, the filter may not be cached
3534 3545 while fromfile(repo) is None:
3535 3546 currentfilter = subsettable.get(currentfilter)
3536 3547 if currentfilter is None:
3537 3548 raise error.Abort(
3538 3549 b'No branchmap cached for %s repo' % (filter or b'unfiltered')
3539 3550 )
3540 3551 repo = repo.filtered(currentfilter)
3541 3552 timer, fm = gettimer(ui, opts)
3542 3553
3543 3554 def setup():
3544 3555 if clearrevlogs:
3545 3556 clearchangelog(repo)
3546 3557
3547 3558 def bench():
3548 3559 fromfile(repo)
3549 3560
3550 3561 timer(bench, setup=setup)
3551 3562 fm.end()
3552 3563
3553 3564
3554 3565 @command(b'perfloadmarkers')
3555 3566 def perfloadmarkers(ui, repo):
3556 3567 """benchmark the time to parse the on-disk markers for a repo
3557 3568
3558 3569 Result is the number of markers in the repo."""
3559 3570 timer, fm = gettimer(ui)
3560 3571 svfs = getsvfs(repo)
3561 3572 timer(lambda: len(obsolete.obsstore(svfs)))
3562 3573 fm.end()
3563 3574
3564 3575
3565 3576 @command(
3566 3577 b'perflrucachedict',
3567 3578 formatteropts
3568 3579 + [
3569 3580 (b'', b'costlimit', 0, b'maximum total cost of items in cache'),
3570 3581 (b'', b'mincost', 0, b'smallest cost of items in cache'),
3571 3582 (b'', b'maxcost', 100, b'maximum cost of items in cache'),
3572 3583 (b'', b'size', 4, b'size of cache'),
3573 3584 (b'', b'gets', 10000, b'number of key lookups'),
3574 3585 (b'', b'sets', 10000, b'number of key sets'),
3575 3586 (b'', b'mixed', 10000, b'number of mixed mode operations'),
3576 3587 (
3577 3588 b'',
3578 3589 b'mixedgetfreq',
3579 3590 50,
3580 3591 b'frequency of get vs set ops in mixed mode',
3581 3592 ),
3582 3593 ],
3583 3594 norepo=True,
3584 3595 )
3585 3596 def perflrucache(
3586 3597 ui,
3587 3598 mincost=0,
3588 3599 maxcost=100,
3589 3600 costlimit=0,
3590 3601 size=4,
3591 3602 gets=10000,
3592 3603 sets=10000,
3593 3604 mixed=10000,
3594 3605 mixedgetfreq=50,
3595 3606 **opts
3596 3607 ):
3597 3608 opts = _byteskwargs(opts)
3598 3609
3599 3610 def doinit():
3600 3611 for i in _xrange(10000):
3601 3612 util.lrucachedict(size)
3602 3613
3603 3614 costrange = list(range(mincost, maxcost + 1))
3604 3615
3605 3616 values = []
3606 3617 for i in _xrange(size):
3607 3618 values.append(random.randint(0, _maxint))
3608 3619
3609 3620 # Get mode fills the cache and tests raw lookup performance with no
3610 3621 # eviction.
3611 3622 getseq = []
3612 3623 for i in _xrange(gets):
3613 3624 getseq.append(random.choice(values))
3614 3625
3615 3626 def dogets():
3616 3627 d = util.lrucachedict(size)
3617 3628 for v in values:
3618 3629 d[v] = v
3619 3630 for key in getseq:
3620 3631 value = d[key]
3621 3632 value # silence pyflakes warning
3622 3633
3623 3634 def dogetscost():
3624 3635 d = util.lrucachedict(size, maxcost=costlimit)
3625 3636 for i, v in enumerate(values):
3626 3637 d.insert(v, v, cost=costs[i])
3627 3638 for key in getseq:
3628 3639 try:
3629 3640 value = d[key]
3630 3641 value # silence pyflakes warning
3631 3642 except KeyError:
3632 3643 pass
3633 3644
3634 3645 # Set mode tests insertion speed with cache eviction.
3635 3646 setseq = []
3636 3647 costs = []
3637 3648 for i in _xrange(sets):
3638 3649 setseq.append(random.randint(0, _maxint))
3639 3650 costs.append(random.choice(costrange))
3640 3651
3641 3652 def doinserts():
3642 3653 d = util.lrucachedict(size)
3643 3654 for v in setseq:
3644 3655 d.insert(v, v)
3645 3656
3646 3657 def doinsertscost():
3647 3658 d = util.lrucachedict(size, maxcost=costlimit)
3648 3659 for i, v in enumerate(setseq):
3649 3660 d.insert(v, v, cost=costs[i])
3650 3661
3651 3662 def dosets():
3652 3663 d = util.lrucachedict(size)
3653 3664 for v in setseq:
3654 3665 d[v] = v
3655 3666
3656 3667 # Mixed mode randomly performs gets and sets with eviction.
3657 3668 mixedops = []
3658 3669 for i in _xrange(mixed):
3659 3670 r = random.randint(0, 100)
3660 3671 if r < mixedgetfreq:
3661 3672 op = 0
3662 3673 else:
3663 3674 op = 1
3664 3675
3665 3676 mixedops.append(
3666 3677 (op, random.randint(0, size * 2), random.choice(costrange))
3667 3678 )
3668 3679
3669 3680 def domixed():
3670 3681 d = util.lrucachedict(size)
3671 3682
3672 3683 for op, v, cost in mixedops:
3673 3684 if op == 0:
3674 3685 try:
3675 3686 d[v]
3676 3687 except KeyError:
3677 3688 pass
3678 3689 else:
3679 3690 d[v] = v
3680 3691
3681 3692 def domixedcost():
3682 3693 d = util.lrucachedict(size, maxcost=costlimit)
3683 3694
3684 3695 for op, v, cost in mixedops:
3685 3696 if op == 0:
3686 3697 try:
3687 3698 d[v]
3688 3699 except KeyError:
3689 3700 pass
3690 3701 else:
3691 3702 d.insert(v, v, cost=cost)
3692 3703
3693 3704 benches = [
3694 3705 (doinit, b'init'),
3695 3706 ]
3696 3707
3697 3708 if costlimit:
3698 3709 benches.extend(
3699 3710 [
3700 3711 (dogetscost, b'gets w/ cost limit'),
3701 3712 (doinsertscost, b'inserts w/ cost limit'),
3702 3713 (domixedcost, b'mixed w/ cost limit'),
3703 3714 ]
3704 3715 )
3705 3716 else:
3706 3717 benches.extend(
3707 3718 [
3708 3719 (dogets, b'gets'),
3709 3720 (doinserts, b'inserts'),
3710 3721 (dosets, b'sets'),
3711 3722 (domixed, b'mixed'),
3712 3723 ]
3713 3724 )
3714 3725
3715 3726 for fn, title in benches:
3716 3727 timer, fm = gettimer(ui, opts)
3717 3728 timer(fn, title=title)
3718 3729 fm.end()
3719 3730
3720 3731
3721 3732 @command(b'perfwrite', formatteropts)
3722 3733 def perfwrite(ui, repo, **opts):
3723 3734 """microbenchmark ui.write
3724 3735 """
3725 3736 opts = _byteskwargs(opts)
3726 3737
3727 3738 timer, fm = gettimer(ui, opts)
3728 3739
3729 3740 def write():
3730 3741 for i in range(100000):
3731 3742 ui.writenoi18n(b'Testing write performance\n')
3732 3743
3733 3744 timer(write)
3734 3745 fm.end()
3735 3746
3736 3747
3737 3748 def uisetup(ui):
3738 3749 if util.safehasattr(cmdutil, b'openrevlog') and not util.safehasattr(
3739 3750 commands, b'debugrevlogopts'
3740 3751 ):
3741 3752 # for "historical portability":
3742 3753 # In this case, Mercurial should be 1.9 (or a79fea6b3e77) -
3743 3754 # 3.7 (or 5606f7d0d063). Therefore, '--dir' option for
3744 3755 # openrevlog() should cause failure, because it has been
3745 3756 # available since 3.5 (or 49c583ca48c4).
3746 3757 def openrevlog(orig, repo, cmd, file_, opts):
3747 3758 if opts.get(b'dir') and not util.safehasattr(repo, b'dirlog'):
3748 3759 raise error.Abort(
3749 3760 b"This version doesn't support --dir option",
3750 3761 hint=b"use 3.5 or later",
3751 3762 )
3752 3763 return orig(repo, cmd, file_, opts)
3753 3764
3754 3765 extensions.wrapfunction(cmdutil, b'openrevlog', openrevlog)
3755 3766
3756 3767
3757 3768 @command(
3758 3769 b'perfprogress',
3759 3770 formatteropts
3760 3771 + [
3761 3772 (b'', b'topic', b'topic', b'topic for progress messages'),
3762 3773 (b'c', b'total', 1000000, b'total value we are progressing to'),
3763 3774 ],
3764 3775 norepo=True,
3765 3776 )
3766 3777 def perfprogress(ui, topic=None, total=None, **opts):
3767 3778 """printing of progress bars"""
3768 3779 opts = _byteskwargs(opts)
3769 3780
3770 3781 timer, fm = gettimer(ui, opts)
3771 3782
3772 3783 def doprogress():
3773 3784 with ui.makeprogress(topic, total=total) as progress:
3774 3785 for i in _xrange(total):
3775 3786 progress.increment()
3776 3787
3777 3788 timer(doprogress)
3778 3789 fm.end()
@@ -1,396 +1,397 b''
1 1 #require test-repo
2 2
3 3 Set vars:
4 4
5 5 $ . "$TESTDIR/helpers-testrepo.sh"
6 6 $ CONTRIBDIR="$TESTDIR/../contrib"
7 7
8 8 Prepare repo:
9 9
10 10 $ hg init
11 11
12 12 $ echo this is file a > a
13 13 $ hg add a
14 14 $ hg commit -m first
15 15
16 16 $ echo adding to file a >> a
17 17 $ hg commit -m second
18 18
19 19 $ echo adding more to file a >> a
20 20 $ hg commit -m third
21 21
22 22 $ hg up -r 0
23 23 1 files updated, 0 files merged, 0 files removed, 0 files unresolved
24 24 $ echo merge-this >> a
25 25 $ hg commit -m merge-able
26 26 created new head
27 27
28 28 $ hg up -r 2
29 29 1 files updated, 0 files merged, 0 files removed, 0 files unresolved
30 30
31 31 perfstatus
32 32
33 33 $ cat >> $HGRCPATH << EOF
34 34 > [extensions]
35 35 > perf=$CONTRIBDIR/perf.py
36 36 > [perf]
37 37 > presleep=0
38 38 > stub=on
39 39 > parentscount=1
40 40 > EOF
41 41 $ hg help -e perf
42 42 perf extension - helper extension to measure performance
43 43
44 44 Configurations
45 45 ==============
46 46
47 47 "perf"
48 48 ------
49 49
50 50 "all-timing"
51 51 When set, additional statistics will be reported for each benchmark: best,
52 52 worst, median average. If not set only the best timing is reported
53 53 (default: off).
54 54
55 55 "presleep"
56 56 number of second to wait before any group of runs (default: 1)
57 57
58 58 "pre-run"
59 59 number of run to perform before starting measurement.
60 60
61 61 "profile-benchmark"
62 62 Enable profiling for the benchmarked section. (The first iteration is
63 63 benchmarked)
64 64
65 65 "run-limits"
66 66 Control the number of runs each benchmark will perform. The option value
67 67 should be a list of '<time>-<numberofrun>' pairs. After each run the
68 68 conditions are considered in order with the following logic:
69 69
70 70 If benchmark has been running for <time> seconds, and we have performed
71 71 <numberofrun> iterations, stop the benchmark,
72 72
73 73 The default value is: '3.0-100, 10.0-3'
74 74
75 75 "stub"
76 76 When set, benchmarks will only be run once, useful for testing (default:
77 77 off)
78 78
79 79 list of commands:
80 80
81 81 perfaddremove
82 82 (no help text available)
83 83 perfancestors
84 84 (no help text available)
85 85 perfancestorset
86 86 (no help text available)
87 87 perfannotate (no help text available)
88 88 perfbdiff benchmark a bdiff between revisions
89 89 perfbookmarks
90 90 benchmark parsing bookmarks from disk to memory
91 91 perfbranchmap
92 92 benchmark the update of a branchmap
93 93 perfbranchmapload
94 94 benchmark reading the branchmap
95 95 perfbranchmapupdate
96 96 benchmark branchmap update from for <base> revs to <target>
97 97 revs
98 98 perfbundleread
99 99 Benchmark reading of bundle files.
100 100 perfcca (no help text available)
101 101 perfchangegroupchangelog
102 102 Benchmark producing a changelog group for a changegroup.
103 103 perfchangeset
104 104 (no help text available)
105 105 perfctxfiles (no help text available)
106 106 perfdiffwd Profile diff of working directory changes
107 107 perfdirfoldmap
108 108 benchmap a 'dirstate._map.dirfoldmap.get()' request
109 109 perfdirs (no help text available)
110 perfdirstate benchmap the time necessary to load a dirstate from scratch
110 perfdirstate benchmap the time of various distate operations
111 111 perfdirstatedirs
112 112 benchmap a 'dirstate.hasdir' call from an empty 'dirs' cache
113 113 perfdirstatefoldmap
114 114 benchmap a 'dirstate._map.filefoldmap.get()' request
115 115 perfdirstatewrite
116 116 benchmap the time it take to write a dirstate on disk
117 117 perfdiscovery
118 118 benchmark discovery between local repo and the peer at given
119 119 path
120 120 perffncacheencode
121 121 (no help text available)
122 122 perffncacheload
123 123 (no help text available)
124 124 perffncachewrite
125 125 (no help text available)
126 126 perfheads benchmark the computation of a changelog heads
127 127 perfhelper-mergecopies
128 128 find statistics about potential parameters for
129 129 'perfmergecopies'
130 130 perfhelper-pathcopies
131 131 find statistic about potential parameters for the
132 132 'perftracecopies'
133 133 perfignore benchmark operation related to computing ignore
134 134 perfindex benchmark index creation time followed by a lookup
135 135 perflinelogedits
136 136 (no help text available)
137 137 perfloadmarkers
138 138 benchmark the time to parse the on-disk markers for a repo
139 139 perflog (no help text available)
140 140 perflookup (no help text available)
141 141 perflrucachedict
142 142 (no help text available)
143 143 perfmanifest benchmark the time to read a manifest from disk and return a
144 144 usable
145 145 perfmergecalculate
146 146 (no help text available)
147 147 perfmergecopies
148 148 measure runtime of 'copies.mergecopies'
149 149 perfmoonwalk benchmark walking the changelog backwards
150 150 perfnodelookup
151 151 (no help text available)
152 152 perfnodemap benchmark the time necessary to look up revision from a cold
153 153 nodemap
154 154 perfparents benchmark the time necessary to fetch one changeset's parents.
155 155 perfpathcopies
156 156 benchmark the copy tracing logic
157 157 perfphases benchmark phasesets computation
158 158 perfphasesremote
159 159 benchmark time needed to analyse phases of the remote server
160 160 perfprogress printing of progress bars
161 161 perfrawfiles (no help text available)
162 162 perfrevlogchunks
163 163 Benchmark operations on revlog chunks.
164 164 perfrevlogindex
165 165 Benchmark operations against a revlog index.
166 166 perfrevlogrevision
167 167 Benchmark obtaining a revlog revision.
168 168 perfrevlogrevisions
169 169 Benchmark reading a series of revisions from a revlog.
170 170 perfrevlogwrite
171 171 Benchmark writing a series of revisions to a revlog.
172 172 perfrevrange (no help text available)
173 173 perfrevset benchmark the execution time of a revset
174 174 perfstartup (no help text available)
175 175 perfstatus benchmark the performance of a single status call
176 176 perftags (no help text available)
177 177 perftemplating
178 178 test the rendering time of a given template
179 179 perfunidiff benchmark a unified diff between revisions
180 180 perfvolatilesets
181 181 benchmark the computation of various volatile set
182 182 perfwalk (no help text available)
183 183 perfwrite microbenchmark ui.write
184 184
185 185 (use 'hg help -v perf' to show built-in aliases and global options)
186 186 $ hg perfaddremove
187 187 $ hg perfancestors
188 188 $ hg perfancestorset 2
189 189 $ hg perfannotate a
190 190 $ hg perfbdiff -c 1
191 191 $ hg perfbdiff --alldata 1
192 192 $ hg perfunidiff -c 1
193 193 $ hg perfunidiff --alldata 1
194 194 $ hg perfbookmarks
195 195 $ hg perfbranchmap
196 196 $ hg perfbranchmapload
197 197 $ hg perfbranchmapupdate --base "not tip" --target "tip"
198 198 benchmark of branchmap with 3 revisions with 1 new ones
199 199 $ hg perfcca
200 200 $ hg perfchangegroupchangelog
201 201 $ hg perfchangegroupchangelog --cgversion 01
202 202 $ hg perfchangeset 2
203 203 $ hg perfctxfiles 2
204 204 $ hg perfdiffwd
205 205 $ hg perfdirfoldmap
206 206 $ hg perfdirs
207 207 $ hg perfdirstate
208 $ hg perfdirstate --iteration
208 209 $ hg perfdirstatedirs
209 210 $ hg perfdirstatefoldmap
210 211 $ hg perfdirstatewrite
211 212 #if repofncache
212 213 $ hg perffncacheencode
213 214 $ hg perffncacheload
214 215 $ hg debugrebuildfncache
215 216 fncache already up to date
216 217 $ hg perffncachewrite
217 218 $ hg debugrebuildfncache
218 219 fncache already up to date
219 220 #endif
220 221 $ hg perfheads
221 222 $ hg perfignore
222 223 $ hg perfindex
223 224 $ hg perflinelogedits -n 1
224 225 $ hg perfloadmarkers
225 226 $ hg perflog
226 227 $ hg perflookup 2
227 228 $ hg perflrucache
228 229 $ hg perfmanifest 2
229 230 $ hg perfmanifest -m 44fe2c8352bb3a478ffd7d8350bbc721920134d1
230 231 $ hg perfmanifest -m 44fe2c8352bb
231 232 abort: manifest revision must be integer or full node
232 233 [255]
233 234 $ hg perfmergecalculate -r 3
234 235 $ hg perfmoonwalk
235 236 $ hg perfnodelookup 2
236 237 $ hg perfpathcopies 1 2
237 238 $ hg perfprogress --total 1000
238 239 $ hg perfrawfiles 2
239 240 $ hg perfrevlogindex -c
240 241 #if reporevlogstore
241 242 $ hg perfrevlogrevisions .hg/store/data/a.i
242 243 #endif
243 244 $ hg perfrevlogrevision -m 0
244 245 $ hg perfrevlogchunks -c
245 246 $ hg perfrevrange
246 247 $ hg perfrevset 'all()'
247 248 $ hg perfstartup
248 249 $ hg perfstatus
249 250 $ hg perftags
250 251 $ hg perftemplating
251 252 $ hg perfvolatilesets
252 253 $ hg perfwalk
253 254 $ hg perfparents
254 255 $ hg perfdiscovery -q .
255 256
256 257 Test run control
257 258 ----------------
258 259
259 260 Simple single entry
260 261
261 262 $ hg perfparents --config perf.stub=no --config perf.run-limits='0.000000001-15'
262 263 ! wall * comb * user * sys * (best of 15) (glob)
263 264
264 265 Multiple entries
265 266
266 267 $ hg perfparents --config perf.stub=no --config perf.run-limits='500000-1, 0.000000001-5'
267 268 ! wall * comb * user * sys * (best of 5) (glob)
268 269
269 270 error case are ignored
270 271
271 272 $ hg perfparents --config perf.stub=no --config perf.run-limits='500, 0.000000001-5'
272 273 malformatted run limit entry, missing "-": 500
273 274 ! wall * comb * user * sys * (best of 5) (glob)
274 275 $ hg perfparents --config perf.stub=no --config perf.run-limits='aaa-12, 0.000000001-5'
275 276 malformatted run limit entry, could not convert string to float: aaa: aaa-12 (no-py3 !)
276 277 malformatted run limit entry, could not convert string to float: 'aaa': aaa-12 (py3 !)
277 278 ! wall * comb * user * sys * (best of 5) (glob)
278 279 $ hg perfparents --config perf.stub=no --config perf.run-limits='12-aaaaaa, 0.000000001-5'
279 280 malformatted run limit entry, invalid literal for int() with base 10: 'aaaaaa': 12-aaaaaa
280 281 ! wall * comb * user * sys * (best of 5) (glob)
281 282
282 283 test actual output
283 284 ------------------
284 285
285 286 normal output:
286 287
287 288 $ hg perfheads --config perf.stub=no
288 289 ! wall * comb * user * sys * (best of *) (glob)
289 290
290 291 detailed output:
291 292
292 293 $ hg perfheads --config perf.all-timing=yes --config perf.stub=no
293 294 ! wall * comb * user * sys * (best of *) (glob)
294 295 ! wall * comb * user * sys * (max of *) (glob)
295 296 ! wall * comb * user * sys * (avg of *) (glob)
296 297 ! wall * comb * user * sys * (median of *) (glob)
297 298
298 299 test json output
299 300 ----------------
300 301
301 302 normal output:
302 303
303 304 $ hg perfheads --template json --config perf.stub=no
304 305 [
305 306 {
306 307 "comb": *, (glob)
307 308 "count": *, (glob)
308 309 "sys": *, (glob)
309 310 "user": *, (glob)
310 311 "wall": * (glob)
311 312 }
312 313 ]
313 314
314 315 detailed output:
315 316
316 317 $ hg perfheads --template json --config perf.all-timing=yes --config perf.stub=no
317 318 [
318 319 {
319 320 "avg.comb": *, (glob)
320 321 "avg.count": *, (glob)
321 322 "avg.sys": *, (glob)
322 323 "avg.user": *, (glob)
323 324 "avg.wall": *, (glob)
324 325 "comb": *, (glob)
325 326 "count": *, (glob)
326 327 "max.comb": *, (glob)
327 328 "max.count": *, (glob)
328 329 "max.sys": *, (glob)
329 330 "max.user": *, (glob)
330 331 "max.wall": *, (glob)
331 332 "median.comb": *, (glob)
332 333 "median.count": *, (glob)
333 334 "median.sys": *, (glob)
334 335 "median.user": *, (glob)
335 336 "median.wall": *, (glob)
336 337 "sys": *, (glob)
337 338 "user": *, (glob)
338 339 "wall": * (glob)
339 340 }
340 341 ]
341 342
342 343 Test pre-run feature
343 344 --------------------
344 345
345 346 (perf discovery has some spurious output)
346 347
347 348 $ hg perfdiscovery . --config perf.stub=no --config perf.run-limits='0.000000001-1' --config perf.pre-run=0
348 349 ! wall * comb * user * sys * (best of 1) (glob)
349 350 searching for changes
350 351 $ hg perfdiscovery . --config perf.stub=no --config perf.run-limits='0.000000001-1' --config perf.pre-run=1
351 352 ! wall * comb * user * sys * (best of 1) (glob)
352 353 searching for changes
353 354 searching for changes
354 355 $ hg perfdiscovery . --config perf.stub=no --config perf.run-limits='0.000000001-1' --config perf.pre-run=3
355 356 ! wall * comb * user * sys * (best of 1) (glob)
356 357 searching for changes
357 358 searching for changes
358 359 searching for changes
359 360 searching for changes
360 361
361 362 test profile-benchmark option
362 363 ------------------------------
363 364
364 365 Function to check that statprof ran
365 366 $ statprofran () {
366 367 > egrep 'Sample count:|No samples recorded' > /dev/null
367 368 > }
368 369 $ hg perfdiscovery . --config perf.stub=no --config perf.run-limits='0.000000001-1' --config perf.profile-benchmark=yes 2>&1 | statprofran
369 370
370 371 Check perf.py for historical portability
371 372 ----------------------------------------
372 373
373 374 $ cd "$TESTDIR/.."
374 375
375 376 $ (testrepohg files -r 1.2 glob:mercurial/*.c glob:mercurial/*.py;
376 377 > testrepohg files -r tip glob:mercurial/*.c glob:mercurial/*.py) |
377 378 > "$TESTDIR"/check-perf-code.py contrib/perf.py
378 379 contrib/perf.py:\d+: (re)
379 380 > from mercurial import (
380 381 import newer module separately in try clause for early Mercurial
381 382 contrib/perf.py:\d+: (re)
382 383 > from mercurial import (
383 384 import newer module separately in try clause for early Mercurial
384 385 contrib/perf.py:\d+: (re)
385 386 > origindexpath = orig.opener.join(orig.indexfile)
386 387 use getvfs()/getsvfs() for early Mercurial
387 388 contrib/perf.py:\d+: (re)
388 389 > origdatapath = orig.opener.join(orig.datafile)
389 390 use getvfs()/getsvfs() for early Mercurial
390 391 contrib/perf.py:\d+: (re)
391 392 > vfs = vfsmod.vfs(tmpdir)
392 393 use getvfs()/getsvfs() for early Mercurial
393 394 contrib/perf.py:\d+: (re)
394 395 > vfs.options = getattr(orig.opener, 'options', None)
395 396 use getvfs()/getsvfs() for early Mercurial
396 397 [1]
General Comments 0
You need to be logged in to leave comments. Login now