~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/repofmt/knitrepo.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2010-03-25 00:02:51 UTC
  • mfrom: (5106.1.1 version-bump)
  • Revision ID: pqm@pqm.ubuntu.com-20100325000251-bwsv5c5d3l9x3lnn
(Jelmer) Bump API version for 2.2.0.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2007-2010 Canonical Ltd
 
1
# Copyright (C) 2005, 2006, 2007 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
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
 
 
19
17
from bzrlib.lazy_import import lazy_import
20
18
lazy_import(globals(), """
21
 
import itertools
22
 
 
23
19
from bzrlib import (
24
 
    controldir,
 
20
    bzrdir,
25
21
    errors,
26
22
    knit as _mod_knit,
27
23
    lockable_files,
28
24
    lockdir,
29
25
    osutils,
30
26
    revision as _mod_revision,
31
 
    trace,
32
27
    transactions,
33
28
    versionedfile,
34
29
    xml5,
36
31
    xml7,
37
32
    )
38
33
""")
 
34
from bzrlib import (
 
35
    symbol_versioning,
 
36
    )
39
37
from bzrlib.decorators import needs_read_lock, needs_write_lock
40
38
from bzrlib.repository import (
41
 
    InterRepository,
42
 
    IsInWriteGroupError,
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
 
39
    CommitBuilder,
 
40
    MetaDirRepository,
 
41
    MetaDirRepositoryFormat,
 
42
    RepositoryFormat,
 
43
    RootCommitBuilder,
 
44
    )
 
45
from bzrlib.trace import mutter, mutter_callsite
53
46
 
54
47
 
55
48
class _KnitParentsProvider(object):
109
102
        return result
110
103
 
111
104
 
112
 
class KnitRepository(MetaDirVersionedFileRepository):
 
105
class KnitRepository(MetaDirRepository):
113
106
    """Knit format repository."""
114
107
 
115
108
    # These attributes are inherited from the Repository base class. Setting
121
114
 
122
115
    def __init__(self, _format, a_bzrdir, control_files, _commit_builder_class,
123
116
        _serializer):
124
 
        super(KnitRepository, self).__init__(_format, a_bzrdir, control_files)
 
117
        MetaDirRepository.__init__(self, _format, a_bzrdir, control_files)
125
118
        self._commit_builder_class = _commit_builder_class
126
119
        self._serializer = _serializer
127
120
        self._reconcile_fixes_text_parents = True
183
176
        result.get_parent_map([('A',)])
184
177
        return result
185
178
 
 
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
 
186
204
    @needs_read_lock
187
205
    def get_revision(self, revision_id):
188
206
        """Return the Revision object for a named revision"""
192
210
    def _refresh_data(self):
193
211
        if not self.is_locked():
194
212
            return
195
 
        if self.is_in_write_group():
196
 
            raise IsInWriteGroupError(self)
197
213
        # Create a new transaction to force all knits to see the scope change.
198
214
        # This is safe because we're outside a write group.
199
215
        self.control_files._finish_transaction()
213
229
    def _make_parents_provider(self):
214
230
        return _KnitsParentsProvider(self.revisions)
215
231
 
216
 
 
217
 
class RepositoryFormatKnit(MetaDirVersionedFileRepositoryFormat):
 
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):
218
270
    """Bzr repository knit format (generalized).
219
271
 
220
272
    This repository format has:
249
301
    _fetch_order = 'topological'
250
302
    _fetch_uses_deltas = True
251
303
    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
256
304
 
257
305
    def _get_inventories(self, repo_transport, repo, name='inventory'):
258
306
        mapper = versionedfile.ConstantMapper(name)
294
342
        :param shared: If true the repository will be initialized as a shared
295
343
                       repository.
296
344
        """
297
 
        trace.mutter('creating repository in %s.', a_bzrdir.transport.base)
 
345
        mutter('creating repository in %s.', a_bzrdir.transport.base)
298
346
        dirs = ['knits']
299
347
        files = []
300
348
        utf8_files = [('format', self.get_format_string())]
312
360
        result.revisions.get_parent_map([('A',)])
313
361
        result.signatures.get_parent_map([('A',)])
314
362
        result.unlock()
315
 
        self._run_post_repo_init_hooks(result, a_bzrdir, shared)
316
363
        return result
317
364
 
318
365
    def open(self, a_bzrdir, _found=False, _override_transport=None):
323
370
                                    than normal. I.e. during 'upgrade'.
324
371
        """
325
372
        if not _found:
326
 
            format = RepositoryFormatMetaDir.find_format(a_bzrdir)
 
373
            format = RepositoryFormat.find_format(a_bzrdir)
327
374
        if _override_transport is not None:
328
375
            repo_transport = _override_transport
329
376
        else:
361
408
    """
362
409
 
363
410
    repository_class = KnitRepository
364
 
    _commit_builder_class = VersionedFileCommitBuilder
 
411
    _commit_builder_class = CommitBuilder
365
412
    @property
366
413
    def _serializer(self):
367
414
        return xml5.serializer_v5
369
416
    def __ne__(self, other):
370
417
        return self.__class__ is not other.__class__
371
418
 
372
 
    @classmethod
373
 
    def get_format_string(cls):
 
419
    def get_format_string(self):
374
420
        """See RepositoryFormat.get_format_string()."""
375
421
        return "Bazaar-NG Knit Repository Format 1"
376
422
 
396
442
    """
397
443
 
398
444
    repository_class = KnitRepository
399
 
    _commit_builder_class = VersionedFileRootCommitBuilder
 
445
    _commit_builder_class = RootCommitBuilder
400
446
    rich_root_data = True
401
447
    experimental = True
402
448
    supports_tree_reference = True
405
451
        return xml7.serializer_v7
406
452
 
407
453
    def _get_matching_bzrdir(self):
408
 
        return controldir.format_registry.make_bzrdir('dirstate-with-subtree')
 
454
        return bzrdir.format_registry.make_bzrdir('dirstate-with-subtree')
409
455
 
410
456
    def _ignore_setting_bzrdir(self, format):
411
457
        pass
412
458
 
413
459
    _matchingbzrdir = property(_get_matching_bzrdir, _ignore_setting_bzrdir)
414
460
 
415
 
    @classmethod
416
 
    def get_format_string(cls):
 
461
    def get_format_string(self):
417
462
        """See RepositoryFormat.get_format_string()."""
418
463
        return "Bazaar Knit Repository Format 3 (bzr 0.15)\n"
419
464
 
439
484
    """
440
485
 
441
486
    repository_class = KnitRepository
442
 
    _commit_builder_class = VersionedFileRootCommitBuilder
 
487
    _commit_builder_class = RootCommitBuilder
443
488
    rich_root_data = True
444
489
    supports_tree_reference = False
445
490
    @property
447
492
        return xml6.serializer_v6
448
493
 
449
494
    def _get_matching_bzrdir(self):
450
 
        return controldir.format_registry.make_bzrdir('rich-root')
 
495
        return bzrdir.format_registry.make_bzrdir('rich-root')
451
496
 
452
497
    def _ignore_setting_bzrdir(self, format):
453
498
        pass
454
499
 
455
500
    _matchingbzrdir = property(_get_matching_bzrdir, _ignore_setting_bzrdir)
456
501
 
457
 
    @classmethod
458
 
    def get_format_string(cls):
 
502
    def get_format_string(self):
459
503
        """See RepositoryFormat.get_format_string()."""
460
504
        return 'Bazaar Knit Repository Format 4 (bzr 1.0)\n'
461
505
 
462
506
    def get_format_description(self):
463
507
        """See RepositoryFormat.get_format_description()."""
464
508
        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)