← Back to branch summary

~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-09-01 08:02:42 UTC
  • mfrom: (5390.3.3 faster-revert-593560)
  • Revision ID: pqm@pqm.ubuntu.com-20100901080242-esg62ody4frwmy66
(spiv) Avoid repeatedly calling self.target.all_file_ids() in
 InterTree.iter_changes. (Andrew Bennetts)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/repofmt/knitrepo.py
1
 
# Copyright (C) 2005, 2006, 2007 Canonical Ltd
 
1
# Copyright (C) 2007-2010 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
24
24
    lockdir,
25
25
    osutils,
26
26
    revision as _mod_revision,
 
27
    trace,
27
28
    transactions,
28
29
    versionedfile,
29
30
    xml5,
37
38
from bzrlib.decorators import needs_read_lock, needs_write_lock
38
39
from bzrlib.repository import (
39
40
    CommitBuilder,
 
41
    IsInWriteGroupError,
40
42
    MetaDirRepository,
41
43
    MetaDirRepositoryFormat,
42
44
    RepositoryFormat,
43
45
    RootCommitBuilder,
44
46
    )
45
 
from bzrlib.trace import mutter, mutter_callsite
46
47
 
47
48
 
48
49
class _KnitParentsProvider(object):
54
55
        return 'KnitParentsProvider(%r)' % self._knit
55
56
 
56
57
    def get_parent_map(self, keys):
57
 
        """See graph._StackedParentsProvider.get_parent_map"""
 
58
        """See graph.StackedParentsProvider.get_parent_map"""
58
59
        parent_map = {}
59
60
        for revision_id in keys:
60
61
            if revision_id is None:
85
86
        return 'KnitsParentsProvider(%r)' % self._knit
86
87
 
87
88
    def get_parent_map(self, keys):
88
 
        """See graph._StackedParentsProvider.get_parent_map"""
 
89
        """See graph.StackedParentsProvider.get_parent_map"""
89
90
        parent_map = self._knit.get_parent_map(
90
91
            [self._prefix + (key,) for key in keys])
91
92
        result = {}
210
211
    def _refresh_data(self):
211
212
        if not self.is_locked():
212
213
            return
 
214
        if self.is_in_write_group():
 
215
            raise IsInWriteGroupError(self)
213
216
        # Create a new transaction to force all knits to see the scope change.
214
217
        # This is safe because we're outside a write group.
215
218
        self.control_files._finish_transaction()
229
232
    def _make_parents_provider(self):
230
233
        return _KnitsParentsProvider(self.revisions)
231
234
 
232
 
    def _find_inconsistent_revision_parents(self):
 
235
    def _find_inconsistent_revision_parents(self, revisions_iterator=None):
233
236
        """Find revisions with different parent lists in the revision object
234
237
        and in the index graph.
235
238
 
 
239
        :param revisions_iterator: None, or an iterator of (revid,
 
240
            Revision-or-None). This iterator controls the revisions checked.
236
241
        :returns: an iterator yielding tuples of (revison-id, parents-in-index,
237
242
            parents-in-revision).
238
243
        """
239
244
        if not self.is_locked():
240
245
            raise AssertionError()
241
246
        vf = self.revisions
242
 
        for index_version in vf.keys():
243
 
            parent_map = vf.get_parent_map([index_version])
 
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,)])
244
253
            parents_according_to_index = tuple(parent[-1] for parent in
245
 
                parent_map[index_version])
246
 
            revision = self.get_revision(index_version[-1])
 
254
                parent_map[(revid,)])
247
255
            parents_according_to_revision = tuple(revision.parent_ids)
248
256
            if parents_according_to_index != parents_according_to_revision:
249
 
                yield (index_version[-1], parents_according_to_index,
 
257
                yield (revid, parents_according_to_index,
250
258
                    parents_according_to_revision)
251
259
 
252
260
    def _check_for_inconsistent_revision_parents(self):
337
345
        :param shared: If true the repository will be initialized as a shared
338
346
                       repository.
339
347
        """
340
 
        mutter('creating repository in %s.', a_bzrdir.transport.base)
 
348
        trace.mutter('creating repository in %s.', a_bzrdir.transport.base)
341
349
        dirs = ['knits']
342
350
        files = []
343
351
        utf8_files = [('format', self.get_format_string())]
355
363
        result.revisions.get_parent_map([('A',)])
356
364
        result.signatures.get_parent_map([('A',)])
357
365
        result.unlock()
 
366
        self._run_post_repo_init_hooks(result, a_bzrdir, shared)
358
367
        return result
359
368
 
360
369
    def open(self, a_bzrdir, _found=False, _override_transport=None):
419
428
        """See RepositoryFormat.get_format_description()."""
420
429
        return "Knit repository format 1"
421
430
 
422
 
    def check_conversion_target(self, target_format):
423
 
        pass
424
 
 
425
431
 
426
432
class RepositoryFormatKnit3(RepositoryFormatKnit):
427
433
    """Bzr repository knit format 3.
442
448
    repository_class = KnitRepository
443
449
    _commit_builder_class = RootCommitBuilder
444
450
    rich_root_data = True
 
451
    experimental = True
445
452
    supports_tree_reference = True
446
453
    @property
447
454
    def _serializer(self):
455
462
 
456
463
    _matchingbzrdir = property(_get_matching_bzrdir, _ignore_setting_bzrdir)
457
464
 
458
 
    def check_conversion_target(self, target_format):
459
 
        if not target_format.rich_root_data:
460
 
            raise errors.BadConversionTarget(
461
 
                'Does not support rich root data.', target_format)
462
 
        if not getattr(target_format, 'supports_tree_reference', False):
463
 
            raise errors.BadConversionTarget(
464
 
                'Does not support nested trees', target_format)
465
 
 
466
465
    def get_format_string(self):
467
466
        """See RepositoryFormat.get_format_string()."""
468
467
        return "Bazaar Knit Repository Format 3 (bzr 0.15)\n"
504
503
 
505
504
    _matchingbzrdir = property(_get_matching_bzrdir, _ignore_setting_bzrdir)
506
505
 
507
 
    def check_conversion_target(self, target_format):
508
 
        if not target_format.rich_root_data:
509
 
            raise errors.BadConversionTarget(
510
 
                'Does not support rich root data.', target_format)
511
 
 
512
506
    def get_format_string(self):
513
507
        """See RepositoryFormat.get_format_string()."""
514
508
        return 'Bazaar Knit Repository Format 4 (bzr 1.0)\n'