##// END OF EJS Templates
obsolete: rename _addprecursors into _addpredecessors...
Boris Feld -
r33698:32d4f815 default
parent child Browse files
Show More
@@ -1,1034 +1,1041 b''
1 # obsolete.py - obsolete markers handling
1 # obsolete.py - obsolete markers handling
2 #
2 #
3 # Copyright 2012 Pierre-Yves David <pierre-yves.david@ens-lyon.org>
3 # Copyright 2012 Pierre-Yves David <pierre-yves.david@ens-lyon.org>
4 # Logilab SA <contact@logilab.fr>
4 # Logilab SA <contact@logilab.fr>
5 #
5 #
6 # This software may be used and distributed according to the terms of the
6 # This software may be used and distributed according to the terms of the
7 # GNU General Public License version 2 or any later version.
7 # GNU General Public License version 2 or any later version.
8
8
9 """Obsolete marker handling
9 """Obsolete marker handling
10
10
11 An obsolete marker maps an old changeset to a list of new
11 An obsolete marker maps an old changeset to a list of new
12 changesets. If the list of new changesets is empty, the old changeset
12 changesets. If the list of new changesets is empty, the old changeset
13 is said to be "killed". Otherwise, the old changeset is being
13 is said to be "killed". Otherwise, the old changeset is being
14 "replaced" by the new changesets.
14 "replaced" by the new changesets.
15
15
16 Obsolete markers can be used to record and distribute changeset graph
16 Obsolete markers can be used to record and distribute changeset graph
17 transformations performed by history rewrite operations, and help
17 transformations performed by history rewrite operations, and help
18 building new tools to reconcile conflicting rewrite actions. To
18 building new tools to reconcile conflicting rewrite actions. To
19 facilitate conflict resolution, markers include various annotations
19 facilitate conflict resolution, markers include various annotations
20 besides old and news changeset identifiers, such as creation date or
20 besides old and news changeset identifiers, such as creation date or
21 author name.
21 author name.
22
22
23 The old obsoleted changeset is called a "precursor" and possible
23 The old obsoleted changeset is called a "precursor" and possible
24 replacements are called "successors". Markers that used changeset X as
24 replacements are called "successors". Markers that used changeset X as
25 a precursor are called "successor markers of X" because they hold
25 a precursor are called "successor markers of X" because they hold
26 information about the successors of X. Markers that use changeset Y as
26 information about the successors of X. Markers that use changeset Y as
27 a successors are call "precursor markers of Y" because they hold
27 a successors are call "precursor markers of Y" because they hold
28 information about the precursors of Y.
28 information about the precursors of Y.
29
29
30 Examples:
30 Examples:
31
31
32 - When changeset A is replaced by changeset A', one marker is stored:
32 - When changeset A is replaced by changeset A', one marker is stored:
33
33
34 (A, (A',))
34 (A, (A',))
35
35
36 - When changesets A and B are folded into a new changeset C, two markers are
36 - When changesets A and B are folded into a new changeset C, two markers are
37 stored:
37 stored:
38
38
39 (A, (C,)) and (B, (C,))
39 (A, (C,)) and (B, (C,))
40
40
41 - When changeset A is simply "pruned" from the graph, a marker is created:
41 - When changeset A is simply "pruned" from the graph, a marker is created:
42
42
43 (A, ())
43 (A, ())
44
44
45 - When changeset A is split into B and C, a single marker is used:
45 - When changeset A is split into B and C, a single marker is used:
46
46
47 (A, (B, C))
47 (A, (B, C))
48
48
49 We use a single marker to distinguish the "split" case from the "divergence"
49 We use a single marker to distinguish the "split" case from the "divergence"
50 case. If two independent operations rewrite the same changeset A in to A' and
50 case. If two independent operations rewrite the same changeset A in to A' and
51 A'', we have an error case: divergent rewriting. We can detect it because
51 A'', we have an error case: divergent rewriting. We can detect it because
52 two markers will be created independently:
52 two markers will be created independently:
53
53
54 (A, (B,)) and (A, (C,))
54 (A, (B,)) and (A, (C,))
55
55
56 Format
56 Format
57 ------
57 ------
58
58
59 Markers are stored in an append-only file stored in
59 Markers are stored in an append-only file stored in
60 '.hg/store/obsstore'.
60 '.hg/store/obsstore'.
61
61
62 The file starts with a version header:
62 The file starts with a version header:
63
63
64 - 1 unsigned byte: version number, starting at zero.
64 - 1 unsigned byte: version number, starting at zero.
65
65
66 The header is followed by the markers. Marker format depend of the version. See
66 The header is followed by the markers. Marker format depend of the version. See
67 comment associated with each format for details.
67 comment associated with each format for details.
68
68
69 """
69 """
70 from __future__ import absolute_import
70 from __future__ import absolute_import
71
71
72 import errno
72 import errno
73 import struct
73 import struct
74
74
75 from .i18n import _
75 from .i18n import _
76 from . import (
76 from . import (
77 error,
77 error,
78 node,
78 node,
79 obsutil,
79 obsutil,
80 phases,
80 phases,
81 policy,
81 policy,
82 util,
82 util,
83 )
83 )
84
84
85 parsers = policy.importmod(r'parsers')
85 parsers = policy.importmod(r'parsers')
86
86
87 _pack = struct.pack
87 _pack = struct.pack
88 _unpack = struct.unpack
88 _unpack = struct.unpack
89 _calcsize = struct.calcsize
89 _calcsize = struct.calcsize
90 propertycache = util.propertycache
90 propertycache = util.propertycache
91
91
92 # the obsolete feature is not mature enough to be enabled by default.
92 # the obsolete feature is not mature enough to be enabled by default.
93 # you have to rely on third party extension extension to enable this.
93 # you have to rely on third party extension extension to enable this.
94 _enabled = False
94 _enabled = False
95
95
96 # Options for obsolescence
96 # Options for obsolescence
97 createmarkersopt = 'createmarkers'
97 createmarkersopt = 'createmarkers'
98 allowunstableopt = 'allowunstable'
98 allowunstableopt = 'allowunstable'
99 exchangeopt = 'exchange'
99 exchangeopt = 'exchange'
100
100
101 def isenabled(repo, option):
101 def isenabled(repo, option):
102 """Returns True if the given repository has the given obsolete option
102 """Returns True if the given repository has the given obsolete option
103 enabled.
103 enabled.
104 """
104 """
105 result = set(repo.ui.configlist('experimental', 'evolution'))
105 result = set(repo.ui.configlist('experimental', 'evolution'))
106 if 'all' in result:
106 if 'all' in result:
107 return True
107 return True
108
108
109 # For migration purposes, temporarily return true if the config hasn't been
109 # For migration purposes, temporarily return true if the config hasn't been
110 # set but _enabled is true.
110 # set but _enabled is true.
111 if len(result) == 0 and _enabled:
111 if len(result) == 0 and _enabled:
112 return True
112 return True
113
113
114 # createmarkers must be enabled if other options are enabled
114 # createmarkers must be enabled if other options are enabled
115 if ((allowunstableopt in result or exchangeopt in result) and
115 if ((allowunstableopt in result or exchangeopt in result) and
116 not createmarkersopt in result):
116 not createmarkersopt in result):
117 raise error.Abort(_("'createmarkers' obsolete option must be enabled "
117 raise error.Abort(_("'createmarkers' obsolete option must be enabled "
118 "if other obsolete options are enabled"))
118 "if other obsolete options are enabled"))
119
119
120 return option in result
120 return option in result
121
121
122 ### obsolescence marker flag
122 ### obsolescence marker flag
123
123
124 ## bumpedfix flag
124 ## bumpedfix flag
125 #
125 #
126 # When a changeset A' succeed to a changeset A which became public, we call A'
126 # When a changeset A' succeed to a changeset A which became public, we call A'
127 # "bumped" because it's a successors of a public changesets
127 # "bumped" because it's a successors of a public changesets
128 #
128 #
129 # o A' (bumped)
129 # o A' (bumped)
130 # |`:
130 # |`:
131 # | o A
131 # | o A
132 # |/
132 # |/
133 # o Z
133 # o Z
134 #
134 #
135 # The way to solve this situation is to create a new changeset Ad as children
135 # The way to solve this situation is to create a new changeset Ad as children
136 # of A. This changeset have the same content than A'. So the diff from A to A'
136 # of A. This changeset have the same content than A'. So the diff from A to A'
137 # is the same than the diff from A to Ad. Ad is marked as a successors of A'
137 # is the same than the diff from A to Ad. Ad is marked as a successors of A'
138 #
138 #
139 # o Ad
139 # o Ad
140 # |`:
140 # |`:
141 # | x A'
141 # | x A'
142 # |'|
142 # |'|
143 # o | A
143 # o | A
144 # |/
144 # |/
145 # o Z
145 # o Z
146 #
146 #
147 # But by transitivity Ad is also a successors of A. To avoid having Ad marked
147 # But by transitivity Ad is also a successors of A. To avoid having Ad marked
148 # as bumped too, we add the `bumpedfix` flag to the marker. <A', (Ad,)>.
148 # as bumped too, we add the `bumpedfix` flag to the marker. <A', (Ad,)>.
149 # This flag mean that the successors express the changes between the public and
149 # This flag mean that the successors express the changes between the public and
150 # bumped version and fix the situation, breaking the transitivity of
150 # bumped version and fix the situation, breaking the transitivity of
151 # "bumped" here.
151 # "bumped" here.
152 bumpedfix = 1
152 bumpedfix = 1
153 usingsha256 = 2
153 usingsha256 = 2
154
154
155 ## Parsing and writing of version "0"
155 ## Parsing and writing of version "0"
156 #
156 #
157 # The header is followed by the markers. Each marker is made of:
157 # The header is followed by the markers. Each marker is made of:
158 #
158 #
159 # - 1 uint8 : number of new changesets "N", can be zero.
159 # - 1 uint8 : number of new changesets "N", can be zero.
160 #
160 #
161 # - 1 uint32: metadata size "M" in bytes.
161 # - 1 uint32: metadata size "M" in bytes.
162 #
162 #
163 # - 1 byte: a bit field. It is reserved for flags used in common
163 # - 1 byte: a bit field. It is reserved for flags used in common
164 # obsolete marker operations, to avoid repeated decoding of metadata
164 # obsolete marker operations, to avoid repeated decoding of metadata
165 # entries.
165 # entries.
166 #
166 #
167 # - 20 bytes: obsoleted changeset identifier.
167 # - 20 bytes: obsoleted changeset identifier.
168 #
168 #
169 # - N*20 bytes: new changesets identifiers.
169 # - N*20 bytes: new changesets identifiers.
170 #
170 #
171 # - M bytes: metadata as a sequence of nul-terminated strings. Each
171 # - M bytes: metadata as a sequence of nul-terminated strings. Each
172 # string contains a key and a value, separated by a colon ':', without
172 # string contains a key and a value, separated by a colon ':', without
173 # additional encoding. Keys cannot contain '\0' or ':' and values
173 # additional encoding. Keys cannot contain '\0' or ':' and values
174 # cannot contain '\0'.
174 # cannot contain '\0'.
175 _fm0version = 0
175 _fm0version = 0
176 _fm0fixed = '>BIB20s'
176 _fm0fixed = '>BIB20s'
177 _fm0node = '20s'
177 _fm0node = '20s'
178 _fm0fsize = _calcsize(_fm0fixed)
178 _fm0fsize = _calcsize(_fm0fixed)
179 _fm0fnodesize = _calcsize(_fm0node)
179 _fm0fnodesize = _calcsize(_fm0node)
180
180
181 def _fm0readmarkers(data, off, stop):
181 def _fm0readmarkers(data, off, stop):
182 # Loop on markers
182 # Loop on markers
183 while off < stop:
183 while off < stop:
184 # read fixed part
184 # read fixed part
185 cur = data[off:off + _fm0fsize]
185 cur = data[off:off + _fm0fsize]
186 off += _fm0fsize
186 off += _fm0fsize
187 numsuc, mdsize, flags, pre = _unpack(_fm0fixed, cur)
187 numsuc, mdsize, flags, pre = _unpack(_fm0fixed, cur)
188 # read replacement
188 # read replacement
189 sucs = ()
189 sucs = ()
190 if numsuc:
190 if numsuc:
191 s = (_fm0fnodesize * numsuc)
191 s = (_fm0fnodesize * numsuc)
192 cur = data[off:off + s]
192 cur = data[off:off + s]
193 sucs = _unpack(_fm0node * numsuc, cur)
193 sucs = _unpack(_fm0node * numsuc, cur)
194 off += s
194 off += s
195 # read metadata
195 # read metadata
196 # (metadata will be decoded on demand)
196 # (metadata will be decoded on demand)
197 metadata = data[off:off + mdsize]
197 metadata = data[off:off + mdsize]
198 if len(metadata) != mdsize:
198 if len(metadata) != mdsize:
199 raise error.Abort(_('parsing obsolete marker: metadata is too '
199 raise error.Abort(_('parsing obsolete marker: metadata is too '
200 'short, %d bytes expected, got %d')
200 'short, %d bytes expected, got %d')
201 % (mdsize, len(metadata)))
201 % (mdsize, len(metadata)))
202 off += mdsize
202 off += mdsize
203 metadata = _fm0decodemeta(metadata)
203 metadata = _fm0decodemeta(metadata)
204 try:
204 try:
205 when, offset = metadata.pop('date', '0 0').split(' ')
205 when, offset = metadata.pop('date', '0 0').split(' ')
206 date = float(when), int(offset)
206 date = float(when), int(offset)
207 except ValueError:
207 except ValueError:
208 date = (0., 0)
208 date = (0., 0)
209 parents = None
209 parents = None
210 if 'p2' in metadata:
210 if 'p2' in metadata:
211 parents = (metadata.pop('p1', None), metadata.pop('p2', None))
211 parents = (metadata.pop('p1', None), metadata.pop('p2', None))
212 elif 'p1' in metadata:
212 elif 'p1' in metadata:
213 parents = (metadata.pop('p1', None),)
213 parents = (metadata.pop('p1', None),)
214 elif 'p0' in metadata:
214 elif 'p0' in metadata:
215 parents = ()
215 parents = ()
216 if parents is not None:
216 if parents is not None:
217 try:
217 try:
218 parents = tuple(node.bin(p) for p in parents)
218 parents = tuple(node.bin(p) for p in parents)
219 # if parent content is not a nodeid, drop the data
219 # if parent content is not a nodeid, drop the data
220 for p in parents:
220 for p in parents:
221 if len(p) != 20:
221 if len(p) != 20:
222 parents = None
222 parents = None
223 break
223 break
224 except TypeError:
224 except TypeError:
225 # if content cannot be translated to nodeid drop the data.
225 # if content cannot be translated to nodeid drop the data.
226 parents = None
226 parents = None
227
227
228 metadata = tuple(sorted(metadata.iteritems()))
228 metadata = tuple(sorted(metadata.iteritems()))
229
229
230 yield (pre, sucs, flags, metadata, date, parents)
230 yield (pre, sucs, flags, metadata, date, parents)
231
231
232 def _fm0encodeonemarker(marker):
232 def _fm0encodeonemarker(marker):
233 pre, sucs, flags, metadata, date, parents = marker
233 pre, sucs, flags, metadata, date, parents = marker
234 if flags & usingsha256:
234 if flags & usingsha256:
235 raise error.Abort(_('cannot handle sha256 with old obsstore format'))
235 raise error.Abort(_('cannot handle sha256 with old obsstore format'))
236 metadata = dict(metadata)
236 metadata = dict(metadata)
237 time, tz = date
237 time, tz = date
238 metadata['date'] = '%r %i' % (time, tz)
238 metadata['date'] = '%r %i' % (time, tz)
239 if parents is not None:
239 if parents is not None:
240 if not parents:
240 if not parents:
241 # mark that we explicitly recorded no parents
241 # mark that we explicitly recorded no parents
242 metadata['p0'] = ''
242 metadata['p0'] = ''
243 for i, p in enumerate(parents, 1):
243 for i, p in enumerate(parents, 1):
244 metadata['p%i' % i] = node.hex(p)
244 metadata['p%i' % i] = node.hex(p)
245 metadata = _fm0encodemeta(metadata)
245 metadata = _fm0encodemeta(metadata)
246 numsuc = len(sucs)
246 numsuc = len(sucs)
247 format = _fm0fixed + (_fm0node * numsuc)
247 format = _fm0fixed + (_fm0node * numsuc)
248 data = [numsuc, len(metadata), flags, pre]
248 data = [numsuc, len(metadata), flags, pre]
249 data.extend(sucs)
249 data.extend(sucs)
250 return _pack(format, *data) + metadata
250 return _pack(format, *data) + metadata
251
251
252 def _fm0encodemeta(meta):
252 def _fm0encodemeta(meta):
253 """Return encoded metadata string to string mapping.
253 """Return encoded metadata string to string mapping.
254
254
255 Assume no ':' in key and no '\0' in both key and value."""
255 Assume no ':' in key and no '\0' in both key and value."""
256 for key, value in meta.iteritems():
256 for key, value in meta.iteritems():
257 if ':' in key or '\0' in key:
257 if ':' in key or '\0' in key:
258 raise ValueError("':' and '\0' are forbidden in metadata key'")
258 raise ValueError("':' and '\0' are forbidden in metadata key'")
259 if '\0' in value:
259 if '\0' in value:
260 raise ValueError("':' is forbidden in metadata value'")
260 raise ValueError("':' is forbidden in metadata value'")
261 return '\0'.join(['%s:%s' % (k, meta[k]) for k in sorted(meta)])
261 return '\0'.join(['%s:%s' % (k, meta[k]) for k in sorted(meta)])
262
262
263 def _fm0decodemeta(data):
263 def _fm0decodemeta(data):
264 """Return string to string dictionary from encoded version."""
264 """Return string to string dictionary from encoded version."""
265 d = {}
265 d = {}
266 for l in data.split('\0'):
266 for l in data.split('\0'):
267 if l:
267 if l:
268 key, value = l.split(':')
268 key, value = l.split(':')
269 d[key] = value
269 d[key] = value
270 return d
270 return d
271
271
272 ## Parsing and writing of version "1"
272 ## Parsing and writing of version "1"
273 #
273 #
274 # The header is followed by the markers. Each marker is made of:
274 # The header is followed by the markers. Each marker is made of:
275 #
275 #
276 # - uint32: total size of the marker (including this field)
276 # - uint32: total size of the marker (including this field)
277 #
277 #
278 # - float64: date in seconds since epoch
278 # - float64: date in seconds since epoch
279 #
279 #
280 # - int16: timezone offset in minutes
280 # - int16: timezone offset in minutes
281 #
281 #
282 # - uint16: a bit field. It is reserved for flags used in common
282 # - uint16: a bit field. It is reserved for flags used in common
283 # obsolete marker operations, to avoid repeated decoding of metadata
283 # obsolete marker operations, to avoid repeated decoding of metadata
284 # entries.
284 # entries.
285 #
285 #
286 # - uint8: number of successors "N", can be zero.
286 # - uint8: number of successors "N", can be zero.
287 #
287 #
288 # - uint8: number of parents "P", can be zero.
288 # - uint8: number of parents "P", can be zero.
289 #
289 #
290 # 0: parents data stored but no parent,
290 # 0: parents data stored but no parent,
291 # 1: one parent stored,
291 # 1: one parent stored,
292 # 2: two parents stored,
292 # 2: two parents stored,
293 # 3: no parent data stored
293 # 3: no parent data stored
294 #
294 #
295 # - uint8: number of metadata entries M
295 # - uint8: number of metadata entries M
296 #
296 #
297 # - 20 or 32 bytes: precursor changeset identifier.
297 # - 20 or 32 bytes: precursor changeset identifier.
298 #
298 #
299 # - N*(20 or 32) bytes: successors changesets identifiers.
299 # - N*(20 or 32) bytes: successors changesets identifiers.
300 #
300 #
301 # - P*(20 or 32) bytes: parents of the precursors changesets.
301 # - P*(20 or 32) bytes: parents of the precursors changesets.
302 #
302 #
303 # - M*(uint8, uint8): size of all metadata entries (key and value)
303 # - M*(uint8, uint8): size of all metadata entries (key and value)
304 #
304 #
305 # - remaining bytes: the metadata, each (key, value) pair after the other.
305 # - remaining bytes: the metadata, each (key, value) pair after the other.
306 _fm1version = 1
306 _fm1version = 1
307 _fm1fixed = '>IdhHBBB20s'
307 _fm1fixed = '>IdhHBBB20s'
308 _fm1nodesha1 = '20s'
308 _fm1nodesha1 = '20s'
309 _fm1nodesha256 = '32s'
309 _fm1nodesha256 = '32s'
310 _fm1nodesha1size = _calcsize(_fm1nodesha1)
310 _fm1nodesha1size = _calcsize(_fm1nodesha1)
311 _fm1nodesha256size = _calcsize(_fm1nodesha256)
311 _fm1nodesha256size = _calcsize(_fm1nodesha256)
312 _fm1fsize = _calcsize(_fm1fixed)
312 _fm1fsize = _calcsize(_fm1fixed)
313 _fm1parentnone = 3
313 _fm1parentnone = 3
314 _fm1parentshift = 14
314 _fm1parentshift = 14
315 _fm1parentmask = (_fm1parentnone << _fm1parentshift)
315 _fm1parentmask = (_fm1parentnone << _fm1parentshift)
316 _fm1metapair = 'BB'
316 _fm1metapair = 'BB'
317 _fm1metapairsize = _calcsize(_fm1metapair)
317 _fm1metapairsize = _calcsize(_fm1metapair)
318
318
319 def _fm1purereadmarkers(data, off, stop):
319 def _fm1purereadmarkers(data, off, stop):
320 # make some global constants local for performance
320 # make some global constants local for performance
321 noneflag = _fm1parentnone
321 noneflag = _fm1parentnone
322 sha2flag = usingsha256
322 sha2flag = usingsha256
323 sha1size = _fm1nodesha1size
323 sha1size = _fm1nodesha1size
324 sha2size = _fm1nodesha256size
324 sha2size = _fm1nodesha256size
325 sha1fmt = _fm1nodesha1
325 sha1fmt = _fm1nodesha1
326 sha2fmt = _fm1nodesha256
326 sha2fmt = _fm1nodesha256
327 metasize = _fm1metapairsize
327 metasize = _fm1metapairsize
328 metafmt = _fm1metapair
328 metafmt = _fm1metapair
329 fsize = _fm1fsize
329 fsize = _fm1fsize
330 unpack = _unpack
330 unpack = _unpack
331
331
332 # Loop on markers
332 # Loop on markers
333 ufixed = struct.Struct(_fm1fixed).unpack
333 ufixed = struct.Struct(_fm1fixed).unpack
334
334
335 while off < stop:
335 while off < stop:
336 # read fixed part
336 # read fixed part
337 o1 = off + fsize
337 o1 = off + fsize
338 t, secs, tz, flags, numsuc, numpar, nummeta, prec = ufixed(data[off:o1])
338 t, secs, tz, flags, numsuc, numpar, nummeta, prec = ufixed(data[off:o1])
339
339
340 if flags & sha2flag:
340 if flags & sha2flag:
341 # FIXME: prec was read as a SHA1, needs to be amended
341 # FIXME: prec was read as a SHA1, needs to be amended
342
342
343 # read 0 or more successors
343 # read 0 or more successors
344 if numsuc == 1:
344 if numsuc == 1:
345 o2 = o1 + sha2size
345 o2 = o1 + sha2size
346 sucs = (data[o1:o2],)
346 sucs = (data[o1:o2],)
347 else:
347 else:
348 o2 = o1 + sha2size * numsuc
348 o2 = o1 + sha2size * numsuc
349 sucs = unpack(sha2fmt * numsuc, data[o1:o2])
349 sucs = unpack(sha2fmt * numsuc, data[o1:o2])
350
350
351 # read parents
351 # read parents
352 if numpar == noneflag:
352 if numpar == noneflag:
353 o3 = o2
353 o3 = o2
354 parents = None
354 parents = None
355 elif numpar == 1:
355 elif numpar == 1:
356 o3 = o2 + sha2size
356 o3 = o2 + sha2size
357 parents = (data[o2:o3],)
357 parents = (data[o2:o3],)
358 else:
358 else:
359 o3 = o2 + sha2size * numpar
359 o3 = o2 + sha2size * numpar
360 parents = unpack(sha2fmt * numpar, data[o2:o3])
360 parents = unpack(sha2fmt * numpar, data[o2:o3])
361 else:
361 else:
362 # read 0 or more successors
362 # read 0 or more successors
363 if numsuc == 1:
363 if numsuc == 1:
364 o2 = o1 + sha1size
364 o2 = o1 + sha1size
365 sucs = (data[o1:o2],)
365 sucs = (data[o1:o2],)
366 else:
366 else:
367 o2 = o1 + sha1size * numsuc
367 o2 = o1 + sha1size * numsuc
368 sucs = unpack(sha1fmt * numsuc, data[o1:o2])
368 sucs = unpack(sha1fmt * numsuc, data[o1:o2])
369
369
370 # read parents
370 # read parents
371 if numpar == noneflag:
371 if numpar == noneflag:
372 o3 = o2
372 o3 = o2
373 parents = None
373 parents = None
374 elif numpar == 1:
374 elif numpar == 1:
375 o3 = o2 + sha1size
375 o3 = o2 + sha1size
376 parents = (data[o2:o3],)
376 parents = (data[o2:o3],)
377 else:
377 else:
378 o3 = o2 + sha1size * numpar
378 o3 = o2 + sha1size * numpar
379 parents = unpack(sha1fmt * numpar, data[o2:o3])
379 parents = unpack(sha1fmt * numpar, data[o2:o3])
380
380
381 # read metadata
381 # read metadata
382 off = o3 + metasize * nummeta
382 off = o3 + metasize * nummeta
383 metapairsize = unpack('>' + (metafmt * nummeta), data[o3:off])
383 metapairsize = unpack('>' + (metafmt * nummeta), data[o3:off])
384 metadata = []
384 metadata = []
385 for idx in xrange(0, len(metapairsize), 2):
385 for idx in xrange(0, len(metapairsize), 2):
386 o1 = off + metapairsize[idx]
386 o1 = off + metapairsize[idx]
387 o2 = o1 + metapairsize[idx + 1]
387 o2 = o1 + metapairsize[idx + 1]
388 metadata.append((data[off:o1], data[o1:o2]))
388 metadata.append((data[off:o1], data[o1:o2]))
389 off = o2
389 off = o2
390
390
391 yield (prec, sucs, flags, tuple(metadata), (secs, tz * 60), parents)
391 yield (prec, sucs, flags, tuple(metadata), (secs, tz * 60), parents)
392
392
393 def _fm1encodeonemarker(marker):
393 def _fm1encodeonemarker(marker):
394 pre, sucs, flags, metadata, date, parents = marker
394 pre, sucs, flags, metadata, date, parents = marker
395 # determine node size
395 # determine node size
396 _fm1node = _fm1nodesha1
396 _fm1node = _fm1nodesha1
397 if flags & usingsha256:
397 if flags & usingsha256:
398 _fm1node = _fm1nodesha256
398 _fm1node = _fm1nodesha256
399 numsuc = len(sucs)
399 numsuc = len(sucs)
400 numextranodes = numsuc
400 numextranodes = numsuc
401 if parents is None:
401 if parents is None:
402 numpar = _fm1parentnone
402 numpar = _fm1parentnone
403 else:
403 else:
404 numpar = len(parents)
404 numpar = len(parents)
405 numextranodes += numpar
405 numextranodes += numpar
406 formatnodes = _fm1node * numextranodes
406 formatnodes = _fm1node * numextranodes
407 formatmeta = _fm1metapair * len(metadata)
407 formatmeta = _fm1metapair * len(metadata)
408 format = _fm1fixed + formatnodes + formatmeta
408 format = _fm1fixed + formatnodes + formatmeta
409 # tz is stored in minutes so we divide by 60
409 # tz is stored in minutes so we divide by 60
410 tz = date[1]//60
410 tz = date[1]//60
411 data = [None, date[0], tz, flags, numsuc, numpar, len(metadata), pre]
411 data = [None, date[0], tz, flags, numsuc, numpar, len(metadata), pre]
412 data.extend(sucs)
412 data.extend(sucs)
413 if parents is not None:
413 if parents is not None:
414 data.extend(parents)
414 data.extend(parents)
415 totalsize = _calcsize(format)
415 totalsize = _calcsize(format)
416 for key, value in metadata:
416 for key, value in metadata:
417 lk = len(key)
417 lk = len(key)
418 lv = len(value)
418 lv = len(value)
419 data.append(lk)
419 data.append(lk)
420 data.append(lv)
420 data.append(lv)
421 totalsize += lk + lv
421 totalsize += lk + lv
422 data[0] = totalsize
422 data[0] = totalsize
423 data = [_pack(format, *data)]
423 data = [_pack(format, *data)]
424 for key, value in metadata:
424 for key, value in metadata:
425 data.append(key)
425 data.append(key)
426 data.append(value)
426 data.append(value)
427 return ''.join(data)
427 return ''.join(data)
428
428
429 def _fm1readmarkers(data, off, stop):
429 def _fm1readmarkers(data, off, stop):
430 native = getattr(parsers, 'fm1readmarkers', None)
430 native = getattr(parsers, 'fm1readmarkers', None)
431 if not native:
431 if not native:
432 return _fm1purereadmarkers(data, off, stop)
432 return _fm1purereadmarkers(data, off, stop)
433 return native(data, off, stop)
433 return native(data, off, stop)
434
434
435 # mapping to read/write various marker formats
435 # mapping to read/write various marker formats
436 # <version> -> (decoder, encoder)
436 # <version> -> (decoder, encoder)
437 formats = {_fm0version: (_fm0readmarkers, _fm0encodeonemarker),
437 formats = {_fm0version: (_fm0readmarkers, _fm0encodeonemarker),
438 _fm1version: (_fm1readmarkers, _fm1encodeonemarker)}
438 _fm1version: (_fm1readmarkers, _fm1encodeonemarker)}
439
439
440 def _readmarkerversion(data):
440 def _readmarkerversion(data):
441 return _unpack('>B', data[0:1])[0]
441 return _unpack('>B', data[0:1])[0]
442
442
443 @util.nogc
443 @util.nogc
444 def _readmarkers(data, off=None, stop=None):
444 def _readmarkers(data, off=None, stop=None):
445 """Read and enumerate markers from raw data"""
445 """Read and enumerate markers from raw data"""
446 diskversion = _readmarkerversion(data)
446 diskversion = _readmarkerversion(data)
447 if not off:
447 if not off:
448 off = 1 # skip 1 byte version number
448 off = 1 # skip 1 byte version number
449 if stop is None:
449 if stop is None:
450 stop = len(data)
450 stop = len(data)
451 if diskversion not in formats:
451 if diskversion not in formats:
452 msg = _('parsing obsolete marker: unknown version %r') % diskversion
452 msg = _('parsing obsolete marker: unknown version %r') % diskversion
453 raise error.UnknownVersion(msg, version=diskversion)
453 raise error.UnknownVersion(msg, version=diskversion)
454 return diskversion, formats[diskversion][0](data, off, stop)
454 return diskversion, formats[diskversion][0](data, off, stop)
455
455
456 def encodeheader(version=_fm0version):
456 def encodeheader(version=_fm0version):
457 return _pack('>B', version)
457 return _pack('>B', version)
458
458
459 def encodemarkers(markers, addheader=False, version=_fm0version):
459 def encodemarkers(markers, addheader=False, version=_fm0version):
460 # Kept separate from flushmarkers(), it will be reused for
460 # Kept separate from flushmarkers(), it will be reused for
461 # markers exchange.
461 # markers exchange.
462 encodeone = formats[version][1]
462 encodeone = formats[version][1]
463 if addheader:
463 if addheader:
464 yield encodeheader(version)
464 yield encodeheader(version)
465 for marker in markers:
465 for marker in markers:
466 yield encodeone(marker)
466 yield encodeone(marker)
467
467
468 @util.nogc
468 @util.nogc
469 def _addsuccessors(successors, markers):
469 def _addsuccessors(successors, markers):
470 for mark in markers:
470 for mark in markers:
471 successors.setdefault(mark[0], set()).add(mark)
471 successors.setdefault(mark[0], set()).add(mark)
472
472
473 def _addprecursors(*args, **kwargs):
474 msg = ("'obsolete._addprecursors' is deprecated, "
475 "use 'obsolete._addpredecessors'")
476 util.nouideprecwarn(msg, '4.4')
477
478 return _addpredecessors(*args, **kwargs)
479
473 @util.nogc
480 @util.nogc
474 def _addprecursors(precursors, markers):
481 def _addpredecessors(predecessors, markers):
475 for mark in markers:
482 for mark in markers:
476 for suc in mark[1]:
483 for suc in mark[1]:
477 precursors.setdefault(suc, set()).add(mark)
484 predecessors.setdefault(suc, set()).add(mark)
478
485
479 @util.nogc
486 @util.nogc
480 def _addchildren(children, markers):
487 def _addchildren(children, markers):
481 for mark in markers:
488 for mark in markers:
482 parents = mark[5]
489 parents = mark[5]
483 if parents is not None:
490 if parents is not None:
484 for p in parents:
491 for p in parents:
485 children.setdefault(p, set()).add(mark)
492 children.setdefault(p, set()).add(mark)
486
493
487 def _checkinvalidmarkers(markers):
494 def _checkinvalidmarkers(markers):
488 """search for marker with invalid data and raise error if needed
495 """search for marker with invalid data and raise error if needed
489
496
490 Exist as a separated function to allow the evolve extension for a more
497 Exist as a separated function to allow the evolve extension for a more
491 subtle handling.
498 subtle handling.
492 """
499 """
493 for mark in markers:
500 for mark in markers:
494 if node.nullid in mark[1]:
501 if node.nullid in mark[1]:
495 raise error.Abort(_('bad obsolescence marker detected: '
502 raise error.Abort(_('bad obsolescence marker detected: '
496 'invalid successors nullid'))
503 'invalid successors nullid'))
497
504
498 class obsstore(object):
505 class obsstore(object):
499 """Store obsolete markers
506 """Store obsolete markers
500
507
501 Markers can be accessed with two mappings:
508 Markers can be accessed with two mappings:
502 - precursors[x] -> set(markers on precursors edges of x)
509 - precursors[x] -> set(markers on precursors edges of x)
503 - successors[x] -> set(markers on successors edges of x)
510 - successors[x] -> set(markers on successors edges of x)
504 - children[x] -> set(markers on precursors edges of children(x)
511 - children[x] -> set(markers on precursors edges of children(x)
505 """
512 """
506
513
507 fields = ('prec', 'succs', 'flag', 'meta', 'date', 'parents')
514 fields = ('prec', 'succs', 'flag', 'meta', 'date', 'parents')
508 # prec: nodeid, precursor changesets
515 # prec: nodeid, precursor changesets
509 # succs: tuple of nodeid, successor changesets (0-N length)
516 # succs: tuple of nodeid, successor changesets (0-N length)
510 # flag: integer, flag field carrying modifier for the markers (see doc)
517 # flag: integer, flag field carrying modifier for the markers (see doc)
511 # meta: binary blob, encoded metadata dictionary
518 # meta: binary blob, encoded metadata dictionary
512 # date: (float, int) tuple, date of marker creation
519 # date: (float, int) tuple, date of marker creation
513 # parents: (tuple of nodeid) or None, parents of precursors
520 # parents: (tuple of nodeid) or None, parents of precursors
514 # None is used when no data has been recorded
521 # None is used when no data has been recorded
515
522
516 def __init__(self, svfs, defaultformat=_fm1version, readonly=False):
523 def __init__(self, svfs, defaultformat=_fm1version, readonly=False):
517 # caches for various obsolescence related cache
524 # caches for various obsolescence related cache
518 self.caches = {}
525 self.caches = {}
519 self.svfs = svfs
526 self.svfs = svfs
520 self._defaultformat = defaultformat
527 self._defaultformat = defaultformat
521 self._readonly = readonly
528 self._readonly = readonly
522
529
523 def __iter__(self):
530 def __iter__(self):
524 return iter(self._all)
531 return iter(self._all)
525
532
526 def __len__(self):
533 def __len__(self):
527 return len(self._all)
534 return len(self._all)
528
535
529 def __nonzero__(self):
536 def __nonzero__(self):
530 if not self._cached('_all'):
537 if not self._cached('_all'):
531 try:
538 try:
532 return self.svfs.stat('obsstore').st_size > 1
539 return self.svfs.stat('obsstore').st_size > 1
533 except OSError as inst:
540 except OSError as inst:
534 if inst.errno != errno.ENOENT:
541 if inst.errno != errno.ENOENT:
535 raise
542 raise
536 # just build an empty _all list if no obsstore exists, which
543 # just build an empty _all list if no obsstore exists, which
537 # avoids further stat() syscalls
544 # avoids further stat() syscalls
538 pass
545 pass
539 return bool(self._all)
546 return bool(self._all)
540
547
541 __bool__ = __nonzero__
548 __bool__ = __nonzero__
542
549
543 @property
550 @property
544 def readonly(self):
551 def readonly(self):
545 """True if marker creation is disabled
552 """True if marker creation is disabled
546
553
547 Remove me in the future when obsolete marker is always on."""
554 Remove me in the future when obsolete marker is always on."""
548 return self._readonly
555 return self._readonly
549
556
550 def create(self, transaction, prec, succs=(), flag=0, parents=None,
557 def create(self, transaction, prec, succs=(), flag=0, parents=None,
551 date=None, metadata=None, ui=None):
558 date=None, metadata=None, ui=None):
552 """obsolete: add a new obsolete marker
559 """obsolete: add a new obsolete marker
553
560
554 * ensuring it is hashable
561 * ensuring it is hashable
555 * check mandatory metadata
562 * check mandatory metadata
556 * encode metadata
563 * encode metadata
557
564
558 If you are a human writing code creating marker you want to use the
565 If you are a human writing code creating marker you want to use the
559 `createmarkers` function in this module instead.
566 `createmarkers` function in this module instead.
560
567
561 return True if a new marker have been added, False if the markers
568 return True if a new marker have been added, False if the markers
562 already existed (no op).
569 already existed (no op).
563 """
570 """
564 if metadata is None:
571 if metadata is None:
565 metadata = {}
572 metadata = {}
566 if date is None:
573 if date is None:
567 if 'date' in metadata:
574 if 'date' in metadata:
568 # as a courtesy for out-of-tree extensions
575 # as a courtesy for out-of-tree extensions
569 date = util.parsedate(metadata.pop('date'))
576 date = util.parsedate(metadata.pop('date'))
570 elif ui is not None:
577 elif ui is not None:
571 date = ui.configdate('devel', 'default-date')
578 date = ui.configdate('devel', 'default-date')
572 if date is None:
579 if date is None:
573 date = util.makedate()
580 date = util.makedate()
574 else:
581 else:
575 date = util.makedate()
582 date = util.makedate()
576 if len(prec) != 20:
583 if len(prec) != 20:
577 raise ValueError(prec)
584 raise ValueError(prec)
578 for succ in succs:
585 for succ in succs:
579 if len(succ) != 20:
586 if len(succ) != 20:
580 raise ValueError(succ)
587 raise ValueError(succ)
581 if prec in succs:
588 if prec in succs:
582 raise ValueError(_('in-marker cycle with %s') % node.hex(prec))
589 raise ValueError(_('in-marker cycle with %s') % node.hex(prec))
583
590
584 metadata = tuple(sorted(metadata.iteritems()))
591 metadata = tuple(sorted(metadata.iteritems()))
585
592
586 marker = (bytes(prec), tuple(succs), int(flag), metadata, date, parents)
593 marker = (bytes(prec), tuple(succs), int(flag), metadata, date, parents)
587 return bool(self.add(transaction, [marker]))
594 return bool(self.add(transaction, [marker]))
588
595
589 def add(self, transaction, markers):
596 def add(self, transaction, markers):
590 """Add new markers to the store
597 """Add new markers to the store
591
598
592 Take care of filtering duplicate.
599 Take care of filtering duplicate.
593 Return the number of new marker."""
600 Return the number of new marker."""
594 if self._readonly:
601 if self._readonly:
595 raise error.Abort(_('creating obsolete markers is not enabled on '
602 raise error.Abort(_('creating obsolete markers is not enabled on '
596 'this repo'))
603 'this repo'))
597 known = set()
604 known = set()
598 getsuccessors = self.successors.get
605 getsuccessors = self.successors.get
599 new = []
606 new = []
600 for m in markers:
607 for m in markers:
601 if m not in getsuccessors(m[0], ()) and m not in known:
608 if m not in getsuccessors(m[0], ()) and m not in known:
602 known.add(m)
609 known.add(m)
603 new.append(m)
610 new.append(m)
604 if new:
611 if new:
605 f = self.svfs('obsstore', 'ab')
612 f = self.svfs('obsstore', 'ab')
606 try:
613 try:
607 offset = f.tell()
614 offset = f.tell()
608 transaction.add('obsstore', offset)
615 transaction.add('obsstore', offset)
609 # offset == 0: new file - add the version header
616 # offset == 0: new file - add the version header
610 data = b''.join(encodemarkers(new, offset == 0, self._version))
617 data = b''.join(encodemarkers(new, offset == 0, self._version))
611 f.write(data)
618 f.write(data)
612 finally:
619 finally:
613 # XXX: f.close() == filecache invalidation == obsstore rebuilt.
620 # XXX: f.close() == filecache invalidation == obsstore rebuilt.
614 # call 'filecacheentry.refresh()' here
621 # call 'filecacheentry.refresh()' here
615 f.close()
622 f.close()
616 addedmarkers = transaction.changes.get('obsmarkers')
623 addedmarkers = transaction.changes.get('obsmarkers')
617 if addedmarkers is not None:
624 if addedmarkers is not None:
618 addedmarkers.update(new)
625 addedmarkers.update(new)
619 self._addmarkers(new, data)
626 self._addmarkers(new, data)
620 # new marker *may* have changed several set. invalidate the cache.
627 # new marker *may* have changed several set. invalidate the cache.
621 self.caches.clear()
628 self.caches.clear()
622 # records the number of new markers for the transaction hooks
629 # records the number of new markers for the transaction hooks
623 previous = int(transaction.hookargs.get('new_obsmarkers', '0'))
630 previous = int(transaction.hookargs.get('new_obsmarkers', '0'))
624 transaction.hookargs['new_obsmarkers'] = str(previous + len(new))
631 transaction.hookargs['new_obsmarkers'] = str(previous + len(new))
625 return len(new)
632 return len(new)
626
633
627 def mergemarkers(self, transaction, data):
634 def mergemarkers(self, transaction, data):
628 """merge a binary stream of markers inside the obsstore
635 """merge a binary stream of markers inside the obsstore
629
636
630 Returns the number of new markers added."""
637 Returns the number of new markers added."""
631 version, markers = _readmarkers(data)
638 version, markers = _readmarkers(data)
632 return self.add(transaction, markers)
639 return self.add(transaction, markers)
633
640
634 @propertycache
641 @propertycache
635 def _data(self):
642 def _data(self):
636 return self.svfs.tryread('obsstore')
643 return self.svfs.tryread('obsstore')
637
644
638 @propertycache
645 @propertycache
639 def _version(self):
646 def _version(self):
640 if len(self._data) >= 1:
647 if len(self._data) >= 1:
641 return _readmarkerversion(self._data)
648 return _readmarkerversion(self._data)
642 else:
649 else:
643 return self._defaultformat
650 return self._defaultformat
644
651
645 @propertycache
652 @propertycache
646 def _all(self):
653 def _all(self):
647 data = self._data
654 data = self._data
648 if not data:
655 if not data:
649 return []
656 return []
650 self._version, markers = _readmarkers(data)
657 self._version, markers = _readmarkers(data)
651 markers = list(markers)
658 markers = list(markers)
652 _checkinvalidmarkers(markers)
659 _checkinvalidmarkers(markers)
653 return markers
660 return markers
654
661
655 @propertycache
662 @propertycache
656 def successors(self):
663 def successors(self):
657 successors = {}
664 successors = {}
658 _addsuccessors(successors, self._all)
665 _addsuccessors(successors, self._all)
659 return successors
666 return successors
660
667
661 @propertycache
668 @propertycache
662 def precursors(self):
669 def precursors(self):
663 precursors = {}
670 predecessors = {}
664 _addprecursors(precursors, self._all)
671 _addpredecessors(predecessors, self._all)
665 return precursors
672 return predecessors
666
673
667 @propertycache
674 @propertycache
668 def children(self):
675 def children(self):
669 children = {}
676 children = {}
670 _addchildren(children, self._all)
677 _addchildren(children, self._all)
671 return children
678 return children
672
679
673 def _cached(self, attr):
680 def _cached(self, attr):
674 return attr in self.__dict__
681 return attr in self.__dict__
675
682
676 def _addmarkers(self, markers, rawdata):
683 def _addmarkers(self, markers, rawdata):
677 markers = list(markers) # to allow repeated iteration
684 markers = list(markers) # to allow repeated iteration
678 self._data = self._data + rawdata
685 self._data = self._data + rawdata
679 self._all.extend(markers)
686 self._all.extend(markers)
680 if self._cached('successors'):
687 if self._cached('successors'):
681 _addsuccessors(self.successors, markers)
688 _addsuccessors(self.successors, markers)
682 if self._cached('precursors'):
689 if self._cached('precursors'):
683 _addprecursors(self.precursors, markers)
690 _addpredecessors(self.precursors, markers)
684 if self._cached('children'):
691 if self._cached('children'):
685 _addchildren(self.children, markers)
692 _addchildren(self.children, markers)
686 _checkinvalidmarkers(markers)
693 _checkinvalidmarkers(markers)
687
694
688 def relevantmarkers(self, nodes):
695 def relevantmarkers(self, nodes):
689 """return a set of all obsolescence markers relevant to a set of nodes.
696 """return a set of all obsolescence markers relevant to a set of nodes.
690
697
691 "relevant" to a set of nodes mean:
698 "relevant" to a set of nodes mean:
692
699
693 - marker that use this changeset as successor
700 - marker that use this changeset as successor
694 - prune marker of direct children on this changeset
701 - prune marker of direct children on this changeset
695 - recursive application of the two rules on precursors of these markers
702 - recursive application of the two rules on precursors of these markers
696
703
697 It is a set so you cannot rely on order."""
704 It is a set so you cannot rely on order."""
698
705
699 pendingnodes = set(nodes)
706 pendingnodes = set(nodes)
700 seenmarkers = set()
707 seenmarkers = set()
701 seennodes = set(pendingnodes)
708 seennodes = set(pendingnodes)
702 precursorsmarkers = self.precursors
709 precursorsmarkers = self.precursors
703 succsmarkers = self.successors
710 succsmarkers = self.successors
704 children = self.children
711 children = self.children
705 while pendingnodes:
712 while pendingnodes:
706 direct = set()
713 direct = set()
707 for current in pendingnodes:
714 for current in pendingnodes:
708 direct.update(precursorsmarkers.get(current, ()))
715 direct.update(precursorsmarkers.get(current, ()))
709 pruned = [m for m in children.get(current, ()) if not m[1]]
716 pruned = [m for m in children.get(current, ()) if not m[1]]
710 direct.update(pruned)
717 direct.update(pruned)
711 pruned = [m for m in succsmarkers.get(current, ()) if not m[1]]
718 pruned = [m for m in succsmarkers.get(current, ()) if not m[1]]
712 direct.update(pruned)
719 direct.update(pruned)
713 direct -= seenmarkers
720 direct -= seenmarkers
714 pendingnodes = set([m[0] for m in direct])
721 pendingnodes = set([m[0] for m in direct])
715 seenmarkers |= direct
722 seenmarkers |= direct
716 pendingnodes -= seennodes
723 pendingnodes -= seennodes
717 seennodes |= pendingnodes
724 seennodes |= pendingnodes
718 return seenmarkers
725 return seenmarkers
719
726
720 def makestore(ui, repo):
727 def makestore(ui, repo):
721 """Create an obsstore instance from a repo."""
728 """Create an obsstore instance from a repo."""
722 # read default format for new obsstore.
729 # read default format for new obsstore.
723 # developer config: format.obsstore-version
730 # developer config: format.obsstore-version
724 defaultformat = ui.configint('format', 'obsstore-version')
731 defaultformat = ui.configint('format', 'obsstore-version')
725 # rely on obsstore class default when possible.
732 # rely on obsstore class default when possible.
726 kwargs = {}
733 kwargs = {}
727 if defaultformat is not None:
734 if defaultformat is not None:
728 kwargs['defaultformat'] = defaultformat
735 kwargs['defaultformat'] = defaultformat
729 readonly = not isenabled(repo, createmarkersopt)
736 readonly = not isenabled(repo, createmarkersopt)
730 store = obsstore(repo.svfs, readonly=readonly, **kwargs)
737 store = obsstore(repo.svfs, readonly=readonly, **kwargs)
731 if store and readonly:
738 if store and readonly:
732 ui.warn(_('obsolete feature not enabled but %i markers found!\n')
739 ui.warn(_('obsolete feature not enabled but %i markers found!\n')
733 % len(list(store)))
740 % len(list(store)))
734 return store
741 return store
735
742
736 def commonversion(versions):
743 def commonversion(versions):
737 """Return the newest version listed in both versions and our local formats.
744 """Return the newest version listed in both versions and our local formats.
738
745
739 Returns None if no common version exists.
746 Returns None if no common version exists.
740 """
747 """
741 versions.sort(reverse=True)
748 versions.sort(reverse=True)
742 # search for highest version known on both side
749 # search for highest version known on both side
743 for v in versions:
750 for v in versions:
744 if v in formats:
751 if v in formats:
745 return v
752 return v
746 return None
753 return None
747
754
748 # arbitrary picked to fit into 8K limit from HTTP server
755 # arbitrary picked to fit into 8K limit from HTTP server
749 # you have to take in account:
756 # you have to take in account:
750 # - the version header
757 # - the version header
751 # - the base85 encoding
758 # - the base85 encoding
752 _maxpayload = 5300
759 _maxpayload = 5300
753
760
754 def _pushkeyescape(markers):
761 def _pushkeyescape(markers):
755 """encode markers into a dict suitable for pushkey exchange
762 """encode markers into a dict suitable for pushkey exchange
756
763
757 - binary data is base85 encoded
764 - binary data is base85 encoded
758 - split in chunks smaller than 5300 bytes"""
765 - split in chunks smaller than 5300 bytes"""
759 keys = {}
766 keys = {}
760 parts = []
767 parts = []
761 currentlen = _maxpayload * 2 # ensure we create a new part
768 currentlen = _maxpayload * 2 # ensure we create a new part
762 for marker in markers:
769 for marker in markers:
763 nextdata = _fm0encodeonemarker(marker)
770 nextdata = _fm0encodeonemarker(marker)
764 if (len(nextdata) + currentlen > _maxpayload):
771 if (len(nextdata) + currentlen > _maxpayload):
765 currentpart = []
772 currentpart = []
766 currentlen = 0
773 currentlen = 0
767 parts.append(currentpart)
774 parts.append(currentpart)
768 currentpart.append(nextdata)
775 currentpart.append(nextdata)
769 currentlen += len(nextdata)
776 currentlen += len(nextdata)
770 for idx, part in enumerate(reversed(parts)):
777 for idx, part in enumerate(reversed(parts)):
771 data = ''.join([_pack('>B', _fm0version)] + part)
778 data = ''.join([_pack('>B', _fm0version)] + part)
772 keys['dump%i' % idx] = util.b85encode(data)
779 keys['dump%i' % idx] = util.b85encode(data)
773 return keys
780 return keys
774
781
775 def listmarkers(repo):
782 def listmarkers(repo):
776 """List markers over pushkey"""
783 """List markers over pushkey"""
777 if not repo.obsstore:
784 if not repo.obsstore:
778 return {}
785 return {}
779 return _pushkeyescape(sorted(repo.obsstore))
786 return _pushkeyescape(sorted(repo.obsstore))
780
787
781 def pushmarker(repo, key, old, new):
788 def pushmarker(repo, key, old, new):
782 """Push markers over pushkey"""
789 """Push markers over pushkey"""
783 if not key.startswith('dump'):
790 if not key.startswith('dump'):
784 repo.ui.warn(_('unknown key: %r') % key)
791 repo.ui.warn(_('unknown key: %r') % key)
785 return False
792 return False
786 if old:
793 if old:
787 repo.ui.warn(_('unexpected old value for %r') % key)
794 repo.ui.warn(_('unexpected old value for %r') % key)
788 return False
795 return False
789 data = util.b85decode(new)
796 data = util.b85decode(new)
790 lock = repo.lock()
797 lock = repo.lock()
791 try:
798 try:
792 tr = repo.transaction('pushkey: obsolete markers')
799 tr = repo.transaction('pushkey: obsolete markers')
793 try:
800 try:
794 repo.obsstore.mergemarkers(tr, data)
801 repo.obsstore.mergemarkers(tr, data)
795 repo.invalidatevolatilesets()
802 repo.invalidatevolatilesets()
796 tr.close()
803 tr.close()
797 return True
804 return True
798 finally:
805 finally:
799 tr.release()
806 tr.release()
800 finally:
807 finally:
801 lock.release()
808 lock.release()
802
809
803 # keep compatibility for the 4.3 cycle
810 # keep compatibility for the 4.3 cycle
804 def allprecursors(obsstore, nodes, ignoreflags=0):
811 def allprecursors(obsstore, nodes, ignoreflags=0):
805 movemsg = 'obsolete.allprecursors moved to obsutil.allprecursors'
812 movemsg = 'obsolete.allprecursors moved to obsutil.allprecursors'
806 util.nouideprecwarn(movemsg, '4.3')
813 util.nouideprecwarn(movemsg, '4.3')
807 return obsutil.allprecursors(obsstore, nodes, ignoreflags)
814 return obsutil.allprecursors(obsstore, nodes, ignoreflags)
808
815
809 def allsuccessors(obsstore, nodes, ignoreflags=0):
816 def allsuccessors(obsstore, nodes, ignoreflags=0):
810 movemsg = 'obsolete.allsuccessors moved to obsutil.allsuccessors'
817 movemsg = 'obsolete.allsuccessors moved to obsutil.allsuccessors'
811 util.nouideprecwarn(movemsg, '4.3')
818 util.nouideprecwarn(movemsg, '4.3')
812 return obsutil.allsuccessors(obsstore, nodes, ignoreflags)
819 return obsutil.allsuccessors(obsstore, nodes, ignoreflags)
813
820
814 def marker(repo, data):
821 def marker(repo, data):
815 movemsg = 'obsolete.marker moved to obsutil.marker'
822 movemsg = 'obsolete.marker moved to obsutil.marker'
816 repo.ui.deprecwarn(movemsg, '4.3')
823 repo.ui.deprecwarn(movemsg, '4.3')
817 return obsutil.marker(repo, data)
824 return obsutil.marker(repo, data)
818
825
819 def getmarkers(repo, nodes=None, exclusive=False):
826 def getmarkers(repo, nodes=None, exclusive=False):
820 movemsg = 'obsolete.getmarkers moved to obsutil.getmarkers'
827 movemsg = 'obsolete.getmarkers moved to obsutil.getmarkers'
821 repo.ui.deprecwarn(movemsg, '4.3')
828 repo.ui.deprecwarn(movemsg, '4.3')
822 return obsutil.getmarkers(repo, nodes=nodes, exclusive=exclusive)
829 return obsutil.getmarkers(repo, nodes=nodes, exclusive=exclusive)
823
830
824 def exclusivemarkers(repo, nodes):
831 def exclusivemarkers(repo, nodes):
825 movemsg = 'obsolete.exclusivemarkers moved to obsutil.exclusivemarkers'
832 movemsg = 'obsolete.exclusivemarkers moved to obsutil.exclusivemarkers'
826 repo.ui.deprecwarn(movemsg, '4.3')
833 repo.ui.deprecwarn(movemsg, '4.3')
827 return obsutil.exclusivemarkers(repo, nodes)
834 return obsutil.exclusivemarkers(repo, nodes)
828
835
829 def foreground(repo, nodes):
836 def foreground(repo, nodes):
830 movemsg = 'obsolete.foreground moved to obsutil.foreground'
837 movemsg = 'obsolete.foreground moved to obsutil.foreground'
831 repo.ui.deprecwarn(movemsg, '4.3')
838 repo.ui.deprecwarn(movemsg, '4.3')
832 return obsutil.foreground(repo, nodes)
839 return obsutil.foreground(repo, nodes)
833
840
834 def successorssets(repo, initialnode, cache=None):
841 def successorssets(repo, initialnode, cache=None):
835 movemsg = 'obsolete.successorssets moved to obsutil.successorssets'
842 movemsg = 'obsolete.successorssets moved to obsutil.successorssets'
836 repo.ui.deprecwarn(movemsg, '4.3')
843 repo.ui.deprecwarn(movemsg, '4.3')
837 return obsutil.successorssets(repo, initialnode, cache=cache)
844 return obsutil.successorssets(repo, initialnode, cache=cache)
838
845
839 # mapping of 'set-name' -> <function to compute this set>
846 # mapping of 'set-name' -> <function to compute this set>
840 cachefuncs = {}
847 cachefuncs = {}
841 def cachefor(name):
848 def cachefor(name):
842 """Decorator to register a function as computing the cache for a set"""
849 """Decorator to register a function as computing the cache for a set"""
843 def decorator(func):
850 def decorator(func):
844 if name in cachefuncs:
851 if name in cachefuncs:
845 msg = "duplicated registration for volatileset '%s' (existing: %r)"
852 msg = "duplicated registration for volatileset '%s' (existing: %r)"
846 raise error.ProgrammingError(msg % (name, cachefuncs[name]))
853 raise error.ProgrammingError(msg % (name, cachefuncs[name]))
847 cachefuncs[name] = func
854 cachefuncs[name] = func
848 return func
855 return func
849 return decorator
856 return decorator
850
857
851 def getrevs(repo, name):
858 def getrevs(repo, name):
852 """Return the set of revision that belong to the <name> set
859 """Return the set of revision that belong to the <name> set
853
860
854 Such access may compute the set and cache it for future use"""
861 Such access may compute the set and cache it for future use"""
855 repo = repo.unfiltered()
862 repo = repo.unfiltered()
856 if not repo.obsstore:
863 if not repo.obsstore:
857 return frozenset()
864 return frozenset()
858 if name not in repo.obsstore.caches:
865 if name not in repo.obsstore.caches:
859 repo.obsstore.caches[name] = cachefuncs[name](repo)
866 repo.obsstore.caches[name] = cachefuncs[name](repo)
860 return repo.obsstore.caches[name]
867 return repo.obsstore.caches[name]
861
868
862 # To be simple we need to invalidate obsolescence cache when:
869 # To be simple we need to invalidate obsolescence cache when:
863 #
870 #
864 # - new changeset is added:
871 # - new changeset is added:
865 # - public phase is changed
872 # - public phase is changed
866 # - obsolescence marker are added
873 # - obsolescence marker are added
867 # - strip is used a repo
874 # - strip is used a repo
868 def clearobscaches(repo):
875 def clearobscaches(repo):
869 """Remove all obsolescence related cache from a repo
876 """Remove all obsolescence related cache from a repo
870
877
871 This remove all cache in obsstore is the obsstore already exist on the
878 This remove all cache in obsstore is the obsstore already exist on the
872 repo.
879 repo.
873
880
874 (We could be smarter here given the exact event that trigger the cache
881 (We could be smarter here given the exact event that trigger the cache
875 clearing)"""
882 clearing)"""
876 # only clear cache is there is obsstore data in this repo
883 # only clear cache is there is obsstore data in this repo
877 if 'obsstore' in repo._filecache:
884 if 'obsstore' in repo._filecache:
878 repo.obsstore.caches.clear()
885 repo.obsstore.caches.clear()
879
886
880 def _mutablerevs(repo):
887 def _mutablerevs(repo):
881 """the set of mutable revision in the repository"""
888 """the set of mutable revision in the repository"""
882 return repo._phasecache.getrevset(repo, (phases.draft, phases.secret))
889 return repo._phasecache.getrevset(repo, (phases.draft, phases.secret))
883
890
884 @cachefor('obsolete')
891 @cachefor('obsolete')
885 def _computeobsoleteset(repo):
892 def _computeobsoleteset(repo):
886 """the set of obsolete revisions"""
893 """the set of obsolete revisions"""
887 getnode = repo.changelog.node
894 getnode = repo.changelog.node
888 notpublic = _mutablerevs(repo)
895 notpublic = _mutablerevs(repo)
889 isobs = repo.obsstore.successors.__contains__
896 isobs = repo.obsstore.successors.__contains__
890 obs = set(r for r in notpublic if isobs(getnode(r)))
897 obs = set(r for r in notpublic if isobs(getnode(r)))
891 return obs
898 return obs
892
899
893 @cachefor('unstable')
900 @cachefor('unstable')
894 def _computeunstableset(repo):
901 def _computeunstableset(repo):
895 """the set of non obsolete revisions with obsolete parents"""
902 """the set of non obsolete revisions with obsolete parents"""
896 pfunc = repo.changelog.parentrevs
903 pfunc = repo.changelog.parentrevs
897 mutable = _mutablerevs(repo)
904 mutable = _mutablerevs(repo)
898 obsolete = getrevs(repo, 'obsolete')
905 obsolete = getrevs(repo, 'obsolete')
899 others = mutable - obsolete
906 others = mutable - obsolete
900 unstable = set()
907 unstable = set()
901 for r in sorted(others):
908 for r in sorted(others):
902 # A rev is unstable if one of its parent is obsolete or unstable
909 # A rev is unstable if one of its parent is obsolete or unstable
903 # this works since we traverse following growing rev order
910 # this works since we traverse following growing rev order
904 for p in pfunc(r):
911 for p in pfunc(r):
905 if p in obsolete or p in unstable:
912 if p in obsolete or p in unstable:
906 unstable.add(r)
913 unstable.add(r)
907 break
914 break
908 return unstable
915 return unstable
909
916
910 @cachefor('suspended')
917 @cachefor('suspended')
911 def _computesuspendedset(repo):
918 def _computesuspendedset(repo):
912 """the set of obsolete parents with non obsolete descendants"""
919 """the set of obsolete parents with non obsolete descendants"""
913 suspended = repo.changelog.ancestors(getrevs(repo, 'unstable'))
920 suspended = repo.changelog.ancestors(getrevs(repo, 'unstable'))
914 return set(r for r in getrevs(repo, 'obsolete') if r in suspended)
921 return set(r for r in getrevs(repo, 'obsolete') if r in suspended)
915
922
916 @cachefor('extinct')
923 @cachefor('extinct')
917 def _computeextinctset(repo):
924 def _computeextinctset(repo):
918 """the set of obsolete parents without non obsolete descendants"""
925 """the set of obsolete parents without non obsolete descendants"""
919 return getrevs(repo, 'obsolete') - getrevs(repo, 'suspended')
926 return getrevs(repo, 'obsolete') - getrevs(repo, 'suspended')
920
927
921
928
922 @cachefor('bumped')
929 @cachefor('bumped')
923 def _computebumpedset(repo):
930 def _computebumpedset(repo):
924 """the set of revs trying to obsolete public revisions"""
931 """the set of revs trying to obsolete public revisions"""
925 bumped = set()
932 bumped = set()
926 # util function (avoid attribute lookup in the loop)
933 # util function (avoid attribute lookup in the loop)
927 phase = repo._phasecache.phase # would be faster to grab the full list
934 phase = repo._phasecache.phase # would be faster to grab the full list
928 public = phases.public
935 public = phases.public
929 cl = repo.changelog
936 cl = repo.changelog
930 torev = cl.nodemap.get
937 torev = cl.nodemap.get
931 for ctx in repo.set('(not public()) and (not obsolete())'):
938 for ctx in repo.set('(not public()) and (not obsolete())'):
932 rev = ctx.rev()
939 rev = ctx.rev()
933 # We only evaluate mutable, non-obsolete revision
940 # We only evaluate mutable, non-obsolete revision
934 node = ctx.node()
941 node = ctx.node()
935 # (future) A cache of precursors may worth if split is very common
942 # (future) A cache of precursors may worth if split is very common
936 for pnode in obsutil.allprecursors(repo.obsstore, [node],
943 for pnode in obsutil.allprecursors(repo.obsstore, [node],
937 ignoreflags=bumpedfix):
944 ignoreflags=bumpedfix):
938 prev = torev(pnode) # unfiltered! but so is phasecache
945 prev = torev(pnode) # unfiltered! but so is phasecache
939 if (prev is not None) and (phase(repo, prev) <= public):
946 if (prev is not None) and (phase(repo, prev) <= public):
940 # we have a public precursor
947 # we have a public precursor
941 bumped.add(rev)
948 bumped.add(rev)
942 break # Next draft!
949 break # Next draft!
943 return bumped
950 return bumped
944
951
945 @cachefor('divergent')
952 @cachefor('divergent')
946 def _computedivergentset(repo):
953 def _computedivergentset(repo):
947 """the set of rev that compete to be the final successors of some revision.
954 """the set of rev that compete to be the final successors of some revision.
948 """
955 """
949 divergent = set()
956 divergent = set()
950 obsstore = repo.obsstore
957 obsstore = repo.obsstore
951 newermap = {}
958 newermap = {}
952 for ctx in repo.set('(not public()) - obsolete()'):
959 for ctx in repo.set('(not public()) - obsolete()'):
953 mark = obsstore.precursors.get(ctx.node(), ())
960 mark = obsstore.precursors.get(ctx.node(), ())
954 toprocess = set(mark)
961 toprocess = set(mark)
955 seen = set()
962 seen = set()
956 while toprocess:
963 while toprocess:
957 prec = toprocess.pop()[0]
964 prec = toprocess.pop()[0]
958 if prec in seen:
965 if prec in seen:
959 continue # emergency cycle hanging prevention
966 continue # emergency cycle hanging prevention
960 seen.add(prec)
967 seen.add(prec)
961 if prec not in newermap:
968 if prec not in newermap:
962 obsutil.successorssets(repo, prec, cache=newermap)
969 obsutil.successorssets(repo, prec, cache=newermap)
963 newer = [n for n in newermap[prec] if n]
970 newer = [n for n in newermap[prec] if n]
964 if len(newer) > 1:
971 if len(newer) > 1:
965 divergent.add(ctx.rev())
972 divergent.add(ctx.rev())
966 break
973 break
967 toprocess.update(obsstore.precursors.get(prec, ()))
974 toprocess.update(obsstore.precursors.get(prec, ()))
968 return divergent
975 return divergent
969
976
970
977
971 def createmarkers(repo, relations, flag=0, date=None, metadata=None,
978 def createmarkers(repo, relations, flag=0, date=None, metadata=None,
972 operation=None):
979 operation=None):
973 """Add obsolete markers between changesets in a repo
980 """Add obsolete markers between changesets in a repo
974
981
975 <relations> must be an iterable of (<old>, (<new>, ...)[,{metadata}])
982 <relations> must be an iterable of (<old>, (<new>, ...)[,{metadata}])
976 tuple. `old` and `news` are changectx. metadata is an optional dictionary
983 tuple. `old` and `news` are changectx. metadata is an optional dictionary
977 containing metadata for this marker only. It is merged with the global
984 containing metadata for this marker only. It is merged with the global
978 metadata specified through the `metadata` argument of this function,
985 metadata specified through the `metadata` argument of this function,
979
986
980 Trying to obsolete a public changeset will raise an exception.
987 Trying to obsolete a public changeset will raise an exception.
981
988
982 Current user and date are used except if specified otherwise in the
989 Current user and date are used except if specified otherwise in the
983 metadata attribute.
990 metadata attribute.
984
991
985 This function operates within a transaction of its own, but does
992 This function operates within a transaction of its own, but does
986 not take any lock on the repo.
993 not take any lock on the repo.
987 """
994 """
988 # prepare metadata
995 # prepare metadata
989 if metadata is None:
996 if metadata is None:
990 metadata = {}
997 metadata = {}
991 if 'user' not in metadata:
998 if 'user' not in metadata:
992 metadata['user'] = repo.ui.username()
999 metadata['user'] = repo.ui.username()
993 useoperation = repo.ui.configbool('experimental',
1000 useoperation = repo.ui.configbool('experimental',
994 'evolution.track-operation')
1001 'evolution.track-operation')
995 if useoperation and operation:
1002 if useoperation and operation:
996 metadata['operation'] = operation
1003 metadata['operation'] = operation
997 tr = repo.transaction('add-obsolescence-marker')
1004 tr = repo.transaction('add-obsolescence-marker')
998 try:
1005 try:
999 markerargs = []
1006 markerargs = []
1000 for rel in relations:
1007 for rel in relations:
1001 prec = rel[0]
1008 prec = rel[0]
1002 sucs = rel[1]
1009 sucs = rel[1]
1003 localmetadata = metadata.copy()
1010 localmetadata = metadata.copy()
1004 if 2 < len(rel):
1011 if 2 < len(rel):
1005 localmetadata.update(rel[2])
1012 localmetadata.update(rel[2])
1006
1013
1007 if not prec.mutable():
1014 if not prec.mutable():
1008 raise error.Abort(_("cannot obsolete public changeset: %s")
1015 raise error.Abort(_("cannot obsolete public changeset: %s")
1009 % prec,
1016 % prec,
1010 hint="see 'hg help phases' for details")
1017 hint="see 'hg help phases' for details")
1011 nprec = prec.node()
1018 nprec = prec.node()
1012 nsucs = tuple(s.node() for s in sucs)
1019 nsucs = tuple(s.node() for s in sucs)
1013 npare = None
1020 npare = None
1014 if not nsucs:
1021 if not nsucs:
1015 npare = tuple(p.node() for p in prec.parents())
1022 npare = tuple(p.node() for p in prec.parents())
1016 if nprec in nsucs:
1023 if nprec in nsucs:
1017 raise error.Abort(_("changeset %s cannot obsolete itself")
1024 raise error.Abort(_("changeset %s cannot obsolete itself")
1018 % prec)
1025 % prec)
1019
1026
1020 # Creating the marker causes the hidden cache to become invalid,
1027 # Creating the marker causes the hidden cache to become invalid,
1021 # which causes recomputation when we ask for prec.parents() above.
1028 # which causes recomputation when we ask for prec.parents() above.
1022 # Resulting in n^2 behavior. So let's prepare all of the args
1029 # Resulting in n^2 behavior. So let's prepare all of the args
1023 # first, then create the markers.
1030 # first, then create the markers.
1024 markerargs.append((nprec, nsucs, npare, localmetadata))
1031 markerargs.append((nprec, nsucs, npare, localmetadata))
1025
1032
1026 for args in markerargs:
1033 for args in markerargs:
1027 nprec, nsucs, npare, localmetadata = args
1034 nprec, nsucs, npare, localmetadata = args
1028 repo.obsstore.create(tr, nprec, nsucs, flag, parents=npare,
1035 repo.obsstore.create(tr, nprec, nsucs, flag, parents=npare,
1029 date=date, metadata=localmetadata,
1036 date=date, metadata=localmetadata,
1030 ui=repo.ui)
1037 ui=repo.ui)
1031 repo.filteredrevcache.clear()
1038 repo.filteredrevcache.clear()
1032 tr.close()
1039 tr.close()
1033 finally:
1040 finally:
1034 tr.release()
1041 tr.release()
General Comments 0
You need to be logged in to leave comments. Login now