~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/repofmt/knitrepo.py

  • Committer: Patch Queue Manager
  • Date: 2011-09-22 14:12:18 UTC
  • mfrom: (6155.3.1 jam)
  • Revision ID: pqm@pqm.ubuntu.com-20110922141218-86s4uu6nqvourw4f
(jameinel) Cleanup comments bzrlib/smart/__init__.py (John A Meinel)

Show diffs side-by-side

added added

removed removed

Lines of Context:
16
16
 
17
17
from bzrlib.lazy_import import lazy_import
18
18
lazy_import(globals(), """
 
19
import itertools
 
20
 
19
21
from bzrlib import (
20
22
    bzrdir,
21
23
    errors,
32
34
    xml7,
33
35
    )
34
36
""")
35
 
from bzrlib import (
36
 
    symbol_versioning,
37
 
    )
38
37
from bzrlib.decorators import needs_read_lock, needs_write_lock
39
38
from bzrlib.repository import (
40
 
    CommitBuilder,
 
39
    InterRepository,
41
40
    IsInWriteGroupError,
42
 
    MetaDirRepository,
43
 
    MetaDirRepositoryFormat,
44
41
    RepositoryFormat,
45
 
    RootCommitBuilder,
46
 
    )
 
42
    )
 
43
from bzrlib.vf_repository import (
 
44
    InterSameDataRepository,
 
45
    MetaDirVersionedFileRepository,
 
46
    MetaDirVersionedFileRepositoryFormat,
 
47
    VersionedFileCommitBuilder,
 
48
    VersionedFileRootCommitBuilder,
 
49
    )
 
50
from bzrlib import symbol_versioning
47
51
 
48
52
 
49
53
class _KnitParentsProvider(object):
103
107
        return result
104
108
 
105
109
 
106
 
class KnitRepository(MetaDirRepository):
 
110
class KnitRepository(MetaDirVersionedFileRepository):
107
111
    """Knit format repository."""
108
112
 
109
113
    # These attributes are inherited from the Repository base class. Setting
115
119
 
116
120
    def __init__(self, _format, a_bzrdir, control_files, _commit_builder_class,
117
121
        _serializer):
118
 
        MetaDirRepository.__init__(self, _format, a_bzrdir, control_files)
 
122
        super(KnitRepository, self).__init__(_format, a_bzrdir, control_files)
119
123
        self._commit_builder_class = _commit_builder_class
120
124
        self._serializer = _serializer
121
125
        self._reconcile_fixes_text_parents = True
177
181
        result.get_parent_map([('A',)])
178
182
        return result
179
183
 
180
 
    def fileid_involved_between_revs(self, from_revid, to_revid):
181
 
        """Find file_id(s) which are involved in the changes between revisions.
182
 
 
183
 
        This determines the set of revisions which are involved, and then
184
 
        finds all file ids affected by those revisions.
185
 
        """
186
 
        vf = self._get_revision_vf()
187
 
        from_set = set(vf.get_ancestry(from_revid))
188
 
        to_set = set(vf.get_ancestry(to_revid))
189
 
        changed = to_set.difference(from_set)
190
 
        return self._fileid_involved_by_set(changed)
191
 
 
192
 
    def fileid_involved(self, last_revid=None):
193
 
        """Find all file_ids modified in the ancestry of last_revid.
194
 
 
195
 
        :param last_revid: If None, last_revision() will be used.
196
 
        """
197
 
        if not last_revid:
198
 
            changed = set(self.all_revision_ids())
199
 
        else:
200
 
            changed = set(self.get_ancestry(last_revid))
201
 
        if None in changed:
202
 
            changed.remove(None)
203
 
        return self._fileid_involved_by_set(changed)
204
 
 
205
184
    @needs_read_lock
206
185
    def get_revision(self, revision_id):
207
186
        """Return the Revision object for a named revision"""
232
211
    def _make_parents_provider(self):
233
212
        return _KnitsParentsProvider(self.revisions)
234
213
 
235
 
    def _find_inconsistent_revision_parents(self, revisions_iterator=None):
236
 
        """Find revisions with different parent lists in the revision object
237
 
        and in the index graph.
238
 
 
239
 
        :param revisions_iterator: None, or an iterator of (revid,
240
 
            Revision-or-None). This iterator controls the revisions checked.
241
 
        :returns: an iterator yielding tuples of (revison-id, parents-in-index,
242
 
            parents-in-revision).
243
 
        """
244
 
        if not self.is_locked():
245
 
            raise AssertionError()
246
 
        vf = self.revisions
247
 
        if revisions_iterator is None:
248
 
            revisions_iterator = self._iter_revisions(None)
249
 
        for revid, revision in revisions_iterator:
250
 
            if revision is None:
251
 
                pass
252
 
            parent_map = vf.get_parent_map([(revid,)])
253
 
            parents_according_to_index = tuple(parent[-1] for parent in
254
 
                parent_map[(revid,)])
255
 
            parents_according_to_revision = tuple(revision.parent_ids)
256
 
            if parents_according_to_index != parents_according_to_revision:
257
 
                yield (revid, parents_according_to_index,
258
 
                    parents_according_to_revision)
259
 
 
260
 
    def _check_for_inconsistent_revision_parents(self):
261
 
        inconsistencies = list(self._find_inconsistent_revision_parents())
262
 
        if inconsistencies:
263
 
            raise errors.BzrCheckError(
264
 
                "Revision knit has inconsistent parents.")
265
 
 
266
 
    def revision_graph_can_have_wrong_parents(self):
267
 
        # The revision.kndx could potentially claim a revision has a different
268
 
        # parent to the revision text.
269
 
        return True
270
 
 
271
 
 
272
 
class RepositoryFormatKnit(MetaDirRepositoryFormat):
 
214
 
 
215
class RepositoryFormatKnit(MetaDirVersionedFileRepositoryFormat):
273
216
    """Bzr repository knit format (generalized).
274
217
 
275
218
    This repository format has:
304
247
    _fetch_order = 'topological'
305
248
    _fetch_uses_deltas = True
306
249
    fast_deltas = False
 
250
    supports_funky_characters = True
 
251
    # The revision.kndx could potentially claim a revision has a different
 
252
    # parent to the revision text.
 
253
    revision_graph_can_have_wrong_parents = True
307
254
 
308
255
    def _get_inventories(self, repo_transport, repo, name='inventory'):
309
256
        mapper = versionedfile.ConstantMapper(name)
412
359
    """
413
360
 
414
361
    repository_class = KnitRepository
415
 
    _commit_builder_class = CommitBuilder
 
362
    _commit_builder_class = VersionedFileCommitBuilder
416
363
    @property
417
364
    def _serializer(self):
418
365
        return xml5.serializer_v5
446
393
    """
447
394
 
448
395
    repository_class = KnitRepository
449
 
    _commit_builder_class = RootCommitBuilder
 
396
    _commit_builder_class = VersionedFileRootCommitBuilder
450
397
    rich_root_data = True
451
398
    experimental = True
452
399
    supports_tree_reference = True
488
435
    """
489
436
 
490
437
    repository_class = KnitRepository
491
 
    _commit_builder_class = RootCommitBuilder
 
438
    _commit_builder_class = VersionedFileRootCommitBuilder
492
439
    rich_root_data = True
493
440
    supports_tree_reference = False
494
441
    @property
510
457
    def get_format_description(self):
511
458
        """See RepositoryFormat.get_format_description()."""
512
459
        return "Knit repository format 4"
 
460
 
 
461
 
 
462
class InterKnitRepo(InterSameDataRepository):
 
463
    """Optimised code paths between Knit based repositories."""
 
464
 
 
465
    @classmethod
 
466
    def _get_repo_format_to_test(self):
 
467
        return RepositoryFormatKnit1()
 
468
 
 
469
    @staticmethod
 
470
    def is_compatible(source, target):
 
471
        """Be compatible with known Knit formats.
 
472
 
 
473
        We don't test for the stores being of specific types because that
 
474
        could lead to confusing results, and there is no need to be
 
475
        overly general.
 
476
        """
 
477
        try:
 
478
            are_knits = (isinstance(source._format, RepositoryFormatKnit) and
 
479
                isinstance(target._format, RepositoryFormatKnit))
 
480
        except AttributeError:
 
481
            return False
 
482
        return are_knits and InterRepository._same_model(source, target)
 
483
 
 
484
    @needs_read_lock
 
485
    def search_missing_revision_ids(self,
 
486
            revision_id=symbol_versioning.DEPRECATED_PARAMETER,
 
487
            find_ghosts=True, revision_ids=None, if_present_ids=None,
 
488
            limit=None):
 
489
        """See InterRepository.search_missing_revision_ids()."""
 
490
        if symbol_versioning.deprecated_passed(revision_id):
 
491
            symbol_versioning.warn(
 
492
                'search_missing_revision_ids(revision_id=...) was '
 
493
                'deprecated in 2.4.  Use revision_ids=[...] instead.',
 
494
                DeprecationWarning, stacklevel=2)
 
495
            if revision_ids is not None:
 
496
                raise AssertionError(
 
497
                    'revision_ids is mutually exclusive with revision_id')
 
498
            if revision_id is not None:
 
499
                revision_ids = [revision_id]
 
500
        del revision_id
 
501
        source_ids_set = self._present_source_revisions_for(
 
502
            revision_ids, if_present_ids)
 
503
        # source_ids is the worst possible case we may need to pull.
 
504
        # now we want to filter source_ids against what we actually
 
505
        # have in target, but don't try to check for existence where we know
 
506
        # we do not have a revision as that would be pointless.
 
507
        target_ids = set(self.target.all_revision_ids())
 
508
        possibly_present_revisions = target_ids.intersection(source_ids_set)
 
509
        actually_present_revisions = set(
 
510
            self.target._eliminate_revisions_not_present(possibly_present_revisions))
 
511
        required_revisions = source_ids_set.difference(actually_present_revisions)
 
512
        if revision_ids is not None:
 
513
            # we used get_ancestry to determine source_ids then we are assured all
 
514
            # revisions referenced are present as they are installed in topological order.
 
515
            # and the tip revision was validated by get_ancestry.
 
516
            result_set = required_revisions
 
517
        else:
 
518
            # if we just grabbed the possibly available ids, then
 
519
            # we only have an estimate of whats available and need to validate
 
520
            # that against the revision records.
 
521
            result_set = set(
 
522
                self.source._eliminate_revisions_not_present(required_revisions))
 
523
        if limit is not None:
 
524
            topo_ordered = self.source.get_graph().iter_topo_order(result_set)
 
525
            result_set = set(itertools.islice(topo_ordered, limit))
 
526
        return self.source.revision_ids_to_search_result(result_set)
 
527
 
 
528
 
 
529
InterRepository.register_optimiser(InterKnitRepo)