~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/repofmt/knitrepo.py

  • Committer: Patch Queue Manager
  • Date: 2014-09-22 19:14:34 UTC
  • mfrom: (6598.1.2 bzr)
  • Revision ID: pqm@pqm.ubuntu.com-20140922191434-6bbnpnxi5jab4vim
(richard-wilbur) Allows launchpad APIs to use proxies by default(Paul Gear).
 (Paul Gear)

Show diffs side-by-side

added added

removed removed

Lines of Context:
14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
 
17
from __future__ import absolute_import
 
18
 
17
19
from bzrlib.lazy_import import lazy_import
18
20
lazy_import(globals(), """
 
21
import itertools
 
22
 
19
23
from bzrlib import (
20
 
    bzrdir,
 
24
    controldir,
21
25
    errors,
22
26
    knit as _mod_knit,
23
27
    lockable_files,
32
36
    xml7,
33
37
    )
34
38
""")
35
 
from bzrlib import (
36
 
    symbol_versioning,
37
 
    )
38
39
from bzrlib.decorators import needs_read_lock, needs_write_lock
39
40
from bzrlib.repository import (
40
 
    CommitBuilder,
 
41
    InterRepository,
41
42
    IsInWriteGroupError,
42
 
    MetaDirRepository,
43
 
    MetaDirRepositoryFormat,
44
 
    RepositoryFormat,
45
 
    RootCommitBuilder,
46
 
    )
 
43
    RepositoryFormatMetaDir,
 
44
    )
 
45
from bzrlib.vf_repository import (
 
46
    InterSameDataRepository,
 
47
    MetaDirVersionedFileRepository,
 
48
    MetaDirVersionedFileRepositoryFormat,
 
49
    VersionedFileCommitBuilder,
 
50
    VersionedFileRootCommitBuilder,
 
51
    )
 
52
from bzrlib import symbol_versioning
47
53
 
48
54
 
49
55
class _KnitParentsProvider(object):
103
109
        return result
104
110
 
105
111
 
106
 
class KnitRepository(MetaDirRepository):
 
112
class KnitRepository(MetaDirVersionedFileRepository):
107
113
    """Knit format repository."""
108
114
 
109
115
    # These attributes are inherited from the Repository base class. Setting
115
121
 
116
122
    def __init__(self, _format, a_bzrdir, control_files, _commit_builder_class,
117
123
        _serializer):
118
 
        MetaDirRepository.__init__(self, _format, a_bzrdir, control_files)
 
124
        super(KnitRepository, self).__init__(_format, a_bzrdir, control_files)
119
125
        self._commit_builder_class = _commit_builder_class
120
126
        self._serializer = _serializer
121
127
        self._reconcile_fixes_text_parents = True
177
183
        result.get_parent_map([('A',)])
178
184
        return result
179
185
 
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
186
    @needs_read_lock
206
187
    def get_revision(self, revision_id):
207
188
        """Return the Revision object for a named revision"""
232
213
    def _make_parents_provider(self):
233
214
        return _KnitsParentsProvider(self.revisions)
234
215
 
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):
 
216
 
 
217
class RepositoryFormatKnit(MetaDirVersionedFileRepositoryFormat):
273
218
    """Bzr repository knit format (generalized).
274
219
 
275
220
    This repository format has:
304
249
    _fetch_order = 'topological'
305
250
    _fetch_uses_deltas = True
306
251
    fast_deltas = False
 
252
    supports_funky_characters = True
 
253
    # The revision.kndx could potentially claim a revision has a different
 
254
    # parent to the revision text.
 
255
    revision_graph_can_have_wrong_parents = True
307
256
 
308
257
    def _get_inventories(self, repo_transport, repo, name='inventory'):
309
258
        mapper = versionedfile.ConstantMapper(name)
374
323
                                    than normal. I.e. during 'upgrade'.
375
324
        """
376
325
        if not _found:
377
 
            format = RepositoryFormat.find_format(a_bzrdir)
 
326
            format = RepositoryFormatMetaDir.find_format(a_bzrdir)
378
327
        if _override_transport is not None:
379
328
            repo_transport = _override_transport
380
329
        else:
412
361
    """
413
362
 
414
363
    repository_class = KnitRepository
415
 
    _commit_builder_class = CommitBuilder
 
364
    _commit_builder_class = VersionedFileCommitBuilder
416
365
    @property
417
366
    def _serializer(self):
418
367
        return xml5.serializer_v5
420
369
    def __ne__(self, other):
421
370
        return self.__class__ is not other.__class__
422
371
 
423
 
    def get_format_string(self):
 
372
    @classmethod
 
373
    def get_format_string(cls):
424
374
        """See RepositoryFormat.get_format_string()."""
425
375
        return "Bazaar-NG Knit Repository Format 1"
426
376
 
446
396
    """
447
397
 
448
398
    repository_class = KnitRepository
449
 
    _commit_builder_class = RootCommitBuilder
 
399
    _commit_builder_class = VersionedFileRootCommitBuilder
450
400
    rich_root_data = True
451
401
    experimental = True
452
402
    supports_tree_reference = True
455
405
        return xml7.serializer_v7
456
406
 
457
407
    def _get_matching_bzrdir(self):
458
 
        return bzrdir.format_registry.make_bzrdir('dirstate-with-subtree')
 
408
        return controldir.format_registry.make_bzrdir('dirstate-with-subtree')
459
409
 
460
410
    def _ignore_setting_bzrdir(self, format):
461
411
        pass
462
412
 
463
413
    _matchingbzrdir = property(_get_matching_bzrdir, _ignore_setting_bzrdir)
464
414
 
465
 
    def get_format_string(self):
 
415
    @classmethod
 
416
    def get_format_string(cls):
466
417
        """See RepositoryFormat.get_format_string()."""
467
418
        return "Bazaar Knit Repository Format 3 (bzr 0.15)\n"
468
419
 
488
439
    """
489
440
 
490
441
    repository_class = KnitRepository
491
 
    _commit_builder_class = RootCommitBuilder
 
442
    _commit_builder_class = VersionedFileRootCommitBuilder
492
443
    rich_root_data = True
493
444
    supports_tree_reference = False
494
445
    @property
496
447
        return xml6.serializer_v6
497
448
 
498
449
    def _get_matching_bzrdir(self):
499
 
        return bzrdir.format_registry.make_bzrdir('rich-root')
 
450
        return controldir.format_registry.make_bzrdir('rich-root')
500
451
 
501
452
    def _ignore_setting_bzrdir(self, format):
502
453
        pass
503
454
 
504
455
    _matchingbzrdir = property(_get_matching_bzrdir, _ignore_setting_bzrdir)
505
456
 
506
 
    def get_format_string(self):
 
457
    @classmethod
 
458
    def get_format_string(cls):
507
459
        """See RepositoryFormat.get_format_string()."""
508
460
        return 'Bazaar Knit Repository Format 4 (bzr 1.0)\n'
509
461
 
510
462
    def get_format_description(self):
511
463
        """See RepositoryFormat.get_format_description()."""
512
464
        return "Knit repository format 4"
 
465
 
 
466
 
 
467
class InterKnitRepo(InterSameDataRepository):
 
468
    """Optimised code paths between Knit based repositories."""
 
469
 
 
470
    @classmethod
 
471
    def _get_repo_format_to_test(self):
 
472
        return RepositoryFormatKnit1()
 
473
 
 
474
    @staticmethod
 
475
    def is_compatible(source, target):
 
476
        """Be compatible with known Knit formats.
 
477
 
 
478
        We don't test for the stores being of specific types because that
 
479
        could lead to confusing results, and there is no need to be
 
480
        overly general.
 
481
        """
 
482
        try:
 
483
            are_knits = (isinstance(source._format, RepositoryFormatKnit) and
 
484
                isinstance(target._format, RepositoryFormatKnit))
 
485
        except AttributeError:
 
486
            return False
 
487
        return are_knits and InterRepository._same_model(source, target)
 
488
 
 
489
    @needs_read_lock
 
490
    def search_missing_revision_ids(self,
 
491
            find_ghosts=True, revision_ids=None, if_present_ids=None,
 
492
            limit=None):
 
493
        """See InterRepository.search_missing_revision_ids()."""
 
494
        source_ids_set = self._present_source_revisions_for(
 
495
            revision_ids, if_present_ids)
 
496
        # source_ids is the worst possible case we may need to pull.
 
497
        # now we want to filter source_ids against what we actually
 
498
        # have in target, but don't try to check for existence where we know
 
499
        # we do not have a revision as that would be pointless.
 
500
        target_ids = set(self.target.all_revision_ids())
 
501
        possibly_present_revisions = target_ids.intersection(source_ids_set)
 
502
        actually_present_revisions = set(
 
503
            self.target._eliminate_revisions_not_present(possibly_present_revisions))
 
504
        required_revisions = source_ids_set.difference(actually_present_revisions)
 
505
        if revision_ids is not None:
 
506
            # we used get_ancestry to determine source_ids then we are assured all
 
507
            # revisions referenced are present as they are installed in topological order.
 
508
            # and the tip revision was validated by get_ancestry.
 
509
            result_set = required_revisions
 
510
        else:
 
511
            # if we just grabbed the possibly available ids, then
 
512
            # we only have an estimate of whats available and need to validate
 
513
            # that against the revision records.
 
514
            result_set = set(
 
515
                self.source._eliminate_revisions_not_present(required_revisions))
 
516
        if limit is not None:
 
517
            topo_ordered = self.source.get_graph().iter_topo_order(result_set)
 
518
            result_set = set(itertools.islice(topo_ordered, limit))
 
519
        return self.source.revision_ids_to_search_result(result_set)
 
520
 
 
521
 
 
522
InterRepository.register_optimiser(InterKnitRepo)