~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/repofmt/groupcompress_repo.py

(gz) Change minimum required testtools version for selftest to 0.9.5 for
 unicode fixes (Martin [gz])

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2008-2011 Canonical Ltd
 
1
# Copyright (C) 2008, 2009, 2010 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
26
26
    errors,
27
27
    index as _mod_index,
28
28
    inventory,
 
29
    knit,
29
30
    osutils,
30
31
    pack,
31
32
    revision as _mod_revision,
43
44
    GroupCompressVersionedFiles,
44
45
    )
45
46
from bzrlib.repofmt.pack_repo import (
46
 
    _DirectPackAccess,
47
47
    Pack,
48
48
    NewPack,
49
 
    PackRepository,
 
49
    KnitPackRepository,
 
50
    KnitPackStreamSource,
50
51
    PackRootCommitBuilder,
51
52
    RepositoryPackCollection,
52
53
    RepositoryFormatPack,
53
54
    ResumedPack,
54
55
    Packer,
55
56
    )
56
 
from bzrlib.repository import (
57
 
    StreamSource,
58
 
    )
59
57
from bzrlib.static_tuple import StaticTuple
60
58
 
61
59
 
354
352
        """Build a VersionedFiles instance on top of this group of packs."""
355
353
        index_name = index_name + '_index'
356
354
        index_to_pack = {}
357
 
        access = _DirectPackAccess(index_to_pack,
358
 
                                   reload_func=self._reload_func)
 
355
        access = knit._DirectPackAccess(index_to_pack,
 
356
                                        reload_func=self._reload_func)
359
357
        if for_write:
360
358
            # Use new_pack
361
359
            if self.new_pack is None:
423
421
        inventory_keys = source_vf.keys()
424
422
        missing_inventories = set(self.revision_keys).difference(inventory_keys)
425
423
        if missing_inventories:
426
 
            # Go back to the original repo, to see if these are really missing
427
 
            # https://bugs.launchpad.net/bzr/+bug/437003
428
 
            # If we are packing a subset of the repo, it is fine to just have
429
 
            # the data in another Pack file, which is not included in this pack
430
 
            # operation.
431
 
            inv_index = self._pack_collection.repo.inventories._index
432
 
            pmap = inv_index.get_parent_map(missing_inventories)
433
 
            really_missing = missing_inventories.difference(pmap)
434
 
            if really_missing:
435
 
                missing_inventories = sorted(really_missing)
436
 
                raise ValueError('We are missing inventories for revisions: %s'
437
 
                    % (missing_inventories,))
 
424
            missing_inventories = sorted(missing_inventories)
 
425
            raise ValueError('We are missing inventories for revisions: %s'
 
426
                % (missing_inventories,))
438
427
        self._copy_stream(source_vf, target_vf, inventory_keys,
439
428
                          'inventories', self._get_filtered_inv_stream, 2)
440
429
 
605
594
    def __init__(self, *args, **kwargs):
606
595
        super(GCCHKCanonicalizingPacker, self).__init__(*args, **kwargs)
607
596
        self._data_changed = False
608
 
 
 
597
    
609
598
    def _exhaust_stream(self, source_vf, keys, message, vf_to_stream, pb_offset):
610
599
        """Create and exhaust a stream, but don't insert it.
611
 
 
 
600
        
612
601
        This is useful to get the side-effects of generating a stream.
613
602
        """
614
603
        self.pb.update('scanning %s' % (message,), pb_offset)
703
692
 
704
693
    pack_factory = GCPack
705
694
    resumed_pack_factory = ResumedGCPack
706
 
    normal_packer_class = GCCHKPacker
707
 
    optimising_packer_class = GCCHKPacker
708
695
 
709
696
    def _check_new_inventories(self):
710
697
        """Detect missing inventories or chk root entries for the new revisions
792
779
                % (sorted(missing_text_keys),))
793
780
        return problems
794
781
 
795
 
 
796
 
class CHKInventoryRepository(PackRepository):
797
 
    """subclass of PackRepository that uses CHK based inventories."""
 
782
    def _execute_pack_operations(self, pack_operations,
 
783
                                 _packer_class=GCCHKPacker,
 
784
                                 reload_func=None):
 
785
        """Execute a series of pack operations.
 
786
 
 
787
        :param pack_operations: A list of [revision_count, packs_to_combine].
 
788
        :param _packer_class: The class of packer to use (default: Packer).
 
789
        :return: None.
 
790
        """
 
791
        # XXX: Copied across from RepositoryPackCollection simply because we
 
792
        #      want to override the _packer_class ... :(
 
793
        for revision_count, packs in pack_operations:
 
794
            # we may have no-ops from the setup logic
 
795
            if len(packs) == 0:
 
796
                continue
 
797
            packer = GCCHKPacker(self, packs, '.autopack',
 
798
                                 reload_func=reload_func)
 
799
            try:
 
800
                result = packer.pack()
 
801
            except errors.RetryWithNewPacks:
 
802
                # An exception is propagating out of this context, make sure
 
803
                # this packer has cleaned up. Packer() doesn't set its new_pack
 
804
                # state into the RepositoryPackCollection object, so we only
 
805
                # have access to it directly here.
 
806
                if packer.new_pack is not None:
 
807
                    packer.new_pack.abort()
 
808
                raise
 
809
            if result is None:
 
810
                return
 
811
            for pack in packs:
 
812
                self._remove_pack_from_memory(pack)
 
813
        # record the newly available packs and stop advertising the old
 
814
        # packs
 
815
        to_be_obsoleted = []
 
816
        for _, packs in pack_operations:
 
817
            to_be_obsoleted.extend(packs)
 
818
        result = self._save_pack_names(clear_obsolete_packs=True,
 
819
                                       obsolete_packs=to_be_obsoleted)
 
820
        return result
 
821
 
 
822
 
 
823
class CHKInventoryRepository(KnitPackRepository):
 
824
    """subclass of KnitPackRepository that uses CHK based inventories."""
798
825
 
799
826
    def __init__(self, _format, a_bzrdir, control_files, _commit_builder_class,
800
827
        _serializer):
801
828
        """Overridden to change pack collection class."""
802
 
        super(CHKInventoryRepository, self).__init__(_format, a_bzrdir,
803
 
            control_files, _commit_builder_class, _serializer)
 
829
        KnitPackRepository.__init__(self, _format, a_bzrdir, control_files,
 
830
            _commit_builder_class, _serializer)
 
831
        # and now replace everything it did :)
804
832
        index_transport = self._transport.clone('indices')
805
833
        self._pack_collection = GCRepositoryPackCollection(self,
806
834
            self._transport, index_transport,
1109
1137
            return GroupCHKStreamSource(self, to_format)
1110
1138
        return super(CHKInventoryRepository, self)._get_source(to_format)
1111
1139
 
1112
 
    def _find_inconsistent_revision_parents(self, revisions_iterator=None):
1113
 
        """Find revisions with different parent lists in the revision object
1114
 
        and in the index graph.
1115
 
 
1116
 
        :param revisions_iterator: None, or an iterator of (revid,
1117
 
            Revision-or-None). This iterator controls the revisions checked.
1118
 
        :returns: an iterator yielding tuples of (revison-id, parents-in-index,
1119
 
            parents-in-revision).
1120
 
        """
1121
 
        if not self.is_locked():
1122
 
            raise AssertionError()
1123
 
        vf = self.revisions
1124
 
        if revisions_iterator is None:
1125
 
            revisions_iterator = self._iter_revisions(None)
1126
 
        for revid, revision in revisions_iterator:
1127
 
            if revision is None:
1128
 
                pass
1129
 
            parent_map = vf.get_parent_map([(revid,)])
1130
 
            parents_according_to_index = tuple(parent[-1] for parent in
1131
 
                parent_map[(revid,)])
1132
 
            parents_according_to_revision = tuple(revision.parent_ids)
1133
 
            if parents_according_to_index != parents_according_to_revision:
1134
 
                yield (revid, parents_according_to_index,
1135
 
                    parents_according_to_revision)
1136
 
 
1137
 
    def _check_for_inconsistent_revision_parents(self):
1138
 
        inconsistencies = list(self._find_inconsistent_revision_parents())
1139
 
        if inconsistencies:
1140
 
            raise errors.BzrCheckError(
1141
 
                "Revision index has inconsistent parents.")
1142
 
 
1143
 
 
1144
 
class GroupCHKStreamSource(StreamSource):
 
1140
 
 
1141
class GroupCHKStreamSource(KnitPackStreamSource):
1145
1142
    """Used when both the source and target repo are GroupCHK repos."""
1146
1143
 
1147
1144
    def __init__(self, from_repository, to_format):
1234
1231
            self._chk_p_id_roots = None
1235
1232
        yield 'chk_bytes', _get_parent_id_basename_to_file_id_pages()
1236
1233
 
1237
 
    def _get_text_stream(self):
1238
 
        # Note: We know we don't have to handle adding root keys, because both
1239
 
        # the source and target are the identical network name.
1240
 
        text_stream = self.from_repository.texts.get_record_stream(
1241
 
                        self._text_keys, self._text_fetch_order, False)
1242
 
        return ('texts', text_stream)
1243
 
 
1244
1234
    def get_stream(self, search):
1245
1235
        def wrap_and_count(pb, rc, stream):
1246
1236
            """Yield records from stream while showing progress."""