/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to breezy/repofmt/knitrepo.py

  • Committer: Jelmer Vernooij
  • Date: 2017-06-10 00:21:41 UTC
  • mto: This revision was merged to the branch mainline in revision 6675.
  • Revision ID: jelmer@jelmer.uk-20170610002141-m1z5k7fs8laesa65
Fix import.

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 bzrlib.lazy_import import lazy_import
 
17
from __future__ import absolute_import
 
18
 
 
19
from ..lazy_import import lazy_import
18
20
lazy_import(globals(), """
19
 
from bzrlib import (
20
 
    bzrdir,
 
21
import itertools
 
22
 
 
23
from breezy import (
 
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
 
from bzrlib.decorators import needs_read_lock, needs_write_lock
39
 
from bzrlib.repository import (
40
 
    CommitBuilder,
41
 
    MetaDirRepository,
42
 
    MetaDirRepositoryFormat,
43
 
    RepositoryFormat,
44
 
    RootCommitBuilder,
 
39
from ..decorators import needs_read_lock, needs_write_lock
 
40
from ..repository import (
 
41
    InterRepository,
 
42
    IsInWriteGroupError,
 
43
    )
 
44
from ..bzrrepository import (
 
45
    RepositoryFormatMetaDir,
 
46
    )
 
47
from ..vf_repository import (
 
48
    InterSameDataRepository,
 
49
    MetaDirVersionedFileRepository,
 
50
    MetaDirVersionedFileRepositoryFormat,
 
51
    VersionedFileCommitBuilder,
 
52
    VersionedFileRootCommitBuilder,
45
53
    )
46
54
 
47
55
 
102
110
        return result
103
111
 
104
112
 
105
 
class KnitRepository(MetaDirRepository):
 
113
class KnitRepository(MetaDirVersionedFileRepository):
106
114
    """Knit format repository."""
107
115
 
108
116
    # These attributes are inherited from the Repository base class. Setting
114
122
 
115
123
    def __init__(self, _format, a_bzrdir, control_files, _commit_builder_class,
116
124
        _serializer):
117
 
        MetaDirRepository.__init__(self, _format, a_bzrdir, control_files)
 
125
        super(KnitRepository, self).__init__(_format, a_bzrdir, control_files)
118
126
        self._commit_builder_class = _commit_builder_class
119
127
        self._serializer = _serializer
120
128
        self._reconcile_fixes_text_parents = True
176
184
        result.get_parent_map([('A',)])
177
185
        return result
178
186
 
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
187
    @needs_read_lock
205
188
    def get_revision(self, revision_id):
206
189
        """Return the Revision object for a named revision"""
210
193
    def _refresh_data(self):
211
194
        if not self.is_locked():
212
195
            return
 
196
        if self.is_in_write_group():
 
197
            raise IsInWriteGroupError(self)
213
198
        # Create a new transaction to force all knits to see the scope change.
214
199
        # This is safe because we're outside a write group.
215
200
        self.control_files._finish_transaction()
221
206
    @needs_write_lock
222
207
    def reconcile(self, other=None, thorough=False):
223
208
        """Reconcile this repository."""
224
 
        from bzrlib.reconcile import KnitReconciler
 
209
        from breezy.reconcile import KnitReconciler
225
210
        reconciler = KnitReconciler(self, thorough=thorough)
226
211
        reconciler.reconcile()
227
212
        return reconciler
229
214
    def _make_parents_provider(self):
230
215
        return _KnitsParentsProvider(self.revisions)
231
216
 
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):
 
217
 
 
218
class RepositoryFormatKnit(MetaDirVersionedFileRepositoryFormat):
270
219
    """Bzr repository knit format (generalized).
271
220
 
272
221
    This repository format has:
301
250
    _fetch_order = 'topological'
302
251
    _fetch_uses_deltas = True
303
252
    fast_deltas = False
 
253
    supports_funky_characters = True
 
254
    # The revision.kndx could potentially claim a revision has a different
 
255
    # parent to the revision text.
 
256
    revision_graph_can_have_wrong_parents = True
304
257
 
305
258
    def _get_inventories(self, repo_transport, repo, name='inventory'):
306
259
        mapper = versionedfile.ConstantMapper(name)
371
324
                                    than normal. I.e. during 'upgrade'.
372
325
        """
373
326
        if not _found:
374
 
            format = RepositoryFormat.find_format(a_bzrdir)
 
327
            format = RepositoryFormatMetaDir.find_format(a_bzrdir)
375
328
        if _override_transport is not None:
376
329
            repo_transport = _override_transport
377
330
        else:
409
362
    """
410
363
 
411
364
    repository_class = KnitRepository
412
 
    _commit_builder_class = CommitBuilder
 
365
    _commit_builder_class = VersionedFileCommitBuilder
413
366
    @property
414
367
    def _serializer(self):
415
368
        return xml5.serializer_v5
417
370
    def __ne__(self, other):
418
371
        return self.__class__ is not other.__class__
419
372
 
420
 
    def get_format_string(self):
 
373
    @classmethod
 
374
    def get_format_string(cls):
421
375
        """See RepositoryFormat.get_format_string()."""
422
376
        return "Bazaar-NG Knit Repository Format 1"
423
377
 
443
397
    """
444
398
 
445
399
    repository_class = KnitRepository
446
 
    _commit_builder_class = RootCommitBuilder
 
400
    _commit_builder_class = VersionedFileRootCommitBuilder
447
401
    rich_root_data = True
448
402
    experimental = True
449
403
    supports_tree_reference = True
452
406
        return xml7.serializer_v7
453
407
 
454
408
    def _get_matching_bzrdir(self):
455
 
        return bzrdir.format_registry.make_bzrdir('dirstate-with-subtree')
 
409
        return controldir.format_registry.make_bzrdir('dirstate-with-subtree')
456
410
 
457
411
    def _ignore_setting_bzrdir(self, format):
458
412
        pass
459
413
 
460
414
    _matchingbzrdir = property(_get_matching_bzrdir, _ignore_setting_bzrdir)
461
415
 
462
 
    def get_format_string(self):
 
416
    @classmethod
 
417
    def get_format_string(cls):
463
418
        """See RepositoryFormat.get_format_string()."""
464
419
        return "Bazaar Knit Repository Format 3 (bzr 0.15)\n"
465
420
 
485
440
    """
486
441
 
487
442
    repository_class = KnitRepository
488
 
    _commit_builder_class = RootCommitBuilder
 
443
    _commit_builder_class = VersionedFileRootCommitBuilder
489
444
    rich_root_data = True
490
445
    supports_tree_reference = False
491
446
    @property
493
448
        return xml6.serializer_v6
494
449
 
495
450
    def _get_matching_bzrdir(self):
496
 
        return bzrdir.format_registry.make_bzrdir('rich-root')
 
451
        return controldir.format_registry.make_bzrdir('rich-root')
497
452
 
498
453
    def _ignore_setting_bzrdir(self, format):
499
454
        pass
500
455
 
501
456
    _matchingbzrdir = property(_get_matching_bzrdir, _ignore_setting_bzrdir)
502
457
 
503
 
    def get_format_string(self):
 
458
    @classmethod
 
459
    def get_format_string(cls):
504
460
        """See RepositoryFormat.get_format_string()."""
505
461
        return 'Bazaar Knit Repository Format 4 (bzr 1.0)\n'
506
462
 
507
463
    def get_format_description(self):
508
464
        """See RepositoryFormat.get_format_description()."""
509
465
        return "Knit repository format 4"
 
466
 
 
467
 
 
468
class InterKnitRepo(InterSameDataRepository):
 
469
    """Optimised code paths between Knit based repositories."""
 
470
 
 
471
    @classmethod
 
472
    def _get_repo_format_to_test(self):
 
473
        return RepositoryFormatKnit1()
 
474
 
 
475
    @staticmethod
 
476
    def is_compatible(source, target):
 
477
        """Be compatible with known Knit formats.
 
478
 
 
479
        We don't test for the stores being of specific types because that
 
480
        could lead to confusing results, and there is no need to be
 
481
        overly general.
 
482
        """
 
483
        try:
 
484
            are_knits = (isinstance(source._format, RepositoryFormatKnit) and
 
485
                isinstance(target._format, RepositoryFormatKnit))
 
486
        except AttributeError:
 
487
            return False
 
488
        return are_knits and InterRepository._same_model(source, target)
 
489
 
 
490
    @needs_read_lock
 
491
    def search_missing_revision_ids(self,
 
492
            find_ghosts=True, revision_ids=None, if_present_ids=None,
 
493
            limit=None):
 
494
        """See InterRepository.search_missing_revision_ids()."""
 
495
        source_ids_set = self._present_source_revisions_for(
 
496
            revision_ids, if_present_ids)
 
497
        # source_ids is the worst possible case we may need to pull.
 
498
        # now we want to filter source_ids against what we actually
 
499
        # have in target, but don't try to check for existence where we know
 
500
        # we do not have a revision as that would be pointless.
 
501
        target_ids = set(self.target.all_revision_ids())
 
502
        possibly_present_revisions = target_ids.intersection(source_ids_set)
 
503
        actually_present_revisions = set(
 
504
            self.target._eliminate_revisions_not_present(possibly_present_revisions))
 
505
        required_revisions = source_ids_set.difference(actually_present_revisions)
 
506
        if revision_ids is not None:
 
507
            # we used get_ancestry to determine source_ids then we are assured all
 
508
            # revisions referenced are present as they are installed in topological order.
 
509
            # and the tip revision was validated by get_ancestry.
 
510
            result_set = required_revisions
 
511
        else:
 
512
            # if we just grabbed the possibly available ids, then
 
513
            # we only have an estimate of whats available and need to validate
 
514
            # that against the revision records.
 
515
            result_set = set(
 
516
                self.source._eliminate_revisions_not_present(required_revisions))
 
517
        if limit is not None:
 
518
            topo_ordered = self.source.get_graph().iter_topo_order(result_set)
 
519
            result_set = set(itertools.islice(topo_ordered, limit))
 
520
        return self.source.revision_ids_to_search_result(result_set)
 
521
 
 
522
 
 
523
InterRepository.register_optimiser(InterKnitRepo)