##// END OF EJS Templates
perf: make `perfphasesremote` use the new `index.has_node` api...
marmoute -
r43951:3350d7ae default
parent child Browse files
Show More
@@ -1,3838 +1,3840 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 '_tagscache' in vars(repo):
695 695 del repo.__dict__['_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, '_clcachekey', None)
730 730 object.__setattr__(repo, '_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 [
764 764 (b'u', b'unknown', False, b'ask status to look for unknown files'),
765 765 (b'', b'dirstate', False, b'benchmark the internal dirstate call'),
766 766 ]
767 767 + formatteropts,
768 768 )
769 769 def perfstatus(ui, repo, **opts):
770 770 """benchmark the performance of a single status call
771 771
772 772 The repository data are preserved between each call.
773 773
774 774 By default, only the status of the tracked file are requested. If
775 775 `--unknown` is passed, the "unknown" files are also tracked.
776 776 """
777 777 opts = _byteskwargs(opts)
778 778 # m = match.always(repo.root, repo.getcwd())
779 779 # timer(lambda: sum(map(len, repo.dirstate.status(m, [], False, False,
780 780 # False))))
781 781 timer, fm = gettimer(ui, opts)
782 782 if opts[b'dirstate']:
783 783 dirstate = repo.dirstate
784 784 m = scmutil.matchall(repo)
785 785 unknown = opts[b'unknown']
786 786
787 787 def status_dirstate():
788 788 s = dirstate.status(
789 789 m, subrepos=[], ignored=False, clean=False, unknown=unknown
790 790 )
791 791 sum(map(len, s))
792 792
793 793 timer(status_dirstate)
794 794 else:
795 795 timer(lambda: sum(map(len, repo.status(unknown=opts[b'unknown']))))
796 796 fm.end()
797 797
798 798
799 799 @command(b'perfaddremove', formatteropts)
800 800 def perfaddremove(ui, repo, **opts):
801 801 opts = _byteskwargs(opts)
802 802 timer, fm = gettimer(ui, opts)
803 803 try:
804 804 oldquiet = repo.ui.quiet
805 805 repo.ui.quiet = True
806 806 matcher = scmutil.match(repo[None])
807 807 opts[b'dry_run'] = True
808 808 if b'uipathfn' in getargspec(scmutil.addremove).args:
809 809 uipathfn = scmutil.getuipathfn(repo)
810 810 timer(lambda: scmutil.addremove(repo, matcher, b"", uipathfn, opts))
811 811 else:
812 812 timer(lambda: scmutil.addremove(repo, matcher, b"", opts))
813 813 finally:
814 814 repo.ui.quiet = oldquiet
815 815 fm.end()
816 816
817 817
818 818 def clearcaches(cl):
819 819 # behave somewhat consistently across internal API changes
820 820 if util.safehasattr(cl, b'clearcaches'):
821 821 cl.clearcaches()
822 822 elif util.safehasattr(cl, b'_nodecache'):
823 823 # <= hg-5.2
824 824 from mercurial.node import nullid, nullrev
825 825
826 826 cl._nodecache = {nullid: nullrev}
827 827 cl._nodepos = None
828 828
829 829
830 830 @command(b'perfheads', formatteropts)
831 831 def perfheads(ui, repo, **opts):
832 832 """benchmark the computation of a changelog heads"""
833 833 opts = _byteskwargs(opts)
834 834 timer, fm = gettimer(ui, opts)
835 835 cl = repo.changelog
836 836
837 837 def s():
838 838 clearcaches(cl)
839 839
840 840 def d():
841 841 len(cl.headrevs())
842 842
843 843 timer(d, setup=s)
844 844 fm.end()
845 845
846 846
847 847 @command(
848 848 b'perftags',
849 849 formatteropts
850 850 + [(b'', b'clear-revlogs', False, b'refresh changelog and manifest'),],
851 851 )
852 852 def perftags(ui, repo, **opts):
853 853 opts = _byteskwargs(opts)
854 854 timer, fm = gettimer(ui, opts)
855 855 repocleartagscache = repocleartagscachefunc(repo)
856 856 clearrevlogs = opts[b'clear_revlogs']
857 857
858 858 def s():
859 859 if clearrevlogs:
860 860 clearchangelog(repo)
861 861 clearfilecache(repo.unfiltered(), 'manifest')
862 862 repocleartagscache()
863 863
864 864 def t():
865 865 return len(repo.tags())
866 866
867 867 timer(t, setup=s)
868 868 fm.end()
869 869
870 870
871 871 @command(b'perfancestors', formatteropts)
872 872 def perfancestors(ui, repo, **opts):
873 873 opts = _byteskwargs(opts)
874 874 timer, fm = gettimer(ui, opts)
875 875 heads = repo.changelog.headrevs()
876 876
877 877 def d():
878 878 for a in repo.changelog.ancestors(heads):
879 879 pass
880 880
881 881 timer(d)
882 882 fm.end()
883 883
884 884
885 885 @command(b'perfancestorset', formatteropts)
886 886 def perfancestorset(ui, repo, revset, **opts):
887 887 opts = _byteskwargs(opts)
888 888 timer, fm = gettimer(ui, opts)
889 889 revs = repo.revs(revset)
890 890 heads = repo.changelog.headrevs()
891 891
892 892 def d():
893 893 s = repo.changelog.ancestors(heads)
894 894 for rev in revs:
895 895 rev in s
896 896
897 897 timer(d)
898 898 fm.end()
899 899
900 900
901 901 @command(b'perfdiscovery', formatteropts, b'PATH')
902 902 def perfdiscovery(ui, repo, path, **opts):
903 903 """benchmark discovery between local repo and the peer at given path
904 904 """
905 905 repos = [repo, None]
906 906 timer, fm = gettimer(ui, opts)
907 907 path = ui.expandpath(path)
908 908
909 909 def s():
910 910 repos[1] = hg.peer(ui, opts, path)
911 911
912 912 def d():
913 913 setdiscovery.findcommonheads(ui, *repos)
914 914
915 915 timer(d, setup=s)
916 916 fm.end()
917 917
918 918
919 919 @command(
920 920 b'perfbookmarks',
921 921 formatteropts
922 922 + [(b'', b'clear-revlogs', False, b'refresh changelog and manifest'),],
923 923 )
924 924 def perfbookmarks(ui, repo, **opts):
925 925 """benchmark parsing bookmarks from disk to memory"""
926 926 opts = _byteskwargs(opts)
927 927 timer, fm = gettimer(ui, opts)
928 928
929 929 clearrevlogs = opts[b'clear_revlogs']
930 930
931 931 def s():
932 932 if clearrevlogs:
933 933 clearchangelog(repo)
934 934 clearfilecache(repo, b'_bookmarks')
935 935
936 936 def d():
937 937 repo._bookmarks
938 938
939 939 timer(d, setup=s)
940 940 fm.end()
941 941
942 942
943 943 @command(b'perfbundleread', formatteropts, b'BUNDLE')
944 944 def perfbundleread(ui, repo, bundlepath, **opts):
945 945 """Benchmark reading of bundle files.
946 946
947 947 This command is meant to isolate the I/O part of bundle reading as
948 948 much as possible.
949 949 """
950 950 from mercurial import (
951 951 bundle2,
952 952 exchange,
953 953 streamclone,
954 954 )
955 955
956 956 opts = _byteskwargs(opts)
957 957
958 958 def makebench(fn):
959 959 def run():
960 960 with open(bundlepath, b'rb') as fh:
961 961 bundle = exchange.readbundle(ui, fh, bundlepath)
962 962 fn(bundle)
963 963
964 964 return run
965 965
966 966 def makereadnbytes(size):
967 967 def run():
968 968 with open(bundlepath, b'rb') as fh:
969 969 bundle = exchange.readbundle(ui, fh, bundlepath)
970 970 while bundle.read(size):
971 971 pass
972 972
973 973 return run
974 974
975 975 def makestdioread(size):
976 976 def run():
977 977 with open(bundlepath, b'rb') as fh:
978 978 while fh.read(size):
979 979 pass
980 980
981 981 return run
982 982
983 983 # bundle1
984 984
985 985 def deltaiter(bundle):
986 986 for delta in bundle.deltaiter():
987 987 pass
988 988
989 989 def iterchunks(bundle):
990 990 for chunk in bundle.getchunks():
991 991 pass
992 992
993 993 # bundle2
994 994
995 995 def forwardchunks(bundle):
996 996 for chunk in bundle._forwardchunks():
997 997 pass
998 998
999 999 def iterparts(bundle):
1000 1000 for part in bundle.iterparts():
1001 1001 pass
1002 1002
1003 1003 def iterpartsseekable(bundle):
1004 1004 for part in bundle.iterparts(seekable=True):
1005 1005 pass
1006 1006
1007 1007 def seek(bundle):
1008 1008 for part in bundle.iterparts(seekable=True):
1009 1009 part.seek(0, os.SEEK_END)
1010 1010
1011 1011 def makepartreadnbytes(size):
1012 1012 def run():
1013 1013 with open(bundlepath, b'rb') as fh:
1014 1014 bundle = exchange.readbundle(ui, fh, bundlepath)
1015 1015 for part in bundle.iterparts():
1016 1016 while part.read(size):
1017 1017 pass
1018 1018
1019 1019 return run
1020 1020
1021 1021 benches = [
1022 1022 (makestdioread(8192), b'read(8k)'),
1023 1023 (makestdioread(16384), b'read(16k)'),
1024 1024 (makestdioread(32768), b'read(32k)'),
1025 1025 (makestdioread(131072), b'read(128k)'),
1026 1026 ]
1027 1027
1028 1028 with open(bundlepath, b'rb') as fh:
1029 1029 bundle = exchange.readbundle(ui, fh, bundlepath)
1030 1030
1031 1031 if isinstance(bundle, changegroup.cg1unpacker):
1032 1032 benches.extend(
1033 1033 [
1034 1034 (makebench(deltaiter), b'cg1 deltaiter()'),
1035 1035 (makebench(iterchunks), b'cg1 getchunks()'),
1036 1036 (makereadnbytes(8192), b'cg1 read(8k)'),
1037 1037 (makereadnbytes(16384), b'cg1 read(16k)'),
1038 1038 (makereadnbytes(32768), b'cg1 read(32k)'),
1039 1039 (makereadnbytes(131072), b'cg1 read(128k)'),
1040 1040 ]
1041 1041 )
1042 1042 elif isinstance(bundle, bundle2.unbundle20):
1043 1043 benches.extend(
1044 1044 [
1045 1045 (makebench(forwardchunks), b'bundle2 forwardchunks()'),
1046 1046 (makebench(iterparts), b'bundle2 iterparts()'),
1047 1047 (
1048 1048 makebench(iterpartsseekable),
1049 1049 b'bundle2 iterparts() seekable',
1050 1050 ),
1051 1051 (makebench(seek), b'bundle2 part seek()'),
1052 1052 (makepartreadnbytes(8192), b'bundle2 part read(8k)'),
1053 1053 (makepartreadnbytes(16384), b'bundle2 part read(16k)'),
1054 1054 (makepartreadnbytes(32768), b'bundle2 part read(32k)'),
1055 1055 (makepartreadnbytes(131072), b'bundle2 part read(128k)'),
1056 1056 ]
1057 1057 )
1058 1058 elif isinstance(bundle, streamclone.streamcloneapplier):
1059 1059 raise error.Abort(b'stream clone bundles not supported')
1060 1060 else:
1061 1061 raise error.Abort(b'unhandled bundle type: %s' % type(bundle))
1062 1062
1063 1063 for fn, title in benches:
1064 1064 timer, fm = gettimer(ui, opts)
1065 1065 timer(fn, title=title)
1066 1066 fm.end()
1067 1067
1068 1068
1069 1069 @command(
1070 1070 b'perfchangegroupchangelog',
1071 1071 formatteropts
1072 1072 + [
1073 1073 (b'', b'cgversion', b'02', b'changegroup version'),
1074 1074 (b'r', b'rev', b'', b'revisions to add to changegroup'),
1075 1075 ],
1076 1076 )
1077 1077 def perfchangegroupchangelog(ui, repo, cgversion=b'02', rev=None, **opts):
1078 1078 """Benchmark producing a changelog group for a changegroup.
1079 1079
1080 1080 This measures the time spent processing the changelog during a
1081 1081 bundle operation. This occurs during `hg bundle` and on a server
1082 1082 processing a `getbundle` wire protocol request (handles clones
1083 1083 and pull requests).
1084 1084
1085 1085 By default, all revisions are added to the changegroup.
1086 1086 """
1087 1087 opts = _byteskwargs(opts)
1088 1088 cl = repo.changelog
1089 1089 nodes = [cl.lookup(r) for r in repo.revs(rev or b'all()')]
1090 1090 bundler = changegroup.getbundler(cgversion, repo)
1091 1091
1092 1092 def d():
1093 1093 state, chunks = bundler._generatechangelog(cl, nodes)
1094 1094 for chunk in chunks:
1095 1095 pass
1096 1096
1097 1097 timer, fm = gettimer(ui, opts)
1098 1098
1099 1099 # Terminal printing can interfere with timing. So disable it.
1100 1100 with ui.configoverride({(b'progress', b'disable'): True}):
1101 1101 timer(d)
1102 1102
1103 1103 fm.end()
1104 1104
1105 1105
1106 1106 @command(b'perfdirs', formatteropts)
1107 1107 def perfdirs(ui, repo, **opts):
1108 1108 opts = _byteskwargs(opts)
1109 1109 timer, fm = gettimer(ui, opts)
1110 1110 dirstate = repo.dirstate
1111 1111 b'a' in dirstate
1112 1112
1113 1113 def d():
1114 1114 dirstate.hasdir(b'a')
1115 1115 del dirstate._map._dirs
1116 1116
1117 1117 timer(d)
1118 1118 fm.end()
1119 1119
1120 1120
1121 1121 @command(
1122 1122 b'perfdirstate',
1123 1123 [
1124 1124 (
1125 1125 b'',
1126 1126 b'iteration',
1127 1127 None,
1128 1128 b'benchmark a full iteration for the dirstate',
1129 1129 ),
1130 1130 (
1131 1131 b'',
1132 1132 b'contains',
1133 1133 None,
1134 1134 b'benchmark a large amount of `nf in dirstate` calls',
1135 1135 ),
1136 1136 ]
1137 1137 + formatteropts,
1138 1138 )
1139 1139 def perfdirstate(ui, repo, **opts):
1140 1140 """benchmap the time of various distate operations
1141 1141
1142 1142 By default benchmark the time necessary to load a dirstate from scratch.
1143 1143 The dirstate is loaded to the point were a "contains" request can be
1144 1144 answered.
1145 1145 """
1146 1146 opts = _byteskwargs(opts)
1147 1147 timer, fm = gettimer(ui, opts)
1148 1148 b"a" in repo.dirstate
1149 1149
1150 1150 if opts[b'iteration'] and opts[b'contains']:
1151 1151 msg = b'only specify one of --iteration or --contains'
1152 1152 raise error.Abort(msg)
1153 1153
1154 1154 if opts[b'iteration']:
1155 1155 setup = None
1156 1156 dirstate = repo.dirstate
1157 1157
1158 1158 def d():
1159 1159 for f in dirstate:
1160 1160 pass
1161 1161
1162 1162 elif opts[b'contains']:
1163 1163 setup = None
1164 1164 dirstate = repo.dirstate
1165 1165 allfiles = list(dirstate)
1166 1166 # also add file path that will be "missing" from the dirstate
1167 1167 allfiles.extend([f[::-1] for f in allfiles])
1168 1168
1169 1169 def d():
1170 1170 for f in allfiles:
1171 1171 f in dirstate
1172 1172
1173 1173 else:
1174 1174
1175 1175 def setup():
1176 1176 repo.dirstate.invalidate()
1177 1177
1178 1178 def d():
1179 1179 b"a" in repo.dirstate
1180 1180
1181 1181 timer(d, setup=setup)
1182 1182 fm.end()
1183 1183
1184 1184
1185 1185 @command(b'perfdirstatedirs', formatteropts)
1186 1186 def perfdirstatedirs(ui, repo, **opts):
1187 1187 """benchmap a 'dirstate.hasdir' call from an empty `dirs` cache
1188 1188 """
1189 1189 opts = _byteskwargs(opts)
1190 1190 timer, fm = gettimer(ui, opts)
1191 1191 repo.dirstate.hasdir(b"a")
1192 1192
1193 1193 def setup():
1194 1194 del repo.dirstate._map._dirs
1195 1195
1196 1196 def d():
1197 1197 repo.dirstate.hasdir(b"a")
1198 1198
1199 1199 timer(d, setup=setup)
1200 1200 fm.end()
1201 1201
1202 1202
1203 1203 @command(b'perfdirstatefoldmap', formatteropts)
1204 1204 def perfdirstatefoldmap(ui, repo, **opts):
1205 1205 """benchmap a `dirstate._map.filefoldmap.get()` request
1206 1206
1207 1207 The dirstate filefoldmap cache is dropped between every request.
1208 1208 """
1209 1209 opts = _byteskwargs(opts)
1210 1210 timer, fm = gettimer(ui, opts)
1211 1211 dirstate = repo.dirstate
1212 1212 dirstate._map.filefoldmap.get(b'a')
1213 1213
1214 1214 def setup():
1215 1215 del dirstate._map.filefoldmap
1216 1216
1217 1217 def d():
1218 1218 dirstate._map.filefoldmap.get(b'a')
1219 1219
1220 1220 timer(d, setup=setup)
1221 1221 fm.end()
1222 1222
1223 1223
1224 1224 @command(b'perfdirfoldmap', formatteropts)
1225 1225 def perfdirfoldmap(ui, repo, **opts):
1226 1226 """benchmap a `dirstate._map.dirfoldmap.get()` request
1227 1227
1228 1228 The dirstate dirfoldmap cache is dropped between every request.
1229 1229 """
1230 1230 opts = _byteskwargs(opts)
1231 1231 timer, fm = gettimer(ui, opts)
1232 1232 dirstate = repo.dirstate
1233 1233 dirstate._map.dirfoldmap.get(b'a')
1234 1234
1235 1235 def setup():
1236 1236 del dirstate._map.dirfoldmap
1237 1237 del dirstate._map._dirs
1238 1238
1239 1239 def d():
1240 1240 dirstate._map.dirfoldmap.get(b'a')
1241 1241
1242 1242 timer(d, setup=setup)
1243 1243 fm.end()
1244 1244
1245 1245
1246 1246 @command(b'perfdirstatewrite', formatteropts)
1247 1247 def perfdirstatewrite(ui, repo, **opts):
1248 1248 """benchmap the time it take to write a dirstate on disk
1249 1249 """
1250 1250 opts = _byteskwargs(opts)
1251 1251 timer, fm = gettimer(ui, opts)
1252 1252 ds = repo.dirstate
1253 1253 b"a" in ds
1254 1254
1255 1255 def setup():
1256 1256 ds._dirty = True
1257 1257
1258 1258 def d():
1259 1259 ds.write(repo.currenttransaction())
1260 1260
1261 1261 timer(d, setup=setup)
1262 1262 fm.end()
1263 1263
1264 1264
1265 1265 def _getmergerevs(repo, opts):
1266 1266 """parse command argument to return rev involved in merge
1267 1267
1268 1268 input: options dictionnary with `rev`, `from` and `bse`
1269 1269 output: (localctx, otherctx, basectx)
1270 1270 """
1271 1271 if opts[b'from']:
1272 1272 fromrev = scmutil.revsingle(repo, opts[b'from'])
1273 1273 wctx = repo[fromrev]
1274 1274 else:
1275 1275 wctx = repo[None]
1276 1276 # we don't want working dir files to be stat'd in the benchmark, so
1277 1277 # prime that cache
1278 1278 wctx.dirty()
1279 1279 rctx = scmutil.revsingle(repo, opts[b'rev'], opts[b'rev'])
1280 1280 if opts[b'base']:
1281 1281 fromrev = scmutil.revsingle(repo, opts[b'base'])
1282 1282 ancestor = repo[fromrev]
1283 1283 else:
1284 1284 ancestor = wctx.ancestor(rctx)
1285 1285 return (wctx, rctx, ancestor)
1286 1286
1287 1287
1288 1288 @command(
1289 1289 b'perfmergecalculate',
1290 1290 [
1291 1291 (b'r', b'rev', b'.', b'rev to merge against'),
1292 1292 (b'', b'from', b'', b'rev to merge from'),
1293 1293 (b'', b'base', b'', b'the revision to use as base'),
1294 1294 ]
1295 1295 + formatteropts,
1296 1296 )
1297 1297 def perfmergecalculate(ui, repo, **opts):
1298 1298 opts = _byteskwargs(opts)
1299 1299 timer, fm = gettimer(ui, opts)
1300 1300
1301 1301 wctx, rctx, ancestor = _getmergerevs(repo, opts)
1302 1302
1303 1303 def d():
1304 1304 # acceptremote is True because we don't want prompts in the middle of
1305 1305 # our benchmark
1306 1306 merge.calculateupdates(
1307 1307 repo,
1308 1308 wctx,
1309 1309 rctx,
1310 1310 [ancestor],
1311 1311 branchmerge=False,
1312 1312 force=False,
1313 1313 acceptremote=True,
1314 1314 followcopies=True,
1315 1315 )
1316 1316
1317 1317 timer(d)
1318 1318 fm.end()
1319 1319
1320 1320
1321 1321 @command(
1322 1322 b'perfmergecopies',
1323 1323 [
1324 1324 (b'r', b'rev', b'.', b'rev to merge against'),
1325 1325 (b'', b'from', b'', b'rev to merge from'),
1326 1326 (b'', b'base', b'', b'the revision to use as base'),
1327 1327 ]
1328 1328 + formatteropts,
1329 1329 )
1330 1330 def perfmergecopies(ui, repo, **opts):
1331 1331 """measure runtime of `copies.mergecopies`"""
1332 1332 opts = _byteskwargs(opts)
1333 1333 timer, fm = gettimer(ui, opts)
1334 1334 wctx, rctx, ancestor = _getmergerevs(repo, opts)
1335 1335
1336 1336 def d():
1337 1337 # acceptremote is True because we don't want prompts in the middle of
1338 1338 # our benchmark
1339 1339 copies.mergecopies(repo, wctx, rctx, ancestor)
1340 1340
1341 1341 timer(d)
1342 1342 fm.end()
1343 1343
1344 1344
1345 1345 @command(b'perfpathcopies', [], b"REV REV")
1346 1346 def perfpathcopies(ui, repo, rev1, rev2, **opts):
1347 1347 """benchmark the copy tracing logic"""
1348 1348 opts = _byteskwargs(opts)
1349 1349 timer, fm = gettimer(ui, opts)
1350 1350 ctx1 = scmutil.revsingle(repo, rev1, rev1)
1351 1351 ctx2 = scmutil.revsingle(repo, rev2, rev2)
1352 1352
1353 1353 def d():
1354 1354 copies.pathcopies(ctx1, ctx2)
1355 1355
1356 1356 timer(d)
1357 1357 fm.end()
1358 1358
1359 1359
1360 1360 @command(
1361 1361 b'perfphases',
1362 1362 [(b'', b'full', False, b'include file reading time too'),],
1363 1363 b"",
1364 1364 )
1365 1365 def perfphases(ui, repo, **opts):
1366 1366 """benchmark phasesets computation"""
1367 1367 opts = _byteskwargs(opts)
1368 1368 timer, fm = gettimer(ui, opts)
1369 1369 _phases = repo._phasecache
1370 1370 full = opts.get(b'full')
1371 1371
1372 1372 def d():
1373 1373 phases = _phases
1374 1374 if full:
1375 1375 clearfilecache(repo, b'_phasecache')
1376 1376 phases = repo._phasecache
1377 1377 phases.invalidate()
1378 1378 phases.loadphaserevs(repo)
1379 1379
1380 1380 timer(d)
1381 1381 fm.end()
1382 1382
1383 1383
1384 1384 @command(b'perfphasesremote', [], b"[DEST]")
1385 1385 def perfphasesremote(ui, repo, dest=None, **opts):
1386 1386 """benchmark time needed to analyse phases of the remote server"""
1387 1387 from mercurial.node import bin
1388 1388 from mercurial import (
1389 1389 exchange,
1390 1390 hg,
1391 1391 phases,
1392 1392 )
1393 1393
1394 1394 opts = _byteskwargs(opts)
1395 1395 timer, fm = gettimer(ui, opts)
1396 1396
1397 1397 path = ui.paths.getpath(dest, default=(b'default-push', b'default'))
1398 1398 if not path:
1399 1399 raise error.Abort(
1400 1400 b'default repository not configured!',
1401 1401 hint=b"see 'hg help config.paths'",
1402 1402 )
1403 1403 dest = path.pushloc or path.loc
1404 1404 ui.statusnoi18n(b'analysing phase of %s\n' % util.hidepassword(dest))
1405 1405 other = hg.peer(repo, opts, dest)
1406 1406
1407 1407 # easier to perform discovery through the operation
1408 1408 op = exchange.pushoperation(repo, other)
1409 1409 exchange._pushdiscoverychangeset(op)
1410 1410
1411 1411 remotesubset = op.fallbackheads
1412 1412
1413 1413 with other.commandexecutor() as e:
1414 1414 remotephases = e.callcommand(
1415 1415 b'listkeys', {b'namespace': b'phases'}
1416 1416 ).result()
1417 1417 del other
1418 1418 publishing = remotephases.get(b'publishing', False)
1419 1419 if publishing:
1420 1420 ui.statusnoi18n(b'publishing: yes\n')
1421 1421 else:
1422 1422 ui.statusnoi18n(b'publishing: no\n')
1423 1423
1424 nodemap = repo.changelog.nodemap
1424 has_node = getattr(repo.changelog.index, 'has_node', None)
1425 if has_node is None:
1426 has_node = repo.changelog.nodemap.__contains__
1425 1427 nonpublishroots = 0
1426 1428 for nhex, phase in remotephases.iteritems():
1427 1429 if nhex == b'publishing': # ignore data related to publish option
1428 1430 continue
1429 1431 node = bin(nhex)
1430 if node in nodemap and int(phase):
1432 if has_node(node) and int(phase):
1431 1433 nonpublishroots += 1
1432 1434 ui.statusnoi18n(b'number of roots: %d\n' % len(remotephases))
1433 1435 ui.statusnoi18n(b'number of known non public roots: %d\n' % nonpublishroots)
1434 1436
1435 1437 def d():
1436 1438 phases.remotephasessummary(repo, remotesubset, remotephases)
1437 1439
1438 1440 timer(d)
1439 1441 fm.end()
1440 1442
1441 1443
1442 1444 @command(
1443 1445 b'perfmanifest',
1444 1446 [
1445 1447 (b'm', b'manifest-rev', False, b'Look up a manifest node revision'),
1446 1448 (b'', b'clear-disk', False, b'clear on-disk caches too'),
1447 1449 ]
1448 1450 + formatteropts,
1449 1451 b'REV|NODE',
1450 1452 )
1451 1453 def perfmanifest(ui, repo, rev, manifest_rev=False, clear_disk=False, **opts):
1452 1454 """benchmark the time to read a manifest from disk and return a usable
1453 1455 dict-like object
1454 1456
1455 1457 Manifest caches are cleared before retrieval."""
1456 1458 opts = _byteskwargs(opts)
1457 1459 timer, fm = gettimer(ui, opts)
1458 1460 if not manifest_rev:
1459 1461 ctx = scmutil.revsingle(repo, rev, rev)
1460 1462 t = ctx.manifestnode()
1461 1463 else:
1462 1464 from mercurial.node import bin
1463 1465
1464 1466 if len(rev) == 40:
1465 1467 t = bin(rev)
1466 1468 else:
1467 1469 try:
1468 1470 rev = int(rev)
1469 1471
1470 1472 if util.safehasattr(repo.manifestlog, b'getstorage'):
1471 1473 t = repo.manifestlog.getstorage(b'').node(rev)
1472 1474 else:
1473 1475 t = repo.manifestlog._revlog.lookup(rev)
1474 1476 except ValueError:
1475 1477 raise error.Abort(
1476 1478 b'manifest revision must be integer or full node'
1477 1479 )
1478 1480
1479 1481 def d():
1480 1482 repo.manifestlog.clearcaches(clear_persisted_data=clear_disk)
1481 1483 repo.manifestlog[t].read()
1482 1484
1483 1485 timer(d)
1484 1486 fm.end()
1485 1487
1486 1488
1487 1489 @command(b'perfchangeset', formatteropts)
1488 1490 def perfchangeset(ui, repo, rev, **opts):
1489 1491 opts = _byteskwargs(opts)
1490 1492 timer, fm = gettimer(ui, opts)
1491 1493 n = scmutil.revsingle(repo, rev).node()
1492 1494
1493 1495 def d():
1494 1496 repo.changelog.read(n)
1495 1497 # repo.changelog._cache = None
1496 1498
1497 1499 timer(d)
1498 1500 fm.end()
1499 1501
1500 1502
1501 1503 @command(b'perfignore', formatteropts)
1502 1504 def perfignore(ui, repo, **opts):
1503 1505 """benchmark operation related to computing ignore"""
1504 1506 opts = _byteskwargs(opts)
1505 1507 timer, fm = gettimer(ui, opts)
1506 1508 dirstate = repo.dirstate
1507 1509
1508 1510 def setupone():
1509 1511 dirstate.invalidate()
1510 1512 clearfilecache(dirstate, b'_ignore')
1511 1513
1512 1514 def runone():
1513 1515 dirstate._ignore
1514 1516
1515 1517 timer(runone, setup=setupone, title=b"load")
1516 1518 fm.end()
1517 1519
1518 1520
1519 1521 @command(
1520 1522 b'perfindex',
1521 1523 [
1522 1524 (b'', b'rev', [], b'revision to be looked up (default tip)'),
1523 1525 (b'', b'no-lookup', None, b'do not revision lookup post creation'),
1524 1526 ]
1525 1527 + formatteropts,
1526 1528 )
1527 1529 def perfindex(ui, repo, **opts):
1528 1530 """benchmark index creation time followed by a lookup
1529 1531
1530 1532 The default is to look `tip` up. Depending on the index implementation,
1531 1533 the revision looked up can matters. For example, an implementation
1532 1534 scanning the index will have a faster lookup time for `--rev tip` than for
1533 1535 `--rev 0`. The number of looked up revisions and their order can also
1534 1536 matters.
1535 1537
1536 1538 Example of useful set to test:
1537 1539 * tip
1538 1540 * 0
1539 1541 * -10:
1540 1542 * :10
1541 1543 * -10: + :10
1542 1544 * :10: + -10:
1543 1545 * -10000:
1544 1546 * -10000: + 0
1545 1547
1546 1548 It is not currently possible to check for lookup of a missing node. For
1547 1549 deeper lookup benchmarking, checkout the `perfnodemap` command."""
1548 1550 import mercurial.revlog
1549 1551
1550 1552 opts = _byteskwargs(opts)
1551 1553 timer, fm = gettimer(ui, opts)
1552 1554 mercurial.revlog._prereadsize = 2 ** 24 # disable lazy parser in old hg
1553 1555 if opts[b'no_lookup']:
1554 1556 if opts['rev']:
1555 1557 raise error.Abort('--no-lookup and --rev are mutually exclusive')
1556 1558 nodes = []
1557 1559 elif not opts[b'rev']:
1558 1560 nodes = [repo[b"tip"].node()]
1559 1561 else:
1560 1562 revs = scmutil.revrange(repo, opts[b'rev'])
1561 1563 cl = repo.changelog
1562 1564 nodes = [cl.node(r) for r in revs]
1563 1565
1564 1566 unfi = repo.unfiltered()
1565 1567 # find the filecache func directly
1566 1568 # This avoid polluting the benchmark with the filecache logic
1567 1569 makecl = unfi.__class__.changelog.func
1568 1570
1569 1571 def setup():
1570 1572 # probably not necessary, but for good measure
1571 1573 clearchangelog(unfi)
1572 1574
1573 1575 def d():
1574 1576 cl = makecl(unfi)
1575 1577 for n in nodes:
1576 1578 cl.rev(n)
1577 1579
1578 1580 timer(d, setup=setup)
1579 1581 fm.end()
1580 1582
1581 1583
1582 1584 @command(
1583 1585 b'perfnodemap',
1584 1586 [
1585 1587 (b'', b'rev', [], b'revision to be looked up (default tip)'),
1586 1588 (b'', b'clear-caches', True, b'clear revlog cache between calls'),
1587 1589 ]
1588 1590 + formatteropts,
1589 1591 )
1590 1592 def perfnodemap(ui, repo, **opts):
1591 1593 """benchmark the time necessary to look up revision from a cold nodemap
1592 1594
1593 1595 Depending on the implementation, the amount and order of revision we look
1594 1596 up can varies. Example of useful set to test:
1595 1597 * tip
1596 1598 * 0
1597 1599 * -10:
1598 1600 * :10
1599 1601 * -10: + :10
1600 1602 * :10: + -10:
1601 1603 * -10000:
1602 1604 * -10000: + 0
1603 1605
1604 1606 The command currently focus on valid binary lookup. Benchmarking for
1605 1607 hexlookup, prefix lookup and missing lookup would also be valuable.
1606 1608 """
1607 1609 import mercurial.revlog
1608 1610
1609 1611 opts = _byteskwargs(opts)
1610 1612 timer, fm = gettimer(ui, opts)
1611 1613 mercurial.revlog._prereadsize = 2 ** 24 # disable lazy parser in old hg
1612 1614
1613 1615 unfi = repo.unfiltered()
1614 1616 clearcaches = opts['clear_caches']
1615 1617 # find the filecache func directly
1616 1618 # This avoid polluting the benchmark with the filecache logic
1617 1619 makecl = unfi.__class__.changelog.func
1618 1620 if not opts[b'rev']:
1619 1621 raise error.Abort('use --rev to specify revisions to look up')
1620 1622 revs = scmutil.revrange(repo, opts[b'rev'])
1621 1623 cl = repo.changelog
1622 1624 nodes = [cl.node(r) for r in revs]
1623 1625
1624 1626 # use a list to pass reference to a nodemap from one closure to the next
1625 1627 nodeget = [None]
1626 1628
1627 1629 def setnodeget():
1628 1630 # probably not necessary, but for good measure
1629 1631 clearchangelog(unfi)
1630 1632 nodeget[0] = makecl(unfi).nodemap.get
1631 1633
1632 1634 def d():
1633 1635 get = nodeget[0]
1634 1636 for n in nodes:
1635 1637 get(n)
1636 1638
1637 1639 setup = None
1638 1640 if clearcaches:
1639 1641
1640 1642 def setup():
1641 1643 setnodeget()
1642 1644
1643 1645 else:
1644 1646 setnodeget()
1645 1647 d() # prewarm the data structure
1646 1648 timer(d, setup=setup)
1647 1649 fm.end()
1648 1650
1649 1651
1650 1652 @command(b'perfstartup', formatteropts)
1651 1653 def perfstartup(ui, repo, **opts):
1652 1654 opts = _byteskwargs(opts)
1653 1655 timer, fm = gettimer(ui, opts)
1654 1656
1655 1657 def d():
1656 1658 if os.name != 'nt':
1657 1659 os.system(
1658 1660 b"HGRCPATH= %s version -q > /dev/null" % fsencode(sys.argv[0])
1659 1661 )
1660 1662 else:
1661 1663 os.environ['HGRCPATH'] = r' '
1662 1664 os.system("%s version -q > NUL" % sys.argv[0])
1663 1665
1664 1666 timer(d)
1665 1667 fm.end()
1666 1668
1667 1669
1668 1670 @command(b'perfparents', formatteropts)
1669 1671 def perfparents(ui, repo, **opts):
1670 1672 """benchmark the time necessary to fetch one changeset's parents.
1671 1673
1672 1674 The fetch is done using the `node identifier`, traversing all object layers
1673 1675 from the repository object. The first N revisions will be used for this
1674 1676 benchmark. N is controlled by the ``perf.parentscount`` config option
1675 1677 (default: 1000).
1676 1678 """
1677 1679 opts = _byteskwargs(opts)
1678 1680 timer, fm = gettimer(ui, opts)
1679 1681 # control the number of commits perfparents iterates over
1680 1682 # experimental config: perf.parentscount
1681 1683 count = getint(ui, b"perf", b"parentscount", 1000)
1682 1684 if len(repo.changelog) < count:
1683 1685 raise error.Abort(b"repo needs %d commits for this test" % count)
1684 1686 repo = repo.unfiltered()
1685 1687 nl = [repo.changelog.node(i) for i in _xrange(count)]
1686 1688
1687 1689 def d():
1688 1690 for n in nl:
1689 1691 repo.changelog.parents(n)
1690 1692
1691 1693 timer(d)
1692 1694 fm.end()
1693 1695
1694 1696
1695 1697 @command(b'perfctxfiles', formatteropts)
1696 1698 def perfctxfiles(ui, repo, x, **opts):
1697 1699 opts = _byteskwargs(opts)
1698 1700 x = int(x)
1699 1701 timer, fm = gettimer(ui, opts)
1700 1702
1701 1703 def d():
1702 1704 len(repo[x].files())
1703 1705
1704 1706 timer(d)
1705 1707 fm.end()
1706 1708
1707 1709
1708 1710 @command(b'perfrawfiles', formatteropts)
1709 1711 def perfrawfiles(ui, repo, x, **opts):
1710 1712 opts = _byteskwargs(opts)
1711 1713 x = int(x)
1712 1714 timer, fm = gettimer(ui, opts)
1713 1715 cl = repo.changelog
1714 1716
1715 1717 def d():
1716 1718 len(cl.read(x)[3])
1717 1719
1718 1720 timer(d)
1719 1721 fm.end()
1720 1722
1721 1723
1722 1724 @command(b'perflookup', formatteropts)
1723 1725 def perflookup(ui, repo, rev, **opts):
1724 1726 opts = _byteskwargs(opts)
1725 1727 timer, fm = gettimer(ui, opts)
1726 1728 timer(lambda: len(repo.lookup(rev)))
1727 1729 fm.end()
1728 1730
1729 1731
1730 1732 @command(
1731 1733 b'perflinelogedits',
1732 1734 [
1733 1735 (b'n', b'edits', 10000, b'number of edits'),
1734 1736 (b'', b'max-hunk-lines', 10, b'max lines in a hunk'),
1735 1737 ],
1736 1738 norepo=True,
1737 1739 )
1738 1740 def perflinelogedits(ui, **opts):
1739 1741 from mercurial import linelog
1740 1742
1741 1743 opts = _byteskwargs(opts)
1742 1744
1743 1745 edits = opts[b'edits']
1744 1746 maxhunklines = opts[b'max_hunk_lines']
1745 1747
1746 1748 maxb1 = 100000
1747 1749 random.seed(0)
1748 1750 randint = random.randint
1749 1751 currentlines = 0
1750 1752 arglist = []
1751 1753 for rev in _xrange(edits):
1752 1754 a1 = randint(0, currentlines)
1753 1755 a2 = randint(a1, min(currentlines, a1 + maxhunklines))
1754 1756 b1 = randint(0, maxb1)
1755 1757 b2 = randint(b1, b1 + maxhunklines)
1756 1758 currentlines += (b2 - b1) - (a2 - a1)
1757 1759 arglist.append((rev, a1, a2, b1, b2))
1758 1760
1759 1761 def d():
1760 1762 ll = linelog.linelog()
1761 1763 for args in arglist:
1762 1764 ll.replacelines(*args)
1763 1765
1764 1766 timer, fm = gettimer(ui, opts)
1765 1767 timer(d)
1766 1768 fm.end()
1767 1769
1768 1770
1769 1771 @command(b'perfrevrange', formatteropts)
1770 1772 def perfrevrange(ui, repo, *specs, **opts):
1771 1773 opts = _byteskwargs(opts)
1772 1774 timer, fm = gettimer(ui, opts)
1773 1775 revrange = scmutil.revrange
1774 1776 timer(lambda: len(revrange(repo, specs)))
1775 1777 fm.end()
1776 1778
1777 1779
1778 1780 @command(b'perfnodelookup', formatteropts)
1779 1781 def perfnodelookup(ui, repo, rev, **opts):
1780 1782 opts = _byteskwargs(opts)
1781 1783 timer, fm = gettimer(ui, opts)
1782 1784 import mercurial.revlog
1783 1785
1784 1786 mercurial.revlog._prereadsize = 2 ** 24 # disable lazy parser in old hg
1785 1787 n = scmutil.revsingle(repo, rev).node()
1786 1788 cl = mercurial.revlog.revlog(getsvfs(repo), b"00changelog.i")
1787 1789
1788 1790 def d():
1789 1791 cl.rev(n)
1790 1792 clearcaches(cl)
1791 1793
1792 1794 timer(d)
1793 1795 fm.end()
1794 1796
1795 1797
1796 1798 @command(
1797 1799 b'perflog',
1798 1800 [(b'', b'rename', False, b'ask log to follow renames')] + formatteropts,
1799 1801 )
1800 1802 def perflog(ui, repo, rev=None, **opts):
1801 1803 opts = _byteskwargs(opts)
1802 1804 if rev is None:
1803 1805 rev = []
1804 1806 timer, fm = gettimer(ui, opts)
1805 1807 ui.pushbuffer()
1806 1808 timer(
1807 1809 lambda: commands.log(
1808 1810 ui, repo, rev=rev, date=b'', user=b'', copies=opts.get(b'rename')
1809 1811 )
1810 1812 )
1811 1813 ui.popbuffer()
1812 1814 fm.end()
1813 1815
1814 1816
1815 1817 @command(b'perfmoonwalk', formatteropts)
1816 1818 def perfmoonwalk(ui, repo, **opts):
1817 1819 """benchmark walking the changelog backwards
1818 1820
1819 1821 This also loads the changelog data for each revision in the changelog.
1820 1822 """
1821 1823 opts = _byteskwargs(opts)
1822 1824 timer, fm = gettimer(ui, opts)
1823 1825
1824 1826 def moonwalk():
1825 1827 for i in repo.changelog.revs(start=(len(repo) - 1), stop=-1):
1826 1828 ctx = repo[i]
1827 1829 ctx.branch() # read changelog data (in addition to the index)
1828 1830
1829 1831 timer(moonwalk)
1830 1832 fm.end()
1831 1833
1832 1834
1833 1835 @command(
1834 1836 b'perftemplating',
1835 1837 [(b'r', b'rev', [], b'revisions to run the template on'),] + formatteropts,
1836 1838 )
1837 1839 def perftemplating(ui, repo, testedtemplate=None, **opts):
1838 1840 """test the rendering time of a given template"""
1839 1841 if makelogtemplater is None:
1840 1842 raise error.Abort(
1841 1843 b"perftemplating not available with this Mercurial",
1842 1844 hint=b"use 4.3 or later",
1843 1845 )
1844 1846
1845 1847 opts = _byteskwargs(opts)
1846 1848
1847 1849 nullui = ui.copy()
1848 1850 nullui.fout = open(os.devnull, 'wb')
1849 1851 nullui.disablepager()
1850 1852 revs = opts.get(b'rev')
1851 1853 if not revs:
1852 1854 revs = [b'all()']
1853 1855 revs = list(scmutil.revrange(repo, revs))
1854 1856
1855 1857 defaulttemplate = (
1856 1858 b'{date|shortdate} [{rev}:{node|short}]'
1857 1859 b' {author|person}: {desc|firstline}\n'
1858 1860 )
1859 1861 if testedtemplate is None:
1860 1862 testedtemplate = defaulttemplate
1861 1863 displayer = makelogtemplater(nullui, repo, testedtemplate)
1862 1864
1863 1865 def format():
1864 1866 for r in revs:
1865 1867 ctx = repo[r]
1866 1868 displayer.show(ctx)
1867 1869 displayer.flush(ctx)
1868 1870
1869 1871 timer, fm = gettimer(ui, opts)
1870 1872 timer(format)
1871 1873 fm.end()
1872 1874
1873 1875
1874 1876 def _displaystats(ui, opts, entries, data):
1875 1877 pass
1876 1878 # use a second formatter because the data are quite different, not sure
1877 1879 # how it flies with the templater.
1878 1880 fm = ui.formatter(b'perf-stats', opts)
1879 1881 for key, title in entries:
1880 1882 values = data[key]
1881 1883 nbvalues = len(data)
1882 1884 values.sort()
1883 1885 stats = {
1884 1886 'key': key,
1885 1887 'title': title,
1886 1888 'nbitems': len(values),
1887 1889 'min': values[0][0],
1888 1890 '10%': values[(nbvalues * 10) // 100][0],
1889 1891 '25%': values[(nbvalues * 25) // 100][0],
1890 1892 '50%': values[(nbvalues * 50) // 100][0],
1891 1893 '75%': values[(nbvalues * 75) // 100][0],
1892 1894 '80%': values[(nbvalues * 80) // 100][0],
1893 1895 '85%': values[(nbvalues * 85) // 100][0],
1894 1896 '90%': values[(nbvalues * 90) // 100][0],
1895 1897 '95%': values[(nbvalues * 95) // 100][0],
1896 1898 '99%': values[(nbvalues * 99) // 100][0],
1897 1899 'max': values[-1][0],
1898 1900 }
1899 1901 fm.startitem()
1900 1902 fm.data(**stats)
1901 1903 # make node pretty for the human output
1902 1904 fm.plain('### %s (%d items)\n' % (title, len(values)))
1903 1905 lines = [
1904 1906 'min',
1905 1907 '10%',
1906 1908 '25%',
1907 1909 '50%',
1908 1910 '75%',
1909 1911 '80%',
1910 1912 '85%',
1911 1913 '90%',
1912 1914 '95%',
1913 1915 '99%',
1914 1916 'max',
1915 1917 ]
1916 1918 for l in lines:
1917 1919 fm.plain('%s: %s\n' % (l, stats[l]))
1918 1920 fm.end()
1919 1921
1920 1922
1921 1923 @command(
1922 1924 b'perfhelper-mergecopies',
1923 1925 formatteropts
1924 1926 + [
1925 1927 (b'r', b'revs', [], b'restrict search to these revisions'),
1926 1928 (b'', b'timing', False, b'provides extra data (costly)'),
1927 1929 (b'', b'stats', False, b'provides statistic about the measured data'),
1928 1930 ],
1929 1931 )
1930 1932 def perfhelpermergecopies(ui, repo, revs=[], **opts):
1931 1933 """find statistics about potential parameters for `perfmergecopies`
1932 1934
1933 1935 This command find (base, p1, p2) triplet relevant for copytracing
1934 1936 benchmarking in the context of a merge. It reports values for some of the
1935 1937 parameters that impact merge copy tracing time during merge.
1936 1938
1937 1939 If `--timing` is set, rename detection is run and the associated timing
1938 1940 will be reported. The extra details come at the cost of slower command
1939 1941 execution.
1940 1942
1941 1943 Since rename detection is only run once, other factors might easily
1942 1944 affect the precision of the timing. However it should give a good
1943 1945 approximation of which revision triplets are very costly.
1944 1946 """
1945 1947 opts = _byteskwargs(opts)
1946 1948 fm = ui.formatter(b'perf', opts)
1947 1949 dotiming = opts[b'timing']
1948 1950 dostats = opts[b'stats']
1949 1951
1950 1952 output_template = [
1951 1953 ("base", "%(base)12s"),
1952 1954 ("p1", "%(p1.node)12s"),
1953 1955 ("p2", "%(p2.node)12s"),
1954 1956 ("p1.nb-revs", "%(p1.nbrevs)12d"),
1955 1957 ("p1.nb-files", "%(p1.nbmissingfiles)12d"),
1956 1958 ("p1.renames", "%(p1.renamedfiles)12d"),
1957 1959 ("p1.time", "%(p1.time)12.3f"),
1958 1960 ("p2.nb-revs", "%(p2.nbrevs)12d"),
1959 1961 ("p2.nb-files", "%(p2.nbmissingfiles)12d"),
1960 1962 ("p2.renames", "%(p2.renamedfiles)12d"),
1961 1963 ("p2.time", "%(p2.time)12.3f"),
1962 1964 ("renames", "%(nbrenamedfiles)12d"),
1963 1965 ("total.time", "%(time)12.3f"),
1964 1966 ]
1965 1967 if not dotiming:
1966 1968 output_template = [
1967 1969 i
1968 1970 for i in output_template
1969 1971 if not ('time' in i[0] or 'renames' in i[0])
1970 1972 ]
1971 1973 header_names = [h for (h, v) in output_template]
1972 1974 output = ' '.join([v for (h, v) in output_template]) + '\n'
1973 1975 header = ' '.join(['%12s'] * len(header_names)) + '\n'
1974 1976 fm.plain(header % tuple(header_names))
1975 1977
1976 1978 if not revs:
1977 1979 revs = ['all()']
1978 1980 revs = scmutil.revrange(repo, revs)
1979 1981
1980 1982 if dostats:
1981 1983 alldata = {
1982 1984 'nbrevs': [],
1983 1985 'nbmissingfiles': [],
1984 1986 }
1985 1987 if dotiming:
1986 1988 alldata['parentnbrenames'] = []
1987 1989 alldata['totalnbrenames'] = []
1988 1990 alldata['parenttime'] = []
1989 1991 alldata['totaltime'] = []
1990 1992
1991 1993 roi = repo.revs('merge() and %ld', revs)
1992 1994 for r in roi:
1993 1995 ctx = repo[r]
1994 1996 p1 = ctx.p1()
1995 1997 p2 = ctx.p2()
1996 1998 bases = repo.changelog._commonancestorsheads(p1.rev(), p2.rev())
1997 1999 for b in bases:
1998 2000 b = repo[b]
1999 2001 p1missing = copies._computeforwardmissing(b, p1)
2000 2002 p2missing = copies._computeforwardmissing(b, p2)
2001 2003 data = {
2002 2004 b'base': b.hex(),
2003 2005 b'p1.node': p1.hex(),
2004 2006 b'p1.nbrevs': len(repo.revs('only(%d, %d)', p1.rev(), b.rev())),
2005 2007 b'p1.nbmissingfiles': len(p1missing),
2006 2008 b'p2.node': p2.hex(),
2007 2009 b'p2.nbrevs': len(repo.revs('only(%d, %d)', p2.rev(), b.rev())),
2008 2010 b'p2.nbmissingfiles': len(p2missing),
2009 2011 }
2010 2012 if dostats:
2011 2013 if p1missing:
2012 2014 alldata['nbrevs'].append(
2013 2015 (data['p1.nbrevs'], b.hex(), p1.hex())
2014 2016 )
2015 2017 alldata['nbmissingfiles'].append(
2016 2018 (data['p1.nbmissingfiles'], b.hex(), p1.hex())
2017 2019 )
2018 2020 if p2missing:
2019 2021 alldata['nbrevs'].append(
2020 2022 (data['p2.nbrevs'], b.hex(), p2.hex())
2021 2023 )
2022 2024 alldata['nbmissingfiles'].append(
2023 2025 (data['p2.nbmissingfiles'], b.hex(), p2.hex())
2024 2026 )
2025 2027 if dotiming:
2026 2028 begin = util.timer()
2027 2029 mergedata = copies.mergecopies(repo, p1, p2, b)
2028 2030 end = util.timer()
2029 2031 # not very stable timing since we did only one run
2030 2032 data['time'] = end - begin
2031 2033 # mergedata contains five dicts: "copy", "movewithdir",
2032 2034 # "diverge", "renamedelete" and "dirmove".
2033 2035 # The first 4 are about renamed file so lets count that.
2034 2036 renames = len(mergedata[0])
2035 2037 renames += len(mergedata[1])
2036 2038 renames += len(mergedata[2])
2037 2039 renames += len(mergedata[3])
2038 2040 data['nbrenamedfiles'] = renames
2039 2041 begin = util.timer()
2040 2042 p1renames = copies.pathcopies(b, p1)
2041 2043 end = util.timer()
2042 2044 data['p1.time'] = end - begin
2043 2045 begin = util.timer()
2044 2046 p2renames = copies.pathcopies(b, p2)
2045 2047 data['p2.time'] = end - begin
2046 2048 end = util.timer()
2047 2049 data['p1.renamedfiles'] = len(p1renames)
2048 2050 data['p2.renamedfiles'] = len(p2renames)
2049 2051
2050 2052 if dostats:
2051 2053 if p1missing:
2052 2054 alldata['parentnbrenames'].append(
2053 2055 (data['p1.renamedfiles'], b.hex(), p1.hex())
2054 2056 )
2055 2057 alldata['parenttime'].append(
2056 2058 (data['p1.time'], b.hex(), p1.hex())
2057 2059 )
2058 2060 if p2missing:
2059 2061 alldata['parentnbrenames'].append(
2060 2062 (data['p2.renamedfiles'], b.hex(), p2.hex())
2061 2063 )
2062 2064 alldata['parenttime'].append(
2063 2065 (data['p2.time'], b.hex(), p2.hex())
2064 2066 )
2065 2067 if p1missing or p2missing:
2066 2068 alldata['totalnbrenames'].append(
2067 2069 (
2068 2070 data['nbrenamedfiles'],
2069 2071 b.hex(),
2070 2072 p1.hex(),
2071 2073 p2.hex(),
2072 2074 )
2073 2075 )
2074 2076 alldata['totaltime'].append(
2075 2077 (data['time'], b.hex(), p1.hex(), p2.hex())
2076 2078 )
2077 2079 fm.startitem()
2078 2080 fm.data(**data)
2079 2081 # make node pretty for the human output
2080 2082 out = data.copy()
2081 2083 out['base'] = fm.hexfunc(b.node())
2082 2084 out['p1.node'] = fm.hexfunc(p1.node())
2083 2085 out['p2.node'] = fm.hexfunc(p2.node())
2084 2086 fm.plain(output % out)
2085 2087
2086 2088 fm.end()
2087 2089 if dostats:
2088 2090 # use a second formatter because the data are quite different, not sure
2089 2091 # how it flies with the templater.
2090 2092 entries = [
2091 2093 ('nbrevs', 'number of revision covered'),
2092 2094 ('nbmissingfiles', 'number of missing files at head'),
2093 2095 ]
2094 2096 if dotiming:
2095 2097 entries.append(
2096 2098 ('parentnbrenames', 'rename from one parent to base')
2097 2099 )
2098 2100 entries.append(('totalnbrenames', 'total number of renames'))
2099 2101 entries.append(('parenttime', 'time for one parent'))
2100 2102 entries.append(('totaltime', 'time for both parents'))
2101 2103 _displaystats(ui, opts, entries, alldata)
2102 2104
2103 2105
2104 2106 @command(
2105 2107 b'perfhelper-pathcopies',
2106 2108 formatteropts
2107 2109 + [
2108 2110 (b'r', b'revs', [], b'restrict search to these revisions'),
2109 2111 (b'', b'timing', False, b'provides extra data (costly)'),
2110 2112 (b'', b'stats', False, b'provides statistic about the measured data'),
2111 2113 ],
2112 2114 )
2113 2115 def perfhelperpathcopies(ui, repo, revs=[], **opts):
2114 2116 """find statistic about potential parameters for the `perftracecopies`
2115 2117
2116 2118 This command find source-destination pair relevant for copytracing testing.
2117 2119 It report value for some of the parameters that impact copy tracing time.
2118 2120
2119 2121 If `--timing` is set, rename detection is run and the associated timing
2120 2122 will be reported. The extra details comes at the cost of a slower command
2121 2123 execution.
2122 2124
2123 2125 Since the rename detection is only run once, other factors might easily
2124 2126 affect the precision of the timing. However it should give a good
2125 2127 approximation of which revision pairs are very costly.
2126 2128 """
2127 2129 opts = _byteskwargs(opts)
2128 2130 fm = ui.formatter(b'perf', opts)
2129 2131 dotiming = opts[b'timing']
2130 2132 dostats = opts[b'stats']
2131 2133
2132 2134 if dotiming:
2133 2135 header = '%12s %12s %12s %12s %12s %12s\n'
2134 2136 output = (
2135 2137 "%(source)12s %(destination)12s "
2136 2138 "%(nbrevs)12d %(nbmissingfiles)12d "
2137 2139 "%(nbrenamedfiles)12d %(time)18.5f\n"
2138 2140 )
2139 2141 header_names = (
2140 2142 "source",
2141 2143 "destination",
2142 2144 "nb-revs",
2143 2145 "nb-files",
2144 2146 "nb-renames",
2145 2147 "time",
2146 2148 )
2147 2149 fm.plain(header % header_names)
2148 2150 else:
2149 2151 header = '%12s %12s %12s %12s\n'
2150 2152 output = (
2151 2153 "%(source)12s %(destination)12s "
2152 2154 "%(nbrevs)12d %(nbmissingfiles)12d\n"
2153 2155 )
2154 2156 fm.plain(header % ("source", "destination", "nb-revs", "nb-files"))
2155 2157
2156 2158 if not revs:
2157 2159 revs = ['all()']
2158 2160 revs = scmutil.revrange(repo, revs)
2159 2161
2160 2162 if dostats:
2161 2163 alldata = {
2162 2164 'nbrevs': [],
2163 2165 'nbmissingfiles': [],
2164 2166 }
2165 2167 if dotiming:
2166 2168 alldata['nbrenames'] = []
2167 2169 alldata['time'] = []
2168 2170
2169 2171 roi = repo.revs('merge() and %ld', revs)
2170 2172 for r in roi:
2171 2173 ctx = repo[r]
2172 2174 p1 = ctx.p1().rev()
2173 2175 p2 = ctx.p2().rev()
2174 2176 bases = repo.changelog._commonancestorsheads(p1, p2)
2175 2177 for p in (p1, p2):
2176 2178 for b in bases:
2177 2179 base = repo[b]
2178 2180 parent = repo[p]
2179 2181 missing = copies._computeforwardmissing(base, parent)
2180 2182 if not missing:
2181 2183 continue
2182 2184 data = {
2183 2185 b'source': base.hex(),
2184 2186 b'destination': parent.hex(),
2185 2187 b'nbrevs': len(repo.revs('only(%d, %d)', p, b)),
2186 2188 b'nbmissingfiles': len(missing),
2187 2189 }
2188 2190 if dostats:
2189 2191 alldata['nbrevs'].append(
2190 2192 (data['nbrevs'], base.hex(), parent.hex(),)
2191 2193 )
2192 2194 alldata['nbmissingfiles'].append(
2193 2195 (data['nbmissingfiles'], base.hex(), parent.hex(),)
2194 2196 )
2195 2197 if dotiming:
2196 2198 begin = util.timer()
2197 2199 renames = copies.pathcopies(base, parent)
2198 2200 end = util.timer()
2199 2201 # not very stable timing since we did only one run
2200 2202 data['time'] = end - begin
2201 2203 data['nbrenamedfiles'] = len(renames)
2202 2204 if dostats:
2203 2205 alldata['time'].append(
2204 2206 (data['time'], base.hex(), parent.hex(),)
2205 2207 )
2206 2208 alldata['nbrenames'].append(
2207 2209 (data['nbrenamedfiles'], base.hex(), parent.hex(),)
2208 2210 )
2209 2211 fm.startitem()
2210 2212 fm.data(**data)
2211 2213 out = data.copy()
2212 2214 out['source'] = fm.hexfunc(base.node())
2213 2215 out['destination'] = fm.hexfunc(parent.node())
2214 2216 fm.plain(output % out)
2215 2217
2216 2218 fm.end()
2217 2219 if dostats:
2218 2220 # use a second formatter because the data are quite different, not sure
2219 2221 # how it flies with the templater.
2220 2222 fm = ui.formatter(b'perf', opts)
2221 2223 entries = [
2222 2224 ('nbrevs', 'number of revision covered'),
2223 2225 ('nbmissingfiles', 'number of missing files at head'),
2224 2226 ]
2225 2227 if dotiming:
2226 2228 entries.append(('nbrenames', 'renamed files'))
2227 2229 entries.append(('time', 'time'))
2228 2230 _displaystats(ui, opts, entries, alldata)
2229 2231
2230 2232
2231 2233 @command(b'perfcca', formatteropts)
2232 2234 def perfcca(ui, repo, **opts):
2233 2235 opts = _byteskwargs(opts)
2234 2236 timer, fm = gettimer(ui, opts)
2235 2237 timer(lambda: scmutil.casecollisionauditor(ui, False, repo.dirstate))
2236 2238 fm.end()
2237 2239
2238 2240
2239 2241 @command(b'perffncacheload', formatteropts)
2240 2242 def perffncacheload(ui, repo, **opts):
2241 2243 opts = _byteskwargs(opts)
2242 2244 timer, fm = gettimer(ui, opts)
2243 2245 s = repo.store
2244 2246
2245 2247 def d():
2246 2248 s.fncache._load()
2247 2249
2248 2250 timer(d)
2249 2251 fm.end()
2250 2252
2251 2253
2252 2254 @command(b'perffncachewrite', formatteropts)
2253 2255 def perffncachewrite(ui, repo, **opts):
2254 2256 opts = _byteskwargs(opts)
2255 2257 timer, fm = gettimer(ui, opts)
2256 2258 s = repo.store
2257 2259 lock = repo.lock()
2258 2260 s.fncache._load()
2259 2261 tr = repo.transaction(b'perffncachewrite')
2260 2262 tr.addbackup(b'fncache')
2261 2263
2262 2264 def d():
2263 2265 s.fncache._dirty = True
2264 2266 s.fncache.write(tr)
2265 2267
2266 2268 timer(d)
2267 2269 tr.close()
2268 2270 lock.release()
2269 2271 fm.end()
2270 2272
2271 2273
2272 2274 @command(b'perffncacheencode', formatteropts)
2273 2275 def perffncacheencode(ui, repo, **opts):
2274 2276 opts = _byteskwargs(opts)
2275 2277 timer, fm = gettimer(ui, opts)
2276 2278 s = repo.store
2277 2279 s.fncache._load()
2278 2280
2279 2281 def d():
2280 2282 for p in s.fncache.entries:
2281 2283 s.encode(p)
2282 2284
2283 2285 timer(d)
2284 2286 fm.end()
2285 2287
2286 2288
2287 2289 def _bdiffworker(q, blocks, xdiff, ready, done):
2288 2290 while not done.is_set():
2289 2291 pair = q.get()
2290 2292 while pair is not None:
2291 2293 if xdiff:
2292 2294 mdiff.bdiff.xdiffblocks(*pair)
2293 2295 elif blocks:
2294 2296 mdiff.bdiff.blocks(*pair)
2295 2297 else:
2296 2298 mdiff.textdiff(*pair)
2297 2299 q.task_done()
2298 2300 pair = q.get()
2299 2301 q.task_done() # for the None one
2300 2302 with ready:
2301 2303 ready.wait()
2302 2304
2303 2305
2304 2306 def _manifestrevision(repo, mnode):
2305 2307 ml = repo.manifestlog
2306 2308
2307 2309 if util.safehasattr(ml, b'getstorage'):
2308 2310 store = ml.getstorage(b'')
2309 2311 else:
2310 2312 store = ml._revlog
2311 2313
2312 2314 return store.revision(mnode)
2313 2315
2314 2316
2315 2317 @command(
2316 2318 b'perfbdiff',
2317 2319 revlogopts
2318 2320 + formatteropts
2319 2321 + [
2320 2322 (
2321 2323 b'',
2322 2324 b'count',
2323 2325 1,
2324 2326 b'number of revisions to test (when using --startrev)',
2325 2327 ),
2326 2328 (b'', b'alldata', False, b'test bdiffs for all associated revisions'),
2327 2329 (b'', b'threads', 0, b'number of thread to use (disable with 0)'),
2328 2330 (b'', b'blocks', False, b'test computing diffs into blocks'),
2329 2331 (b'', b'xdiff', False, b'use xdiff algorithm'),
2330 2332 ],
2331 2333 b'-c|-m|FILE REV',
2332 2334 )
2333 2335 def perfbdiff(ui, repo, file_, rev=None, count=None, threads=0, **opts):
2334 2336 """benchmark a bdiff between revisions
2335 2337
2336 2338 By default, benchmark a bdiff between its delta parent and itself.
2337 2339
2338 2340 With ``--count``, benchmark bdiffs between delta parents and self for N
2339 2341 revisions starting at the specified revision.
2340 2342
2341 2343 With ``--alldata``, assume the requested revision is a changeset and
2342 2344 measure bdiffs for all changes related to that changeset (manifest
2343 2345 and filelogs).
2344 2346 """
2345 2347 opts = _byteskwargs(opts)
2346 2348
2347 2349 if opts[b'xdiff'] and not opts[b'blocks']:
2348 2350 raise error.CommandError(b'perfbdiff', b'--xdiff requires --blocks')
2349 2351
2350 2352 if opts[b'alldata']:
2351 2353 opts[b'changelog'] = True
2352 2354
2353 2355 if opts.get(b'changelog') or opts.get(b'manifest'):
2354 2356 file_, rev = None, file_
2355 2357 elif rev is None:
2356 2358 raise error.CommandError(b'perfbdiff', b'invalid arguments')
2357 2359
2358 2360 blocks = opts[b'blocks']
2359 2361 xdiff = opts[b'xdiff']
2360 2362 textpairs = []
2361 2363
2362 2364 r = cmdutil.openrevlog(repo, b'perfbdiff', file_, opts)
2363 2365
2364 2366 startrev = r.rev(r.lookup(rev))
2365 2367 for rev in range(startrev, min(startrev + count, len(r) - 1)):
2366 2368 if opts[b'alldata']:
2367 2369 # Load revisions associated with changeset.
2368 2370 ctx = repo[rev]
2369 2371 mtext = _manifestrevision(repo, ctx.manifestnode())
2370 2372 for pctx in ctx.parents():
2371 2373 pman = _manifestrevision(repo, pctx.manifestnode())
2372 2374 textpairs.append((pman, mtext))
2373 2375
2374 2376 # Load filelog revisions by iterating manifest delta.
2375 2377 man = ctx.manifest()
2376 2378 pman = ctx.p1().manifest()
2377 2379 for filename, change in pman.diff(man).items():
2378 2380 fctx = repo.file(filename)
2379 2381 f1 = fctx.revision(change[0][0] or -1)
2380 2382 f2 = fctx.revision(change[1][0] or -1)
2381 2383 textpairs.append((f1, f2))
2382 2384 else:
2383 2385 dp = r.deltaparent(rev)
2384 2386 textpairs.append((r.revision(dp), r.revision(rev)))
2385 2387
2386 2388 withthreads = threads > 0
2387 2389 if not withthreads:
2388 2390
2389 2391 def d():
2390 2392 for pair in textpairs:
2391 2393 if xdiff:
2392 2394 mdiff.bdiff.xdiffblocks(*pair)
2393 2395 elif blocks:
2394 2396 mdiff.bdiff.blocks(*pair)
2395 2397 else:
2396 2398 mdiff.textdiff(*pair)
2397 2399
2398 2400 else:
2399 2401 q = queue()
2400 2402 for i in _xrange(threads):
2401 2403 q.put(None)
2402 2404 ready = threading.Condition()
2403 2405 done = threading.Event()
2404 2406 for i in _xrange(threads):
2405 2407 threading.Thread(
2406 2408 target=_bdiffworker, args=(q, blocks, xdiff, ready, done)
2407 2409 ).start()
2408 2410 q.join()
2409 2411
2410 2412 def d():
2411 2413 for pair in textpairs:
2412 2414 q.put(pair)
2413 2415 for i in _xrange(threads):
2414 2416 q.put(None)
2415 2417 with ready:
2416 2418 ready.notify_all()
2417 2419 q.join()
2418 2420
2419 2421 timer, fm = gettimer(ui, opts)
2420 2422 timer(d)
2421 2423 fm.end()
2422 2424
2423 2425 if withthreads:
2424 2426 done.set()
2425 2427 for i in _xrange(threads):
2426 2428 q.put(None)
2427 2429 with ready:
2428 2430 ready.notify_all()
2429 2431
2430 2432
2431 2433 @command(
2432 2434 b'perfunidiff',
2433 2435 revlogopts
2434 2436 + formatteropts
2435 2437 + [
2436 2438 (
2437 2439 b'',
2438 2440 b'count',
2439 2441 1,
2440 2442 b'number of revisions to test (when using --startrev)',
2441 2443 ),
2442 2444 (b'', b'alldata', False, b'test unidiffs for all associated revisions'),
2443 2445 ],
2444 2446 b'-c|-m|FILE REV',
2445 2447 )
2446 2448 def perfunidiff(ui, repo, file_, rev=None, count=None, **opts):
2447 2449 """benchmark a unified diff between revisions
2448 2450
2449 2451 This doesn't include any copy tracing - it's just a unified diff
2450 2452 of the texts.
2451 2453
2452 2454 By default, benchmark a diff between its delta parent and itself.
2453 2455
2454 2456 With ``--count``, benchmark diffs between delta parents and self for N
2455 2457 revisions starting at the specified revision.
2456 2458
2457 2459 With ``--alldata``, assume the requested revision is a changeset and
2458 2460 measure diffs for all changes related to that changeset (manifest
2459 2461 and filelogs).
2460 2462 """
2461 2463 opts = _byteskwargs(opts)
2462 2464 if opts[b'alldata']:
2463 2465 opts[b'changelog'] = True
2464 2466
2465 2467 if opts.get(b'changelog') or opts.get(b'manifest'):
2466 2468 file_, rev = None, file_
2467 2469 elif rev is None:
2468 2470 raise error.CommandError(b'perfunidiff', b'invalid arguments')
2469 2471
2470 2472 textpairs = []
2471 2473
2472 2474 r = cmdutil.openrevlog(repo, b'perfunidiff', file_, opts)
2473 2475
2474 2476 startrev = r.rev(r.lookup(rev))
2475 2477 for rev in range(startrev, min(startrev + count, len(r) - 1)):
2476 2478 if opts[b'alldata']:
2477 2479 # Load revisions associated with changeset.
2478 2480 ctx = repo[rev]
2479 2481 mtext = _manifestrevision(repo, ctx.manifestnode())
2480 2482 for pctx in ctx.parents():
2481 2483 pman = _manifestrevision(repo, pctx.manifestnode())
2482 2484 textpairs.append((pman, mtext))
2483 2485
2484 2486 # Load filelog revisions by iterating manifest delta.
2485 2487 man = ctx.manifest()
2486 2488 pman = ctx.p1().manifest()
2487 2489 for filename, change in pman.diff(man).items():
2488 2490 fctx = repo.file(filename)
2489 2491 f1 = fctx.revision(change[0][0] or -1)
2490 2492 f2 = fctx.revision(change[1][0] or -1)
2491 2493 textpairs.append((f1, f2))
2492 2494 else:
2493 2495 dp = r.deltaparent(rev)
2494 2496 textpairs.append((r.revision(dp), r.revision(rev)))
2495 2497
2496 2498 def d():
2497 2499 for left, right in textpairs:
2498 2500 # The date strings don't matter, so we pass empty strings.
2499 2501 headerlines, hunks = mdiff.unidiff(
2500 2502 left, b'', right, b'', b'left', b'right', binary=False
2501 2503 )
2502 2504 # consume iterators in roughly the way patch.py does
2503 2505 b'\n'.join(headerlines)
2504 2506 b''.join(sum((list(hlines) for hrange, hlines in hunks), []))
2505 2507
2506 2508 timer, fm = gettimer(ui, opts)
2507 2509 timer(d)
2508 2510 fm.end()
2509 2511
2510 2512
2511 2513 @command(b'perfdiffwd', formatteropts)
2512 2514 def perfdiffwd(ui, repo, **opts):
2513 2515 """Profile diff of working directory changes"""
2514 2516 opts = _byteskwargs(opts)
2515 2517 timer, fm = gettimer(ui, opts)
2516 2518 options = {
2517 2519 'w': 'ignore_all_space',
2518 2520 'b': 'ignore_space_change',
2519 2521 'B': 'ignore_blank_lines',
2520 2522 }
2521 2523
2522 2524 for diffopt in ('', 'w', 'b', 'B', 'wB'):
2523 2525 opts = dict((options[c], b'1') for c in diffopt)
2524 2526
2525 2527 def d():
2526 2528 ui.pushbuffer()
2527 2529 commands.diff(ui, repo, **opts)
2528 2530 ui.popbuffer()
2529 2531
2530 2532 diffopt = diffopt.encode('ascii')
2531 2533 title = b'diffopts: %s' % (diffopt and (b'-' + diffopt) or b'none')
2532 2534 timer(d, title=title)
2533 2535 fm.end()
2534 2536
2535 2537
2536 2538 @command(b'perfrevlogindex', revlogopts + formatteropts, b'-c|-m|FILE')
2537 2539 def perfrevlogindex(ui, repo, file_=None, **opts):
2538 2540 """Benchmark operations against a revlog index.
2539 2541
2540 2542 This tests constructing a revlog instance, reading index data,
2541 2543 parsing index data, and performing various operations related to
2542 2544 index data.
2543 2545 """
2544 2546
2545 2547 opts = _byteskwargs(opts)
2546 2548
2547 2549 rl = cmdutil.openrevlog(repo, b'perfrevlogindex', file_, opts)
2548 2550
2549 2551 opener = getattr(rl, 'opener') # trick linter
2550 2552 indexfile = rl.indexfile
2551 2553 data = opener.read(indexfile)
2552 2554
2553 2555 header = struct.unpack(b'>I', data[0:4])[0]
2554 2556 version = header & 0xFFFF
2555 2557 if version == 1:
2556 2558 revlogio = revlog.revlogio()
2557 2559 inline = header & (1 << 16)
2558 2560 else:
2559 2561 raise error.Abort(b'unsupported revlog version: %d' % version)
2560 2562
2561 2563 rllen = len(rl)
2562 2564
2563 2565 node0 = rl.node(0)
2564 2566 node25 = rl.node(rllen // 4)
2565 2567 node50 = rl.node(rllen // 2)
2566 2568 node75 = rl.node(rllen // 4 * 3)
2567 2569 node100 = rl.node(rllen - 1)
2568 2570
2569 2571 allrevs = range(rllen)
2570 2572 allrevsrev = list(reversed(allrevs))
2571 2573 allnodes = [rl.node(rev) for rev in range(rllen)]
2572 2574 allnodesrev = list(reversed(allnodes))
2573 2575
2574 2576 def constructor():
2575 2577 revlog.revlog(opener, indexfile)
2576 2578
2577 2579 def read():
2578 2580 with opener(indexfile) as fh:
2579 2581 fh.read()
2580 2582
2581 2583 def parseindex():
2582 2584 revlogio.parseindex(data, inline)
2583 2585
2584 2586 def getentry(revornode):
2585 2587 index = revlogio.parseindex(data, inline)[0]
2586 2588 index[revornode]
2587 2589
2588 2590 def getentries(revs, count=1):
2589 2591 index = revlogio.parseindex(data, inline)[0]
2590 2592
2591 2593 for i in range(count):
2592 2594 for rev in revs:
2593 2595 index[rev]
2594 2596
2595 2597 def resolvenode(node):
2596 2598 nodemap = getattr(revlogio.parseindex(data, inline)[0], 'nodemap', None)
2597 2599 # This only works for the C code.
2598 2600 if nodemap is None:
2599 2601 return
2600 2602
2601 2603 try:
2602 2604 nodemap[node]
2603 2605 except error.RevlogError:
2604 2606 pass
2605 2607
2606 2608 def resolvenodes(nodes, count=1):
2607 2609 nodemap = getattr(revlogio.parseindex(data, inline)[0], 'nodemap', None)
2608 2610 if nodemap is None:
2609 2611 return
2610 2612
2611 2613 for i in range(count):
2612 2614 for node in nodes:
2613 2615 try:
2614 2616 nodemap[node]
2615 2617 except error.RevlogError:
2616 2618 pass
2617 2619
2618 2620 benches = [
2619 2621 (constructor, b'revlog constructor'),
2620 2622 (read, b'read'),
2621 2623 (parseindex, b'create index object'),
2622 2624 (lambda: getentry(0), b'retrieve index entry for rev 0'),
2623 2625 (lambda: resolvenode(b'a' * 20), b'look up missing node'),
2624 2626 (lambda: resolvenode(node0), b'look up node at rev 0'),
2625 2627 (lambda: resolvenode(node25), b'look up node at 1/4 len'),
2626 2628 (lambda: resolvenode(node50), b'look up node at 1/2 len'),
2627 2629 (lambda: resolvenode(node75), b'look up node at 3/4 len'),
2628 2630 (lambda: resolvenode(node100), b'look up node at tip'),
2629 2631 # 2x variation is to measure caching impact.
2630 2632 (lambda: resolvenodes(allnodes), b'look up all nodes (forward)'),
2631 2633 (lambda: resolvenodes(allnodes, 2), b'look up all nodes 2x (forward)'),
2632 2634 (lambda: resolvenodes(allnodesrev), b'look up all nodes (reverse)'),
2633 2635 (
2634 2636 lambda: resolvenodes(allnodesrev, 2),
2635 2637 b'look up all nodes 2x (reverse)',
2636 2638 ),
2637 2639 (lambda: getentries(allrevs), b'retrieve all index entries (forward)'),
2638 2640 (
2639 2641 lambda: getentries(allrevs, 2),
2640 2642 b'retrieve all index entries 2x (forward)',
2641 2643 ),
2642 2644 (
2643 2645 lambda: getentries(allrevsrev),
2644 2646 b'retrieve all index entries (reverse)',
2645 2647 ),
2646 2648 (
2647 2649 lambda: getentries(allrevsrev, 2),
2648 2650 b'retrieve all index entries 2x (reverse)',
2649 2651 ),
2650 2652 ]
2651 2653
2652 2654 for fn, title in benches:
2653 2655 timer, fm = gettimer(ui, opts)
2654 2656 timer(fn, title=title)
2655 2657 fm.end()
2656 2658
2657 2659
2658 2660 @command(
2659 2661 b'perfrevlogrevisions',
2660 2662 revlogopts
2661 2663 + formatteropts
2662 2664 + [
2663 2665 (b'd', b'dist', 100, b'distance between the revisions'),
2664 2666 (b's', b'startrev', 0, b'revision to start reading at'),
2665 2667 (b'', b'reverse', False, b'read in reverse'),
2666 2668 ],
2667 2669 b'-c|-m|FILE',
2668 2670 )
2669 2671 def perfrevlogrevisions(
2670 2672 ui, repo, file_=None, startrev=0, reverse=False, **opts
2671 2673 ):
2672 2674 """Benchmark reading a series of revisions from a revlog.
2673 2675
2674 2676 By default, we read every ``-d/--dist`` revision from 0 to tip of
2675 2677 the specified revlog.
2676 2678
2677 2679 The start revision can be defined via ``-s/--startrev``.
2678 2680 """
2679 2681 opts = _byteskwargs(opts)
2680 2682
2681 2683 rl = cmdutil.openrevlog(repo, b'perfrevlogrevisions', file_, opts)
2682 2684 rllen = getlen(ui)(rl)
2683 2685
2684 2686 if startrev < 0:
2685 2687 startrev = rllen + startrev
2686 2688
2687 2689 def d():
2688 2690 rl.clearcaches()
2689 2691
2690 2692 beginrev = startrev
2691 2693 endrev = rllen
2692 2694 dist = opts[b'dist']
2693 2695
2694 2696 if reverse:
2695 2697 beginrev, endrev = endrev - 1, beginrev - 1
2696 2698 dist = -1 * dist
2697 2699
2698 2700 for x in _xrange(beginrev, endrev, dist):
2699 2701 # Old revisions don't support passing int.
2700 2702 n = rl.node(x)
2701 2703 rl.revision(n)
2702 2704
2703 2705 timer, fm = gettimer(ui, opts)
2704 2706 timer(d)
2705 2707 fm.end()
2706 2708
2707 2709
2708 2710 @command(
2709 2711 b'perfrevlogwrite',
2710 2712 revlogopts
2711 2713 + formatteropts
2712 2714 + [
2713 2715 (b's', b'startrev', 1000, b'revision to start writing at'),
2714 2716 (b'', b'stoprev', -1, b'last revision to write'),
2715 2717 (b'', b'count', 3, b'number of passes to perform'),
2716 2718 (b'', b'details', False, b'print timing for every revisions tested'),
2717 2719 (b'', b'source', b'full', b'the kind of data feed in the revlog'),
2718 2720 (b'', b'lazydeltabase', True, b'try the provided delta first'),
2719 2721 (b'', b'clear-caches', True, b'clear revlog cache between calls'),
2720 2722 ],
2721 2723 b'-c|-m|FILE',
2722 2724 )
2723 2725 def perfrevlogwrite(ui, repo, file_=None, startrev=1000, stoprev=-1, **opts):
2724 2726 """Benchmark writing a series of revisions to a revlog.
2725 2727
2726 2728 Possible source values are:
2727 2729 * `full`: add from a full text (default).
2728 2730 * `parent-1`: add from a delta to the first parent
2729 2731 * `parent-2`: add from a delta to the second parent if it exists
2730 2732 (use a delta from the first parent otherwise)
2731 2733 * `parent-smallest`: add from the smallest delta (either p1 or p2)
2732 2734 * `storage`: add from the existing precomputed deltas
2733 2735
2734 2736 Note: This performance command measures performance in a custom way. As a
2735 2737 result some of the global configuration of the 'perf' command does not
2736 2738 apply to it:
2737 2739
2738 2740 * ``pre-run``: disabled
2739 2741
2740 2742 * ``profile-benchmark``: disabled
2741 2743
2742 2744 * ``run-limits``: disabled use --count instead
2743 2745 """
2744 2746 opts = _byteskwargs(opts)
2745 2747
2746 2748 rl = cmdutil.openrevlog(repo, b'perfrevlogwrite', file_, opts)
2747 2749 rllen = getlen(ui)(rl)
2748 2750 if startrev < 0:
2749 2751 startrev = rllen + startrev
2750 2752 if stoprev < 0:
2751 2753 stoprev = rllen + stoprev
2752 2754
2753 2755 lazydeltabase = opts['lazydeltabase']
2754 2756 source = opts['source']
2755 2757 clearcaches = opts['clear_caches']
2756 2758 validsource = (
2757 2759 b'full',
2758 2760 b'parent-1',
2759 2761 b'parent-2',
2760 2762 b'parent-smallest',
2761 2763 b'storage',
2762 2764 )
2763 2765 if source not in validsource:
2764 2766 raise error.Abort('invalid source type: %s' % source)
2765 2767
2766 2768 ### actually gather results
2767 2769 count = opts['count']
2768 2770 if count <= 0:
2769 2771 raise error.Abort('invalide run count: %d' % count)
2770 2772 allresults = []
2771 2773 for c in range(count):
2772 2774 timing = _timeonewrite(
2773 2775 ui,
2774 2776 rl,
2775 2777 source,
2776 2778 startrev,
2777 2779 stoprev,
2778 2780 c + 1,
2779 2781 lazydeltabase=lazydeltabase,
2780 2782 clearcaches=clearcaches,
2781 2783 )
2782 2784 allresults.append(timing)
2783 2785
2784 2786 ### consolidate the results in a single list
2785 2787 results = []
2786 2788 for idx, (rev, t) in enumerate(allresults[0]):
2787 2789 ts = [t]
2788 2790 for other in allresults[1:]:
2789 2791 orev, ot = other[idx]
2790 2792 assert orev == rev
2791 2793 ts.append(ot)
2792 2794 results.append((rev, ts))
2793 2795 resultcount = len(results)
2794 2796
2795 2797 ### Compute and display relevant statistics
2796 2798
2797 2799 # get a formatter
2798 2800 fm = ui.formatter(b'perf', opts)
2799 2801 displayall = ui.configbool(b"perf", b"all-timing", False)
2800 2802
2801 2803 # print individual details if requested
2802 2804 if opts['details']:
2803 2805 for idx, item in enumerate(results, 1):
2804 2806 rev, data = item
2805 2807 title = 'revisions #%d of %d, rev %d' % (idx, resultcount, rev)
2806 2808 formatone(fm, data, title=title, displayall=displayall)
2807 2809
2808 2810 # sorts results by median time
2809 2811 results.sort(key=lambda x: sorted(x[1])[len(x[1]) // 2])
2810 2812 # list of (name, index) to display)
2811 2813 relevants = [
2812 2814 ("min", 0),
2813 2815 ("10%", resultcount * 10 // 100),
2814 2816 ("25%", resultcount * 25 // 100),
2815 2817 ("50%", resultcount * 70 // 100),
2816 2818 ("75%", resultcount * 75 // 100),
2817 2819 ("90%", resultcount * 90 // 100),
2818 2820 ("95%", resultcount * 95 // 100),
2819 2821 ("99%", resultcount * 99 // 100),
2820 2822 ("99.9%", resultcount * 999 // 1000),
2821 2823 ("99.99%", resultcount * 9999 // 10000),
2822 2824 ("99.999%", resultcount * 99999 // 100000),
2823 2825 ("max", -1),
2824 2826 ]
2825 2827 if not ui.quiet:
2826 2828 for name, idx in relevants:
2827 2829 data = results[idx]
2828 2830 title = '%s of %d, rev %d' % (name, resultcount, data[0])
2829 2831 formatone(fm, data[1], title=title, displayall=displayall)
2830 2832
2831 2833 # XXX summing that many float will not be very precise, we ignore this fact
2832 2834 # for now
2833 2835 totaltime = []
2834 2836 for item in allresults:
2835 2837 totaltime.append(
2836 2838 (
2837 2839 sum(x[1][0] for x in item),
2838 2840 sum(x[1][1] for x in item),
2839 2841 sum(x[1][2] for x in item),
2840 2842 )
2841 2843 )
2842 2844 formatone(
2843 2845 fm,
2844 2846 totaltime,
2845 2847 title="total time (%d revs)" % resultcount,
2846 2848 displayall=displayall,
2847 2849 )
2848 2850 fm.end()
2849 2851
2850 2852
2851 2853 class _faketr(object):
2852 2854 def add(s, x, y, z=None):
2853 2855 return None
2854 2856
2855 2857
2856 2858 def _timeonewrite(
2857 2859 ui,
2858 2860 orig,
2859 2861 source,
2860 2862 startrev,
2861 2863 stoprev,
2862 2864 runidx=None,
2863 2865 lazydeltabase=True,
2864 2866 clearcaches=True,
2865 2867 ):
2866 2868 timings = []
2867 2869 tr = _faketr()
2868 2870 with _temprevlog(ui, orig, startrev) as dest:
2869 2871 dest._lazydeltabase = lazydeltabase
2870 2872 revs = list(orig.revs(startrev, stoprev))
2871 2873 total = len(revs)
2872 2874 topic = 'adding'
2873 2875 if runidx is not None:
2874 2876 topic += ' (run #%d)' % runidx
2875 2877 # Support both old and new progress API
2876 2878 if util.safehasattr(ui, 'makeprogress'):
2877 2879 progress = ui.makeprogress(topic, unit='revs', total=total)
2878 2880
2879 2881 def updateprogress(pos):
2880 2882 progress.update(pos)
2881 2883
2882 2884 def completeprogress():
2883 2885 progress.complete()
2884 2886
2885 2887 else:
2886 2888
2887 2889 def updateprogress(pos):
2888 2890 ui.progress(topic, pos, unit='revs', total=total)
2889 2891
2890 2892 def completeprogress():
2891 2893 ui.progress(topic, None, unit='revs', total=total)
2892 2894
2893 2895 for idx, rev in enumerate(revs):
2894 2896 updateprogress(idx)
2895 2897 addargs, addkwargs = _getrevisionseed(orig, rev, tr, source)
2896 2898 if clearcaches:
2897 2899 dest.index.clearcaches()
2898 2900 dest.clearcaches()
2899 2901 with timeone() as r:
2900 2902 dest.addrawrevision(*addargs, **addkwargs)
2901 2903 timings.append((rev, r[0]))
2902 2904 updateprogress(total)
2903 2905 completeprogress()
2904 2906 return timings
2905 2907
2906 2908
2907 2909 def _getrevisionseed(orig, rev, tr, source):
2908 2910 from mercurial.node import nullid
2909 2911
2910 2912 linkrev = orig.linkrev(rev)
2911 2913 node = orig.node(rev)
2912 2914 p1, p2 = orig.parents(node)
2913 2915 flags = orig.flags(rev)
2914 2916 cachedelta = None
2915 2917 text = None
2916 2918
2917 2919 if source == b'full':
2918 2920 text = orig.revision(rev)
2919 2921 elif source == b'parent-1':
2920 2922 baserev = orig.rev(p1)
2921 2923 cachedelta = (baserev, orig.revdiff(p1, rev))
2922 2924 elif source == b'parent-2':
2923 2925 parent = p2
2924 2926 if p2 == nullid:
2925 2927 parent = p1
2926 2928 baserev = orig.rev(parent)
2927 2929 cachedelta = (baserev, orig.revdiff(parent, rev))
2928 2930 elif source == b'parent-smallest':
2929 2931 p1diff = orig.revdiff(p1, rev)
2930 2932 parent = p1
2931 2933 diff = p1diff
2932 2934 if p2 != nullid:
2933 2935 p2diff = orig.revdiff(p2, rev)
2934 2936 if len(p1diff) > len(p2diff):
2935 2937 parent = p2
2936 2938 diff = p2diff
2937 2939 baserev = orig.rev(parent)
2938 2940 cachedelta = (baserev, diff)
2939 2941 elif source == b'storage':
2940 2942 baserev = orig.deltaparent(rev)
2941 2943 cachedelta = (baserev, orig.revdiff(orig.node(baserev), rev))
2942 2944
2943 2945 return (
2944 2946 (text, tr, linkrev, p1, p2),
2945 2947 {'node': node, 'flags': flags, 'cachedelta': cachedelta},
2946 2948 )
2947 2949
2948 2950
2949 2951 @contextlib.contextmanager
2950 2952 def _temprevlog(ui, orig, truncaterev):
2951 2953 from mercurial import vfs as vfsmod
2952 2954
2953 2955 if orig._inline:
2954 2956 raise error.Abort('not supporting inline revlog (yet)')
2955 2957 revlogkwargs = {}
2956 2958 k = 'upperboundcomp'
2957 2959 if util.safehasattr(orig, k):
2958 2960 revlogkwargs[k] = getattr(orig, k)
2959 2961
2960 2962 origindexpath = orig.opener.join(orig.indexfile)
2961 2963 origdatapath = orig.opener.join(orig.datafile)
2962 2964 indexname = 'revlog.i'
2963 2965 dataname = 'revlog.d'
2964 2966
2965 2967 tmpdir = tempfile.mkdtemp(prefix='tmp-hgperf-')
2966 2968 try:
2967 2969 # copy the data file in a temporary directory
2968 2970 ui.debug('copying data in %s\n' % tmpdir)
2969 2971 destindexpath = os.path.join(tmpdir, 'revlog.i')
2970 2972 destdatapath = os.path.join(tmpdir, 'revlog.d')
2971 2973 shutil.copyfile(origindexpath, destindexpath)
2972 2974 shutil.copyfile(origdatapath, destdatapath)
2973 2975
2974 2976 # remove the data we want to add again
2975 2977 ui.debug('truncating data to be rewritten\n')
2976 2978 with open(destindexpath, 'ab') as index:
2977 2979 index.seek(0)
2978 2980 index.truncate(truncaterev * orig._io.size)
2979 2981 with open(destdatapath, 'ab') as data:
2980 2982 data.seek(0)
2981 2983 data.truncate(orig.start(truncaterev))
2982 2984
2983 2985 # instantiate a new revlog from the temporary copy
2984 2986 ui.debug('truncating adding to be rewritten\n')
2985 2987 vfs = vfsmod.vfs(tmpdir)
2986 2988 vfs.options = getattr(orig.opener, 'options', None)
2987 2989
2988 2990 dest = revlog.revlog(
2989 2991 vfs, indexfile=indexname, datafile=dataname, **revlogkwargs
2990 2992 )
2991 2993 if dest._inline:
2992 2994 raise error.Abort('not supporting inline revlog (yet)')
2993 2995 # make sure internals are initialized
2994 2996 dest.revision(len(dest) - 1)
2995 2997 yield dest
2996 2998 del dest, vfs
2997 2999 finally:
2998 3000 shutil.rmtree(tmpdir, True)
2999 3001
3000 3002
3001 3003 @command(
3002 3004 b'perfrevlogchunks',
3003 3005 revlogopts
3004 3006 + formatteropts
3005 3007 + [
3006 3008 (b'e', b'engines', b'', b'compression engines to use'),
3007 3009 (b's', b'startrev', 0, b'revision to start at'),
3008 3010 ],
3009 3011 b'-c|-m|FILE',
3010 3012 )
3011 3013 def perfrevlogchunks(ui, repo, file_=None, engines=None, startrev=0, **opts):
3012 3014 """Benchmark operations on revlog chunks.
3013 3015
3014 3016 Logically, each revlog is a collection of fulltext revisions. However,
3015 3017 stored within each revlog are "chunks" of possibly compressed data. This
3016 3018 data needs to be read and decompressed or compressed and written.
3017 3019
3018 3020 This command measures the time it takes to read+decompress and recompress
3019 3021 chunks in a revlog. It effectively isolates I/O and compression performance.
3020 3022 For measurements of higher-level operations like resolving revisions,
3021 3023 see ``perfrevlogrevisions`` and ``perfrevlogrevision``.
3022 3024 """
3023 3025 opts = _byteskwargs(opts)
3024 3026
3025 3027 rl = cmdutil.openrevlog(repo, b'perfrevlogchunks', file_, opts)
3026 3028
3027 3029 # _chunkraw was renamed to _getsegmentforrevs.
3028 3030 try:
3029 3031 segmentforrevs = rl._getsegmentforrevs
3030 3032 except AttributeError:
3031 3033 segmentforrevs = rl._chunkraw
3032 3034
3033 3035 # Verify engines argument.
3034 3036 if engines:
3035 3037 engines = set(e.strip() for e in engines.split(b','))
3036 3038 for engine in engines:
3037 3039 try:
3038 3040 util.compressionengines[engine]
3039 3041 except KeyError:
3040 3042 raise error.Abort(b'unknown compression engine: %s' % engine)
3041 3043 else:
3042 3044 engines = []
3043 3045 for e in util.compengines:
3044 3046 engine = util.compengines[e]
3045 3047 try:
3046 3048 if engine.available():
3047 3049 engine.revlogcompressor().compress(b'dummy')
3048 3050 engines.append(e)
3049 3051 except NotImplementedError:
3050 3052 pass
3051 3053
3052 3054 revs = list(rl.revs(startrev, len(rl) - 1))
3053 3055
3054 3056 def rlfh(rl):
3055 3057 if rl._inline:
3056 3058 return getsvfs(repo)(rl.indexfile)
3057 3059 else:
3058 3060 return getsvfs(repo)(rl.datafile)
3059 3061
3060 3062 def doread():
3061 3063 rl.clearcaches()
3062 3064 for rev in revs:
3063 3065 segmentforrevs(rev, rev)
3064 3066
3065 3067 def doreadcachedfh():
3066 3068 rl.clearcaches()
3067 3069 fh = rlfh(rl)
3068 3070 for rev in revs:
3069 3071 segmentforrevs(rev, rev, df=fh)
3070 3072
3071 3073 def doreadbatch():
3072 3074 rl.clearcaches()
3073 3075 segmentforrevs(revs[0], revs[-1])
3074 3076
3075 3077 def doreadbatchcachedfh():
3076 3078 rl.clearcaches()
3077 3079 fh = rlfh(rl)
3078 3080 segmentforrevs(revs[0], revs[-1], df=fh)
3079 3081
3080 3082 def dochunk():
3081 3083 rl.clearcaches()
3082 3084 fh = rlfh(rl)
3083 3085 for rev in revs:
3084 3086 rl._chunk(rev, df=fh)
3085 3087
3086 3088 chunks = [None]
3087 3089
3088 3090 def dochunkbatch():
3089 3091 rl.clearcaches()
3090 3092 fh = rlfh(rl)
3091 3093 # Save chunks as a side-effect.
3092 3094 chunks[0] = rl._chunks(revs, df=fh)
3093 3095
3094 3096 def docompress(compressor):
3095 3097 rl.clearcaches()
3096 3098
3097 3099 try:
3098 3100 # Swap in the requested compression engine.
3099 3101 oldcompressor = rl._compressor
3100 3102 rl._compressor = compressor
3101 3103 for chunk in chunks[0]:
3102 3104 rl.compress(chunk)
3103 3105 finally:
3104 3106 rl._compressor = oldcompressor
3105 3107
3106 3108 benches = [
3107 3109 (lambda: doread(), b'read'),
3108 3110 (lambda: doreadcachedfh(), b'read w/ reused fd'),
3109 3111 (lambda: doreadbatch(), b'read batch'),
3110 3112 (lambda: doreadbatchcachedfh(), b'read batch w/ reused fd'),
3111 3113 (lambda: dochunk(), b'chunk'),
3112 3114 (lambda: dochunkbatch(), b'chunk batch'),
3113 3115 ]
3114 3116
3115 3117 for engine in sorted(engines):
3116 3118 compressor = util.compengines[engine].revlogcompressor()
3117 3119 benches.append(
3118 3120 (
3119 3121 functools.partial(docompress, compressor),
3120 3122 b'compress w/ %s' % engine,
3121 3123 )
3122 3124 )
3123 3125
3124 3126 for fn, title in benches:
3125 3127 timer, fm = gettimer(ui, opts)
3126 3128 timer(fn, title=title)
3127 3129 fm.end()
3128 3130
3129 3131
3130 3132 @command(
3131 3133 b'perfrevlogrevision',
3132 3134 revlogopts
3133 3135 + formatteropts
3134 3136 + [(b'', b'cache', False, b'use caches instead of clearing')],
3135 3137 b'-c|-m|FILE REV',
3136 3138 )
3137 3139 def perfrevlogrevision(ui, repo, file_, rev=None, cache=None, **opts):
3138 3140 """Benchmark obtaining a revlog revision.
3139 3141
3140 3142 Obtaining a revlog revision consists of roughly the following steps:
3141 3143
3142 3144 1. Compute the delta chain
3143 3145 2. Slice the delta chain if applicable
3144 3146 3. Obtain the raw chunks for that delta chain
3145 3147 4. Decompress each raw chunk
3146 3148 5. Apply binary patches to obtain fulltext
3147 3149 6. Verify hash of fulltext
3148 3150
3149 3151 This command measures the time spent in each of these phases.
3150 3152 """
3151 3153 opts = _byteskwargs(opts)
3152 3154
3153 3155 if opts.get(b'changelog') or opts.get(b'manifest'):
3154 3156 file_, rev = None, file_
3155 3157 elif rev is None:
3156 3158 raise error.CommandError(b'perfrevlogrevision', b'invalid arguments')
3157 3159
3158 3160 r = cmdutil.openrevlog(repo, b'perfrevlogrevision', file_, opts)
3159 3161
3160 3162 # _chunkraw was renamed to _getsegmentforrevs.
3161 3163 try:
3162 3164 segmentforrevs = r._getsegmentforrevs
3163 3165 except AttributeError:
3164 3166 segmentforrevs = r._chunkraw
3165 3167
3166 3168 node = r.lookup(rev)
3167 3169 rev = r.rev(node)
3168 3170
3169 3171 def getrawchunks(data, chain):
3170 3172 start = r.start
3171 3173 length = r.length
3172 3174 inline = r._inline
3173 3175 iosize = r._io.size
3174 3176 buffer = util.buffer
3175 3177
3176 3178 chunks = []
3177 3179 ladd = chunks.append
3178 3180 for idx, item in enumerate(chain):
3179 3181 offset = start(item[0])
3180 3182 bits = data[idx]
3181 3183 for rev in item:
3182 3184 chunkstart = start(rev)
3183 3185 if inline:
3184 3186 chunkstart += (rev + 1) * iosize
3185 3187 chunklength = length(rev)
3186 3188 ladd(buffer(bits, chunkstart - offset, chunklength))
3187 3189
3188 3190 return chunks
3189 3191
3190 3192 def dodeltachain(rev):
3191 3193 if not cache:
3192 3194 r.clearcaches()
3193 3195 r._deltachain(rev)
3194 3196
3195 3197 def doread(chain):
3196 3198 if not cache:
3197 3199 r.clearcaches()
3198 3200 for item in slicedchain:
3199 3201 segmentforrevs(item[0], item[-1])
3200 3202
3201 3203 def doslice(r, chain, size):
3202 3204 for s in slicechunk(r, chain, targetsize=size):
3203 3205 pass
3204 3206
3205 3207 def dorawchunks(data, chain):
3206 3208 if not cache:
3207 3209 r.clearcaches()
3208 3210 getrawchunks(data, chain)
3209 3211
3210 3212 def dodecompress(chunks):
3211 3213 decomp = r.decompress
3212 3214 for chunk in chunks:
3213 3215 decomp(chunk)
3214 3216
3215 3217 def dopatch(text, bins):
3216 3218 if not cache:
3217 3219 r.clearcaches()
3218 3220 mdiff.patches(text, bins)
3219 3221
3220 3222 def dohash(text):
3221 3223 if not cache:
3222 3224 r.clearcaches()
3223 3225 r.checkhash(text, node, rev=rev)
3224 3226
3225 3227 def dorevision():
3226 3228 if not cache:
3227 3229 r.clearcaches()
3228 3230 r.revision(node)
3229 3231
3230 3232 try:
3231 3233 from mercurial.revlogutils.deltas import slicechunk
3232 3234 except ImportError:
3233 3235 slicechunk = getattr(revlog, '_slicechunk', None)
3234 3236
3235 3237 size = r.length(rev)
3236 3238 chain = r._deltachain(rev)[0]
3237 3239 if not getattr(r, '_withsparseread', False):
3238 3240 slicedchain = (chain,)
3239 3241 else:
3240 3242 slicedchain = tuple(slicechunk(r, chain, targetsize=size))
3241 3243 data = [segmentforrevs(seg[0], seg[-1])[1] for seg in slicedchain]
3242 3244 rawchunks = getrawchunks(data, slicedchain)
3243 3245 bins = r._chunks(chain)
3244 3246 text = bytes(bins[0])
3245 3247 bins = bins[1:]
3246 3248 text = mdiff.patches(text, bins)
3247 3249
3248 3250 benches = [
3249 3251 (lambda: dorevision(), b'full'),
3250 3252 (lambda: dodeltachain(rev), b'deltachain'),
3251 3253 (lambda: doread(chain), b'read'),
3252 3254 ]
3253 3255
3254 3256 if getattr(r, '_withsparseread', False):
3255 3257 slicing = (lambda: doslice(r, chain, size), b'slice-sparse-chain')
3256 3258 benches.append(slicing)
3257 3259
3258 3260 benches.extend(
3259 3261 [
3260 3262 (lambda: dorawchunks(data, slicedchain), b'rawchunks'),
3261 3263 (lambda: dodecompress(rawchunks), b'decompress'),
3262 3264 (lambda: dopatch(text, bins), b'patch'),
3263 3265 (lambda: dohash(text), b'hash'),
3264 3266 ]
3265 3267 )
3266 3268
3267 3269 timer, fm = gettimer(ui, opts)
3268 3270 for fn, title in benches:
3269 3271 timer(fn, title=title)
3270 3272 fm.end()
3271 3273
3272 3274
3273 3275 @command(
3274 3276 b'perfrevset',
3275 3277 [
3276 3278 (b'C', b'clear', False, b'clear volatile cache between each call.'),
3277 3279 (b'', b'contexts', False, b'obtain changectx for each revision'),
3278 3280 ]
3279 3281 + formatteropts,
3280 3282 b"REVSET",
3281 3283 )
3282 3284 def perfrevset(ui, repo, expr, clear=False, contexts=False, **opts):
3283 3285 """benchmark the execution time of a revset
3284 3286
3285 3287 Use the --clean option if need to evaluate the impact of build volatile
3286 3288 revisions set cache on the revset execution. Volatile cache hold filtered
3287 3289 and obsolete related cache."""
3288 3290 opts = _byteskwargs(opts)
3289 3291
3290 3292 timer, fm = gettimer(ui, opts)
3291 3293
3292 3294 def d():
3293 3295 if clear:
3294 3296 repo.invalidatevolatilesets()
3295 3297 if contexts:
3296 3298 for ctx in repo.set(expr):
3297 3299 pass
3298 3300 else:
3299 3301 for r in repo.revs(expr):
3300 3302 pass
3301 3303
3302 3304 timer(d)
3303 3305 fm.end()
3304 3306
3305 3307
3306 3308 @command(
3307 3309 b'perfvolatilesets',
3308 3310 [(b'', b'clear-obsstore', False, b'drop obsstore between each call.'),]
3309 3311 + formatteropts,
3310 3312 )
3311 3313 def perfvolatilesets(ui, repo, *names, **opts):
3312 3314 """benchmark the computation of various volatile set
3313 3315
3314 3316 Volatile set computes element related to filtering and obsolescence."""
3315 3317 opts = _byteskwargs(opts)
3316 3318 timer, fm = gettimer(ui, opts)
3317 3319 repo = repo.unfiltered()
3318 3320
3319 3321 def getobs(name):
3320 3322 def d():
3321 3323 repo.invalidatevolatilesets()
3322 3324 if opts[b'clear_obsstore']:
3323 3325 clearfilecache(repo, b'obsstore')
3324 3326 obsolete.getrevs(repo, name)
3325 3327
3326 3328 return d
3327 3329
3328 3330 allobs = sorted(obsolete.cachefuncs)
3329 3331 if names:
3330 3332 allobs = [n for n in allobs if n in names]
3331 3333
3332 3334 for name in allobs:
3333 3335 timer(getobs(name), title=name)
3334 3336
3335 3337 def getfiltered(name):
3336 3338 def d():
3337 3339 repo.invalidatevolatilesets()
3338 3340 if opts[b'clear_obsstore']:
3339 3341 clearfilecache(repo, b'obsstore')
3340 3342 repoview.filterrevs(repo, name)
3341 3343
3342 3344 return d
3343 3345
3344 3346 allfilter = sorted(repoview.filtertable)
3345 3347 if names:
3346 3348 allfilter = [n for n in allfilter if n in names]
3347 3349
3348 3350 for name in allfilter:
3349 3351 timer(getfiltered(name), title=name)
3350 3352 fm.end()
3351 3353
3352 3354
3353 3355 @command(
3354 3356 b'perfbranchmap',
3355 3357 [
3356 3358 (b'f', b'full', False, b'Includes build time of subset'),
3357 3359 (
3358 3360 b'',
3359 3361 b'clear-revbranch',
3360 3362 False,
3361 3363 b'purge the revbranch cache between computation',
3362 3364 ),
3363 3365 ]
3364 3366 + formatteropts,
3365 3367 )
3366 3368 def perfbranchmap(ui, repo, *filternames, **opts):
3367 3369 """benchmark the update of a branchmap
3368 3370
3369 3371 This benchmarks the full repo.branchmap() call with read and write disabled
3370 3372 """
3371 3373 opts = _byteskwargs(opts)
3372 3374 full = opts.get(b"full", False)
3373 3375 clear_revbranch = opts.get(b"clear_revbranch", False)
3374 3376 timer, fm = gettimer(ui, opts)
3375 3377
3376 3378 def getbranchmap(filtername):
3377 3379 """generate a benchmark function for the filtername"""
3378 3380 if filtername is None:
3379 3381 view = repo
3380 3382 else:
3381 3383 view = repo.filtered(filtername)
3382 3384 if util.safehasattr(view._branchcaches, '_per_filter'):
3383 3385 filtered = view._branchcaches._per_filter
3384 3386 else:
3385 3387 # older versions
3386 3388 filtered = view._branchcaches
3387 3389
3388 3390 def d():
3389 3391 if clear_revbranch:
3390 3392 repo.revbranchcache()._clear()
3391 3393 if full:
3392 3394 view._branchcaches.clear()
3393 3395 else:
3394 3396 filtered.pop(filtername, None)
3395 3397 view.branchmap()
3396 3398
3397 3399 return d
3398 3400
3399 3401 # add filter in smaller subset to bigger subset
3400 3402 possiblefilters = set(repoview.filtertable)
3401 3403 if filternames:
3402 3404 possiblefilters &= set(filternames)
3403 3405 subsettable = getbranchmapsubsettable()
3404 3406 allfilters = []
3405 3407 while possiblefilters:
3406 3408 for name in possiblefilters:
3407 3409 subset = subsettable.get(name)
3408 3410 if subset not in possiblefilters:
3409 3411 break
3410 3412 else:
3411 3413 assert False, b'subset cycle %s!' % possiblefilters
3412 3414 allfilters.append(name)
3413 3415 possiblefilters.remove(name)
3414 3416
3415 3417 # warm the cache
3416 3418 if not full:
3417 3419 for name in allfilters:
3418 3420 repo.filtered(name).branchmap()
3419 3421 if not filternames or b'unfiltered' in filternames:
3420 3422 # add unfiltered
3421 3423 allfilters.append(None)
3422 3424
3423 3425 if util.safehasattr(branchmap.branchcache, 'fromfile'):
3424 3426 branchcacheread = safeattrsetter(branchmap.branchcache, b'fromfile')
3425 3427 branchcacheread.set(classmethod(lambda *args: None))
3426 3428 else:
3427 3429 # older versions
3428 3430 branchcacheread = safeattrsetter(branchmap, b'read')
3429 3431 branchcacheread.set(lambda *args: None)
3430 3432 branchcachewrite = safeattrsetter(branchmap.branchcache, b'write')
3431 3433 branchcachewrite.set(lambda *args: None)
3432 3434 try:
3433 3435 for name in allfilters:
3434 3436 printname = name
3435 3437 if name is None:
3436 3438 printname = b'unfiltered'
3437 3439 timer(getbranchmap(name), title=str(printname))
3438 3440 finally:
3439 3441 branchcacheread.restore()
3440 3442 branchcachewrite.restore()
3441 3443 fm.end()
3442 3444
3443 3445
3444 3446 @command(
3445 3447 b'perfbranchmapupdate',
3446 3448 [
3447 3449 (b'', b'base', [], b'subset of revision to start from'),
3448 3450 (b'', b'target', [], b'subset of revision to end with'),
3449 3451 (b'', b'clear-caches', False, b'clear cache between each runs'),
3450 3452 ]
3451 3453 + formatteropts,
3452 3454 )
3453 3455 def perfbranchmapupdate(ui, repo, base=(), target=(), **opts):
3454 3456 """benchmark branchmap update from for <base> revs to <target> revs
3455 3457
3456 3458 If `--clear-caches` is passed, the following items will be reset before
3457 3459 each update:
3458 3460 * the changelog instance and associated indexes
3459 3461 * the rev-branch-cache instance
3460 3462
3461 3463 Examples:
3462 3464
3463 3465 # update for the one last revision
3464 3466 $ hg perfbranchmapupdate --base 'not tip' --target 'tip'
3465 3467
3466 3468 $ update for change coming with a new branch
3467 3469 $ hg perfbranchmapupdate --base 'stable' --target 'default'
3468 3470 """
3469 3471 from mercurial import branchmap
3470 3472 from mercurial import repoview
3471 3473
3472 3474 opts = _byteskwargs(opts)
3473 3475 timer, fm = gettimer(ui, opts)
3474 3476 clearcaches = opts[b'clear_caches']
3475 3477 unfi = repo.unfiltered()
3476 3478 x = [None] # used to pass data between closure
3477 3479
3478 3480 # we use a `list` here to avoid possible side effect from smartset
3479 3481 baserevs = list(scmutil.revrange(repo, base))
3480 3482 targetrevs = list(scmutil.revrange(repo, target))
3481 3483 if not baserevs:
3482 3484 raise error.Abort(b'no revisions selected for --base')
3483 3485 if not targetrevs:
3484 3486 raise error.Abort(b'no revisions selected for --target')
3485 3487
3486 3488 # make sure the target branchmap also contains the one in the base
3487 3489 targetrevs = list(set(baserevs) | set(targetrevs))
3488 3490 targetrevs.sort()
3489 3491
3490 3492 cl = repo.changelog
3491 3493 allbaserevs = list(cl.ancestors(baserevs, inclusive=True))
3492 3494 allbaserevs.sort()
3493 3495 alltargetrevs = frozenset(cl.ancestors(targetrevs, inclusive=True))
3494 3496
3495 3497 newrevs = list(alltargetrevs.difference(allbaserevs))
3496 3498 newrevs.sort()
3497 3499
3498 3500 allrevs = frozenset(unfi.changelog.revs())
3499 3501 basefilterrevs = frozenset(allrevs.difference(allbaserevs))
3500 3502 targetfilterrevs = frozenset(allrevs.difference(alltargetrevs))
3501 3503
3502 3504 def basefilter(repo, visibilityexceptions=None):
3503 3505 return basefilterrevs
3504 3506
3505 3507 def targetfilter(repo, visibilityexceptions=None):
3506 3508 return targetfilterrevs
3507 3509
3508 3510 msg = b'benchmark of branchmap with %d revisions with %d new ones\n'
3509 3511 ui.status(msg % (len(allbaserevs), len(newrevs)))
3510 3512 if targetfilterrevs:
3511 3513 msg = b'(%d revisions still filtered)\n'
3512 3514 ui.status(msg % len(targetfilterrevs))
3513 3515
3514 3516 try:
3515 3517 repoview.filtertable[b'__perf_branchmap_update_base'] = basefilter
3516 3518 repoview.filtertable[b'__perf_branchmap_update_target'] = targetfilter
3517 3519
3518 3520 baserepo = repo.filtered(b'__perf_branchmap_update_base')
3519 3521 targetrepo = repo.filtered(b'__perf_branchmap_update_target')
3520 3522
3521 3523 # try to find an existing branchmap to reuse
3522 3524 subsettable = getbranchmapsubsettable()
3523 3525 candidatefilter = subsettable.get(None)
3524 3526 while candidatefilter is not None:
3525 3527 candidatebm = repo.filtered(candidatefilter).branchmap()
3526 3528 if candidatebm.validfor(baserepo):
3527 3529 filtered = repoview.filterrevs(repo, candidatefilter)
3528 3530 missing = [r for r in allbaserevs if r in filtered]
3529 3531 base = candidatebm.copy()
3530 3532 base.update(baserepo, missing)
3531 3533 break
3532 3534 candidatefilter = subsettable.get(candidatefilter)
3533 3535 else:
3534 3536 # no suitable subset where found
3535 3537 base = branchmap.branchcache()
3536 3538 base.update(baserepo, allbaserevs)
3537 3539
3538 3540 def setup():
3539 3541 x[0] = base.copy()
3540 3542 if clearcaches:
3541 3543 unfi._revbranchcache = None
3542 3544 clearchangelog(repo)
3543 3545
3544 3546 def bench():
3545 3547 x[0].update(targetrepo, newrevs)
3546 3548
3547 3549 timer(bench, setup=setup)
3548 3550 fm.end()
3549 3551 finally:
3550 3552 repoview.filtertable.pop(b'__perf_branchmap_update_base', None)
3551 3553 repoview.filtertable.pop(b'__perf_branchmap_update_target', None)
3552 3554
3553 3555
3554 3556 @command(
3555 3557 b'perfbranchmapload',
3556 3558 [
3557 3559 (b'f', b'filter', b'', b'Specify repoview filter'),
3558 3560 (b'', b'list', False, b'List brachmap filter caches'),
3559 3561 (b'', b'clear-revlogs', False, b'refresh changelog and manifest'),
3560 3562 ]
3561 3563 + formatteropts,
3562 3564 )
3563 3565 def perfbranchmapload(ui, repo, filter=b'', list=False, **opts):
3564 3566 """benchmark reading the branchmap"""
3565 3567 opts = _byteskwargs(opts)
3566 3568 clearrevlogs = opts[b'clear_revlogs']
3567 3569
3568 3570 if list:
3569 3571 for name, kind, st in repo.cachevfs.readdir(stat=True):
3570 3572 if name.startswith(b'branch2'):
3571 3573 filtername = name.partition(b'-')[2] or b'unfiltered'
3572 3574 ui.status(
3573 3575 b'%s - %s\n' % (filtername, util.bytecount(st.st_size))
3574 3576 )
3575 3577 return
3576 3578 if not filter:
3577 3579 filter = None
3578 3580 subsettable = getbranchmapsubsettable()
3579 3581 if filter is None:
3580 3582 repo = repo.unfiltered()
3581 3583 else:
3582 3584 repo = repoview.repoview(repo, filter)
3583 3585
3584 3586 repo.branchmap() # make sure we have a relevant, up to date branchmap
3585 3587
3586 3588 try:
3587 3589 fromfile = branchmap.branchcache.fromfile
3588 3590 except AttributeError:
3589 3591 # older versions
3590 3592 fromfile = branchmap.read
3591 3593
3592 3594 currentfilter = filter
3593 3595 # try once without timer, the filter may not be cached
3594 3596 while fromfile(repo) is None:
3595 3597 currentfilter = subsettable.get(currentfilter)
3596 3598 if currentfilter is None:
3597 3599 raise error.Abort(
3598 3600 b'No branchmap cached for %s repo' % (filter or b'unfiltered')
3599 3601 )
3600 3602 repo = repo.filtered(currentfilter)
3601 3603 timer, fm = gettimer(ui, opts)
3602 3604
3603 3605 def setup():
3604 3606 if clearrevlogs:
3605 3607 clearchangelog(repo)
3606 3608
3607 3609 def bench():
3608 3610 fromfile(repo)
3609 3611
3610 3612 timer(bench, setup=setup)
3611 3613 fm.end()
3612 3614
3613 3615
3614 3616 @command(b'perfloadmarkers')
3615 3617 def perfloadmarkers(ui, repo):
3616 3618 """benchmark the time to parse the on-disk markers for a repo
3617 3619
3618 3620 Result is the number of markers in the repo."""
3619 3621 timer, fm = gettimer(ui)
3620 3622 svfs = getsvfs(repo)
3621 3623 timer(lambda: len(obsolete.obsstore(svfs)))
3622 3624 fm.end()
3623 3625
3624 3626
3625 3627 @command(
3626 3628 b'perflrucachedict',
3627 3629 formatteropts
3628 3630 + [
3629 3631 (b'', b'costlimit', 0, b'maximum total cost of items in cache'),
3630 3632 (b'', b'mincost', 0, b'smallest cost of items in cache'),
3631 3633 (b'', b'maxcost', 100, b'maximum cost of items in cache'),
3632 3634 (b'', b'size', 4, b'size of cache'),
3633 3635 (b'', b'gets', 10000, b'number of key lookups'),
3634 3636 (b'', b'sets', 10000, b'number of key sets'),
3635 3637 (b'', b'mixed', 10000, b'number of mixed mode operations'),
3636 3638 (
3637 3639 b'',
3638 3640 b'mixedgetfreq',
3639 3641 50,
3640 3642 b'frequency of get vs set ops in mixed mode',
3641 3643 ),
3642 3644 ],
3643 3645 norepo=True,
3644 3646 )
3645 3647 def perflrucache(
3646 3648 ui,
3647 3649 mincost=0,
3648 3650 maxcost=100,
3649 3651 costlimit=0,
3650 3652 size=4,
3651 3653 gets=10000,
3652 3654 sets=10000,
3653 3655 mixed=10000,
3654 3656 mixedgetfreq=50,
3655 3657 **opts
3656 3658 ):
3657 3659 opts = _byteskwargs(opts)
3658 3660
3659 3661 def doinit():
3660 3662 for i in _xrange(10000):
3661 3663 util.lrucachedict(size)
3662 3664
3663 3665 costrange = list(range(mincost, maxcost + 1))
3664 3666
3665 3667 values = []
3666 3668 for i in _xrange(size):
3667 3669 values.append(random.randint(0, _maxint))
3668 3670
3669 3671 # Get mode fills the cache and tests raw lookup performance with no
3670 3672 # eviction.
3671 3673 getseq = []
3672 3674 for i in _xrange(gets):
3673 3675 getseq.append(random.choice(values))
3674 3676
3675 3677 def dogets():
3676 3678 d = util.lrucachedict(size)
3677 3679 for v in values:
3678 3680 d[v] = v
3679 3681 for key in getseq:
3680 3682 value = d[key]
3681 3683 value # silence pyflakes warning
3682 3684
3683 3685 def dogetscost():
3684 3686 d = util.lrucachedict(size, maxcost=costlimit)
3685 3687 for i, v in enumerate(values):
3686 3688 d.insert(v, v, cost=costs[i])
3687 3689 for key in getseq:
3688 3690 try:
3689 3691 value = d[key]
3690 3692 value # silence pyflakes warning
3691 3693 except KeyError:
3692 3694 pass
3693 3695
3694 3696 # Set mode tests insertion speed with cache eviction.
3695 3697 setseq = []
3696 3698 costs = []
3697 3699 for i in _xrange(sets):
3698 3700 setseq.append(random.randint(0, _maxint))
3699 3701 costs.append(random.choice(costrange))
3700 3702
3701 3703 def doinserts():
3702 3704 d = util.lrucachedict(size)
3703 3705 for v in setseq:
3704 3706 d.insert(v, v)
3705 3707
3706 3708 def doinsertscost():
3707 3709 d = util.lrucachedict(size, maxcost=costlimit)
3708 3710 for i, v in enumerate(setseq):
3709 3711 d.insert(v, v, cost=costs[i])
3710 3712
3711 3713 def dosets():
3712 3714 d = util.lrucachedict(size)
3713 3715 for v in setseq:
3714 3716 d[v] = v
3715 3717
3716 3718 # Mixed mode randomly performs gets and sets with eviction.
3717 3719 mixedops = []
3718 3720 for i in _xrange(mixed):
3719 3721 r = random.randint(0, 100)
3720 3722 if r < mixedgetfreq:
3721 3723 op = 0
3722 3724 else:
3723 3725 op = 1
3724 3726
3725 3727 mixedops.append(
3726 3728 (op, random.randint(0, size * 2), random.choice(costrange))
3727 3729 )
3728 3730
3729 3731 def domixed():
3730 3732 d = util.lrucachedict(size)
3731 3733
3732 3734 for op, v, cost in mixedops:
3733 3735 if op == 0:
3734 3736 try:
3735 3737 d[v]
3736 3738 except KeyError:
3737 3739 pass
3738 3740 else:
3739 3741 d[v] = v
3740 3742
3741 3743 def domixedcost():
3742 3744 d = util.lrucachedict(size, maxcost=costlimit)
3743 3745
3744 3746 for op, v, cost in mixedops:
3745 3747 if op == 0:
3746 3748 try:
3747 3749 d[v]
3748 3750 except KeyError:
3749 3751 pass
3750 3752 else:
3751 3753 d.insert(v, v, cost=cost)
3752 3754
3753 3755 benches = [
3754 3756 (doinit, b'init'),
3755 3757 ]
3756 3758
3757 3759 if costlimit:
3758 3760 benches.extend(
3759 3761 [
3760 3762 (dogetscost, b'gets w/ cost limit'),
3761 3763 (doinsertscost, b'inserts w/ cost limit'),
3762 3764 (domixedcost, b'mixed w/ cost limit'),
3763 3765 ]
3764 3766 )
3765 3767 else:
3766 3768 benches.extend(
3767 3769 [
3768 3770 (dogets, b'gets'),
3769 3771 (doinserts, b'inserts'),
3770 3772 (dosets, b'sets'),
3771 3773 (domixed, b'mixed'),
3772 3774 ]
3773 3775 )
3774 3776
3775 3777 for fn, title in benches:
3776 3778 timer, fm = gettimer(ui, opts)
3777 3779 timer(fn, title=title)
3778 3780 fm.end()
3779 3781
3780 3782
3781 3783 @command(b'perfwrite', formatteropts)
3782 3784 def perfwrite(ui, repo, **opts):
3783 3785 """microbenchmark ui.write
3784 3786 """
3785 3787 opts = _byteskwargs(opts)
3786 3788
3787 3789 timer, fm = gettimer(ui, opts)
3788 3790
3789 3791 def write():
3790 3792 for i in range(100000):
3791 3793 ui.writenoi18n(b'Testing write performance\n')
3792 3794
3793 3795 timer(write)
3794 3796 fm.end()
3795 3797
3796 3798
3797 3799 def uisetup(ui):
3798 3800 if util.safehasattr(cmdutil, b'openrevlog') and not util.safehasattr(
3799 3801 commands, b'debugrevlogopts'
3800 3802 ):
3801 3803 # for "historical portability":
3802 3804 # In this case, Mercurial should be 1.9 (or a79fea6b3e77) -
3803 3805 # 3.7 (or 5606f7d0d063). Therefore, '--dir' option for
3804 3806 # openrevlog() should cause failure, because it has been
3805 3807 # available since 3.5 (or 49c583ca48c4).
3806 3808 def openrevlog(orig, repo, cmd, file_, opts):
3807 3809 if opts.get(b'dir') and not util.safehasattr(repo, b'dirlog'):
3808 3810 raise error.Abort(
3809 3811 b"This version doesn't support --dir option",
3810 3812 hint=b"use 3.5 or later",
3811 3813 )
3812 3814 return orig(repo, cmd, file_, opts)
3813 3815
3814 3816 extensions.wrapfunction(cmdutil, b'openrevlog', openrevlog)
3815 3817
3816 3818
3817 3819 @command(
3818 3820 b'perfprogress',
3819 3821 formatteropts
3820 3822 + [
3821 3823 (b'', b'topic', b'topic', b'topic for progress messages'),
3822 3824 (b'c', b'total', 1000000, b'total value we are progressing to'),
3823 3825 ],
3824 3826 norepo=True,
3825 3827 )
3826 3828 def perfprogress(ui, topic=None, total=None, **opts):
3827 3829 """printing of progress bars"""
3828 3830 opts = _byteskwargs(opts)
3829 3831
3830 3832 timer, fm = gettimer(ui, opts)
3831 3833
3832 3834 def doprogress():
3833 3835 with ui.makeprogress(topic, total=total) as progress:
3834 3836 for i in _xrange(total):
3835 3837 progress.increment()
3836 3838
3837 3839 timer(doprogress)
3838 3840 fm.end()
General Comments 0
You need to be logged in to leave comments. Login now