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