/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to breezy/bzr/groupcompress.py

  • Committer: Jelmer Vernooij
  • Date: 2020-07-18 23:14:00 UTC
  • mfrom: (7490.40.62 work)
  • mto: This revision was merged to the branch mainline in revision 7519.
  • Revision ID: jelmer@jelmer.uk-20200718231400-jaes9qltn8oi8xss
Merge lp:brz/3.1.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2008, 2009, 2010 Canonical Ltd
 
1
# Copyright (C) 2008-2011 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
18
18
 
19
19
import time
20
20
import zlib
21
 
try:
22
 
    import pylzma
23
 
except ImportError:
24
 
    pylzma = None
25
21
 
26
 
from bzrlib import (
 
22
from ..lazy_import import lazy_import
 
23
lazy_import(globals(), """
 
24
from breezy import (
27
25
    annotate,
 
26
    config,
28
27
    debug,
29
 
    errors,
30
 
    graph as _mod_graph,
31
 
    knit,
32
28
    osutils,
33
 
    pack,
34
29
    static_tuple,
35
30
    trace,
36
 
    )
37
 
from bzrlib.btree_index import BTreeBuilder
38
 
from bzrlib.lru_cache import LRUSizeCache
39
 
from bzrlib.tsort import topo_sort
40
 
from bzrlib.versionedfile import (
 
31
    tsort,
 
32
    )
 
33
from breezy.bzr import (
 
34
    knit,
 
35
    pack,
 
36
    pack_repo,
 
37
    )
 
38
 
 
39
from breezy.i18n import gettext
 
40
""")
 
41
 
 
42
from .. import (
 
43
    errors,
 
44
    )
 
45
from .btree_index import BTreeBuilder
 
46
from ..lru_cache import LRUSizeCache
 
47
from .versionedfile import (
 
48
    _KeyRefs,
41
49
    adapter_registry,
42
50
    AbsentContentFactory,
43
51
    ChunkedContentFactory,
 
52
    ExistingContent,
44
53
    FulltextContentFactory,
45
 
    VersionedFiles,
 
54
    VersionedFilesWithFallbacks,
 
55
    UnavailableRepresentation,
46
56
    )
47
57
 
48
58
# Minimum number of uncompressed bytes to try fetch at once when retrieving
49
59
# groupcompress blocks.
50
60
BATCH_SIZE = 2**16
51
61
 
52
 
_USE_LZMA = False and (pylzma is not None)
 
62
# osutils.sha_string(b'')
 
63
_null_sha1 = b'da39a3ee5e6b4b0d3255bfef95601890afd80709'
53
64
 
54
 
# osutils.sha_string('')
55
 
_null_sha1 = 'da39a3ee5e6b4b0d3255bfef95601890afd80709'
56
65
 
57
66
def sort_gc_optimal(parent_map):
58
67
    """Sort and group the keys in parent_map into groupcompress order.
65
74
    # groupcompress ordering is approximately reverse topological,
66
75
    # properly grouped by file-id.
67
76
    per_prefix_map = {}
68
 
    for key, value in parent_map.iteritems():
69
 
        if isinstance(key, str) or len(key) == 1:
70
 
            prefix = ''
 
77
    for key, value in parent_map.items():
 
78
        if isinstance(key, bytes) or len(key) == 1:
 
79
            prefix = b''
71
80
        else:
72
81
            prefix = key[0]
73
82
        try:
77
86
 
78
87
    present_keys = []
79
88
    for prefix in sorted(per_prefix_map):
80
 
        present_keys.extend(reversed(topo_sort(per_prefix_map[prefix])))
 
89
        present_keys.extend(reversed(tsort.topo_sort(per_prefix_map[prefix])))
81
90
    return present_keys
82
91
 
83
92
 
 
93
class DecompressCorruption(errors.BzrError):
 
94
 
 
95
    _fmt = "Corruption while decompressing repository file%(orig_error)s"
 
96
 
 
97
    def __init__(self, orig_error=None):
 
98
        if orig_error is not None:
 
99
            self.orig_error = ", %s" % (orig_error,)
 
100
        else:
 
101
            self.orig_error = ""
 
102
        errors.BzrError.__init__(self)
 
103
 
 
104
 
84
105
# The max zlib window size is 32kB, so if we set 'max_size' output of the
85
106
# decompressor to the requested bytes + 32kB, then we should guarantee
86
107
# num_bytes coming out.
87
 
_ZLIB_DECOMP_WINDOW = 32*1024
 
108
_ZLIB_DECOMP_WINDOW = 32 * 1024
 
109
 
88
110
 
89
111
class GroupCompressBlock(object):
90
112
    """An object which maintains the internal structure of the compressed data.
93
115
    """
94
116
 
95
117
    # Group Compress Block v1 Zlib
96
 
    GCB_HEADER = 'gcb1z\n'
 
118
    GCB_HEADER = b'gcb1z\n'
97
119
    # Group Compress Block v1 Lzma
98
 
    GCB_LZ_HEADER = 'gcb1l\n'
 
120
    GCB_LZ_HEADER = b'gcb1l\n'
99
121
    GCB_KNOWN_HEADERS = (GCB_HEADER, GCB_LZ_HEADER)
100
122
 
101
123
    def __init__(self):
102
124
        # map by key? or just order in file?
103
125
        self._compressor_name = None
104
 
        self._z_content = None
 
126
        self._z_content_chunks = None
105
127
        self._z_content_decompressor = None
106
128
        self._z_content_length = None
107
129
        self._content_length = None
132
154
        # Expand the content if required
133
155
        if self._content is None:
134
156
            if self._content_chunks is not None:
135
 
                self._content = ''.join(self._content_chunks)
 
157
                self._content = b''.join(self._content_chunks)
136
158
                self._content_chunks = None
137
159
        if self._content is None:
138
 
            if self._z_content is None:
 
160
            # We join self._z_content_chunks here, because if we are
 
161
            # decompressing, then it is *very* likely that we have a single
 
162
            # chunk
 
163
            if self._z_content_chunks is None:
139
164
                raise AssertionError('No content to decompress')
140
 
            if self._z_content == '':
141
 
                self._content = ''
 
165
            z_content = b''.join(self._z_content_chunks)
 
166
            if z_content == b'':
 
167
                self._content = b''
142
168
            elif self._compressor_name == 'lzma':
143
169
                # We don't do partial lzma decomp yet
144
 
                self._content = pylzma.decompress(self._z_content)
 
170
                import pylzma
 
171
                self._content = pylzma.decompress(z_content)
145
172
            elif self._compressor_name == 'zlib':
146
173
                # Start a zlib decompressor
147
174
                if num_bytes * 4 > self._content_length * 3:
148
175
                    # If we are requesting more that 3/4ths of the content,
149
176
                    # just extract the whole thing in a single pass
150
177
                    num_bytes = self._content_length
151
 
                    self._content = zlib.decompress(self._z_content)
 
178
                    self._content = zlib.decompress(z_content)
152
179
                else:
153
180
                    self._z_content_decompressor = zlib.decompressobj()
154
181
                    # Seed the decompressor with the uncompressed bytes, so
155
182
                    # that the rest of the code is simplified
156
183
                    self._content = self._z_content_decompressor.decompress(
157
 
                        self._z_content, num_bytes + _ZLIB_DECOMP_WINDOW)
 
184
                        z_content, num_bytes + _ZLIB_DECOMP_WINDOW)
158
185
                    if not self._z_content_decompressor.unconsumed_tail:
159
186
                        self._z_content_decompressor = None
160
187
            else:
187
214
            # The stream is finished
188
215
            self._z_content_decompressor = None
189
216
 
190
 
    def _parse_bytes(self, bytes, pos):
 
217
    def _parse_bytes(self, data, pos):
191
218
        """Read the various lengths from the header.
192
219
 
193
220
        This also populates the various 'compressed' buffers.
197
224
        # At present, we have 2 integers for the compressed and uncompressed
198
225
        # content. In base10 (ascii) 14 bytes can represent > 1TB, so to avoid
199
226
        # checking too far, cap the search to 14 bytes.
200
 
        pos2 = bytes.index('\n', pos, pos + 14)
201
 
        self._z_content_length = int(bytes[pos:pos2])
202
 
        pos = pos2 + 1
203
 
        pos2 = bytes.index('\n', pos, pos + 14)
204
 
        self._content_length = int(bytes[pos:pos2])
205
 
        pos = pos2 + 1
206
 
        if len(bytes) != (pos + self._z_content_length):
 
227
        pos2 = data.index(b'\n', pos, pos + 14)
 
228
        self._z_content_length = int(data[pos:pos2])
 
229
        pos = pos2 + 1
 
230
        pos2 = data.index(b'\n', pos, pos + 14)
 
231
        self._content_length = int(data[pos:pos2])
 
232
        pos = pos2 + 1
 
233
        if len(data) != (pos + self._z_content_length):
207
234
            # XXX: Define some GCCorrupt error ?
208
235
            raise AssertionError('Invalid bytes: (%d) != %d + %d' %
209
 
                                 (len(bytes), pos, self._z_content_length))
210
 
        self._z_content = bytes[pos:]
 
236
                                 (len(data), pos, self._z_content_length))
 
237
        self._z_content_chunks = (data[pos:],)
 
238
 
 
239
    @property
 
240
    def _z_content(self):
 
241
        """Return z_content_chunks as a simple string.
 
242
 
 
243
        Meant only to be used by the test suite.
 
244
        """
 
245
        if self._z_content_chunks is not None:
 
246
            return b''.join(self._z_content_chunks)
 
247
        return None
211
248
 
212
249
    @classmethod
213
250
    def from_bytes(cls, bytes):
214
251
        out = cls()
215
 
        if bytes[:6] not in cls.GCB_KNOWN_HEADERS:
 
252
        header = bytes[:6]
 
253
        if header not in cls.GCB_KNOWN_HEADERS:
216
254
            raise ValueError('bytes did not start with any of %r'
217
255
                             % (cls.GCB_KNOWN_HEADERS,))
218
 
        # XXX: why not testing the whole header ?
219
 
        if bytes[4] == 'z':
 
256
        if header == cls.GCB_HEADER:
220
257
            out._compressor_name = 'zlib'
221
 
        elif bytes[4] == 'l':
 
258
        elif header == cls.GCB_LZ_HEADER:
222
259
            out._compressor_name = 'lzma'
223
260
        else:
224
 
            raise ValueError('unknown compressor: %r' % (bytes,))
 
261
            raise ValueError('unknown compressor: %r' % (header,))
225
262
        out._parse_bytes(bytes, 6)
226
263
        return out
227
264
 
233
270
        :return: The bytes for the content
234
271
        """
235
272
        if start == end == 0:
236
 
            return ''
 
273
            return []
237
274
        self._ensure_content(end)
238
275
        # The bytes are 'f' or 'd' for the type, then a variable-length
239
276
        # base128 integer for the content size, then the actual content
240
277
        # We know that the variable-length integer won't be longer than 5
241
278
        # bytes (it takes 5 bytes to encode 2^32)
242
 
        c = self._content[start]
243
 
        if c == 'f':
 
279
        c = self._content[start:start + 1]
 
280
        if c == b'f':
244
281
            type = 'fulltext'
245
282
        else:
246
 
            if c != 'd':
 
283
            if c != b'd':
247
284
                raise ValueError('Unknown content control code: %s'
248
285
                                 % (c,))
249
286
            type = 'delta'
250
287
        content_len, len_len = decode_base128_int(
251
 
                            self._content[start + 1:start + 6])
 
288
            self._content[start + 1:start + 6])
252
289
        content_start = start + 1 + len_len
253
290
        if end != content_start + content_len:
254
291
            raise ValueError('end != len according to field header'
255
 
                ' %s != %s' % (end, content_start + content_len))
256
 
        if c == 'f':
257
 
            bytes = self._content[content_start:end]
258
 
        elif c == 'd':
259
 
            bytes = apply_delta_to_source(self._content, content_start, end)
260
 
        return bytes
 
292
                             ' %s != %s' % (end, content_start + content_len))
 
293
        if c == b'f':
 
294
            return [self._content[content_start:end]]
 
295
        # Must be type delta as checked above
 
296
        return [apply_delta_to_source(self._content, content_start, end)]
261
297
 
262
298
    def set_chunked_content(self, content_chunks, length):
263
299
        """Set the content of this block to the given chunks."""
269
305
        self._content_length = length
270
306
        self._content_chunks = content_chunks
271
307
        self._content = None
272
 
        self._z_content = None
 
308
        self._z_content_chunks = None
273
309
 
274
310
    def set_content(self, content):
275
311
        """Set the content of this block."""
276
312
        self._content_length = len(content)
277
313
        self._content = content
278
 
        self._z_content = None
279
 
 
280
 
    def _create_z_content_using_lzma(self):
281
 
        if self._content_chunks is not None:
282
 
            self._content = ''.join(self._content_chunks)
283
 
            self._content_chunks = None
284
 
        if self._content is None:
285
 
            raise AssertionError('Nothing to compress')
286
 
        self._z_content = pylzma.compress(self._content)
287
 
        self._z_content_length = len(self._z_content)
288
 
 
289
 
    def _create_z_content_from_chunks(self):
 
314
        self._z_content_chunks = None
 
315
 
 
316
    def _create_z_content_from_chunks(self, chunks):
290
317
        compressor = zlib.compressobj(zlib.Z_DEFAULT_COMPRESSION)
291
 
        compressed_chunks = map(compressor.compress, self._content_chunks)
 
318
        # Peak in this point is 1 fulltext, 1 compressed text, + zlib overhead
 
319
        # (measured peak is maybe 30MB over the above...)
 
320
        compressed_chunks = list(map(compressor.compress, chunks))
292
321
        compressed_chunks.append(compressor.flush())
293
 
        self._z_content = ''.join(compressed_chunks)
294
 
        self._z_content_length = len(self._z_content)
 
322
        # Ignore empty chunks
 
323
        self._z_content_chunks = [c for c in compressed_chunks if c]
 
324
        self._z_content_length = sum(map(len, self._z_content_chunks))
295
325
 
296
326
    def _create_z_content(self):
297
 
        if self._z_content is not None:
298
 
            return
299
 
        if _USE_LZMA:
300
 
            self._create_z_content_using_lzma()
 
327
        if self._z_content_chunks is not None:
301
328
            return
302
329
        if self._content_chunks is not None:
303
 
            self._create_z_content_from_chunks()
304
 
            return
305
 
        self._z_content = zlib.compress(self._content)
306
 
        self._z_content_length = len(self._z_content)
 
330
            chunks = self._content_chunks
 
331
        else:
 
332
            chunks = (self._content,)
 
333
        self._create_z_content_from_chunks(chunks)
 
334
 
 
335
    def to_chunks(self):
 
336
        """Create the byte stream as a series of 'chunks'"""
 
337
        self._create_z_content()
 
338
        header = self.GCB_HEADER
 
339
        chunks = [b'%s%d\n%d\n'
 
340
                  % (header, self._z_content_length, self._content_length),
 
341
                  ]
 
342
        chunks.extend(self._z_content_chunks)
 
343
        total_len = sum(map(len, chunks))
 
344
        return total_len, chunks
307
345
 
308
346
    def to_bytes(self):
309
347
        """Encode the information into a byte stream."""
310
 
        self._create_z_content()
311
 
        if _USE_LZMA:
312
 
            header = self.GCB_LZ_HEADER
313
 
        else:
314
 
            header = self.GCB_HEADER
315
 
        chunks = [header,
316
 
                  '%d\n%d\n' % (self._z_content_length, self._content_length),
317
 
                  self._z_content,
318
 
                 ]
319
 
        return ''.join(chunks)
 
348
        total_len, chunks = self.to_chunks()
 
349
        return b''.join(chunks)
320
350
 
321
351
    def _dump(self, include_text=False):
322
352
        """Take this block, and spit out a human-readable structure.
332
362
        result = []
333
363
        pos = 0
334
364
        while pos < self._content_length:
335
 
            kind = self._content[pos]
 
365
            kind = self._content[pos:pos + 1]
336
366
            pos += 1
337
 
            if kind not in ('f', 'd'):
 
367
            if kind not in (b'f', b'd'):
338
368
                raise ValueError('invalid kind character: %r' % (kind,))
339
369
            content_len, len_len = decode_base128_int(
340
 
                                self._content[pos:pos + 5])
 
370
                self._content[pos:pos + 5])
341
371
            pos += len_len
342
372
            if content_len + pos > self._content_length:
343
373
                raise ValueError('invalid content_len %d for record @ pos %d'
344
374
                                 % (content_len, pos - len_len - 1))
345
 
            if kind == 'f': # Fulltext
 
375
            if kind == b'f':  # Fulltext
346
376
                if include_text:
347
 
                    text = self._content[pos:pos+content_len]
348
 
                    result.append(('f', content_len, text))
 
377
                    text = self._content[pos:pos + content_len]
 
378
                    result.append((b'f', content_len, text))
349
379
                else:
350
 
                    result.append(('f', content_len))
351
 
            elif kind == 'd': # Delta
352
 
                delta_content = self._content[pos:pos+content_len]
 
380
                    result.append((b'f', content_len))
 
381
            elif kind == b'd':  # Delta
 
382
                delta_content = self._content[pos:pos + content_len]
353
383
                delta_info = []
354
384
                # The first entry in a delta is the decompressed length
355
385
                decomp_len, delta_pos = decode_base128_int(delta_content)
356
 
                result.append(('d', content_len, decomp_len, delta_info))
 
386
                result.append((b'd', content_len, decomp_len, delta_info))
357
387
                measured_len = 0
358
388
                while delta_pos < content_len:
359
 
                    c = ord(delta_content[delta_pos])
 
389
                    c = delta_content[delta_pos]
360
390
                    delta_pos += 1
361
 
                    if c & 0x80: # Copy
 
391
                    if c & 0x80:  # Copy
362
392
                        (offset, length,
363
393
                         delta_pos) = decode_copy_instruction(delta_content, c,
364
394
                                                              delta_pos)
365
395
                        if include_text:
366
 
                            text = self._content[offset:offset+length]
367
 
                            delta_info.append(('c', offset, length, text))
 
396
                            text = self._content[offset:offset + length]
 
397
                            delta_info.append((b'c', offset, length, text))
368
398
                        else:
369
 
                            delta_info.append(('c', offset, length))
 
399
                            delta_info.append((b'c', offset, length))
370
400
                        measured_len += length
371
 
                    else: # Insert
 
401
                    else:  # Insert
372
402
                        if include_text:
373
 
                            txt = delta_content[delta_pos:delta_pos+c]
 
403
                            txt = delta_content[delta_pos:delta_pos + c]
374
404
                        else:
375
 
                            txt = ''
376
 
                        delta_info.append(('i', c, txt))
 
405
                            txt = b''
 
406
                        delta_info.append((b'i', c, txt))
377
407
                        measured_len += c
378
408
                        delta_pos += c
379
409
                if delta_pos != content_len:
405
435
        self.key = key
406
436
        self.parents = parents
407
437
        self.sha1 = None
 
438
        self.size = None
408
439
        # Note: This attribute coupled with Manager._factories creates a
409
440
        #       reference cycle. Perhaps we would rather use a weakref(), or
410
441
        #       find an appropriate time to release the ref. After the first
411
442
        #       get_bytes_as call? After Manager.get_record_stream() returns
412
443
        #       the object?
413
444
        self._manager = manager
414
 
        self._bytes = None
 
445
        self._chunks = None
415
446
        self.storage_kind = 'groupcompress-block'
416
447
        if not first:
417
448
            self.storage_kind = 'groupcompress-block-ref'
421
452
 
422
453
    def __repr__(self):
423
454
        return '%s(%s, first=%s)' % (self.__class__.__name__,
424
 
            self.key, self._first)
 
455
                                     self.key, self._first)
 
456
 
 
457
    def _extract_bytes(self):
 
458
        # Grab and cache the raw bytes for this entry
 
459
        # and break the ref-cycle with _manager since we don't need it
 
460
        # anymore
 
461
        try:
 
462
            self._manager._prepare_for_extract()
 
463
        except zlib.error as value:
 
464
            raise DecompressCorruption("zlib: " + str(value))
 
465
        block = self._manager._block
 
466
        self._chunks = block.extract(self.key, self._start, self._end)
 
467
        # There are code paths that first extract as fulltext, and then
 
468
        # extract as storage_kind (smart fetch). So we don't break the
 
469
        # refcycle here, but instead in manager.get_record_stream()
425
470
 
426
471
    def get_bytes_as(self, storage_kind):
427
472
        if storage_kind == self.storage_kind:
429
474
                # wire bytes, something...
430
475
                return self._manager._wire_bytes()
431
476
            else:
432
 
                return ''
433
 
        if storage_kind in ('fulltext', 'chunked'):
434
 
            if self._bytes is None:
435
 
                # Grab and cache the raw bytes for this entry
436
 
                # and break the ref-cycle with _manager since we don't need it
437
 
                # anymore
438
 
                self._manager._prepare_for_extract()
439
 
                block = self._manager._block
440
 
                self._bytes = block.extract(self.key, self._start, self._end)
441
 
                # There are code paths that first extract as fulltext, and then
442
 
                # extract as storage_kind (smart fetch). So we don't break the
443
 
                # refcycle here, but instead in manager.get_record_stream()
 
477
                return b''
 
478
        if storage_kind in ('fulltext', 'chunked', 'lines'):
 
479
            if self._chunks is None:
 
480
                self._extract_bytes()
444
481
            if storage_kind == 'fulltext':
445
 
                return self._bytes
 
482
                return b''.join(self._chunks)
 
483
            elif storage_kind == 'chunked':
 
484
                return self._chunks
446
485
            else:
447
 
                return [self._bytes]
448
 
        raise errors.UnavailableRepresentation(self.key, storage_kind,
 
486
                return osutils.chunks_to_lines(self._chunks)
 
487
        raise UnavailableRepresentation(self.key, storage_kind,
 
488
                                               self.storage_kind)
 
489
 
 
490
    def iter_bytes_as(self, storage_kind):
 
491
        if self._chunks is None:
 
492
            self._extract_bytes()
 
493
        if storage_kind == 'chunked':
 
494
            return iter(self._chunks)
 
495
        elif storage_kind == 'lines':
 
496
            return iter(osutils.chunks_to_lines(self._chunks))
 
497
        raise UnavailableRepresentation(self.key, storage_kind,
449
498
                                               self.storage_kind)
450
499
 
451
500
 
452
501
class _LazyGroupContentManager(object):
453
502
    """This manages a group of _LazyGroupCompressFactory objects."""
454
503
 
455
 
    _max_cut_fraction = 0.75 # We allow a block to be trimmed to 75% of
456
 
                             # current size, and still be considered
457
 
                             # resuable
458
 
    _full_block_size = 4*1024*1024
459
 
    _full_mixed_block_size = 2*1024*1024
460
 
    _full_enough_block_size = 3*1024*1024 # size at which we won't repack
461
 
    _full_enough_mixed_block_size = 2*768*1024 # 1.5MB
 
504
    _max_cut_fraction = 0.75  # We allow a block to be trimmed to 75% of
 
505
    # current size, and still be considered
 
506
    # resuable
 
507
    _full_block_size = 4 * 1024 * 1024
 
508
    _full_mixed_block_size = 2 * 1024 * 1024
 
509
    _full_enough_block_size = 3 * 1024 * 1024  # size at which we won't repack
 
510
    _full_enough_mixed_block_size = 2 * 768 * 1024  # 1.5MB
462
511
 
463
 
    def __init__(self, block):
 
512
    def __init__(self, block, get_compressor_settings=None):
464
513
        self._block = block
465
514
        # We need to preserve the ordering
466
515
        self._factories = []
467
516
        self._last_byte = 0
 
517
        self._get_settings = get_compressor_settings
 
518
        self._compressor_settings = None
 
519
 
 
520
    def _get_compressor_settings(self):
 
521
        if self._compressor_settings is not None:
 
522
            return self._compressor_settings
 
523
        settings = None
 
524
        if self._get_settings is not None:
 
525
            settings = self._get_settings()
 
526
        if settings is None:
 
527
            vf = GroupCompressVersionedFiles
 
528
            settings = vf._DEFAULT_COMPRESSOR_SETTINGS
 
529
        self._compressor_settings = settings
 
530
        return self._compressor_settings
468
531
 
469
532
    def add_factory(self, key, parents, start, end):
470
533
        if not self._factories:
473
536
            first = False
474
537
        # Note that this creates a reference cycle....
475
538
        factory = _LazyGroupCompressFactory(key, parents, self,
476
 
            start, end, first=first)
 
539
                                            start, end, first=first)
477
540
        # max() works here, but as a function call, doing a compare seems to be
478
541
        # significantly faster, timeit says 250ms for max() and 100ms for the
479
542
        # comparison
503
566
        new_block.set_content(self._block._content[:last_byte])
504
567
        self._block = new_block
505
568
 
 
569
    def _make_group_compressor(self):
 
570
        return GroupCompressor(self._get_compressor_settings())
 
571
 
506
572
    def _rebuild_block(self):
507
573
        """Create a new GroupCompressBlock with only the referenced texts."""
508
 
        compressor = GroupCompressor()
 
574
        compressor = self._make_group_compressor()
509
575
        tstart = time.time()
510
576
        old_length = self._block._content_length
511
577
        end_point = 0
512
578
        for factory in self._factories:
513
 
            bytes = factory.get_bytes_as('fulltext')
 
579
            chunks = factory.get_bytes_as('chunked')
 
580
            chunks_len = factory.size
 
581
            if chunks_len is None:
 
582
                chunks_len = sum(map(len, chunks))
514
583
            (found_sha1, start_point, end_point,
515
 
             type) = compressor.compress(factory.key, bytes, factory.sha1)
 
584
             type) = compressor.compress(
 
585
                 factory.key, chunks, chunks_len, factory.sha1)
516
586
            # Now update this factory with the new offsets, etc
517
587
            factory.sha1 = found_sha1
518
588
            factory._start = start_point
523
593
        #       block? It seems hard to come up with a method that it would
524
594
        #       expand, since we do full compression again. Perhaps based on a
525
595
        #       request that ends up poorly ordered?
 
596
        # TODO: If the content would have expanded, then we would want to
 
597
        #       handle a case where we need to split the block.
 
598
        #       Now that we have a user-tweakable option
 
599
        #       (max_bytes_to_index), it is possible that one person set it
 
600
        #       to a very low value, causing poor compression.
526
601
        delta = time.time() - tstart
527
602
        self._block = new_block
528
603
        trace.mutter('creating new compressed block on-the-fly in %.3fs'
651
726
        #   <length of gc block>\n
652
727
        #   <header bytes>
653
728
        #   <gc-block>
654
 
        lines = ['groupcompress-block\n']
 
729
        lines = [b'groupcompress-block\n']
655
730
        # The minimal info we need is the key, the start offset, and the
656
731
        # parents. The length and type are encoded in the record itself.
657
732
        # However, passing in the other bits makes it easier.  The list of
662
737
        # 1 line for end byte
663
738
        header_lines = []
664
739
        for factory in self._factories:
665
 
            key_bytes = '\x00'.join(factory.key)
 
740
            key_bytes = b'\x00'.join(factory.key)
666
741
            parents = factory.parents
667
742
            if parents is None:
668
 
                parent_bytes = 'None:'
 
743
                parent_bytes = b'None:'
669
744
            else:
670
 
                parent_bytes = '\t'.join('\x00'.join(key) for key in parents)
671
 
            record_header = '%s\n%s\n%d\n%d\n' % (
 
745
                parent_bytes = b'\t'.join(b'\x00'.join(key) for key in parents)
 
746
            record_header = b'%s\n%s\n%d\n%d\n' % (
672
747
                key_bytes, parent_bytes, factory._start, factory._end)
673
748
            header_lines.append(record_header)
674
749
            # TODO: Can we break the refcycle at this point and set
675
750
            #       factory._manager = None?
676
 
        header_bytes = ''.join(header_lines)
 
751
        header_bytes = b''.join(header_lines)
677
752
        del header_lines
678
753
        header_bytes_len = len(header_bytes)
679
754
        z_header_bytes = zlib.compress(header_bytes)
680
755
        del header_bytes
681
756
        z_header_bytes_len = len(z_header_bytes)
682
 
        block_bytes = self._block.to_bytes()
683
 
        lines.append('%d\n%d\n%d\n' % (z_header_bytes_len, header_bytes_len,
684
 
                                       len(block_bytes)))
 
757
        block_bytes_len, block_chunks = self._block.to_chunks()
 
758
        lines.append(b'%d\n%d\n%d\n' % (
 
759
            z_header_bytes_len, header_bytes_len, block_bytes_len))
685
760
        lines.append(z_header_bytes)
686
 
        lines.append(block_bytes)
687
 
        del z_header_bytes, block_bytes
688
 
        return ''.join(lines)
 
761
        lines.extend(block_chunks)
 
762
        del z_header_bytes, block_chunks
 
763
        # TODO: This is a point where we will double the memory consumption. To
 
764
        #       avoid this, we probably have to switch to a 'chunked' api
 
765
        return b''.join(lines)
689
766
 
690
767
    @classmethod
691
768
    def from_bytes(cls, bytes):
692
769
        # TODO: This does extra string copying, probably better to do it a
693
 
        #       different way
 
770
        #       different way. At a minimum this creates 2 copies of the
 
771
        #       compressed content
694
772
        (storage_kind, z_header_len, header_len,
695
 
         block_len, rest) = bytes.split('\n', 4)
 
773
         block_len, rest) = bytes.split(b'\n', 4)
696
774
        del bytes
697
 
        if storage_kind != 'groupcompress-block':
 
775
        if storage_kind != b'groupcompress-block':
698
776
            raise ValueError('Unknown storage kind: %s' % (storage_kind,))
699
777
        z_header_len = int(z_header_len)
700
778
        if len(rest) < z_header_len:
712
790
        del rest
713
791
        # So now we have a valid GCB, we just need to parse the factories that
714
792
        # were sent to us
715
 
        header_lines = header.split('\n')
 
793
        header_lines = header.split(b'\n')
716
794
        del header
717
795
        last = header_lines.pop()
718
 
        if last != '':
 
796
        if last != b'':
719
797
            raise ValueError('header lines did not end with a trailing'
720
798
                             ' newline')
721
799
        if len(header_lines) % 4 != 0:
723
801
        block = GroupCompressBlock.from_bytes(block_bytes)
724
802
        del block_bytes
725
803
        result = cls(block)
726
 
        for start in xrange(0, len(header_lines), 4):
 
804
        for start in range(0, len(header_lines), 4):
727
805
            # intern()?
728
 
            key = tuple(header_lines[start].split('\x00'))
729
 
            parents_line = header_lines[start+1]
730
 
            if parents_line == 'None:':
 
806
            key = tuple(header_lines[start].split(b'\x00'))
 
807
            parents_line = header_lines[start + 1]
 
808
            if parents_line == b'None:':
731
809
                parents = None
732
810
            else:
733
 
                parents = tuple([tuple(segment.split('\x00'))
734
 
                                 for segment in parents_line.split('\t')
735
 
                                  if segment])
736
 
            start_offset = int(header_lines[start+2])
737
 
            end_offset = int(header_lines[start+3])
 
811
                parents = tuple([tuple(segment.split(b'\x00'))
 
812
                                 for segment in parents_line.split(b'\t')
 
813
                                 if segment])
 
814
            start_offset = int(header_lines[start + 2])
 
815
            end_offset = int(header_lines[start + 3])
738
816
            result.add_factory(key, parents, start_offset, end_offset)
739
817
        return result
740
818
 
748
826
 
749
827
class _CommonGroupCompressor(object):
750
828
 
751
 
    def __init__(self):
 
829
    def __init__(self, settings=None):
752
830
        """Create a GroupCompressor."""
753
831
        self.chunks = []
754
832
        self._last = None
755
833
        self.endpoint = 0
756
834
        self.input_bytes = 0
757
835
        self.labels_deltas = {}
758
 
        self._delta_index = None # Set by the children
 
836
        self._delta_index = None  # Set by the children
759
837
        self._block = GroupCompressBlock()
 
838
        if settings is None:
 
839
            self._settings = {}
 
840
        else:
 
841
            self._settings = settings
760
842
 
761
 
    def compress(self, key, bytes, expected_sha, nostore_sha=None, soft=False):
 
843
    def compress(self, key, chunks, length, expected_sha, nostore_sha=None,
 
844
                 soft=False):
762
845
        """Compress lines with label key.
763
846
 
764
847
        :param key: A key tuple. It is stored in the output
765
848
            for identification of the text during decompression. If the last
766
 
            element is 'None' it is replaced with the sha1 of the text -
 
849
            element is b'None' it is replaced with the sha1 of the text -
767
850
            e.g. sha1:xxxxxxx.
768
 
        :param bytes: The bytes to be compressed
 
851
        :param chunks: Chunks of bytes to be compressed
 
852
        :param length: Length of chunks
769
853
        :param expected_sha: If non-None, the sha the lines are believed to
770
854
            have. During compression the sha is calculated; a mismatch will
771
855
            cause an error.
779
863
 
780
864
        :seealso VersionedFiles.add_lines:
781
865
        """
782
 
        if not bytes: # empty, like a dir entry, etc
 
866
        if length == 0:  # empty, like a dir entry, etc
783
867
            if nostore_sha == _null_sha1:
784
 
                raise errors.ExistingContent()
 
868
                raise ExistingContent()
785
869
            return _null_sha1, 0, 0, 'fulltext'
786
870
        # we assume someone knew what they were doing when they passed it in
787
871
        if expected_sha is not None:
788
872
            sha1 = expected_sha
789
873
        else:
790
 
            sha1 = osutils.sha_string(bytes)
 
874
            sha1 = osutils.sha_strings(chunks)
791
875
        if nostore_sha is not None:
792
876
            if sha1 == nostore_sha:
793
 
                raise errors.ExistingContent()
 
877
                raise ExistingContent()
794
878
        if key[-1] is None:
795
 
            key = key[:-1] + ('sha1:' + sha1,)
 
879
            key = key[:-1] + (b'sha1:' + sha1,)
796
880
 
797
 
        start, end, type = self._compress(key, bytes, len(bytes) / 2, soft)
 
881
        start, end, type = self._compress(key, chunks, length, length / 2, soft)
798
882
        return sha1, start, end, type
799
883
 
800
 
    def _compress(self, key, bytes, max_delta_size, soft=False):
 
884
    def _compress(self, key, chunks, input_len, max_delta_size, soft=False):
801
885
        """Compress lines with label key.
802
886
 
803
887
        :param key: A key tuple. It is stored in the output for identification
804
888
            of the text during decompression.
805
889
 
806
 
        :param bytes: The bytes to be compressed
 
890
        :param chunks: The chunks of bytes to be compressed
 
891
 
 
892
        :param input_len: The length of the chunks
807
893
 
808
894
        :param max_delta_size: The size above which we issue a fulltext instead
809
895
            of a delta.
820
906
        """Extract a key previously added to the compressor.
821
907
 
822
908
        :param key: The key to extract.
823
 
        :return: An iterable over bytes and the sha1.
 
909
        :return: An iterable over chunks and the sha1.
824
910
        """
825
 
        (start_byte, start_chunk, end_byte, end_chunk) = self.labels_deltas[key]
 
911
        (start_byte, start_chunk, end_byte,
 
912
         end_chunk) = self.labels_deltas[key]
826
913
        delta_chunks = self.chunks[start_chunk:end_chunk]
827
 
        stored_bytes = ''.join(delta_chunks)
828
 
        if stored_bytes[0] == 'f':
 
914
        stored_bytes = b''.join(delta_chunks)
 
915
        kind = stored_bytes[:1]
 
916
        if kind == b'f':
829
917
            fulltext_len, offset = decode_base128_int(stored_bytes[1:10])
830
918
            data_len = fulltext_len + 1 + offset
831
 
            if  data_len != len(stored_bytes):
 
919
            if data_len != len(stored_bytes):
832
920
                raise ValueError('Index claimed fulltext len, but stored bytes'
833
921
                                 ' claim %s != %s'
834
922
                                 % (len(stored_bytes), data_len))
835
 
            bytes = stored_bytes[offset + 1:]
 
923
            data = [stored_bytes[offset + 1:]]
836
924
        else:
 
925
            if kind != b'd':
 
926
                raise ValueError('Unknown content kind, bytes claim %s' % kind)
837
927
            # XXX: This is inefficient at best
838
 
            source = ''.join(self.chunks[:start_chunk])
839
 
            if stored_bytes[0] != 'd':
840
 
                raise ValueError('Unknown content kind, bytes claim %s'
841
 
                                 % (stored_bytes[0],))
 
928
            source = b''.join(self.chunks[:start_chunk])
842
929
            delta_len, offset = decode_base128_int(stored_bytes[1:10])
843
930
            data_len = delta_len + 1 + offset
844
931
            if data_len != len(stored_bytes):
845
932
                raise ValueError('Index claimed delta len, but stored bytes'
846
933
                                 ' claim %s != %s'
847
934
                                 % (len(stored_bytes), data_len))
848
 
            bytes = apply_delta(source, stored_bytes[offset + 1:])
849
 
        bytes_sha1 = osutils.sha_string(bytes)
850
 
        return bytes, bytes_sha1
 
935
            data = [apply_delta(source, stored_bytes[offset + 1:])]
 
936
        data_sha1 = osutils.sha_strings(data)
 
937
        return data, data_sha1
851
938
 
852
939
    def flush(self):
853
940
        """Finish this group, creating a formatted stream.
854
941
 
855
942
        After calling this, the compressor should no longer be used
856
943
        """
857
 
        # TODO: this causes us to 'bloat' to 2x the size of content in the
858
 
        #       group. This has an impact for 'commit' of large objects.
859
 
        #       One possibility is to use self._content_chunks, and be lazy and
860
 
        #       only fill out self._content as a full string when we actually
861
 
        #       need it. That would at least drop the peak memory consumption
862
 
        #       for 'commit' down to ~1x the size of the largest file, at a
863
 
        #       cost of increased complexity within this code. 2x is still <<
864
 
        #       3x the size of the largest file, so we are doing ok.
865
944
        self._block.set_chunked_content(self.chunks, self.endpoint)
866
945
        self.chunks = None
867
946
        self._delta_index = None
885
964
 
886
965
class PythonGroupCompressor(_CommonGroupCompressor):
887
966
 
888
 
    def __init__(self):
 
967
    def __init__(self, settings=None):
889
968
        """Create a GroupCompressor.
890
969
 
891
970
        Used only if the pyrex version is not available.
892
971
        """
893
 
        super(PythonGroupCompressor, self).__init__()
 
972
        super(PythonGroupCompressor, self).__init__(settings)
894
973
        self._delta_index = LinesDeltaIndex([])
895
974
        # The actual content is managed by LinesDeltaIndex
896
975
        self.chunks = self._delta_index.lines
897
976
 
898
 
    def _compress(self, key, bytes, max_delta_size, soft=False):
 
977
    def _compress(self, key, chunks, input_len, max_delta_size, soft=False):
899
978
        """see _CommonGroupCompressor._compress"""
900
 
        input_len = len(bytes)
901
 
        new_lines = osutils.split_lines(bytes)
 
979
        new_lines = osutils.chunks_to_lines(chunks)
902
980
        out_lines, index_lines = self._delta_index.make_delta(
903
981
            new_lines, bytes_length=input_len, soft=soft)
904
982
        delta_length = sum(map(len, out_lines))
905
983
        if delta_length > max_delta_size:
906
984
            # The delta is longer than the fulltext, insert a fulltext
907
985
            type = 'fulltext'
908
 
            out_lines = ['f', encode_base128_int(input_len)]
 
986
            out_lines = [b'f', encode_base128_int(input_len)]
909
987
            out_lines.extend(new_lines)
910
988
            index_lines = [False, False]
911
989
            index_lines.extend([True] * len(new_lines))
912
990
        else:
913
991
            # this is a worthy delta, output it
914
992
            type = 'delta'
915
 
            out_lines[0] = 'd'
 
993
            out_lines[0] = b'd'
916
994
            # Update the delta_length to include those two encoded integers
917
995
            out_lines[1] = encode_base128_int(delta_length)
918
996
        # Before insertion
933
1011
 
934
1012
    It contains code very similar to SequenceMatcher because of having a similar
935
1013
    task. However some key differences apply:
936
 
     - there is no junk, we want a minimal edit not a human readable diff.
937
 
     - we don't filter very common lines (because we don't know where a good
938
 
       range will start, and after the first text we want to be emitting minmal
939
 
       edits only.
940
 
     - we chain the left side, not the right side
941
 
     - we incrementally update the adjacency matrix as new lines are provided.
942
 
     - we look for matches in all of the left side, so the routine which does
943
 
       the analagous task of find_longest_match does not need to filter on the
944
 
       left side.
 
1014
 
 
1015
    * there is no junk, we want a minimal edit not a human readable diff.
 
1016
    * we don't filter very common lines (because we don't know where a good
 
1017
      range will start, and after the first text we want to be emitting minmal
 
1018
      edits only.
 
1019
    * we chain the left side, not the right side
 
1020
    * we incrementally update the adjacency matrix as new lines are provided.
 
1021
    * we look for matches in all of the left side, so the routine which does
 
1022
      the analagous task of find_longest_match does not need to filter on the
 
1023
      left side.
945
1024
    """
946
1025
 
947
 
    def __init__(self):
948
 
        super(PyrexGroupCompressor, self).__init__()
949
 
        self._delta_index = DeltaIndex()
 
1026
    def __init__(self, settings=None):
 
1027
        super(PyrexGroupCompressor, self).__init__(settings)
 
1028
        max_bytes_to_index = self._settings.get('max_bytes_to_index', 0)
 
1029
        self._delta_index = DeltaIndex(max_bytes_to_index=max_bytes_to_index)
950
1030
 
951
 
    def _compress(self, key, bytes, max_delta_size, soft=False):
 
1031
    def _compress(self, key, chunks, input_len, max_delta_size, soft=False):
952
1032
        """see _CommonGroupCompressor._compress"""
953
 
        input_len = len(bytes)
954
1033
        # By having action/label/sha1/len, we can parse the group if the index
955
1034
        # was ever destroyed, we have the key in 'label', we know the final
956
1035
        # bytes are valid from sha1, and we know where to find the end of this
962
1041
        # new_chunks = ['label:%s\nsha1:%s\n' % (label, sha1)]
963
1042
        if self._delta_index._source_offset != self.endpoint:
964
1043
            raise AssertionError('_source_offset != endpoint'
965
 
                ' somehow the DeltaIndex got out of sync with'
966
 
                ' the output lines')
 
1044
                                 ' somehow the DeltaIndex got out of sync with'
 
1045
                                 ' the output lines')
 
1046
        bytes = b''.join(chunks)
967
1047
        delta = self._delta_index.make_delta(bytes, max_delta_size)
968
 
        if (delta is None):
 
1048
        if delta is None:
969
1049
            type = 'fulltext'
970
 
            enc_length = encode_base128_int(len(bytes))
 
1050
            enc_length = encode_base128_int(input_len)
971
1051
            len_mini_header = 1 + len(enc_length)
972
1052
            self._delta_index.add_source(bytes, len_mini_header)
973
 
            new_chunks = ['f', enc_length, bytes]
 
1053
            new_chunks = [b'f', enc_length] + chunks
974
1054
        else:
975
1055
            type = 'delta'
976
1056
            enc_length = encode_base128_int(len(delta))
977
1057
            len_mini_header = 1 + len(enc_length)
978
 
            new_chunks = ['d', enc_length, delta]
 
1058
            new_chunks = [b'd', enc_length, delta]
979
1059
            self._delta_index.add_delta_source(delta, len_mini_header)
980
1060
        # Before insertion
981
1061
        start = self.endpoint
988
1068
                                   self.endpoint, chunk_end)
989
1069
        if not self._delta_index._source_offset == self.endpoint:
990
1070
            raise AssertionError('the delta index is out of sync'
991
 
                'with the output lines %s != %s'
992
 
                % (self._delta_index._source_offset, self.endpoint))
 
1071
                                 'with the output lines %s != %s'
 
1072
                                 % (self._delta_index._source_offset, self.endpoint))
993
1073
        return start, self.endpoint, type
994
1074
 
995
1075
    def _output_chunks(self, new_chunks):
1020
1100
        if graph:
1021
1101
            ref_length = 1
1022
1102
        graph_index = BTreeBuilder(reference_lists=ref_length,
1023
 
            key_elements=keylength)
 
1103
                                   key_elements=keylength)
1024
1104
        stream = transport.open_write_stream('newpack')
1025
1105
        writer = pack.ContainerWriter(stream.write)
1026
1106
        writer.begin()
1027
 
        index = _GCGraphIndex(graph_index, lambda:True, parents=parents,
1028
 
            add_callback=graph_index.add_nodes,
1029
 
            inconsistency_fatal=inconsistency_fatal)
1030
 
        access = knit._DirectPackAccess({})
 
1107
        index = _GCGraphIndex(graph_index, lambda: True, parents=parents,
 
1108
                              add_callback=graph_index.add_nodes,
 
1109
                              inconsistency_fatal=inconsistency_fatal)
 
1110
        access = pack_repo._DirectPackAccess({})
1031
1111
        access.set_writer(writer, graph_index, (transport, 'newpack'))
1032
1112
        result = GroupCompressVersionedFiles(index, access, delta)
1033
1113
        result.stream = stream
1043
1123
 
1044
1124
class _BatchingBlockFetcher(object):
1045
1125
    """Fetch group compress blocks in batches.
1046
 
    
 
1126
 
1047
1127
    :ivar total_bytes: int of expected number of bytes needed to fetch the
1048
1128
        currently pending batch.
1049
1129
    """
1050
1130
 
1051
 
    def __init__(self, gcvf, locations):
 
1131
    def __init__(self, gcvf, locations, get_compressor_settings=None):
1052
1132
        self.gcvf = gcvf
1053
1133
        self.locations = locations
1054
1134
        self.keys = []
1057
1137
        self.total_bytes = 0
1058
1138
        self.last_read_memo = None
1059
1139
        self.manager = None
 
1140
        self._get_compressor_settings = get_compressor_settings
1060
1141
 
1061
1142
    def add_key(self, key):
1062
1143
        """Add another to key to fetch.
1063
 
        
 
1144
 
1064
1145
        :return: The estimated number of bytes needed to fetch the batch so
1065
1146
            far.
1066
1147
        """
1091
1172
            # and then.
1092
1173
            self.batch_memos[read_memo] = cached_block
1093
1174
        return self.total_bytes
1094
 
        
 
1175
 
1095
1176
    def _flush_manager(self):
1096
1177
        if self.manager is not None:
1097
1178
            for factory in self.manager.get_record_stream():
1102
1183
    def yield_factories(self, full_flush=False):
1103
1184
        """Yield factories for keys added since the last yield.  They will be
1104
1185
        returned in the order they were added via add_key.
1105
 
        
 
1186
 
1106
1187
        :param full_flush: by default, some results may not be returned in case
1107
1188
            they can be part of the next batch.  If full_flush is True, then
1108
1189
            all results are returned.
1127
1208
                if memos_to_get_stack and memos_to_get_stack[-1] == read_memo:
1128
1209
                    # The next block from _get_blocks will be the block we
1129
1210
                    # need.
1130
 
                    block_read_memo, block = blocks.next()
 
1211
                    block_read_memo, block = next(blocks)
1131
1212
                    if block_read_memo != read_memo:
1132
1213
                        raise AssertionError(
1133
1214
                            "block_read_memo out of sync with read_memo"
1136
1217
                    memos_to_get_stack.pop()
1137
1218
                else:
1138
1219
                    block = self.batch_memos[read_memo]
1139
 
                self.manager = _LazyGroupContentManager(block)
 
1220
                self.manager = _LazyGroupContentManager(block,
 
1221
                                                        get_compressor_settings=self._get_compressor_settings)
1140
1222
                self.last_read_memo = read_memo
1141
1223
            start, end = index_memo[3:5]
1142
1224
            self.manager.add_factory(key, parents, start, end)
1149
1231
        self.total_bytes = 0
1150
1232
 
1151
1233
 
1152
 
class GroupCompressVersionedFiles(VersionedFiles):
 
1234
class GroupCompressVersionedFiles(VersionedFilesWithFallbacks):
1153
1235
    """A group-compress based VersionedFiles implementation."""
1154
1236
 
1155
 
    def __init__(self, index, access, delta=True, _unadded_refs=None):
 
1237
    # This controls how the GroupCompress DeltaIndex works. Basically, we
 
1238
    # compute hash pointers into the source blocks (so hash(text) => text).
 
1239
    # However each of these references costs some memory in trade against a
 
1240
    # more accurate match result. For very large files, they either are
 
1241
    # pre-compressed and change in bulk whenever they change, or change in just
 
1242
    # local blocks. Either way, 'improved resolution' is not very helpful,
 
1243
    # versus running out of memory trying to track everything. The default max
 
1244
    # gives 100% sampling of a 1MB file.
 
1245
    _DEFAULT_MAX_BYTES_TO_INDEX = 1024 * 1024
 
1246
    _DEFAULT_COMPRESSOR_SETTINGS = {'max_bytes_to_index':
 
1247
                                    _DEFAULT_MAX_BYTES_TO_INDEX}
 
1248
 
 
1249
    def __init__(self, index, access, delta=True, _unadded_refs=None,
 
1250
                 _group_cache=None):
1156
1251
        """Create a GroupCompressVersionedFiles object.
1157
1252
 
1158
1253
        :param index: The index object storing access and graph data.
1159
1254
        :param access: The access object storing raw data.
1160
1255
        :param delta: Whether to delta compress or just entropy compress.
1161
1256
        :param _unadded_refs: private parameter, don't use.
 
1257
        :param _group_cache: private parameter, don't use.
1162
1258
        """
1163
1259
        self._index = index
1164
1260
        self._access = access
1166
1262
        if _unadded_refs is None:
1167
1263
            _unadded_refs = {}
1168
1264
        self._unadded_refs = _unadded_refs
1169
 
        self._group_cache = LRUSizeCache(max_size=50*1024*1024)
1170
 
        self._fallback_vfs = []
 
1265
        if _group_cache is None:
 
1266
            _group_cache = LRUSizeCache(max_size=50 * 1024 * 1024)
 
1267
        self._group_cache = _group_cache
 
1268
        self._immediate_fallback_vfs = []
 
1269
        self._max_bytes_to_index = None
1171
1270
 
1172
1271
    def without_fallbacks(self):
1173
1272
        """Return a clone of this object without any fallbacks configured."""
1174
1273
        return GroupCompressVersionedFiles(self._index, self._access,
1175
 
            self._delta, _unadded_refs=dict(self._unadded_refs))
 
1274
                                           self._delta, _unadded_refs=dict(
 
1275
                                               self._unadded_refs),
 
1276
                                           _group_cache=self._group_cache)
1176
1277
 
1177
1278
    def add_lines(self, key, parents, lines, parent_texts=None,
1178
 
        left_matching_blocks=None, nostore_sha=None, random_id=False,
1179
 
        check_content=True):
 
1279
                  left_matching_blocks=None, nostore_sha=None, random_id=False,
 
1280
                  check_content=True):
1180
1281
        """Add a text to the store.
1181
1282
 
1182
1283
        :param key: The key tuple of the text to add.
1183
1284
        :param parents: The parents key tuples of the text to add.
1184
1285
        :param lines: A list of lines. Each line must be a bytestring. And all
1185
 
            of them except the last must be terminated with \n and contain no
1186
 
            other \n's. The last line may either contain no \n's or a single
1187
 
            terminating \n. If the lines list does meet this constraint the add
1188
 
            routine may error or may succeed - but you will be unable to read
1189
 
            the data back accurately. (Checking the lines have been split
 
1286
            of them except the last must be terminated with \\n and contain no
 
1287
            other \\n's. The last line may either contain no \\n's or a single
 
1288
            terminating \\n. If the lines list does meet this constraint the
 
1289
            add routine may error or may succeed - but you will be unable to
 
1290
            read the data back accurately. (Checking the lines have been split
1190
1291
            correctly is expensive and extremely unlikely to catch bugs so it
1191
1292
            is not done at runtime unless check_content is True.)
1192
1293
        :param parent_texts: An optional dictionary containing the opaque
1211
1312
                 back to future add_lines calls in the parent_texts dictionary.
1212
1313
        """
1213
1314
        self._index._check_write_ok()
1214
 
        self._check_add(key, lines, random_id, check_content)
1215
 
        if parents is None:
1216
 
            # The caller might pass None if there is no graph data, but kndx
1217
 
            # indexes can't directly store that, so we give them
1218
 
            # an empty tuple instead.
1219
 
            parents = ()
1220
 
        # double handling for now. Make it work until then.
1221
 
        length = sum(map(len, lines))
1222
 
        record = ChunkedContentFactory(key, parents, None, lines)
1223
 
        sha1 = list(self._insert_record_stream([record], random_id=random_id,
1224
 
                                               nostore_sha=nostore_sha))[0]
1225
 
        return sha1, length, None
1226
 
 
1227
 
    def _add_text(self, key, parents, text, nostore_sha=None, random_id=False):
1228
 
        """See VersionedFiles._add_text()."""
 
1315
        if check_content:
 
1316
            self._check_lines_not_unicode(lines)
 
1317
            self._check_lines_are_lines(lines)
 
1318
        return self.add_content(
 
1319
            ChunkedContentFactory(
 
1320
                key, parents, osutils.sha_strings(lines), lines, chunks_are_lines=True),
 
1321
            parent_texts, left_matching_blocks, nostore_sha, random_id)
 
1322
 
 
1323
    def add_content(self, factory, parent_texts=None,
 
1324
                    left_matching_blocks=None, nostore_sha=None,
 
1325
                    random_id=False):
 
1326
        """Add a text to the store.
 
1327
 
 
1328
        :param factory: A ContentFactory that can be used to retrieve the key,
 
1329
            parents and contents.
 
1330
        :param parent_texts: An optional dictionary containing the opaque
 
1331
            representations of some or all of the parents of version_id to
 
1332
            allow delta optimisations.  VERY IMPORTANT: the texts must be those
 
1333
            returned by add_lines or data corruption can be caused.
 
1334
        :param left_matching_blocks: a hint about which areas are common
 
1335
            between the text and its left-hand-parent.  The format is
 
1336
            the SequenceMatcher.get_matching_blocks format.
 
1337
        :param nostore_sha: Raise ExistingContent and do not add the lines to
 
1338
            the versioned file if the digest of the lines matches this.
 
1339
        :param random_id: If True a random id has been selected rather than
 
1340
            an id determined by some deterministic process such as a converter
 
1341
            from a foreign VCS. When True the backend may choose not to check
 
1342
            for uniqueness of the resulting key within the versioned file, so
 
1343
            this should only be done when the result is expected to be unique
 
1344
            anyway.
 
1345
        :return: The text sha1, the number of bytes in the text, and an opaque
 
1346
                 representation of the inserted version which can be provided
 
1347
                 back to future add_lines calls in the parent_texts dictionary.
 
1348
        """
1229
1349
        self._index._check_write_ok()
1230
 
        self._check_add(key, None, random_id, check_content=False)
1231
 
        if text.__class__ is not str:
1232
 
            raise errors.BzrBadParameterUnicode("text")
 
1350
        parents = factory.parents
 
1351
        self._check_add(factory.key, random_id)
1233
1352
        if parents is None:
1234
1353
            # The caller might pass None if there is no graph data, but kndx
1235
1354
            # indexes can't directly store that, so we give them
1236
1355
            # an empty tuple instead.
1237
1356
            parents = ()
1238
1357
        # double handling for now. Make it work until then.
1239
 
        length = len(text)
1240
 
        record = FulltextContentFactory(key, parents, None, text)
1241
 
        sha1 = list(self._insert_record_stream([record], random_id=random_id,
1242
 
                                               nostore_sha=nostore_sha))[0]
 
1358
        sha1, length = list(self._insert_record_stream(
 
1359
            [factory], random_id=random_id, nostore_sha=nostore_sha))[0]
1243
1360
        return sha1, length, None
1244
1361
 
1245
1362
    def add_fallback_versioned_files(self, a_versioned_files):
1247
1364
 
1248
1365
        :param a_versioned_files: A VersionedFiles object.
1249
1366
        """
1250
 
        self._fallback_vfs.append(a_versioned_files)
 
1367
        self._immediate_fallback_vfs.append(a_versioned_files)
1251
1368
 
1252
1369
    def annotate(self, key):
1253
1370
        """See VersionedFiles.annotate."""
1262
1379
        if keys is None:
1263
1380
            keys = self.keys()
1264
1381
            for record in self.get_record_stream(keys, 'unordered', True):
1265
 
                record.get_bytes_as('fulltext')
 
1382
                for chunk in record.iter_bytes_as('chunked'):
 
1383
                    pass
1266
1384
        else:
1267
1385
            return self.get_record_stream(keys, 'unordered', True)
1268
1386
 
1272
1390
        self._index._graph_index.clear_cache()
1273
1391
        self._index._int_cache.clear()
1274
1392
 
1275
 
    def _check_add(self, key, lines, random_id, check_content):
 
1393
    def _check_add(self, key, random_id):
1276
1394
        """check that version_id and lines are safe to add."""
1277
1395
        version_id = key[-1]
1278
1396
        if version_id is not None:
1283
1401
        # probably check that the existing content is identical to what is
1284
1402
        # being inserted, and otherwise raise an exception.  This would make
1285
1403
        # the bundle code simpler.
1286
 
        if check_content:
1287
 
            self._check_lines_not_unicode(lines)
1288
 
            self._check_lines_are_lines(lines)
1289
 
 
1290
 
    def get_known_graph_ancestry(self, keys):
1291
 
        """Get a KnownGraph instance with the ancestry of keys."""
1292
 
        # Note that this is identical to
1293
 
        # KnitVersionedFiles.get_known_graph_ancestry, but they don't share
1294
 
        # ancestry.
1295
 
        parent_map, missing_keys = self._index.find_ancestry(keys)
1296
 
        for fallback in self._fallback_vfs:
1297
 
            if not missing_keys:
1298
 
                break
1299
 
            (f_parent_map, f_missing_keys) = fallback._index.find_ancestry(
1300
 
                                                missing_keys)
1301
 
            parent_map.update(f_parent_map)
1302
 
            missing_keys = f_missing_keys
1303
 
        kg = _mod_graph.KnownGraph(parent_map)
1304
 
        return kg
1305
1404
 
1306
1405
    def get_parent_map(self, keys):
1307
1406
        """Get a map of the graph parents of keys.
1323
1422
            and so on.
1324
1423
        """
1325
1424
        result = {}
1326
 
        sources = [self._index] + self._fallback_vfs
 
1425
        sources = [self._index] + self._immediate_fallback_vfs
1327
1426
        source_results = []
1328
1427
        missing = set(keys)
1329
1428
        for source in sources:
1366
1465
                yield read_memo, cached[read_memo]
1367
1466
            except KeyError:
1368
1467
                # Read the block, and cache it.
1369
 
                zdata = raw_records.next()
 
1468
                zdata = next(raw_records)
1370
1469
                block = GroupCompressBlock.from_bytes(zdata)
1371
1470
                self._group_cache[read_memo] = block
1372
1471
                cached[read_memo] = block
1400
1499
        if not keys:
1401
1500
            return
1402
1501
        if (not self._index.has_graph
1403
 
            and ordering in ('topological', 'groupcompress')):
 
1502
                and ordering in ('topological', 'groupcompress')):
1404
1503
            # Cannot topological order when no graph has been stored.
1405
1504
            # but we allow 'as-requested' or 'unordered'
1406
1505
            ordering = 'unordered'
1410
1509
            try:
1411
1510
                keys = set(remaining_keys)
1412
1511
                for content_factory in self._get_remaining_record_stream(keys,
1413
 
                        orig_keys, ordering, include_delta_closure):
 
1512
                                                                         orig_keys, ordering, include_delta_closure):
1414
1513
                    remaining_keys.discard(content_factory.key)
1415
1514
                    yield content_factory
1416
1515
                return
1417
 
            except errors.RetryWithNewPacks, e:
 
1516
            except errors.RetryWithNewPacks as e:
1418
1517
                self._access.reload_or_raise(e)
1419
1518
 
1420
1519
    def _find_from_fallback(self, missing):
1430
1529
        parent_map = {}
1431
1530
        key_to_source_map = {}
1432
1531
        source_results = []
1433
 
        for source in self._fallback_vfs:
 
1532
        for source in self._immediate_fallback_vfs:
1434
1533
            if not missing:
1435
1534
                break
1436
1535
            source_parents = source.get_parent_map(missing)
1446
1545
 
1447
1546
        The returned objects should be in the order defined by 'ordering',
1448
1547
        which can weave between different sources.
 
1548
 
1449
1549
        :param ordering: Must be one of 'topological' or 'groupcompress'
1450
1550
        :return: List of [(source, [keys])] tuples, such that all keys are in
1451
1551
            the defined order, regardless of source.
1452
1552
        """
1453
1553
        if ordering == 'topological':
1454
 
            present_keys = topo_sort(parent_map)
 
1554
            present_keys = tsort.topo_sort(parent_map)
1455
1555
        else:
1456
1556
            # ordering == 'groupcompress'
1457
1557
            # XXX: This only optimizes for the target ordering. We may need
1479
1579
                source = self
1480
1580
            elif key in key_to_source_map:
1481
1581
                source = key_to_source_map[key]
1482
 
            else: # absent
 
1582
            else:  # absent
1483
1583
                continue
1484
1584
            if source is not current_source:
1485
1585
                source_keys.append((source, []))
1493
1593
            # This is the group the bytes are stored in, followed by the
1494
1594
            # location in the group
1495
1595
            return locations[key][0]
1496
 
        present_keys = sorted(locations.iterkeys(), key=get_group)
1497
1596
        # We don't have an ordering for keys in the in-memory object, but
1498
1597
        # lets process the in-memory ones first.
1499
 
        present_keys = list(unadded_keys) + present_keys
 
1598
        present_keys = list(unadded_keys)
 
1599
        present_keys.extend(sorted(locations, key=get_group))
1500
1600
        # Now grab all of the ones from other sources
1501
1601
        source_keys = [(self, present_keys)]
1502
1602
        source_keys.extend(source_result)
1526
1626
            # start with one key, recurse to its oldest parent, then grab
1527
1627
            # everything in the same group, etc.
1528
1628
            parent_map = dict((key, details[2]) for key, details in
1529
 
                locations.iteritems())
 
1629
                              locations.items())
1530
1630
            for key in unadded_keys:
1531
1631
                parent_map[key] = self._unadded_refs[key]
1532
1632
            parent_map.update(fallback_parent_map)
1534
1634
                                                        key_to_source_map)
1535
1635
        elif ordering == 'as-requested':
1536
1636
            source_keys = self._get_as_requested_source_keys(orig_keys,
1537
 
                locations, unadded_keys, key_to_source_map)
 
1637
                                                             locations, unadded_keys, key_to_source_map)
1538
1638
        else:
1539
1639
            # We want to yield the keys in a semi-optimal (read-wise) ordering.
1540
1640
            # Otherwise we thrash the _group_cache and destroy performance
1541
1641
            source_keys = self._get_io_ordered_source_keys(locations,
1542
 
                unadded_keys, source_result)
 
1642
                                                           unadded_keys, source_result)
1543
1643
        for key in missing:
1544
1644
            yield AbsentContentFactory(key)
1545
1645
        # Batch up as many keys as we can until either:
1546
1646
        #  - we encounter an unadded ref, or
1547
1647
        #  - we run out of keys, or
1548
1648
        #  - the total bytes to retrieve for this batch > BATCH_SIZE
1549
 
        batcher = _BatchingBlockFetcher(self, locations)
 
1649
        batcher = _BatchingBlockFetcher(self, locations,
 
1650
                                        get_compressor_settings=self._get_compressor_settings)
1550
1651
        for source, keys in source_keys:
1551
1652
            if source is self:
1552
1653
                for key in keys:
1555
1656
                        # self._compressor.
1556
1657
                        for factory in batcher.yield_factories(full_flush=True):
1557
1658
                            yield factory
1558
 
                        bytes, sha1 = self._compressor.extract(key)
 
1659
                        chunks, sha1 = self._compressor.extract(key)
1559
1660
                        parents = self._unadded_refs[key]
1560
 
                        yield FulltextContentFactory(key, parents, sha1, bytes)
 
1661
                        yield ChunkedContentFactory(key, parents, sha1, chunks)
1561
1662
                        continue
1562
1663
                    if batcher.add_key(key) > BATCH_SIZE:
1563
1664
                        # Ok, this batch is big enough.  Yield some results.
1576
1677
        """See VersionedFiles.get_sha1s()."""
1577
1678
        result = {}
1578
1679
        for record in self.get_record_stream(keys, 'unordered', True):
1579
 
            if record.sha1 != None:
 
1680
            if record.sha1 is not None:
1580
1681
                result[record.key] = record.sha1
1581
1682
            else:
1582
1683
                if record.storage_kind != 'absent':
1583
 
                    result[record.key] = osutils.sha_string(
1584
 
                        record.get_bytes_as('fulltext'))
 
1684
                    result[record.key] = osutils.sha_strings(
 
1685
                        record.iter_bytes_as('chunked'))
1585
1686
        return result
1586
1687
 
1587
1688
    def insert_record_stream(self, stream):
1595
1696
        # test_insert_record_stream_existing_keys fail for groupcompress and
1596
1697
        # groupcompress-nograph, this needs to be revisited while addressing
1597
1698
        # 'bzr branch' performance issues.
1598
 
        for _ in self._insert_record_stream(stream, random_id=False):
 
1699
        for _, _ in self._insert_record_stream(stream, random_id=False):
1599
1700
            pass
1600
1701
 
 
1702
    def _get_compressor_settings(self):
 
1703
        if self._max_bytes_to_index is None:
 
1704
            # TODO: VersionedFiles don't know about their containing
 
1705
            #       repository, so they don't have much of an idea about their
 
1706
            #       location. So for now, this is only a global option.
 
1707
            c = config.GlobalConfig()
 
1708
            val = c.get_user_option('bzr.groupcompress.max_bytes_to_index')
 
1709
            if val is not None:
 
1710
                try:
 
1711
                    val = int(val)
 
1712
                except ValueError as e:
 
1713
                    trace.warning('Value for '
 
1714
                                  '"bzr.groupcompress.max_bytes_to_index"'
 
1715
                                  ' %r is not an integer'
 
1716
                                  % (val,))
 
1717
                    val = None
 
1718
            if val is None:
 
1719
                val = self._DEFAULT_MAX_BYTES_TO_INDEX
 
1720
            self._max_bytes_to_index = val
 
1721
        return {'max_bytes_to_index': self._max_bytes_to_index}
 
1722
 
 
1723
    def _make_group_compressor(self):
 
1724
        return GroupCompressor(self._get_compressor_settings())
 
1725
 
1601
1726
    def _insert_record_stream(self, stream, random_id=False, nostore_sha=None,
1602
1727
                              reuse_blocks=True):
1603
1728
        """Internal core to insert a record stream into this container.
1611
1736
        :param reuse_blocks: If the source is streaming from
1612
1737
            groupcompress-blocks, just insert the blocks as-is, rather than
1613
1738
            expanding the texts and inserting again.
1614
 
        :return: An iterator over the sha1 of the inserted records.
 
1739
        :return: An iterator over (sha1, length) of the inserted records.
1615
1740
        :seealso insert_record_stream:
1616
1741
        :seealso add_lines:
1617
1742
        """
1618
1743
        adapters = {}
 
1744
 
1619
1745
        def get_adapter(adapter_key):
1620
1746
            try:
1621
1747
                return adapters[adapter_key]
1626
1752
                return adapter
1627
1753
        # This will go up to fulltexts for gc to gc fetching, which isn't
1628
1754
        # ideal.
1629
 
        self._compressor = GroupCompressor()
 
1755
        self._compressor = self._make_group_compressor()
1630
1756
        self._unadded_refs = {}
1631
1757
        keys_to_add = []
 
1758
 
1632
1759
        def flush():
1633
 
            bytes = self._compressor.flush().to_bytes()
1634
 
            self._compressor = GroupCompressor()
1635
 
            index, start, length = self._access.add_raw_records(
1636
 
                [(None, len(bytes))], bytes)[0]
 
1760
            bytes_len, chunks = self._compressor.flush().to_chunks()
 
1761
            self._compressor = self._make_group_compressor()
 
1762
            # Note: At this point we still have 1 copy of the fulltext (in
 
1763
            #       record and the var 'bytes'), and this generates 2 copies of
 
1764
            #       the compressed text (one for bytes, one in chunks)
 
1765
            # TODO: Figure out how to indicate that we would be happy to free
 
1766
            #       the fulltext content at this point. Note that sometimes we
 
1767
            #       will want it later (streaming CHK pages), but most of the
 
1768
            #       time we won't (everything else)
 
1769
            index, start, length = self._access.add_raw_record(
 
1770
                None, bytes_len, chunks)
1637
1771
            nodes = []
1638
1772
            for key, reads, refs in keys_to_add:
1639
 
                nodes.append((key, "%d %d %s" % (start, length, reads), refs))
 
1773
                nodes.append((key, b"%d %d %s" % (start, length, reads), refs))
1640
1774
            self._index.add_records(nodes, random_id=random_id)
1641
1775
            self._unadded_refs = {}
1642
1776
            del keys_to_add[:]
1656
1790
                raise errors.RevisionNotPresent(record.key, self)
1657
1791
            if random_id:
1658
1792
                if record.key in inserted_keys:
1659
 
                    trace.note('Insert claimed random_id=True,'
1660
 
                               ' but then inserted %r two times', record.key)
 
1793
                    trace.note(gettext('Insert claimed random_id=True,'
 
1794
                                       ' but then inserted %r two times'), record.key)
1661
1795
                    continue
1662
1796
                inserted_keys.add(record.key)
1663
1797
            if reuse_blocks:
1677
1811
                if record.storage_kind == 'groupcompress-block':
1678
1812
                    # Insert the raw block into the target repo
1679
1813
                    insert_manager = record._manager
1680
 
                    bytes = record._manager._block.to_bytes()
1681
 
                    _, start, length = self._access.add_raw_records(
1682
 
                        [(None, len(bytes))], bytes)[0]
1683
 
                    del bytes
 
1814
                    bytes_len, chunks = record._manager._block.to_chunks()
 
1815
                    _, start, length = self._access.add_raw_record(
 
1816
                        None, bytes_len, chunks)
1684
1817
                    block_start = start
1685
1818
                    block_length = length
1686
1819
                if record.storage_kind in ('groupcompress-block',
1689
1822
                        raise AssertionError('No insert_manager set')
1690
1823
                    if insert_manager is not record._manager:
1691
1824
                        raise AssertionError('insert_manager does not match'
1692
 
                            ' the current record, we cannot be positive'
1693
 
                            ' that the appropriate content was inserted.'
1694
 
                            )
1695
 
                    value = "%d %d %d %d" % (block_start, block_length,
1696
 
                                             record._start, record._end)
 
1825
                                             ' the current record, we cannot be positive'
 
1826
                                             ' that the appropriate content was inserted.'
 
1827
                                             )
 
1828
                    value = b"%d %d %d %d" % (block_start, block_length,
 
1829
                                              record._start, record._end)
1697
1830
                    nodes = [(record.key, value, (record.parents,))]
1698
1831
                    # TODO: Consider buffering up many nodes to be added, not
1699
1832
                    #       sure how much overhead this has, but we're seeing
1701
1834
                    self._index.add_records(nodes, random_id=random_id)
1702
1835
                    continue
1703
1836
            try:
1704
 
                bytes = record.get_bytes_as('fulltext')
1705
 
            except errors.UnavailableRepresentation:
1706
 
                adapter_key = record.storage_kind, 'fulltext'
 
1837
                chunks = record.get_bytes_as('chunked')
 
1838
            except UnavailableRepresentation:
 
1839
                adapter_key = record.storage_kind, 'chunked'
1707
1840
                adapter = get_adapter(adapter_key)
1708
 
                bytes = adapter.get_bytes(record)
 
1841
                chunks = adapter.get_bytes(record, 'chunked')
 
1842
            chunks_len = record.size
 
1843
            if chunks_len is None:
 
1844
                chunks_len = sum(map(len, chunks))
1709
1845
            if len(record.key) > 1:
1710
1846
                prefix = record.key[0]
1711
1847
                soft = (prefix == last_prefix)
1712
1848
            else:
1713
1849
                prefix = None
1714
1850
                soft = False
1715
 
            if max_fulltext_len < len(bytes):
1716
 
                max_fulltext_len = len(bytes)
 
1851
            if max_fulltext_len < chunks_len:
 
1852
                max_fulltext_len = chunks_len
1717
1853
                max_fulltext_prefix = prefix
1718
1854
            (found_sha1, start_point, end_point,
1719
 
             type) = self._compressor.compress(record.key,
1720
 
                                               bytes, record.sha1, soft=soft,
1721
 
                                               nostore_sha=nostore_sha)
1722
 
            # delta_ratio = float(len(bytes)) / (end_point - start_point)
 
1855
             type) = self._compressor.compress(
 
1856
                 record.key, chunks, chunks_len, record.sha1, soft=soft,
 
1857
                 nostore_sha=nostore_sha)
 
1858
            # delta_ratio = float(chunks_len) / (end_point - start_point)
1723
1859
            # Check if we want to continue to include that text
1724
1860
            if (prefix == max_fulltext_prefix
1725
 
                and end_point < 2 * max_fulltext_len):
 
1861
                    and end_point < 2 * max_fulltext_len):
1726
1862
                # As long as we are on the same file_id, we will fill at least
1727
1863
                # 2 * max_fulltext_len
1728
1864
                start_new_block = False
1729
 
            elif end_point > 4*1024*1024:
 
1865
            elif end_point > 4 * 1024 * 1024:
1730
1866
                start_new_block = True
1731
1867
            elif (prefix is not None and prefix != last_prefix
1732
 
                  and end_point > 2*1024*1024):
 
1868
                  and end_point > 2 * 1024 * 1024):
1733
1869
                start_new_block = True
1734
1870
            else:
1735
1871
                start_new_block = False
1737
1873
            if start_new_block:
1738
1874
                self._compressor.pop_last()
1739
1875
                flush()
1740
 
                max_fulltext_len = len(bytes)
 
1876
                max_fulltext_len = chunks_len
1741
1877
                (found_sha1, start_point, end_point,
1742
 
                 type) = self._compressor.compress(record.key, bytes,
1743
 
                                                   record.sha1)
 
1878
                 type) = self._compressor.compress(
 
1879
                     record.key, chunks, chunks_len, record.sha1)
1744
1880
            if record.key[-1] is None:
1745
 
                key = record.key[:-1] + ('sha1:' + found_sha1,)
 
1881
                key = record.key[:-1] + (b'sha1:' + found_sha1,)
1746
1882
            else:
1747
1883
                key = record.key
1748
1884
            self._unadded_refs[key] = record.parents
1749
 
            yield found_sha1
 
1885
            yield found_sha1, chunks_len
1750
1886
            as_st = static_tuple.StaticTuple.from_sequence
1751
1887
            if record.parents is not None:
1752
1888
                parents = as_st([as_st(p) for p in record.parents])
1753
1889
            else:
1754
1890
                parents = None
1755
1891
            refs = static_tuple.StaticTuple(parents)
1756
 
            keys_to_add.append((key, '%d %d' % (start_point, end_point), refs))
 
1892
            keys_to_add.append(
 
1893
                (key, b'%d %d' % (start_point, end_point), refs))
1757
1894
        if len(keys_to_add):
1758
1895
            flush()
1759
1896
        self._compressor = None
1785
1922
        # but we need to setup a list of records to visit.
1786
1923
        # we need key, position, length
1787
1924
        for key_idx, record in enumerate(self.get_record_stream(keys,
1788
 
            'unordered', True)):
 
1925
                                                                'unordered', True)):
1789
1926
            # XXX: todo - optimise to use less than full texts.
1790
1927
            key = record.key
1791
1928
            if pb is not None:
1792
1929
                pb.update('Walking content', key_idx, total)
1793
1930
            if record.storage_kind == 'absent':
1794
1931
                raise errors.RevisionNotPresent(key, self)
1795
 
            lines = osutils.split_lines(record.get_bytes_as('fulltext'))
1796
 
            for line in lines:
 
1932
            for line in record.iter_bytes_as('lines'):
1797
1933
                yield line, key
1798
1934
        if pb is not None:
1799
1935
            pb.update('Walking content', total, total)
1802
1938
        """See VersionedFiles.keys."""
1803
1939
        if 'evil' in debug.debug_flags:
1804
1940
            trace.mutter_callsite(2, "keys scales with size of history")
1805
 
        sources = [self._index] + self._fallback_vfs
 
1941
        sources = [self._index] + self._immediate_fallback_vfs
1806
1942
        result = set()
1807
1943
        for source in sources:
1808
1944
            result.update(source.keys())
1809
1945
        return result
1810
1946
 
1811
1947
 
 
1948
class _GCBuildDetails(object):
 
1949
    """A blob of data about the build details.
 
1950
 
 
1951
    This stores the minimal data, which then allows compatibility with the old
 
1952
    api, without taking as much memory.
 
1953
    """
 
1954
 
 
1955
    __slots__ = ('_index', '_group_start', '_group_end', '_basis_end',
 
1956
                 '_delta_end', '_parents')
 
1957
 
 
1958
    method = 'group'
 
1959
    compression_parent = None
 
1960
 
 
1961
    def __init__(self, parents, position_info):
 
1962
        self._parents = parents
 
1963
        (self._index, self._group_start, self._group_end, self._basis_end,
 
1964
         self._delta_end) = position_info
 
1965
 
 
1966
    def __repr__(self):
 
1967
        return '%s(%s, %s)' % (self.__class__.__name__,
 
1968
                               self.index_memo, self._parents)
 
1969
 
 
1970
    @property
 
1971
    def index_memo(self):
 
1972
        return (self._index, self._group_start, self._group_end,
 
1973
                self._basis_end, self._delta_end)
 
1974
 
 
1975
    @property
 
1976
    def record_details(self):
 
1977
        return static_tuple.StaticTuple(self.method, None)
 
1978
 
 
1979
    def __getitem__(self, offset):
 
1980
        """Compatibility thunk to act like a tuple."""
 
1981
        if offset == 0:
 
1982
            return self.index_memo
 
1983
        elif offset == 1:
 
1984
            return self.compression_parent  # Always None
 
1985
        elif offset == 2:
 
1986
            return self._parents
 
1987
        elif offset == 3:
 
1988
            return self.record_details
 
1989
        else:
 
1990
            raise IndexError('offset out of range')
 
1991
 
 
1992
    def __len__(self):
 
1993
        return 4
 
1994
 
 
1995
 
1812
1996
class _GCGraphIndex(object):
1813
1997
    """Mapper from GroupCompressVersionedFiles needs into GraphIndex storage."""
1814
1998
 
1815
1999
    def __init__(self, graph_index, is_locked, parents=True,
1816
 
        add_callback=None, track_external_parent_refs=False,
1817
 
        inconsistency_fatal=True, track_new_keys=False):
 
2000
                 add_callback=None, track_external_parent_refs=False,
 
2001
                 inconsistency_fatal=True, track_new_keys=False):
1818
2002
        """Construct a _GCGraphIndex on a graph_index.
1819
2003
 
1820
 
        :param graph_index: An implementation of bzrlib.index.GraphIndex.
 
2004
        :param graph_index: An implementation of breezy.index.GraphIndex.
1821
2005
        :param is_locked: A callback, returns True if the index is locked and
1822
2006
            thus usable.
1823
2007
        :param parents: If True, record knits parents, if not do not record
1843
2027
        # repeated over and over, this creates a surplus of ints
1844
2028
        self._int_cache = {}
1845
2029
        if track_external_parent_refs:
1846
 
            self._key_dependencies = knit._KeyRefs(
 
2030
            self._key_dependencies = _KeyRefs(
1847
2031
                track_new_keys=track_new_keys)
1848
2032
        else:
1849
2033
            self._key_dependencies = None
1873
2057
                if refs:
1874
2058
                    for ref in refs:
1875
2059
                        if ref:
1876
 
                            raise errors.KnitCorrupt(self,
1877
 
                                "attempt to add node with parents "
1878
 
                                "in parentless index.")
 
2060
                            raise knit.KnitCorrupt(self,
 
2061
                                                   "attempt to add node with parents "
 
2062
                                                   "in parentless index.")
1879
2063
                    refs = ()
1880
2064
                    changed = True
1881
2065
            keys[key] = (value, refs)
1889
2073
                if node_refs != passed[1]:
1890
2074
                    details = '%s %s %s' % (key, (value, node_refs), passed)
1891
2075
                    if self._inconsistency_fatal:
1892
 
                        raise errors.KnitCorrupt(self, "inconsistent details"
1893
 
                                                 " in add_records: %s" %
1894
 
                                                 details)
 
2076
                        raise knit.KnitCorrupt(self, "inconsistent details"
 
2077
                                               " in add_records: %s" %
 
2078
                                               details)
1895
2079
                    else:
1896
2080
                        trace.warning("inconsistent details in skipped"
1897
2081
                                      " record: %s", details)
1900
2084
        if changed:
1901
2085
            result = []
1902
2086
            if self._parents:
1903
 
                for key, (value, node_refs) in keys.iteritems():
 
2087
                for key, (value, node_refs) in keys.items():
1904
2088
                    result.append((key, value, node_refs))
1905
2089
            else:
1906
 
                for key, (value, node_refs) in keys.iteritems():
 
2090
                for key, (value, node_refs) in keys.items():
1907
2091
                    result.append((key, value))
1908
2092
            records = result
1909
2093
        key_dependencies = self._key_dependencies
1989
2173
        :param keys: An iterable of keys.
1990
2174
        :return: A dict of key:
1991
2175
            (index_memo, compression_parent, parents, record_details).
1992
 
            index_memo
1993
 
                opaque structure to pass to read_records to extract the raw
1994
 
                data
1995
 
            compression_parent
1996
 
                Content that this record is built upon, may be None
1997
 
            parents
1998
 
                Logical parents of this node
1999
 
            record_details
2000
 
                extra information about the content which needs to be passed to
2001
 
                Factory.parse_record
 
2176
 
 
2177
            * index_memo: opaque structure to pass to read_records to extract
 
2178
              the raw data
 
2179
            * compression_parent: Content that this record is built upon, may
 
2180
              be None
 
2181
            * parents: Logical parents of this node
 
2182
            * record_details: extra information about the content which needs
 
2183
              to be passed to Factory.parse_record
2002
2184
        """
2003
2185
        self._check_read()
2004
2186
        result = {}
2009
2191
                parents = None
2010
2192
            else:
2011
2193
                parents = entry[3][0]
2012
 
            method = 'group'
2013
 
            result[key] = (self._node_to_position(entry),
2014
 
                                  None, parents, (method, None))
 
2194
            details = _GCBuildDetails(parents, self._node_to_position(entry))
 
2195
            result[key] = details
2015
2196
        return result
2016
2197
 
2017
2198
    def keys(self):
2024
2205
 
2025
2206
    def _node_to_position(self, node):
2026
2207
        """Convert an index value to position details."""
2027
 
        bits = node[2].split(' ')
 
2208
        bits = node[2].split(b' ')
2028
2209
        # It would be nice not to read the entire gzip.
2029
2210
        # start and stop are put into _int_cache because they are very common.
2030
2211
        # They define the 'group' that an entry is in, and many groups can have
2033
2214
        # each, or about 7MB. Note that it might be even more when you consider
2034
2215
        # how PyInt is allocated in separate slabs. And you can't return a slab
2035
2216
        # to the OS if even 1 int on it is in use. Note though that Python uses
2036
 
        # a LIFO when re-using PyInt slots, which probably causes more
 
2217
        # a LIFO when re-using PyInt slots, which might cause more
2037
2218
        # fragmentation.
2038
2219
        start = int(bits[0])
2039
2220
        start = self._int_cache.setdefault(start, start)
2063
2244
            key_dependencies.add_references(node[1], node[3][0])
2064
2245
 
2065
2246
 
2066
 
from bzrlib._groupcompress_py import (
 
2247
from ._groupcompress_py import (
2067
2248
    apply_delta,
2068
2249
    apply_delta_to_source,
2069
2250
    encode_base128_int,
2072
2253
    LinesDeltaIndex,
2073
2254
    )
2074
2255
try:
2075
 
    from bzrlib._groupcompress_pyx import (
 
2256
    from ._groupcompress_pyx import (
2076
2257
        apply_delta,
2077
2258
        apply_delta_to_source,
2078
2259
        DeltaIndex,
2080
2261
        decode_base128_int,
2081
2262
        )
2082
2263
    GroupCompressor = PyrexGroupCompressor
2083
 
except ImportError, e:
 
2264
except ImportError as e:
2084
2265
    osutils.failed_to_load_extension(e)
2085
2266
    GroupCompressor = PythonGroupCompressor
2086