##// END OF EJS Templates
revlog: move chunk cache preload from revision to _chunks...
Siddharth Agarwal -
r19716:e1797697 default
parent child Browse files
Show More
@@ -1,1345 +1,1349 b''
1 # revlog.py - storage back-end for mercurial
1 # revlog.py - storage back-end for mercurial
2 #
2 #
3 # Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
3 # Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
4 #
4 #
5 # This software may be used and distributed according to the terms of the
5 # This software may be used and distributed according to the terms of the
6 # GNU General Public License version 2 or any later version.
6 # GNU General Public License version 2 or any later version.
7
7
8 """Storage back-end for Mercurial.
8 """Storage back-end for Mercurial.
9
9
10 This provides efficient delta storage with O(1) retrieve and append
10 This provides efficient delta storage with O(1) retrieve and append
11 and O(changes) merge between branches.
11 and O(changes) merge between branches.
12 """
12 """
13
13
14 # import stuff from node for others to import from revlog
14 # import stuff from node for others to import from revlog
15 from node import bin, hex, nullid, nullrev
15 from node import bin, hex, nullid, nullrev
16 from i18n import _
16 from i18n import _
17 import ancestor, mdiff, parsers, error, util, templatefilters
17 import ancestor, mdiff, parsers, error, util, templatefilters
18 import struct, zlib, errno
18 import struct, zlib, errno
19
19
20 _pack = struct.pack
20 _pack = struct.pack
21 _unpack = struct.unpack
21 _unpack = struct.unpack
22 _compress = zlib.compress
22 _compress = zlib.compress
23 _decompress = zlib.decompress
23 _decompress = zlib.decompress
24 _sha = util.sha1
24 _sha = util.sha1
25
25
26 # revlog header flags
26 # revlog header flags
27 REVLOGV0 = 0
27 REVLOGV0 = 0
28 REVLOGNG = 1
28 REVLOGNG = 1
29 REVLOGNGINLINEDATA = (1 << 16)
29 REVLOGNGINLINEDATA = (1 << 16)
30 REVLOGGENERALDELTA = (1 << 17)
30 REVLOGGENERALDELTA = (1 << 17)
31 REVLOG_DEFAULT_FLAGS = REVLOGNGINLINEDATA
31 REVLOG_DEFAULT_FLAGS = REVLOGNGINLINEDATA
32 REVLOG_DEFAULT_FORMAT = REVLOGNG
32 REVLOG_DEFAULT_FORMAT = REVLOGNG
33 REVLOG_DEFAULT_VERSION = REVLOG_DEFAULT_FORMAT | REVLOG_DEFAULT_FLAGS
33 REVLOG_DEFAULT_VERSION = REVLOG_DEFAULT_FORMAT | REVLOG_DEFAULT_FLAGS
34 REVLOGNG_FLAGS = REVLOGNGINLINEDATA | REVLOGGENERALDELTA
34 REVLOGNG_FLAGS = REVLOGNGINLINEDATA | REVLOGGENERALDELTA
35
35
36 # revlog index flags
36 # revlog index flags
37 REVIDX_KNOWN_FLAGS = 0
37 REVIDX_KNOWN_FLAGS = 0
38
38
39 # max size of revlog with inline data
39 # max size of revlog with inline data
40 _maxinline = 131072
40 _maxinline = 131072
41 _chunksize = 1048576
41 _chunksize = 1048576
42
42
43 RevlogError = error.RevlogError
43 RevlogError = error.RevlogError
44 LookupError = error.LookupError
44 LookupError = error.LookupError
45
45
46 def getoffset(q):
46 def getoffset(q):
47 return int(q >> 16)
47 return int(q >> 16)
48
48
49 def gettype(q):
49 def gettype(q):
50 return int(q & 0xFFFF)
50 return int(q & 0xFFFF)
51
51
52 def offset_type(offset, type):
52 def offset_type(offset, type):
53 return long(long(offset) << 16 | type)
53 return long(long(offset) << 16 | type)
54
54
55 nullhash = _sha(nullid)
55 nullhash = _sha(nullid)
56
56
57 def hash(text, p1, p2):
57 def hash(text, p1, p2):
58 """generate a hash from the given text and its parent hashes
58 """generate a hash from the given text and its parent hashes
59
59
60 This hash combines both the current file contents and its history
60 This hash combines both the current file contents and its history
61 in a manner that makes it easy to distinguish nodes with the same
61 in a manner that makes it easy to distinguish nodes with the same
62 content in the revision graph.
62 content in the revision graph.
63 """
63 """
64 # As of now, if one of the parent node is null, p2 is null
64 # As of now, if one of the parent node is null, p2 is null
65 if p2 == nullid:
65 if p2 == nullid:
66 # deep copy of a hash is faster than creating one
66 # deep copy of a hash is faster than creating one
67 s = nullhash.copy()
67 s = nullhash.copy()
68 s.update(p1)
68 s.update(p1)
69 else:
69 else:
70 # none of the parent nodes are nullid
70 # none of the parent nodes are nullid
71 l = [p1, p2]
71 l = [p1, p2]
72 l.sort()
72 l.sort()
73 s = _sha(l[0])
73 s = _sha(l[0])
74 s.update(l[1])
74 s.update(l[1])
75 s.update(text)
75 s.update(text)
76 return s.digest()
76 return s.digest()
77
77
78 def decompress(bin):
78 def decompress(bin):
79 """ decompress the given input """
79 """ decompress the given input """
80 if not bin:
80 if not bin:
81 return bin
81 return bin
82 t = bin[0]
82 t = bin[0]
83 if t == '\0':
83 if t == '\0':
84 return bin
84 return bin
85 if t == 'x':
85 if t == 'x':
86 try:
86 try:
87 return _decompress(bin)
87 return _decompress(bin)
88 except zlib.error, e:
88 except zlib.error, e:
89 raise RevlogError(_("revlog decompress error: %s") % str(e))
89 raise RevlogError(_("revlog decompress error: %s") % str(e))
90 if t == 'u':
90 if t == 'u':
91 return bin[1:]
91 return bin[1:]
92 raise RevlogError(_("unknown compression type %r") % t)
92 raise RevlogError(_("unknown compression type %r") % t)
93
93
94 # index v0:
94 # index v0:
95 # 4 bytes: offset
95 # 4 bytes: offset
96 # 4 bytes: compressed length
96 # 4 bytes: compressed length
97 # 4 bytes: base rev
97 # 4 bytes: base rev
98 # 4 bytes: link rev
98 # 4 bytes: link rev
99 # 32 bytes: parent 1 nodeid
99 # 32 bytes: parent 1 nodeid
100 # 32 bytes: parent 2 nodeid
100 # 32 bytes: parent 2 nodeid
101 # 32 bytes: nodeid
101 # 32 bytes: nodeid
102 indexformatv0 = ">4l20s20s20s"
102 indexformatv0 = ">4l20s20s20s"
103 v0shaoffset = 56
103 v0shaoffset = 56
104
104
105 class revlogoldio(object):
105 class revlogoldio(object):
106 def __init__(self):
106 def __init__(self):
107 self.size = struct.calcsize(indexformatv0)
107 self.size = struct.calcsize(indexformatv0)
108
108
109 def parseindex(self, data, inline):
109 def parseindex(self, data, inline):
110 s = self.size
110 s = self.size
111 index = []
111 index = []
112 nodemap = {nullid: nullrev}
112 nodemap = {nullid: nullrev}
113 n = off = 0
113 n = off = 0
114 l = len(data)
114 l = len(data)
115 while off + s <= l:
115 while off + s <= l:
116 cur = data[off:off + s]
116 cur = data[off:off + s]
117 off += s
117 off += s
118 e = _unpack(indexformatv0, cur)
118 e = _unpack(indexformatv0, cur)
119 # transform to revlogv1 format
119 # transform to revlogv1 format
120 e2 = (offset_type(e[0], 0), e[1], -1, e[2], e[3],
120 e2 = (offset_type(e[0], 0), e[1], -1, e[2], e[3],
121 nodemap.get(e[4], nullrev), nodemap.get(e[5], nullrev), e[6])
121 nodemap.get(e[4], nullrev), nodemap.get(e[5], nullrev), e[6])
122 index.append(e2)
122 index.append(e2)
123 nodemap[e[6]] = n
123 nodemap[e[6]] = n
124 n += 1
124 n += 1
125
125
126 # add the magic null revision at -1
126 # add the magic null revision at -1
127 index.append((0, 0, 0, -1, -1, -1, -1, nullid))
127 index.append((0, 0, 0, -1, -1, -1, -1, nullid))
128
128
129 return index, nodemap, None
129 return index, nodemap, None
130
130
131 def packentry(self, entry, node, version, rev):
131 def packentry(self, entry, node, version, rev):
132 if gettype(entry[0]):
132 if gettype(entry[0]):
133 raise RevlogError(_("index entry flags need RevlogNG"))
133 raise RevlogError(_("index entry flags need RevlogNG"))
134 e2 = (getoffset(entry[0]), entry[1], entry[3], entry[4],
134 e2 = (getoffset(entry[0]), entry[1], entry[3], entry[4],
135 node(entry[5]), node(entry[6]), entry[7])
135 node(entry[5]), node(entry[6]), entry[7])
136 return _pack(indexformatv0, *e2)
136 return _pack(indexformatv0, *e2)
137
137
138 # index ng:
138 # index ng:
139 # 6 bytes: offset
139 # 6 bytes: offset
140 # 2 bytes: flags
140 # 2 bytes: flags
141 # 4 bytes: compressed length
141 # 4 bytes: compressed length
142 # 4 bytes: uncompressed length
142 # 4 bytes: uncompressed length
143 # 4 bytes: base rev
143 # 4 bytes: base rev
144 # 4 bytes: link rev
144 # 4 bytes: link rev
145 # 4 bytes: parent 1 rev
145 # 4 bytes: parent 1 rev
146 # 4 bytes: parent 2 rev
146 # 4 bytes: parent 2 rev
147 # 32 bytes: nodeid
147 # 32 bytes: nodeid
148 indexformatng = ">Qiiiiii20s12x"
148 indexformatng = ">Qiiiiii20s12x"
149 ngshaoffset = 32
149 ngshaoffset = 32
150 versionformat = ">I"
150 versionformat = ">I"
151
151
152 class revlogio(object):
152 class revlogio(object):
153 def __init__(self):
153 def __init__(self):
154 self.size = struct.calcsize(indexformatng)
154 self.size = struct.calcsize(indexformatng)
155
155
156 def parseindex(self, data, inline):
156 def parseindex(self, data, inline):
157 # call the C implementation to parse the index data
157 # call the C implementation to parse the index data
158 index, cache = parsers.parse_index2(data, inline)
158 index, cache = parsers.parse_index2(data, inline)
159 return index, getattr(index, 'nodemap', None), cache
159 return index, getattr(index, 'nodemap', None), cache
160
160
161 def packentry(self, entry, node, version, rev):
161 def packentry(self, entry, node, version, rev):
162 p = _pack(indexformatng, *entry)
162 p = _pack(indexformatng, *entry)
163 if rev == 0:
163 if rev == 0:
164 p = _pack(versionformat, version) + p[4:]
164 p = _pack(versionformat, version) + p[4:]
165 return p
165 return p
166
166
167 class revlog(object):
167 class revlog(object):
168 """
168 """
169 the underlying revision storage object
169 the underlying revision storage object
170
170
171 A revlog consists of two parts, an index and the revision data.
171 A revlog consists of two parts, an index and the revision data.
172
172
173 The index is a file with a fixed record size containing
173 The index is a file with a fixed record size containing
174 information on each revision, including its nodeid (hash), the
174 information on each revision, including its nodeid (hash), the
175 nodeids of its parents, the position and offset of its data within
175 nodeids of its parents, the position and offset of its data within
176 the data file, and the revision it's based on. Finally, each entry
176 the data file, and the revision it's based on. Finally, each entry
177 contains a linkrev entry that can serve as a pointer to external
177 contains a linkrev entry that can serve as a pointer to external
178 data.
178 data.
179
179
180 The revision data itself is a linear collection of data chunks.
180 The revision data itself is a linear collection of data chunks.
181 Each chunk represents a revision and is usually represented as a
181 Each chunk represents a revision and is usually represented as a
182 delta against the previous chunk. To bound lookup time, runs of
182 delta against the previous chunk. To bound lookup time, runs of
183 deltas are limited to about 2 times the length of the original
183 deltas are limited to about 2 times the length of the original
184 version data. This makes retrieval of a version proportional to
184 version data. This makes retrieval of a version proportional to
185 its size, or O(1) relative to the number of revisions.
185 its size, or O(1) relative to the number of revisions.
186
186
187 Both pieces of the revlog are written to in an append-only
187 Both pieces of the revlog are written to in an append-only
188 fashion, which means we never need to rewrite a file to insert or
188 fashion, which means we never need to rewrite a file to insert or
189 remove data, and can use some simple techniques to avoid the need
189 remove data, and can use some simple techniques to avoid the need
190 for locking while reading.
190 for locking while reading.
191 """
191 """
192 def __init__(self, opener, indexfile):
192 def __init__(self, opener, indexfile):
193 """
193 """
194 create a revlog object
194 create a revlog object
195
195
196 opener is a function that abstracts the file opening operation
196 opener is a function that abstracts the file opening operation
197 and can be used to implement COW semantics or the like.
197 and can be used to implement COW semantics or the like.
198 """
198 """
199 self.indexfile = indexfile
199 self.indexfile = indexfile
200 self.datafile = indexfile[:-2] + ".d"
200 self.datafile = indexfile[:-2] + ".d"
201 self.opener = opener
201 self.opener = opener
202 self._cache = None
202 self._cache = None
203 self._basecache = (0, 0)
203 self._basecache = (0, 0)
204 self._chunkcache = (0, '')
204 self._chunkcache = (0, '')
205 self.index = []
205 self.index = []
206 self._pcache = {}
206 self._pcache = {}
207 self._nodecache = {nullid: nullrev}
207 self._nodecache = {nullid: nullrev}
208 self._nodepos = None
208 self._nodepos = None
209
209
210 v = REVLOG_DEFAULT_VERSION
210 v = REVLOG_DEFAULT_VERSION
211 opts = getattr(opener, 'options', None)
211 opts = getattr(opener, 'options', None)
212 if opts is not None:
212 if opts is not None:
213 if 'revlogv1' in opts:
213 if 'revlogv1' in opts:
214 if 'generaldelta' in opts:
214 if 'generaldelta' in opts:
215 v |= REVLOGGENERALDELTA
215 v |= REVLOGGENERALDELTA
216 else:
216 else:
217 v = 0
217 v = 0
218
218
219 i = ''
219 i = ''
220 self._initempty = True
220 self._initempty = True
221 try:
221 try:
222 f = self.opener(self.indexfile)
222 f = self.opener(self.indexfile)
223 i = f.read()
223 i = f.read()
224 f.close()
224 f.close()
225 if len(i) > 0:
225 if len(i) > 0:
226 v = struct.unpack(versionformat, i[:4])[0]
226 v = struct.unpack(versionformat, i[:4])[0]
227 self._initempty = False
227 self._initempty = False
228 except IOError, inst:
228 except IOError, inst:
229 if inst.errno != errno.ENOENT:
229 if inst.errno != errno.ENOENT:
230 raise
230 raise
231
231
232 self.version = v
232 self.version = v
233 self._inline = v & REVLOGNGINLINEDATA
233 self._inline = v & REVLOGNGINLINEDATA
234 self._generaldelta = v & REVLOGGENERALDELTA
234 self._generaldelta = v & REVLOGGENERALDELTA
235 flags = v & ~0xFFFF
235 flags = v & ~0xFFFF
236 fmt = v & 0xFFFF
236 fmt = v & 0xFFFF
237 if fmt == REVLOGV0 and flags:
237 if fmt == REVLOGV0 and flags:
238 raise RevlogError(_("index %s unknown flags %#04x for format v0")
238 raise RevlogError(_("index %s unknown flags %#04x for format v0")
239 % (self.indexfile, flags >> 16))
239 % (self.indexfile, flags >> 16))
240 elif fmt == REVLOGNG and flags & ~REVLOGNG_FLAGS:
240 elif fmt == REVLOGNG and flags & ~REVLOGNG_FLAGS:
241 raise RevlogError(_("index %s unknown flags %#04x for revlogng")
241 raise RevlogError(_("index %s unknown flags %#04x for revlogng")
242 % (self.indexfile, flags >> 16))
242 % (self.indexfile, flags >> 16))
243 elif fmt > REVLOGNG:
243 elif fmt > REVLOGNG:
244 raise RevlogError(_("index %s unknown format %d")
244 raise RevlogError(_("index %s unknown format %d")
245 % (self.indexfile, fmt))
245 % (self.indexfile, fmt))
246
246
247 self._io = revlogio()
247 self._io = revlogio()
248 if self.version == REVLOGV0:
248 if self.version == REVLOGV0:
249 self._io = revlogoldio()
249 self._io = revlogoldio()
250 try:
250 try:
251 d = self._io.parseindex(i, self._inline)
251 d = self._io.parseindex(i, self._inline)
252 except (ValueError, IndexError):
252 except (ValueError, IndexError):
253 raise RevlogError(_("index %s is corrupted") % (self.indexfile))
253 raise RevlogError(_("index %s is corrupted") % (self.indexfile))
254 self.index, nodemap, self._chunkcache = d
254 self.index, nodemap, self._chunkcache = d
255 if nodemap is not None:
255 if nodemap is not None:
256 self.nodemap = self._nodecache = nodemap
256 self.nodemap = self._nodecache = nodemap
257 if not self._chunkcache:
257 if not self._chunkcache:
258 self._chunkclear()
258 self._chunkclear()
259
259
260 def tip(self):
260 def tip(self):
261 return self.node(len(self.index) - 2)
261 return self.node(len(self.index) - 2)
262 def __len__(self):
262 def __len__(self):
263 return len(self.index) - 1
263 return len(self.index) - 1
264 def __iter__(self):
264 def __iter__(self):
265 return iter(xrange(len(self)))
265 return iter(xrange(len(self)))
266 def revs(self, start=0, stop=None):
266 def revs(self, start=0, stop=None):
267 """iterate over all rev in this revlog (from start to stop)"""
267 """iterate over all rev in this revlog (from start to stop)"""
268 step = 1
268 step = 1
269 if stop is not None:
269 if stop is not None:
270 if start > stop:
270 if start > stop:
271 step = -1
271 step = -1
272 stop += step
272 stop += step
273 else:
273 else:
274 stop = len(self)
274 stop = len(self)
275 return xrange(start, stop, step)
275 return xrange(start, stop, step)
276
276
277 @util.propertycache
277 @util.propertycache
278 def nodemap(self):
278 def nodemap(self):
279 self.rev(self.node(0))
279 self.rev(self.node(0))
280 return self._nodecache
280 return self._nodecache
281
281
282 def hasnode(self, node):
282 def hasnode(self, node):
283 try:
283 try:
284 self.rev(node)
284 self.rev(node)
285 return True
285 return True
286 except KeyError:
286 except KeyError:
287 return False
287 return False
288
288
289 def clearcaches(self):
289 def clearcaches(self):
290 try:
290 try:
291 self._nodecache.clearcaches()
291 self._nodecache.clearcaches()
292 except AttributeError:
292 except AttributeError:
293 self._nodecache = {nullid: nullrev}
293 self._nodecache = {nullid: nullrev}
294 self._nodepos = None
294 self._nodepos = None
295
295
296 def rev(self, node):
296 def rev(self, node):
297 try:
297 try:
298 return self._nodecache[node]
298 return self._nodecache[node]
299 except RevlogError:
299 except RevlogError:
300 # parsers.c radix tree lookup failed
300 # parsers.c radix tree lookup failed
301 raise LookupError(node, self.indexfile, _('no node'))
301 raise LookupError(node, self.indexfile, _('no node'))
302 except KeyError:
302 except KeyError:
303 # pure python cache lookup failed
303 # pure python cache lookup failed
304 n = self._nodecache
304 n = self._nodecache
305 i = self.index
305 i = self.index
306 p = self._nodepos
306 p = self._nodepos
307 if p is None:
307 if p is None:
308 p = len(i) - 2
308 p = len(i) - 2
309 for r in xrange(p, -1, -1):
309 for r in xrange(p, -1, -1):
310 v = i[r][7]
310 v = i[r][7]
311 n[v] = r
311 n[v] = r
312 if v == node:
312 if v == node:
313 self._nodepos = r - 1
313 self._nodepos = r - 1
314 return r
314 return r
315 raise LookupError(node, self.indexfile, _('no node'))
315 raise LookupError(node, self.indexfile, _('no node'))
316
316
317 def node(self, rev):
317 def node(self, rev):
318 return self.index[rev][7]
318 return self.index[rev][7]
319 def linkrev(self, rev):
319 def linkrev(self, rev):
320 return self.index[rev][4]
320 return self.index[rev][4]
321 def parents(self, node):
321 def parents(self, node):
322 i = self.index
322 i = self.index
323 d = i[self.rev(node)]
323 d = i[self.rev(node)]
324 return i[d[5]][7], i[d[6]][7] # map revisions to nodes inline
324 return i[d[5]][7], i[d[6]][7] # map revisions to nodes inline
325 def parentrevs(self, rev):
325 def parentrevs(self, rev):
326 return self.index[rev][5:7]
326 return self.index[rev][5:7]
327 def start(self, rev):
327 def start(self, rev):
328 return int(self.index[rev][0] >> 16)
328 return int(self.index[rev][0] >> 16)
329 def end(self, rev):
329 def end(self, rev):
330 return self.start(rev) + self.length(rev)
330 return self.start(rev) + self.length(rev)
331 def length(self, rev):
331 def length(self, rev):
332 return self.index[rev][1]
332 return self.index[rev][1]
333 def chainbase(self, rev):
333 def chainbase(self, rev):
334 index = self.index
334 index = self.index
335 base = index[rev][3]
335 base = index[rev][3]
336 while base != rev:
336 while base != rev:
337 rev = base
337 rev = base
338 base = index[rev][3]
338 base = index[rev][3]
339 return base
339 return base
340 def flags(self, rev):
340 def flags(self, rev):
341 return self.index[rev][0] & 0xFFFF
341 return self.index[rev][0] & 0xFFFF
342 def rawsize(self, rev):
342 def rawsize(self, rev):
343 """return the length of the uncompressed text for a given revision"""
343 """return the length of the uncompressed text for a given revision"""
344 l = self.index[rev][2]
344 l = self.index[rev][2]
345 if l >= 0:
345 if l >= 0:
346 return l
346 return l
347
347
348 t = self.revision(self.node(rev))
348 t = self.revision(self.node(rev))
349 return len(t)
349 return len(t)
350 size = rawsize
350 size = rawsize
351
351
352 def ancestors(self, revs, stoprev=0, inclusive=False):
352 def ancestors(self, revs, stoprev=0, inclusive=False):
353 """Generate the ancestors of 'revs' in reverse topological order.
353 """Generate the ancestors of 'revs' in reverse topological order.
354 Does not generate revs lower than stoprev.
354 Does not generate revs lower than stoprev.
355
355
356 See the documentation for ancestor.lazyancestors for more details."""
356 See the documentation for ancestor.lazyancestors for more details."""
357
357
358 return ancestor.lazyancestors(self, revs, stoprev=stoprev,
358 return ancestor.lazyancestors(self, revs, stoprev=stoprev,
359 inclusive=inclusive)
359 inclusive=inclusive)
360
360
361 def descendants(self, revs):
361 def descendants(self, revs):
362 """Generate the descendants of 'revs' in revision order.
362 """Generate the descendants of 'revs' in revision order.
363
363
364 Yield a sequence of revision numbers starting with a child of
364 Yield a sequence of revision numbers starting with a child of
365 some rev in revs, i.e., each revision is *not* considered a
365 some rev in revs, i.e., each revision is *not* considered a
366 descendant of itself. Results are ordered by revision number (a
366 descendant of itself. Results are ordered by revision number (a
367 topological sort)."""
367 topological sort)."""
368 first = min(revs)
368 first = min(revs)
369 if first == nullrev:
369 if first == nullrev:
370 for i in self:
370 for i in self:
371 yield i
371 yield i
372 return
372 return
373
373
374 seen = set(revs)
374 seen = set(revs)
375 for i in self.revs(start=first + 1):
375 for i in self.revs(start=first + 1):
376 for x in self.parentrevs(i):
376 for x in self.parentrevs(i):
377 if x != nullrev and x in seen:
377 if x != nullrev and x in seen:
378 seen.add(i)
378 seen.add(i)
379 yield i
379 yield i
380 break
380 break
381
381
382 def findcommonmissing(self, common=None, heads=None):
382 def findcommonmissing(self, common=None, heads=None):
383 """Return a tuple of the ancestors of common and the ancestors of heads
383 """Return a tuple of the ancestors of common and the ancestors of heads
384 that are not ancestors of common. In revset terminology, we return the
384 that are not ancestors of common. In revset terminology, we return the
385 tuple:
385 tuple:
386
386
387 ::common, (::heads) - (::common)
387 ::common, (::heads) - (::common)
388
388
389 The list is sorted by revision number, meaning it is
389 The list is sorted by revision number, meaning it is
390 topologically sorted.
390 topologically sorted.
391
391
392 'heads' and 'common' are both lists of node IDs. If heads is
392 'heads' and 'common' are both lists of node IDs. If heads is
393 not supplied, uses all of the revlog's heads. If common is not
393 not supplied, uses all of the revlog's heads. If common is not
394 supplied, uses nullid."""
394 supplied, uses nullid."""
395 if common is None:
395 if common is None:
396 common = [nullid]
396 common = [nullid]
397 if heads is None:
397 if heads is None:
398 heads = self.heads()
398 heads = self.heads()
399
399
400 common = [self.rev(n) for n in common]
400 common = [self.rev(n) for n in common]
401 heads = [self.rev(n) for n in heads]
401 heads = [self.rev(n) for n in heads]
402
402
403 # we want the ancestors, but inclusive
403 # we want the ancestors, but inclusive
404 has = set(self.ancestors(common))
404 has = set(self.ancestors(common))
405 has.add(nullrev)
405 has.add(nullrev)
406 has.update(common)
406 has.update(common)
407
407
408 # take all ancestors from heads that aren't in has
408 # take all ancestors from heads that aren't in has
409 missing = set()
409 missing = set()
410 visit = util.deque(r for r in heads if r not in has)
410 visit = util.deque(r for r in heads if r not in has)
411 while visit:
411 while visit:
412 r = visit.popleft()
412 r = visit.popleft()
413 if r in missing:
413 if r in missing:
414 continue
414 continue
415 else:
415 else:
416 missing.add(r)
416 missing.add(r)
417 for p in self.parentrevs(r):
417 for p in self.parentrevs(r):
418 if p not in has:
418 if p not in has:
419 visit.append(p)
419 visit.append(p)
420 missing = list(missing)
420 missing = list(missing)
421 missing.sort()
421 missing.sort()
422 return has, [self.node(r) for r in missing]
422 return has, [self.node(r) for r in missing]
423
423
424 def findmissingrevs(self, common=None, heads=None):
424 def findmissingrevs(self, common=None, heads=None):
425 """Return the revision numbers of the ancestors of heads that
425 """Return the revision numbers of the ancestors of heads that
426 are not ancestors of common.
426 are not ancestors of common.
427
427
428 More specifically, return a list of revision numbers corresponding to
428 More specifically, return a list of revision numbers corresponding to
429 nodes N such that every N satisfies the following constraints:
429 nodes N such that every N satisfies the following constraints:
430
430
431 1. N is an ancestor of some node in 'heads'
431 1. N is an ancestor of some node in 'heads'
432 2. N is not an ancestor of any node in 'common'
432 2. N is not an ancestor of any node in 'common'
433
433
434 The list is sorted by revision number, meaning it is
434 The list is sorted by revision number, meaning it is
435 topologically sorted.
435 topologically sorted.
436
436
437 'heads' and 'common' are both lists of revision numbers. If heads is
437 'heads' and 'common' are both lists of revision numbers. If heads is
438 not supplied, uses all of the revlog's heads. If common is not
438 not supplied, uses all of the revlog's heads. If common is not
439 supplied, uses nullid."""
439 supplied, uses nullid."""
440 if common is None:
440 if common is None:
441 common = [nullrev]
441 common = [nullrev]
442 if heads is None:
442 if heads is None:
443 heads = self.headrevs()
443 heads = self.headrevs()
444
444
445 return ancestor.missingancestors(heads, common, self.parentrevs)
445 return ancestor.missingancestors(heads, common, self.parentrevs)
446
446
447 def findmissing(self, common=None, heads=None):
447 def findmissing(self, common=None, heads=None):
448 """Return the ancestors of heads that are not ancestors of common.
448 """Return the ancestors of heads that are not ancestors of common.
449
449
450 More specifically, return a list of nodes N such that every N
450 More specifically, return a list of nodes N such that every N
451 satisfies the following constraints:
451 satisfies the following constraints:
452
452
453 1. N is an ancestor of some node in 'heads'
453 1. N is an ancestor of some node in 'heads'
454 2. N is not an ancestor of any node in 'common'
454 2. N is not an ancestor of any node in 'common'
455
455
456 The list is sorted by revision number, meaning it is
456 The list is sorted by revision number, meaning it is
457 topologically sorted.
457 topologically sorted.
458
458
459 'heads' and 'common' are both lists of node IDs. If heads is
459 'heads' and 'common' are both lists of node IDs. If heads is
460 not supplied, uses all of the revlog's heads. If common is not
460 not supplied, uses all of the revlog's heads. If common is not
461 supplied, uses nullid."""
461 supplied, uses nullid."""
462 if common is None:
462 if common is None:
463 common = [nullid]
463 common = [nullid]
464 if heads is None:
464 if heads is None:
465 heads = self.heads()
465 heads = self.heads()
466
466
467 common = [self.rev(n) for n in common]
467 common = [self.rev(n) for n in common]
468 heads = [self.rev(n) for n in heads]
468 heads = [self.rev(n) for n in heads]
469
469
470 return [self.node(r) for r in
470 return [self.node(r) for r in
471 ancestor.missingancestors(heads, common, self.parentrevs)]
471 ancestor.missingancestors(heads, common, self.parentrevs)]
472
472
473 def nodesbetween(self, roots=None, heads=None):
473 def nodesbetween(self, roots=None, heads=None):
474 """Return a topological path from 'roots' to 'heads'.
474 """Return a topological path from 'roots' to 'heads'.
475
475
476 Return a tuple (nodes, outroots, outheads) where 'nodes' is a
476 Return a tuple (nodes, outroots, outheads) where 'nodes' is a
477 topologically sorted list of all nodes N that satisfy both of
477 topologically sorted list of all nodes N that satisfy both of
478 these constraints:
478 these constraints:
479
479
480 1. N is a descendant of some node in 'roots'
480 1. N is a descendant of some node in 'roots'
481 2. N is an ancestor of some node in 'heads'
481 2. N is an ancestor of some node in 'heads'
482
482
483 Every node is considered to be both a descendant and an ancestor
483 Every node is considered to be both a descendant and an ancestor
484 of itself, so every reachable node in 'roots' and 'heads' will be
484 of itself, so every reachable node in 'roots' and 'heads' will be
485 included in 'nodes'.
485 included in 'nodes'.
486
486
487 'outroots' is the list of reachable nodes in 'roots', i.e., the
487 'outroots' is the list of reachable nodes in 'roots', i.e., the
488 subset of 'roots' that is returned in 'nodes'. Likewise,
488 subset of 'roots' that is returned in 'nodes'. Likewise,
489 'outheads' is the subset of 'heads' that is also in 'nodes'.
489 'outheads' is the subset of 'heads' that is also in 'nodes'.
490
490
491 'roots' and 'heads' are both lists of node IDs. If 'roots' is
491 'roots' and 'heads' are both lists of node IDs. If 'roots' is
492 unspecified, uses nullid as the only root. If 'heads' is
492 unspecified, uses nullid as the only root. If 'heads' is
493 unspecified, uses list of all of the revlog's heads."""
493 unspecified, uses list of all of the revlog's heads."""
494 nonodes = ([], [], [])
494 nonodes = ([], [], [])
495 if roots is not None:
495 if roots is not None:
496 roots = list(roots)
496 roots = list(roots)
497 if not roots:
497 if not roots:
498 return nonodes
498 return nonodes
499 lowestrev = min([self.rev(n) for n in roots])
499 lowestrev = min([self.rev(n) for n in roots])
500 else:
500 else:
501 roots = [nullid] # Everybody's a descendant of nullid
501 roots = [nullid] # Everybody's a descendant of nullid
502 lowestrev = nullrev
502 lowestrev = nullrev
503 if (lowestrev == nullrev) and (heads is None):
503 if (lowestrev == nullrev) and (heads is None):
504 # We want _all_ the nodes!
504 # We want _all_ the nodes!
505 return ([self.node(r) for r in self], [nullid], list(self.heads()))
505 return ([self.node(r) for r in self], [nullid], list(self.heads()))
506 if heads is None:
506 if heads is None:
507 # All nodes are ancestors, so the latest ancestor is the last
507 # All nodes are ancestors, so the latest ancestor is the last
508 # node.
508 # node.
509 highestrev = len(self) - 1
509 highestrev = len(self) - 1
510 # Set ancestors to None to signal that every node is an ancestor.
510 # Set ancestors to None to signal that every node is an ancestor.
511 ancestors = None
511 ancestors = None
512 # Set heads to an empty dictionary for later discovery of heads
512 # Set heads to an empty dictionary for later discovery of heads
513 heads = {}
513 heads = {}
514 else:
514 else:
515 heads = list(heads)
515 heads = list(heads)
516 if not heads:
516 if not heads:
517 return nonodes
517 return nonodes
518 ancestors = set()
518 ancestors = set()
519 # Turn heads into a dictionary so we can remove 'fake' heads.
519 # Turn heads into a dictionary so we can remove 'fake' heads.
520 # Also, later we will be using it to filter out the heads we can't
520 # Also, later we will be using it to filter out the heads we can't
521 # find from roots.
521 # find from roots.
522 heads = dict.fromkeys(heads, False)
522 heads = dict.fromkeys(heads, False)
523 # Start at the top and keep marking parents until we're done.
523 # Start at the top and keep marking parents until we're done.
524 nodestotag = set(heads)
524 nodestotag = set(heads)
525 # Remember where the top was so we can use it as a limit later.
525 # Remember where the top was so we can use it as a limit later.
526 highestrev = max([self.rev(n) for n in nodestotag])
526 highestrev = max([self.rev(n) for n in nodestotag])
527 while nodestotag:
527 while nodestotag:
528 # grab a node to tag
528 # grab a node to tag
529 n = nodestotag.pop()
529 n = nodestotag.pop()
530 # Never tag nullid
530 # Never tag nullid
531 if n == nullid:
531 if n == nullid:
532 continue
532 continue
533 # A node's revision number represents its place in a
533 # A node's revision number represents its place in a
534 # topologically sorted list of nodes.
534 # topologically sorted list of nodes.
535 r = self.rev(n)
535 r = self.rev(n)
536 if r >= lowestrev:
536 if r >= lowestrev:
537 if n not in ancestors:
537 if n not in ancestors:
538 # If we are possibly a descendant of one of the roots
538 # If we are possibly a descendant of one of the roots
539 # and we haven't already been marked as an ancestor
539 # and we haven't already been marked as an ancestor
540 ancestors.add(n) # Mark as ancestor
540 ancestors.add(n) # Mark as ancestor
541 # Add non-nullid parents to list of nodes to tag.
541 # Add non-nullid parents to list of nodes to tag.
542 nodestotag.update([p for p in self.parents(n) if
542 nodestotag.update([p for p in self.parents(n) if
543 p != nullid])
543 p != nullid])
544 elif n in heads: # We've seen it before, is it a fake head?
544 elif n in heads: # We've seen it before, is it a fake head?
545 # So it is, real heads should not be the ancestors of
545 # So it is, real heads should not be the ancestors of
546 # any other heads.
546 # any other heads.
547 heads.pop(n)
547 heads.pop(n)
548 if not ancestors:
548 if not ancestors:
549 return nonodes
549 return nonodes
550 # Now that we have our set of ancestors, we want to remove any
550 # Now that we have our set of ancestors, we want to remove any
551 # roots that are not ancestors.
551 # roots that are not ancestors.
552
552
553 # If one of the roots was nullid, everything is included anyway.
553 # If one of the roots was nullid, everything is included anyway.
554 if lowestrev > nullrev:
554 if lowestrev > nullrev:
555 # But, since we weren't, let's recompute the lowest rev to not
555 # But, since we weren't, let's recompute the lowest rev to not
556 # include roots that aren't ancestors.
556 # include roots that aren't ancestors.
557
557
558 # Filter out roots that aren't ancestors of heads
558 # Filter out roots that aren't ancestors of heads
559 roots = [n for n in roots if n in ancestors]
559 roots = [n for n in roots if n in ancestors]
560 # Recompute the lowest revision
560 # Recompute the lowest revision
561 if roots:
561 if roots:
562 lowestrev = min([self.rev(n) for n in roots])
562 lowestrev = min([self.rev(n) for n in roots])
563 else:
563 else:
564 # No more roots? Return empty list
564 # No more roots? Return empty list
565 return nonodes
565 return nonodes
566 else:
566 else:
567 # We are descending from nullid, and don't need to care about
567 # We are descending from nullid, and don't need to care about
568 # any other roots.
568 # any other roots.
569 lowestrev = nullrev
569 lowestrev = nullrev
570 roots = [nullid]
570 roots = [nullid]
571 # Transform our roots list into a set.
571 # Transform our roots list into a set.
572 descendants = set(roots)
572 descendants = set(roots)
573 # Also, keep the original roots so we can filter out roots that aren't
573 # Also, keep the original roots so we can filter out roots that aren't
574 # 'real' roots (i.e. are descended from other roots).
574 # 'real' roots (i.e. are descended from other roots).
575 roots = descendants.copy()
575 roots = descendants.copy()
576 # Our topologically sorted list of output nodes.
576 # Our topologically sorted list of output nodes.
577 orderedout = []
577 orderedout = []
578 # Don't start at nullid since we don't want nullid in our output list,
578 # Don't start at nullid since we don't want nullid in our output list,
579 # and if nullid shows up in descendants, empty parents will look like
579 # and if nullid shows up in descendants, empty parents will look like
580 # they're descendants.
580 # they're descendants.
581 for r in self.revs(start=max(lowestrev, 0), stop=highestrev + 1):
581 for r in self.revs(start=max(lowestrev, 0), stop=highestrev + 1):
582 n = self.node(r)
582 n = self.node(r)
583 isdescendant = False
583 isdescendant = False
584 if lowestrev == nullrev: # Everybody is a descendant of nullid
584 if lowestrev == nullrev: # Everybody is a descendant of nullid
585 isdescendant = True
585 isdescendant = True
586 elif n in descendants:
586 elif n in descendants:
587 # n is already a descendant
587 # n is already a descendant
588 isdescendant = True
588 isdescendant = True
589 # This check only needs to be done here because all the roots
589 # This check only needs to be done here because all the roots
590 # will start being marked is descendants before the loop.
590 # will start being marked is descendants before the loop.
591 if n in roots:
591 if n in roots:
592 # If n was a root, check if it's a 'real' root.
592 # If n was a root, check if it's a 'real' root.
593 p = tuple(self.parents(n))
593 p = tuple(self.parents(n))
594 # If any of its parents are descendants, it's not a root.
594 # If any of its parents are descendants, it's not a root.
595 if (p[0] in descendants) or (p[1] in descendants):
595 if (p[0] in descendants) or (p[1] in descendants):
596 roots.remove(n)
596 roots.remove(n)
597 else:
597 else:
598 p = tuple(self.parents(n))
598 p = tuple(self.parents(n))
599 # A node is a descendant if either of its parents are
599 # A node is a descendant if either of its parents are
600 # descendants. (We seeded the dependents list with the roots
600 # descendants. (We seeded the dependents list with the roots
601 # up there, remember?)
601 # up there, remember?)
602 if (p[0] in descendants) or (p[1] in descendants):
602 if (p[0] in descendants) or (p[1] in descendants):
603 descendants.add(n)
603 descendants.add(n)
604 isdescendant = True
604 isdescendant = True
605 if isdescendant and ((ancestors is None) or (n in ancestors)):
605 if isdescendant and ((ancestors is None) or (n in ancestors)):
606 # Only include nodes that are both descendants and ancestors.
606 # Only include nodes that are both descendants and ancestors.
607 orderedout.append(n)
607 orderedout.append(n)
608 if (ancestors is not None) and (n in heads):
608 if (ancestors is not None) and (n in heads):
609 # We're trying to figure out which heads are reachable
609 # We're trying to figure out which heads are reachable
610 # from roots.
610 # from roots.
611 # Mark this head as having been reached
611 # Mark this head as having been reached
612 heads[n] = True
612 heads[n] = True
613 elif ancestors is None:
613 elif ancestors is None:
614 # Otherwise, we're trying to discover the heads.
614 # Otherwise, we're trying to discover the heads.
615 # Assume this is a head because if it isn't, the next step
615 # Assume this is a head because if it isn't, the next step
616 # will eventually remove it.
616 # will eventually remove it.
617 heads[n] = True
617 heads[n] = True
618 # But, obviously its parents aren't.
618 # But, obviously its parents aren't.
619 for p in self.parents(n):
619 for p in self.parents(n):
620 heads.pop(p, None)
620 heads.pop(p, None)
621 heads = [n for n, flag in heads.iteritems() if flag]
621 heads = [n for n, flag in heads.iteritems() if flag]
622 roots = list(roots)
622 roots = list(roots)
623 assert orderedout
623 assert orderedout
624 assert roots
624 assert roots
625 assert heads
625 assert heads
626 return (orderedout, roots, heads)
626 return (orderedout, roots, heads)
627
627
628 def headrevs(self):
628 def headrevs(self):
629 try:
629 try:
630 return self.index.headrevs()
630 return self.index.headrevs()
631 except AttributeError:
631 except AttributeError:
632 return self._headrevs()
632 return self._headrevs()
633
633
634 def _headrevs(self):
634 def _headrevs(self):
635 count = len(self)
635 count = len(self)
636 if not count:
636 if not count:
637 return [nullrev]
637 return [nullrev]
638 # we won't iter over filtered rev so nobody is a head at start
638 # we won't iter over filtered rev so nobody is a head at start
639 ishead = [0] * (count + 1)
639 ishead = [0] * (count + 1)
640 index = self.index
640 index = self.index
641 for r in self:
641 for r in self:
642 ishead[r] = 1 # I may be an head
642 ishead[r] = 1 # I may be an head
643 e = index[r]
643 e = index[r]
644 ishead[e[5]] = ishead[e[6]] = 0 # my parent are not
644 ishead[e[5]] = ishead[e[6]] = 0 # my parent are not
645 return [r for r, val in enumerate(ishead) if val]
645 return [r for r, val in enumerate(ishead) if val]
646
646
647 def heads(self, start=None, stop=None):
647 def heads(self, start=None, stop=None):
648 """return the list of all nodes that have no children
648 """return the list of all nodes that have no children
649
649
650 if start is specified, only heads that are descendants of
650 if start is specified, only heads that are descendants of
651 start will be returned
651 start will be returned
652 if stop is specified, it will consider all the revs from stop
652 if stop is specified, it will consider all the revs from stop
653 as if they had no children
653 as if they had no children
654 """
654 """
655 if start is None and stop is None:
655 if start is None and stop is None:
656 if not len(self):
656 if not len(self):
657 return [nullid]
657 return [nullid]
658 return [self.node(r) for r in self.headrevs()]
658 return [self.node(r) for r in self.headrevs()]
659
659
660 if start is None:
660 if start is None:
661 start = nullid
661 start = nullid
662 if stop is None:
662 if stop is None:
663 stop = []
663 stop = []
664 stoprevs = set([self.rev(n) for n in stop])
664 stoprevs = set([self.rev(n) for n in stop])
665 startrev = self.rev(start)
665 startrev = self.rev(start)
666 reachable = set((startrev,))
666 reachable = set((startrev,))
667 heads = set((startrev,))
667 heads = set((startrev,))
668
668
669 parentrevs = self.parentrevs
669 parentrevs = self.parentrevs
670 for r in self.revs(start=startrev + 1):
670 for r in self.revs(start=startrev + 1):
671 for p in parentrevs(r):
671 for p in parentrevs(r):
672 if p in reachable:
672 if p in reachable:
673 if r not in stoprevs:
673 if r not in stoprevs:
674 reachable.add(r)
674 reachable.add(r)
675 heads.add(r)
675 heads.add(r)
676 if p in heads and p not in stoprevs:
676 if p in heads and p not in stoprevs:
677 heads.remove(p)
677 heads.remove(p)
678
678
679 return [self.node(r) for r in heads]
679 return [self.node(r) for r in heads]
680
680
681 def children(self, node):
681 def children(self, node):
682 """find the children of a given node"""
682 """find the children of a given node"""
683 c = []
683 c = []
684 p = self.rev(node)
684 p = self.rev(node)
685 for r in self.revs(start=p + 1):
685 for r in self.revs(start=p + 1):
686 prevs = [pr for pr in self.parentrevs(r) if pr != nullrev]
686 prevs = [pr for pr in self.parentrevs(r) if pr != nullrev]
687 if prevs:
687 if prevs:
688 for pr in prevs:
688 for pr in prevs:
689 if pr == p:
689 if pr == p:
690 c.append(self.node(r))
690 c.append(self.node(r))
691 elif p == nullrev:
691 elif p == nullrev:
692 c.append(self.node(r))
692 c.append(self.node(r))
693 return c
693 return c
694
694
695 def descendant(self, start, end):
695 def descendant(self, start, end):
696 if start == nullrev:
696 if start == nullrev:
697 return True
697 return True
698 for i in self.descendants([start]):
698 for i in self.descendants([start]):
699 if i == end:
699 if i == end:
700 return True
700 return True
701 elif i > end:
701 elif i > end:
702 break
702 break
703 return False
703 return False
704
704
705 def ancestor(self, a, b):
705 def ancestor(self, a, b):
706 """calculate the least common ancestor of nodes a and b"""
706 """calculate the least common ancestor of nodes a and b"""
707
707
708 a, b = self.rev(a), self.rev(b)
708 a, b = self.rev(a), self.rev(b)
709 try:
709 try:
710 ancs = self.index.ancestors(a, b)
710 ancs = self.index.ancestors(a, b)
711 except (AttributeError, OverflowError):
711 except (AttributeError, OverflowError):
712 ancs = ancestor.ancestors(self.parentrevs, a, b)
712 ancs = ancestor.ancestors(self.parentrevs, a, b)
713 if ancs:
713 if ancs:
714 # choose a consistent winner when there's a tie
714 # choose a consistent winner when there's a tie
715 return min(map(self.node, ancs))
715 return min(map(self.node, ancs))
716 return nullid
716 return nullid
717
717
718 def _match(self, id):
718 def _match(self, id):
719 if isinstance(id, int):
719 if isinstance(id, int):
720 # rev
720 # rev
721 return self.node(id)
721 return self.node(id)
722 if len(id) == 20:
722 if len(id) == 20:
723 # possibly a binary node
723 # possibly a binary node
724 # odds of a binary node being all hex in ASCII are 1 in 10**25
724 # odds of a binary node being all hex in ASCII are 1 in 10**25
725 try:
725 try:
726 node = id
726 node = id
727 self.rev(node) # quick search the index
727 self.rev(node) # quick search the index
728 return node
728 return node
729 except LookupError:
729 except LookupError:
730 pass # may be partial hex id
730 pass # may be partial hex id
731 try:
731 try:
732 # str(rev)
732 # str(rev)
733 rev = int(id)
733 rev = int(id)
734 if str(rev) != id:
734 if str(rev) != id:
735 raise ValueError
735 raise ValueError
736 if rev < 0:
736 if rev < 0:
737 rev = len(self) + rev
737 rev = len(self) + rev
738 if rev < 0 or rev >= len(self):
738 if rev < 0 or rev >= len(self):
739 raise ValueError
739 raise ValueError
740 return self.node(rev)
740 return self.node(rev)
741 except (ValueError, OverflowError):
741 except (ValueError, OverflowError):
742 pass
742 pass
743 if len(id) == 40:
743 if len(id) == 40:
744 try:
744 try:
745 # a full hex nodeid?
745 # a full hex nodeid?
746 node = bin(id)
746 node = bin(id)
747 self.rev(node)
747 self.rev(node)
748 return node
748 return node
749 except (TypeError, LookupError):
749 except (TypeError, LookupError):
750 pass
750 pass
751
751
752 def _partialmatch(self, id):
752 def _partialmatch(self, id):
753 try:
753 try:
754 n = self.index.partialmatch(id)
754 n = self.index.partialmatch(id)
755 if n and self.hasnode(n):
755 if n and self.hasnode(n):
756 return n
756 return n
757 return None
757 return None
758 except RevlogError:
758 except RevlogError:
759 # parsers.c radix tree lookup gave multiple matches
759 # parsers.c radix tree lookup gave multiple matches
760 # fall through to slow path that filters hidden revisions
760 # fall through to slow path that filters hidden revisions
761 pass
761 pass
762 except (AttributeError, ValueError):
762 except (AttributeError, ValueError):
763 # we are pure python, or key was too short to search radix tree
763 # we are pure python, or key was too short to search radix tree
764 pass
764 pass
765
765
766 if id in self._pcache:
766 if id in self._pcache:
767 return self._pcache[id]
767 return self._pcache[id]
768
768
769 if len(id) < 40:
769 if len(id) < 40:
770 try:
770 try:
771 # hex(node)[:...]
771 # hex(node)[:...]
772 l = len(id) // 2 # grab an even number of digits
772 l = len(id) // 2 # grab an even number of digits
773 prefix = bin(id[:l * 2])
773 prefix = bin(id[:l * 2])
774 nl = [e[7] for e in self.index if e[7].startswith(prefix)]
774 nl = [e[7] for e in self.index if e[7].startswith(prefix)]
775 nl = [n for n in nl if hex(n).startswith(id) and
775 nl = [n for n in nl if hex(n).startswith(id) and
776 self.hasnode(n)]
776 self.hasnode(n)]
777 if len(nl) > 0:
777 if len(nl) > 0:
778 if len(nl) == 1:
778 if len(nl) == 1:
779 self._pcache[id] = nl[0]
779 self._pcache[id] = nl[0]
780 return nl[0]
780 return nl[0]
781 raise LookupError(id, self.indexfile,
781 raise LookupError(id, self.indexfile,
782 _('ambiguous identifier'))
782 _('ambiguous identifier'))
783 return None
783 return None
784 except TypeError:
784 except TypeError:
785 pass
785 pass
786
786
787 def lookup(self, id):
787 def lookup(self, id):
788 """locate a node based on:
788 """locate a node based on:
789 - revision number or str(revision number)
789 - revision number or str(revision number)
790 - nodeid or subset of hex nodeid
790 - nodeid or subset of hex nodeid
791 """
791 """
792 n = self._match(id)
792 n = self._match(id)
793 if n is not None:
793 if n is not None:
794 return n
794 return n
795 n = self._partialmatch(id)
795 n = self._partialmatch(id)
796 if n:
796 if n:
797 return n
797 return n
798
798
799 raise LookupError(id, self.indexfile, _('no match found'))
799 raise LookupError(id, self.indexfile, _('no match found'))
800
800
801 def cmp(self, node, text):
801 def cmp(self, node, text):
802 """compare text with a given file revision
802 """compare text with a given file revision
803
803
804 returns True if text is different than what is stored.
804 returns True if text is different than what is stored.
805 """
805 """
806 p1, p2 = self.parents(node)
806 p1, p2 = self.parents(node)
807 return hash(text, p1, p2) != node
807 return hash(text, p1, p2) != node
808
808
809 def _addchunk(self, offset, data):
809 def _addchunk(self, offset, data):
810 o, d = self._chunkcache
810 o, d = self._chunkcache
811 # try to add to existing cache
811 # try to add to existing cache
812 if o + len(d) == offset and len(d) + len(data) < _chunksize:
812 if o + len(d) == offset and len(d) + len(data) < _chunksize:
813 self._chunkcache = o, d + data
813 self._chunkcache = o, d + data
814 else:
814 else:
815 self._chunkcache = offset, data
815 self._chunkcache = offset, data
816
816
817 def _loadchunk(self, offset, length):
817 def _loadchunk(self, offset, length):
818 if self._inline:
818 if self._inline:
819 df = self.opener(self.indexfile)
819 df = self.opener(self.indexfile)
820 else:
820 else:
821 df = self.opener(self.datafile)
821 df = self.opener(self.datafile)
822
822
823 readahead = max(65536, length)
823 readahead = max(65536, length)
824 df.seek(offset)
824 df.seek(offset)
825 d = df.read(readahead)
825 d = df.read(readahead)
826 df.close()
826 df.close()
827 self._addchunk(offset, d)
827 self._addchunk(offset, d)
828 if readahead > length:
828 if readahead > length:
829 return util.buffer(d, 0, length)
829 return util.buffer(d, 0, length)
830 return d
830 return d
831
831
832 def _getchunk(self, offset, length):
832 def _getchunk(self, offset, length):
833 o, d = self._chunkcache
833 o, d = self._chunkcache
834 l = len(d)
834 l = len(d)
835
835
836 # is it in the cache?
836 # is it in the cache?
837 cachestart = offset - o
837 cachestart = offset - o
838 cacheend = cachestart + length
838 cacheend = cachestart + length
839 if cachestart >= 0 and cacheend <= l:
839 if cachestart >= 0 and cacheend <= l:
840 if cachestart == 0 and cacheend == l:
840 if cachestart == 0 and cacheend == l:
841 return d # avoid a copy
841 return d # avoid a copy
842 return util.buffer(d, cachestart, cacheend - cachestart)
842 return util.buffer(d, cachestart, cacheend - cachestart)
843
843
844 return self._loadchunk(offset, length)
844 return self._loadchunk(offset, length)
845
845
846 def _chunkraw(self, startrev, endrev):
846 def _chunkraw(self, startrev, endrev):
847 start = self.start(startrev)
847 start = self.start(startrev)
848 end = self.end(endrev)
848 end = self.end(endrev)
849 if self._inline:
849 if self._inline:
850 start += (startrev + 1) * self._io.size
850 start += (startrev + 1) * self._io.size
851 end += (endrev + 1) * self._io.size
851 end += (endrev + 1) * self._io.size
852 length = end - start
852 length = end - start
853 return self._getchunk(start, length)
853 return self._getchunk(start, length)
854
854
855 def _chunk(self, rev):
855 def _chunk(self, rev):
856 return decompress(self._chunkraw(rev, rev))
856 return decompress(self._chunkraw(rev, rev))
857
857
858 def _chunks(self, revs):
858 def _chunks(self, revs):
859 '''faster version of [self._chunk(rev) for rev in revs]
859 '''faster version of [self._chunk(rev) for rev in revs]
860
860
861 Assumes that revs is in ascending order.'''
861 Assumes that revs is in ascending order.'''
862 if not revs:
863 return []
862 start = self.start
864 start = self.start
863 length = self.length
865 length = self.length
864 inline = self._inline
866 inline = self._inline
865 iosize = self._io.size
867 iosize = self._io.size
866 buffer = util.buffer
868 buffer = util.buffer
867
869
868 l = []
870 l = []
869 ladd = l.append
871 ladd = l.append
870
872
871 # XXX assume for now that chunkcache is preloaded
873 # preload the cache
874 self._chunkraw(revs[0], revs[-1])
872 offset, data = self._chunkcache
875 offset, data = self._chunkcache
873
876
874 for rev in revs:
877 for rev in revs:
875 chunkstart = start(rev)
878 chunkstart = start(rev)
876 if inline:
879 if inline:
877 chunkstart += (rev + 1) * iosize
880 chunkstart += (rev + 1) * iosize
878 chunklength = length(rev)
881 chunklength = length(rev)
879 ladd(decompress(buffer(data, chunkstart - offset, chunklength)))
882 ladd(decompress(buffer(data, chunkstart - offset, chunklength)))
880
883
881 return l
884 return l
882
885
883 def _chunkbase(self, rev):
886 def _chunkbase(self, rev):
884 return self._chunk(rev)
887 return self._chunk(rev)
885
888
886 def _chunkclear(self):
889 def _chunkclear(self):
887 self._chunkcache = (0, '')
890 self._chunkcache = (0, '')
888
891
889 def deltaparent(self, rev):
892 def deltaparent(self, rev):
890 """return deltaparent of the given revision"""
893 """return deltaparent of the given revision"""
891 base = self.index[rev][3]
894 base = self.index[rev][3]
892 if base == rev:
895 if base == rev:
893 return nullrev
896 return nullrev
894 elif self._generaldelta:
897 elif self._generaldelta:
895 return base
898 return base
896 else:
899 else:
897 return rev - 1
900 return rev - 1
898
901
899 def revdiff(self, rev1, rev2):
902 def revdiff(self, rev1, rev2):
900 """return or calculate a delta between two revisions"""
903 """return or calculate a delta between two revisions"""
901 if rev1 != nullrev and self.deltaparent(rev2) == rev1:
904 if rev1 != nullrev and self.deltaparent(rev2) == rev1:
902 return str(self._chunk(rev2))
905 return str(self._chunk(rev2))
903
906
904 return mdiff.textdiff(self.revision(rev1),
907 return mdiff.textdiff(self.revision(rev1),
905 self.revision(rev2))
908 self.revision(rev2))
906
909
907 def revision(self, nodeorrev):
910 def revision(self, nodeorrev):
908 """return an uncompressed revision of a given node or revision
911 """return an uncompressed revision of a given node or revision
909 number.
912 number.
910 """
913 """
911 if isinstance(nodeorrev, int):
914 if isinstance(nodeorrev, int):
912 rev = nodeorrev
915 rev = nodeorrev
913 node = self.node(rev)
916 node = self.node(rev)
914 else:
917 else:
915 node = nodeorrev
918 node = nodeorrev
916 rev = None
919 rev = None
917
920
918 cachedrev = None
921 cachedrev = None
919 if node == nullid:
922 if node == nullid:
920 return ""
923 return ""
921 if self._cache:
924 if self._cache:
922 if self._cache[0] == node:
925 if self._cache[0] == node:
923 return self._cache[2]
926 return self._cache[2]
924 cachedrev = self._cache[1]
927 cachedrev = self._cache[1]
925
928
926 # look up what we need to read
929 # look up what we need to read
927 text = None
930 text = None
928 if rev is None:
931 if rev is None:
929 rev = self.rev(node)
932 rev = self.rev(node)
930
933
931 # check rev flags
934 # check rev flags
932 if self.flags(rev) & ~REVIDX_KNOWN_FLAGS:
935 if self.flags(rev) & ~REVIDX_KNOWN_FLAGS:
933 raise RevlogError(_('incompatible revision flag %x') %
936 raise RevlogError(_('incompatible revision flag %x') %
934 (self.flags(rev) & ~REVIDX_KNOWN_FLAGS))
937 (self.flags(rev) & ~REVIDX_KNOWN_FLAGS))
935
938
936 # build delta chain
939 # build delta chain
937 chain = []
940 chain = []
938 index = self.index # for performance
941 index = self.index # for performance
939 generaldelta = self._generaldelta
942 generaldelta = self._generaldelta
940 iterrev = rev
943 iterrev = rev
941 e = index[iterrev]
944 e = index[iterrev]
942 while iterrev != e[3] and iterrev != cachedrev:
945 while iterrev != e[3] and iterrev != cachedrev:
943 chain.append(iterrev)
946 chain.append(iterrev)
944 if generaldelta:
947 if generaldelta:
945 iterrev = e[3]
948 iterrev = e[3]
946 else:
949 else:
947 iterrev -= 1
950 iterrev -= 1
948 e = index[iterrev]
951 e = index[iterrev]
949 chain.reverse()
950 base = iterrev
951
952
952 if iterrev == cachedrev:
953 if iterrev == cachedrev:
953 # cache hit
954 # cache hit
954 text = self._cache[2]
955 text = self._cache[2]
956 else:
957 chain.append(iterrev)
958 chain.reverse()
955
959
956 # drop cache to save memory
960 # drop cache to save memory
957 self._cache = None
961 self._cache = None
958
962
959 self._chunkraw(base, rev)
963 bins = self._chunks(chain)
960 if text is None:
964 if text is None:
961 text = str(self._chunkbase(base))
965 text = str(bins[0])
966 bins = bins[1:]
962
967
963 bins = self._chunks(chain)
964 text = mdiff.patches(text, bins)
968 text = mdiff.patches(text, bins)
965
969
966 text = self._checkhash(text, node, rev)
970 text = self._checkhash(text, node, rev)
967
971
968 self._cache = (node, rev, text)
972 self._cache = (node, rev, text)
969 return text
973 return text
970
974
971 def _checkhash(self, text, node, rev):
975 def _checkhash(self, text, node, rev):
972 p1, p2 = self.parents(node)
976 p1, p2 = self.parents(node)
973 self.checkhash(text, p1, p2, node, rev)
977 self.checkhash(text, p1, p2, node, rev)
974 return text
978 return text
975
979
976 def checkhash(self, text, p1, p2, node, rev=None):
980 def checkhash(self, text, p1, p2, node, rev=None):
977 if node != hash(text, p1, p2):
981 if node != hash(text, p1, p2):
978 revornode = rev
982 revornode = rev
979 if revornode is None:
983 if revornode is None:
980 revornode = templatefilters.short(hex(node))
984 revornode = templatefilters.short(hex(node))
981 raise RevlogError(_("integrity check failed on %s:%s")
985 raise RevlogError(_("integrity check failed on %s:%s")
982 % (self.indexfile, revornode))
986 % (self.indexfile, revornode))
983
987
984 def checkinlinesize(self, tr, fp=None):
988 def checkinlinesize(self, tr, fp=None):
985 if not self._inline or (self.start(-2) + self.length(-2)) < _maxinline:
989 if not self._inline or (self.start(-2) + self.length(-2)) < _maxinline:
986 return
990 return
987
991
988 trinfo = tr.find(self.indexfile)
992 trinfo = tr.find(self.indexfile)
989 if trinfo is None:
993 if trinfo is None:
990 raise RevlogError(_("%s not found in the transaction")
994 raise RevlogError(_("%s not found in the transaction")
991 % self.indexfile)
995 % self.indexfile)
992
996
993 trindex = trinfo[2]
997 trindex = trinfo[2]
994 dataoff = self.start(trindex)
998 dataoff = self.start(trindex)
995
999
996 tr.add(self.datafile, dataoff)
1000 tr.add(self.datafile, dataoff)
997
1001
998 if fp:
1002 if fp:
999 fp.flush()
1003 fp.flush()
1000 fp.close()
1004 fp.close()
1001
1005
1002 df = self.opener(self.datafile, 'w')
1006 df = self.opener(self.datafile, 'w')
1003 try:
1007 try:
1004 for r in self:
1008 for r in self:
1005 df.write(self._chunkraw(r, r))
1009 df.write(self._chunkraw(r, r))
1006 finally:
1010 finally:
1007 df.close()
1011 df.close()
1008
1012
1009 fp = self.opener(self.indexfile, 'w', atomictemp=True)
1013 fp = self.opener(self.indexfile, 'w', atomictemp=True)
1010 self.version &= ~(REVLOGNGINLINEDATA)
1014 self.version &= ~(REVLOGNGINLINEDATA)
1011 self._inline = False
1015 self._inline = False
1012 for i in self:
1016 for i in self:
1013 e = self._io.packentry(self.index[i], self.node, self.version, i)
1017 e = self._io.packentry(self.index[i], self.node, self.version, i)
1014 fp.write(e)
1018 fp.write(e)
1015
1019
1016 # if we don't call close, the temp file will never replace the
1020 # if we don't call close, the temp file will never replace the
1017 # real index
1021 # real index
1018 fp.close()
1022 fp.close()
1019
1023
1020 tr.replace(self.indexfile, trindex * self._io.size)
1024 tr.replace(self.indexfile, trindex * self._io.size)
1021 self._chunkclear()
1025 self._chunkclear()
1022
1026
1023 def addrevision(self, text, transaction, link, p1, p2, cachedelta=None,
1027 def addrevision(self, text, transaction, link, p1, p2, cachedelta=None,
1024 node=None):
1028 node=None):
1025 """add a revision to the log
1029 """add a revision to the log
1026
1030
1027 text - the revision data to add
1031 text - the revision data to add
1028 transaction - the transaction object used for rollback
1032 transaction - the transaction object used for rollback
1029 link - the linkrev data to add
1033 link - the linkrev data to add
1030 p1, p2 - the parent nodeids of the revision
1034 p1, p2 - the parent nodeids of the revision
1031 cachedelta - an optional precomputed delta
1035 cachedelta - an optional precomputed delta
1032 node - nodeid of revision; typically node is not specified, and it is
1036 node - nodeid of revision; typically node is not specified, and it is
1033 computed by default as hash(text, p1, p2), however subclasses might
1037 computed by default as hash(text, p1, p2), however subclasses might
1034 use different hashing method (and override checkhash() in such case)
1038 use different hashing method (and override checkhash() in such case)
1035 """
1039 """
1036 if link == nullrev:
1040 if link == nullrev:
1037 raise RevlogError(_("attempted to add linkrev -1 to %s")
1041 raise RevlogError(_("attempted to add linkrev -1 to %s")
1038 % self.indexfile)
1042 % self.indexfile)
1039 node = node or hash(text, p1, p2)
1043 node = node or hash(text, p1, p2)
1040 if node in self.nodemap:
1044 if node in self.nodemap:
1041 return node
1045 return node
1042
1046
1043 dfh = None
1047 dfh = None
1044 if not self._inline:
1048 if not self._inline:
1045 dfh = self.opener(self.datafile, "a")
1049 dfh = self.opener(self.datafile, "a")
1046 ifh = self.opener(self.indexfile, "a+")
1050 ifh = self.opener(self.indexfile, "a+")
1047 try:
1051 try:
1048 return self._addrevision(node, text, transaction, link, p1, p2,
1052 return self._addrevision(node, text, transaction, link, p1, p2,
1049 cachedelta, ifh, dfh)
1053 cachedelta, ifh, dfh)
1050 finally:
1054 finally:
1051 if dfh:
1055 if dfh:
1052 dfh.close()
1056 dfh.close()
1053 ifh.close()
1057 ifh.close()
1054
1058
1055 def compress(self, text):
1059 def compress(self, text):
1056 """ generate a possibly-compressed representation of text """
1060 """ generate a possibly-compressed representation of text """
1057 if not text:
1061 if not text:
1058 return ("", text)
1062 return ("", text)
1059 l = len(text)
1063 l = len(text)
1060 bin = None
1064 bin = None
1061 if l < 44:
1065 if l < 44:
1062 pass
1066 pass
1063 elif l > 1000000:
1067 elif l > 1000000:
1064 # zlib makes an internal copy, thus doubling memory usage for
1068 # zlib makes an internal copy, thus doubling memory usage for
1065 # large files, so lets do this in pieces
1069 # large files, so lets do this in pieces
1066 z = zlib.compressobj()
1070 z = zlib.compressobj()
1067 p = []
1071 p = []
1068 pos = 0
1072 pos = 0
1069 while pos < l:
1073 while pos < l:
1070 pos2 = pos + 2**20
1074 pos2 = pos + 2**20
1071 p.append(z.compress(text[pos:pos2]))
1075 p.append(z.compress(text[pos:pos2]))
1072 pos = pos2
1076 pos = pos2
1073 p.append(z.flush())
1077 p.append(z.flush())
1074 if sum(map(len, p)) < l:
1078 if sum(map(len, p)) < l:
1075 bin = "".join(p)
1079 bin = "".join(p)
1076 else:
1080 else:
1077 bin = _compress(text)
1081 bin = _compress(text)
1078 if bin is None or len(bin) > l:
1082 if bin is None or len(bin) > l:
1079 if text[0] == '\0':
1083 if text[0] == '\0':
1080 return ("", text)
1084 return ("", text)
1081 return ('u', text)
1085 return ('u', text)
1082 return ("", bin)
1086 return ("", bin)
1083
1087
1084 def _addrevision(self, node, text, transaction, link, p1, p2,
1088 def _addrevision(self, node, text, transaction, link, p1, p2,
1085 cachedelta, ifh, dfh):
1089 cachedelta, ifh, dfh):
1086 """internal function to add revisions to the log
1090 """internal function to add revisions to the log
1087
1091
1088 see addrevision for argument descriptions.
1092 see addrevision for argument descriptions.
1089 invariants:
1093 invariants:
1090 - text is optional (can be None); if not set, cachedelta must be set.
1094 - text is optional (can be None); if not set, cachedelta must be set.
1091 if both are set, they must correspond to each other.
1095 if both are set, they must correspond to each other.
1092 """
1096 """
1093 btext = [text]
1097 btext = [text]
1094 def buildtext():
1098 def buildtext():
1095 if btext[0] is not None:
1099 if btext[0] is not None:
1096 return btext[0]
1100 return btext[0]
1097 # flush any pending writes here so we can read it in revision
1101 # flush any pending writes here so we can read it in revision
1098 if dfh:
1102 if dfh:
1099 dfh.flush()
1103 dfh.flush()
1100 ifh.flush()
1104 ifh.flush()
1101 basetext = self.revision(self.node(cachedelta[0]))
1105 basetext = self.revision(self.node(cachedelta[0]))
1102 btext[0] = mdiff.patch(basetext, cachedelta[1])
1106 btext[0] = mdiff.patch(basetext, cachedelta[1])
1103 self.checkhash(btext[0], p1, p2, node)
1107 self.checkhash(btext[0], p1, p2, node)
1104 return btext[0]
1108 return btext[0]
1105
1109
1106 def builddelta(rev):
1110 def builddelta(rev):
1107 # can we use the cached delta?
1111 # can we use the cached delta?
1108 if cachedelta and cachedelta[0] == rev:
1112 if cachedelta and cachedelta[0] == rev:
1109 delta = cachedelta[1]
1113 delta = cachedelta[1]
1110 else:
1114 else:
1111 t = buildtext()
1115 t = buildtext()
1112 ptext = self.revision(self.node(rev))
1116 ptext = self.revision(self.node(rev))
1113 delta = mdiff.textdiff(ptext, t)
1117 delta = mdiff.textdiff(ptext, t)
1114 data = self.compress(delta)
1118 data = self.compress(delta)
1115 l = len(data[1]) + len(data[0])
1119 l = len(data[1]) + len(data[0])
1116 if basecache[0] == rev:
1120 if basecache[0] == rev:
1117 chainbase = basecache[1]
1121 chainbase = basecache[1]
1118 else:
1122 else:
1119 chainbase = self.chainbase(rev)
1123 chainbase = self.chainbase(rev)
1120 dist = l + offset - self.start(chainbase)
1124 dist = l + offset - self.start(chainbase)
1121 if self._generaldelta:
1125 if self._generaldelta:
1122 base = rev
1126 base = rev
1123 else:
1127 else:
1124 base = chainbase
1128 base = chainbase
1125 return dist, l, data, base, chainbase
1129 return dist, l, data, base, chainbase
1126
1130
1127 curr = len(self)
1131 curr = len(self)
1128 prev = curr - 1
1132 prev = curr - 1
1129 base = chainbase = curr
1133 base = chainbase = curr
1130 offset = self.end(prev)
1134 offset = self.end(prev)
1131 flags = 0
1135 flags = 0
1132 d = None
1136 d = None
1133 basecache = self._basecache
1137 basecache = self._basecache
1134 p1r, p2r = self.rev(p1), self.rev(p2)
1138 p1r, p2r = self.rev(p1), self.rev(p2)
1135
1139
1136 # should we try to build a delta?
1140 # should we try to build a delta?
1137 if prev != nullrev:
1141 if prev != nullrev:
1138 if self._generaldelta:
1142 if self._generaldelta:
1139 if p1r >= basecache[1]:
1143 if p1r >= basecache[1]:
1140 d = builddelta(p1r)
1144 d = builddelta(p1r)
1141 elif p2r >= basecache[1]:
1145 elif p2r >= basecache[1]:
1142 d = builddelta(p2r)
1146 d = builddelta(p2r)
1143 else:
1147 else:
1144 d = builddelta(prev)
1148 d = builddelta(prev)
1145 else:
1149 else:
1146 d = builddelta(prev)
1150 d = builddelta(prev)
1147 dist, l, data, base, chainbase = d
1151 dist, l, data, base, chainbase = d
1148
1152
1149 # full versions are inserted when the needed deltas
1153 # full versions are inserted when the needed deltas
1150 # become comparable to the uncompressed text
1154 # become comparable to the uncompressed text
1151 if text is None:
1155 if text is None:
1152 textlen = mdiff.patchedsize(self.rawsize(cachedelta[0]),
1156 textlen = mdiff.patchedsize(self.rawsize(cachedelta[0]),
1153 cachedelta[1])
1157 cachedelta[1])
1154 else:
1158 else:
1155 textlen = len(text)
1159 textlen = len(text)
1156 if d is None or dist > textlen * 2:
1160 if d is None or dist > textlen * 2:
1157 text = buildtext()
1161 text = buildtext()
1158 data = self.compress(text)
1162 data = self.compress(text)
1159 l = len(data[1]) + len(data[0])
1163 l = len(data[1]) + len(data[0])
1160 base = chainbase = curr
1164 base = chainbase = curr
1161
1165
1162 e = (offset_type(offset, flags), l, textlen,
1166 e = (offset_type(offset, flags), l, textlen,
1163 base, link, p1r, p2r, node)
1167 base, link, p1r, p2r, node)
1164 self.index.insert(-1, e)
1168 self.index.insert(-1, e)
1165 self.nodemap[node] = curr
1169 self.nodemap[node] = curr
1166
1170
1167 entry = self._io.packentry(e, self.node, self.version, curr)
1171 entry = self._io.packentry(e, self.node, self.version, curr)
1168 if not self._inline:
1172 if not self._inline:
1169 transaction.add(self.datafile, offset)
1173 transaction.add(self.datafile, offset)
1170 transaction.add(self.indexfile, curr * len(entry))
1174 transaction.add(self.indexfile, curr * len(entry))
1171 if data[0]:
1175 if data[0]:
1172 dfh.write(data[0])
1176 dfh.write(data[0])
1173 dfh.write(data[1])
1177 dfh.write(data[1])
1174 dfh.flush()
1178 dfh.flush()
1175 ifh.write(entry)
1179 ifh.write(entry)
1176 else:
1180 else:
1177 offset += curr * self._io.size
1181 offset += curr * self._io.size
1178 transaction.add(self.indexfile, offset, curr)
1182 transaction.add(self.indexfile, offset, curr)
1179 ifh.write(entry)
1183 ifh.write(entry)
1180 ifh.write(data[0])
1184 ifh.write(data[0])
1181 ifh.write(data[1])
1185 ifh.write(data[1])
1182 self.checkinlinesize(transaction, ifh)
1186 self.checkinlinesize(transaction, ifh)
1183
1187
1184 if type(text) == str: # only accept immutable objects
1188 if type(text) == str: # only accept immutable objects
1185 self._cache = (node, curr, text)
1189 self._cache = (node, curr, text)
1186 self._basecache = (curr, chainbase)
1190 self._basecache = (curr, chainbase)
1187 return node
1191 return node
1188
1192
1189 def addgroup(self, bundle, linkmapper, transaction):
1193 def addgroup(self, bundle, linkmapper, transaction):
1190 """
1194 """
1191 add a delta group
1195 add a delta group
1192
1196
1193 given a set of deltas, add them to the revision log. the
1197 given a set of deltas, add them to the revision log. the
1194 first delta is against its parent, which should be in our
1198 first delta is against its parent, which should be in our
1195 log, the rest are against the previous delta.
1199 log, the rest are against the previous delta.
1196 """
1200 """
1197
1201
1198 # track the base of the current delta log
1202 # track the base of the current delta log
1199 content = []
1203 content = []
1200 node = None
1204 node = None
1201
1205
1202 r = len(self)
1206 r = len(self)
1203 end = 0
1207 end = 0
1204 if r:
1208 if r:
1205 end = self.end(r - 1)
1209 end = self.end(r - 1)
1206 ifh = self.opener(self.indexfile, "a+")
1210 ifh = self.opener(self.indexfile, "a+")
1207 isize = r * self._io.size
1211 isize = r * self._io.size
1208 if self._inline:
1212 if self._inline:
1209 transaction.add(self.indexfile, end + isize, r)
1213 transaction.add(self.indexfile, end + isize, r)
1210 dfh = None
1214 dfh = None
1211 else:
1215 else:
1212 transaction.add(self.indexfile, isize, r)
1216 transaction.add(self.indexfile, isize, r)
1213 transaction.add(self.datafile, end)
1217 transaction.add(self.datafile, end)
1214 dfh = self.opener(self.datafile, "a")
1218 dfh = self.opener(self.datafile, "a")
1215
1219
1216 try:
1220 try:
1217 # loop through our set of deltas
1221 # loop through our set of deltas
1218 chain = None
1222 chain = None
1219 while True:
1223 while True:
1220 chunkdata = bundle.deltachunk(chain)
1224 chunkdata = bundle.deltachunk(chain)
1221 if not chunkdata:
1225 if not chunkdata:
1222 break
1226 break
1223 node = chunkdata['node']
1227 node = chunkdata['node']
1224 p1 = chunkdata['p1']
1228 p1 = chunkdata['p1']
1225 p2 = chunkdata['p2']
1229 p2 = chunkdata['p2']
1226 cs = chunkdata['cs']
1230 cs = chunkdata['cs']
1227 deltabase = chunkdata['deltabase']
1231 deltabase = chunkdata['deltabase']
1228 delta = chunkdata['delta']
1232 delta = chunkdata['delta']
1229
1233
1230 content.append(node)
1234 content.append(node)
1231
1235
1232 link = linkmapper(cs)
1236 link = linkmapper(cs)
1233 if node in self.nodemap:
1237 if node in self.nodemap:
1234 # this can happen if two branches make the same change
1238 # this can happen if two branches make the same change
1235 chain = node
1239 chain = node
1236 continue
1240 continue
1237
1241
1238 for p in (p1, p2):
1242 for p in (p1, p2):
1239 if p not in self.nodemap:
1243 if p not in self.nodemap:
1240 raise LookupError(p, self.indexfile,
1244 raise LookupError(p, self.indexfile,
1241 _('unknown parent'))
1245 _('unknown parent'))
1242
1246
1243 if deltabase not in self.nodemap:
1247 if deltabase not in self.nodemap:
1244 raise LookupError(deltabase, self.indexfile,
1248 raise LookupError(deltabase, self.indexfile,
1245 _('unknown delta base'))
1249 _('unknown delta base'))
1246
1250
1247 baserev = self.rev(deltabase)
1251 baserev = self.rev(deltabase)
1248 chain = self._addrevision(node, None, transaction, link,
1252 chain = self._addrevision(node, None, transaction, link,
1249 p1, p2, (baserev, delta), ifh, dfh)
1253 p1, p2, (baserev, delta), ifh, dfh)
1250 if not dfh and not self._inline:
1254 if not dfh and not self._inline:
1251 # addrevision switched from inline to conventional
1255 # addrevision switched from inline to conventional
1252 # reopen the index
1256 # reopen the index
1253 ifh.close()
1257 ifh.close()
1254 dfh = self.opener(self.datafile, "a")
1258 dfh = self.opener(self.datafile, "a")
1255 ifh = self.opener(self.indexfile, "a")
1259 ifh = self.opener(self.indexfile, "a")
1256 finally:
1260 finally:
1257 if dfh:
1261 if dfh:
1258 dfh.close()
1262 dfh.close()
1259 ifh.close()
1263 ifh.close()
1260
1264
1261 return content
1265 return content
1262
1266
1263 def strip(self, minlink, transaction):
1267 def strip(self, minlink, transaction):
1264 """truncate the revlog on the first revision with a linkrev >= minlink
1268 """truncate the revlog on the first revision with a linkrev >= minlink
1265
1269
1266 This function is called when we're stripping revision minlink and
1270 This function is called when we're stripping revision minlink and
1267 its descendants from the repository.
1271 its descendants from the repository.
1268
1272
1269 We have to remove all revisions with linkrev >= minlink, because
1273 We have to remove all revisions with linkrev >= minlink, because
1270 the equivalent changelog revisions will be renumbered after the
1274 the equivalent changelog revisions will be renumbered after the
1271 strip.
1275 strip.
1272
1276
1273 So we truncate the revlog on the first of these revisions, and
1277 So we truncate the revlog on the first of these revisions, and
1274 trust that the caller has saved the revisions that shouldn't be
1278 trust that the caller has saved the revisions that shouldn't be
1275 removed and that it'll re-add them after this truncation.
1279 removed and that it'll re-add them after this truncation.
1276 """
1280 """
1277 if len(self) == 0:
1281 if len(self) == 0:
1278 return
1282 return
1279
1283
1280 for rev in self:
1284 for rev in self:
1281 if self.index[rev][4] >= minlink:
1285 if self.index[rev][4] >= minlink:
1282 break
1286 break
1283 else:
1287 else:
1284 return
1288 return
1285
1289
1286 # first truncate the files on disk
1290 # first truncate the files on disk
1287 end = self.start(rev)
1291 end = self.start(rev)
1288 if not self._inline:
1292 if not self._inline:
1289 transaction.add(self.datafile, end)
1293 transaction.add(self.datafile, end)
1290 end = rev * self._io.size
1294 end = rev * self._io.size
1291 else:
1295 else:
1292 end += rev * self._io.size
1296 end += rev * self._io.size
1293
1297
1294 transaction.add(self.indexfile, end)
1298 transaction.add(self.indexfile, end)
1295
1299
1296 # then reset internal state in memory to forget those revisions
1300 # then reset internal state in memory to forget those revisions
1297 self._cache = None
1301 self._cache = None
1298 self._chunkclear()
1302 self._chunkclear()
1299 for x in xrange(rev, len(self)):
1303 for x in xrange(rev, len(self)):
1300 del self.nodemap[self.node(x)]
1304 del self.nodemap[self.node(x)]
1301
1305
1302 del self.index[rev:-1]
1306 del self.index[rev:-1]
1303
1307
1304 def checksize(self):
1308 def checksize(self):
1305 expected = 0
1309 expected = 0
1306 if len(self):
1310 if len(self):
1307 expected = max(0, self.end(len(self) - 1))
1311 expected = max(0, self.end(len(self) - 1))
1308
1312
1309 try:
1313 try:
1310 f = self.opener(self.datafile)
1314 f = self.opener(self.datafile)
1311 f.seek(0, 2)
1315 f.seek(0, 2)
1312 actual = f.tell()
1316 actual = f.tell()
1313 f.close()
1317 f.close()
1314 dd = actual - expected
1318 dd = actual - expected
1315 except IOError, inst:
1319 except IOError, inst:
1316 if inst.errno != errno.ENOENT:
1320 if inst.errno != errno.ENOENT:
1317 raise
1321 raise
1318 dd = 0
1322 dd = 0
1319
1323
1320 try:
1324 try:
1321 f = self.opener(self.indexfile)
1325 f = self.opener(self.indexfile)
1322 f.seek(0, 2)
1326 f.seek(0, 2)
1323 actual = f.tell()
1327 actual = f.tell()
1324 f.close()
1328 f.close()
1325 s = self._io.size
1329 s = self._io.size
1326 i = max(0, actual // s)
1330 i = max(0, actual // s)
1327 di = actual - (i * s)
1331 di = actual - (i * s)
1328 if self._inline:
1332 if self._inline:
1329 databytes = 0
1333 databytes = 0
1330 for r in self:
1334 for r in self:
1331 databytes += max(0, self.length(r))
1335 databytes += max(0, self.length(r))
1332 dd = 0
1336 dd = 0
1333 di = actual - len(self) * s - databytes
1337 di = actual - len(self) * s - databytes
1334 except IOError, inst:
1338 except IOError, inst:
1335 if inst.errno != errno.ENOENT:
1339 if inst.errno != errno.ENOENT:
1336 raise
1340 raise
1337 di = 0
1341 di = 0
1338
1342
1339 return (dd, di)
1343 return (dd, di)
1340
1344
1341 def files(self):
1345 def files(self):
1342 res = [self.indexfile]
1346 res = [self.indexfile]
1343 if not self._inline:
1347 if not self._inline:
1344 res.append(self.datafile)
1348 res.append(self.datafile)
1345 return res
1349 return res
General Comments 0
You need to be logged in to leave comments. Login now