~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/repofmt/knitrepo.py

  • Committer: Jelmer Vernooij
  • Date: 2011-10-06 00:14:01 UTC
  • mto: This revision was merged to the branch mainline in revision 6216.
  • Revision ID: jelmer@samba.org-20111006001401-o14rsyp6gdy5vt3o
Fix plugin use of revision_history.

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,
41
 
    MetaDirRepository,
42
 
    MetaDirRepositoryFormat,
 
39
    InterRepository,
 
40
    IsInWriteGroupError,
43
41
    RepositoryFormat,
44
 
    RootCommitBuilder,
45
 
    )
 
42
    )
 
43
from bzrlib.vf_repository import (
 
44
    InterSameDataRepository,
 
45
    MetaDirVersionedFileRepository,
 
46
    MetaDirVersionedFileRepositoryFormat,
 
47
    VersionedFileCommitBuilder,
 
48
    VersionedFileRootCommitBuilder,
 
49
    )
 
50
from bzrlib import symbol_versioning
46
51
 
47
52
 
48
53
class _KnitParentsProvider(object):
102
107
        return result
103
108
 
104
109
 
105
 
class KnitRepository(MetaDirRepository):
 
110
class KnitRepository(MetaDirVersionedFileRepository):
106
111
    """Knit format repository."""
107
112
 
108
113
    # These attributes are inherited from the Repository base class. Setting
114
119
 
115
120
    def __init__(self, _format, a_bzrdir, control_files, _commit_builder_class,
116
121
        _serializer):
117
 
        MetaDirRepository.__init__(self, _format, a_bzrdir, control_files)
 
122
        super(KnitRepository, self).__init__(_format, a_bzrdir, control_files)
118
123
        self._commit_builder_class = _commit_builder_class
119
124
        self._serializer = _serializer
120
125
        self._reconcile_fixes_text_parents = True
176
181
        result.get_parent_map([('A',)])
177
182
        return result
178
183
 
179
 
    def fileid_involved_between_revs(self, from_revid, to_revid):
180
 
        """Find file_id(s) which are involved in the changes between revisions.
181
 
 
182
 
        This determines the set of revisions which are involved, and then
183
 
        finds all file ids affected by those revisions.
184
 
        """
185
 
        vf = self._get_revision_vf()
186
 
        from_set = set(vf.get_ancestry(from_revid))
187
 
        to_set = set(vf.get_ancestry(to_revid))
188
 
        changed = to_set.difference(from_set)
189
 
        return self._fileid_involved_by_set(changed)
190
 
 
191
 
    def fileid_involved(self, last_revid=None):
192
 
        """Find all file_ids modified in the ancestry of last_revid.
193
 
 
194
 
        :param last_revid: If None, last_revision() will be used.
195
 
        """
196
 
        if not last_revid:
197
 
            changed = set(self.all_revision_ids())
198
 
        else:
199
 
            changed = set(self.get_ancestry(last_revid))
200
 
        if None in changed:
201
 
            changed.remove(None)
202
 
        return self._fileid_involved_by_set(changed)
203
 
 
204
184
    @needs_read_lock
205
185
    def get_revision(self, revision_id):
206
186
        """Return the Revision object for a named revision"""
210
190
    def _refresh_data(self):
211
191
        if not self.is_locked():
212
192
            return
 
193
        if self.is_in_write_group():
 
194
            raise IsInWriteGroupError(self)
213
195
        # Create a new transaction to force all knits to see the scope change.
214
196
        # This is safe because we're outside a write group.
215
197
        self.control_files._finish_transaction()
229
211
    def _make_parents_provider(self):
230
212
        return _KnitsParentsProvider(self.revisions)
231
213
 
232
 
    def _find_inconsistent_revision_parents(self, revisions_iterator=None):
233
 
        """Find revisions with different parent lists in the revision object
234
 
        and in the index graph.
235
 
 
236
 
        :param revisions_iterator: None, or an iterator of (revid,
237
 
            Revision-or-None). This iterator controls the revisions checked.
238
 
        :returns: an iterator yielding tuples of (revison-id, parents-in-index,
239
 
            parents-in-revision).
240
 
        """
241
 
        if not self.is_locked():
242
 
            raise AssertionError()
243
 
        vf = self.revisions
244
 
        if revisions_iterator is None:
245
 
            revisions_iterator = self._iter_revisions(None)
246
 
        for revid, revision in revisions_iterator:
247
 
            if revision is None:
248
 
                pass
249
 
            parent_map = vf.get_parent_map([(revid,)])
250
 
            parents_according_to_index = tuple(parent[-1] for parent in
251
 
                parent_map[(revid,)])
252
 
            parents_according_to_revision = tuple(revision.parent_ids)
253
 
            if parents_according_to_index != parents_according_to_revision:
254
 
                yield (revid, parents_according_to_index,
255
 
                    parents_according_to_revision)
256
 
 
257
 
    def _check_for_inconsistent_revision_parents(self):
258
 
        inconsistencies = list(self._find_inconsistent_revision_parents())
259
 
        if inconsistencies:
260
 
            raise errors.BzrCheckError(
261
 
                "Revision knit has inconsistent parents.")
262
 
 
263
 
    def revision_graph_can_have_wrong_parents(self):
264
 
        # The revision.kndx could potentially claim a revision has a different
265
 
        # parent to the revision text.
266
 
        return True
267
 
 
268
 
 
269
 
class RepositoryFormatKnit(MetaDirRepositoryFormat):
 
214
 
 
215
class RepositoryFormatKnit(MetaDirVersionedFileRepositoryFormat):
270
216
    """Bzr repository knit format (generalized).
271
217
 
272
218
    This repository format has:
301
247
    _fetch_order = 'topological'
302
248
    _fetch_uses_deltas = True
303
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
304
254
 
305
255
    def _get_inventories(self, repo_transport, repo, name='inventory'):
306
256
        mapper = versionedfile.ConstantMapper(name)
409
359
    """
410
360
 
411
361
    repository_class = KnitRepository
412
 
    _commit_builder_class = CommitBuilder
 
362
    _commit_builder_class = VersionedFileCommitBuilder
413
363
    @property
414
364
    def _serializer(self):
415
365
        return xml5.serializer_v5
443
393
    """
444
394
 
445
395
    repository_class = KnitRepository
446
 
    _commit_builder_class = RootCommitBuilder
 
396
    _commit_builder_class = VersionedFileRootCommitBuilder
447
397
    rich_root_data = True
448
398
    experimental = True
449
399
    supports_tree_reference = True
485
435
    """
486
436
 
487
437
    repository_class = KnitRepository
488
 
    _commit_builder_class = RootCommitBuilder
 
438
    _commit_builder_class = VersionedFileRootCommitBuilder
489
439
    rich_root_data = True
490
440
    supports_tree_reference = False
491
441
    @property
507
457
    def get_format_description(self):
508
458
        """See RepositoryFormat.get_format_description()."""
509
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)