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