~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/repofmt/knitrepo.py

  • Committer: Jelmer Vernooij
  • Date: 2011-12-16 19:18:39 UTC
  • mto: This revision was merged to the branch mainline in revision 6391.
  • Revision ID: jelmer@samba.org-20111216191839-eg681lxqibi1qxu1
Fix remaining tests.

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
 
    RepositoryFormat,
45
 
    RootCommitBuilder,
46
 
    )
 
41
    RepositoryFormatMetaDir,
 
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)
374
321
                                    than normal. I.e. during 'upgrade'.
375
322
        """
376
323
        if not _found:
377
 
            format = RepositoryFormat.find_format(a_bzrdir)
 
324
            format = RepositoryFormatMetaDir.find_format(a_bzrdir)
378
325
        if _override_transport is not None:
379
326
            repo_transport = _override_transport
380
327
        else:
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
420
367
    def __ne__(self, other):
421
368
        return self.__class__ is not other.__class__
422
369
 
423
 
    def get_format_string(self):
 
370
    @classmethod
 
371
    def get_format_string(cls):
424
372
        """See RepositoryFormat.get_format_string()."""
425
373
        return "Bazaar-NG Knit Repository Format 1"
426
374
 
446
394
    """
447
395
 
448
396
    repository_class = KnitRepository
449
 
    _commit_builder_class = RootCommitBuilder
 
397
    _commit_builder_class = VersionedFileRootCommitBuilder
450
398
    rich_root_data = True
451
399
    experimental = True
452
400
    supports_tree_reference = True
462
410
 
463
411
    _matchingbzrdir = property(_get_matching_bzrdir, _ignore_setting_bzrdir)
464
412
 
465
 
    def get_format_string(self):
 
413
    @classmethod
 
414
    def get_format_string(cls):
466
415
        """See RepositoryFormat.get_format_string()."""
467
416
        return "Bazaar Knit Repository Format 3 (bzr 0.15)\n"
468
417
 
488
437
    """
489
438
 
490
439
    repository_class = KnitRepository
491
 
    _commit_builder_class = RootCommitBuilder
 
440
    _commit_builder_class = VersionedFileRootCommitBuilder
492
441
    rich_root_data = True
493
442
    supports_tree_reference = False
494
443
    @property
503
452
 
504
453
    _matchingbzrdir = property(_get_matching_bzrdir, _ignore_setting_bzrdir)
505
454
 
506
 
    def get_format_string(self):
 
455
    @classmethod
 
456
    def get_format_string(cls):
507
457
        """See RepositoryFormat.get_format_string()."""
508
458
        return 'Bazaar Knit Repository Format 4 (bzr 1.0)\n'
509
459
 
510
460
    def get_format_description(self):
511
461
        """See RepositoryFormat.get_format_description()."""
512
462
        return "Knit repository format 4"
 
463
 
 
464
 
 
465
class InterKnitRepo(InterSameDataRepository):
 
466
    """Optimised code paths between Knit based repositories."""
 
467
 
 
468
    @classmethod
 
469
    def _get_repo_format_to_test(self):
 
470
        return RepositoryFormatKnit1()
 
471
 
 
472
    @staticmethod
 
473
    def is_compatible(source, target):
 
474
        """Be compatible with known Knit formats.
 
475
 
 
476
        We don't test for the stores being of specific types because that
 
477
        could lead to confusing results, and there is no need to be
 
478
        overly general.
 
479
        """
 
480
        try:
 
481
            are_knits = (isinstance(source._format, RepositoryFormatKnit) and
 
482
                isinstance(target._format, RepositoryFormatKnit))
 
483
        except AttributeError:
 
484
            return False
 
485
        return are_knits and InterRepository._same_model(source, target)
 
486
 
 
487
    @needs_read_lock
 
488
    def search_missing_revision_ids(self,
 
489
            revision_id=symbol_versioning.DEPRECATED_PARAMETER,
 
490
            find_ghosts=True, revision_ids=None, if_present_ids=None,
 
491
            limit=None):
 
492
        """See InterRepository.search_missing_revision_ids()."""
 
493
        if symbol_versioning.deprecated_passed(revision_id):
 
494
            symbol_versioning.warn(
 
495
                'search_missing_revision_ids(revision_id=...) was '
 
496
                'deprecated in 2.4.  Use revision_ids=[...] instead.',
 
497
                DeprecationWarning, stacklevel=2)
 
498
            if revision_ids is not None:
 
499
                raise AssertionError(
 
500
                    'revision_ids is mutually exclusive with revision_id')
 
501
            if revision_id is not None:
 
502
                revision_ids = [revision_id]
 
503
        del revision_id
 
504
        source_ids_set = self._present_source_revisions_for(
 
505
            revision_ids, if_present_ids)
 
506
        # source_ids is the worst possible case we may need to pull.
 
507
        # now we want to filter source_ids against what we actually
 
508
        # have in target, but don't try to check for existence where we know
 
509
        # we do not have a revision as that would be pointless.
 
510
        target_ids = set(self.target.all_revision_ids())
 
511
        possibly_present_revisions = target_ids.intersection(source_ids_set)
 
512
        actually_present_revisions = set(
 
513
            self.target._eliminate_revisions_not_present(possibly_present_revisions))
 
514
        required_revisions = source_ids_set.difference(actually_present_revisions)
 
515
        if revision_ids is not None:
 
516
            # we used get_ancestry to determine source_ids then we are assured all
 
517
            # revisions referenced are present as they are installed in topological order.
 
518
            # and the tip revision was validated by get_ancestry.
 
519
            result_set = required_revisions
 
520
        else:
 
521
            # if we just grabbed the possibly available ids, then
 
522
            # we only have an estimate of whats available and need to validate
 
523
            # that against the revision records.
 
524
            result_set = set(
 
525
                self.source._eliminate_revisions_not_present(required_revisions))
 
526
        if limit is not None:
 
527
            topo_ordered = self.source.get_graph().iter_topo_order(result_set)
 
528
            result_set = set(itertools.islice(topo_ordered, limit))
 
529
        return self.source.revision_ids_to_search_result(result_set)
 
530
 
 
531
 
 
532
InterRepository.register_optimiser(InterKnitRepo)