← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/inter.py

Just embed a struct st in the python result object, avoids converting things we don't need converted, and copying values around always.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/inter.py
29
29
    operations with another of similar type - they will always forward to
30
30
    a subclass of InterObject - i.e. 
31
31
    InterVersionedFile.get(other).method_name(parameters).
 
32
 
 
33
    If the source and target objects implement the locking protocol - 
 
34
    lock_read, lock_write, unlock, then the InterObject's lock_read,
 
35
    lock_write and unlock methods may be used (optionally in conjunction with
 
36
    the needs_read_lock and needs_write_lock decorators.)
 
37
 
 
38
    When looking for an inter, the most recently registered types are tested
 
39
    first.  So typically the most generic and slowest InterObjects should be
 
40
    registered first.
32
41
    """
33
42
 
34
 
    # _optimisers = set()
35
 
    # Each concrete InterObject type should have its own optimisers set.
 
43
    # _optimisers = list()
 
44
    # Each concrete InterObject type should have its own optimisers list.
36
45
 
37
46
    def __init__(self, source, target):
38
47
        """Construct a default InterObject instance. Please use 'get'.
47
56
        self.source = source
48
57
        self.target = target
49
58
 
 
59
    def _double_lock(self, lock_source, lock_target):
 
60
        """Take out two locks, rolling back the first if the second throws."""
 
61
        lock_source()
 
62
        try:
 
63
            lock_target()
 
64
        except Exception:
 
65
            # we want to ensure that we don't leave source locked by mistake.
 
66
            # and any error on target should not confuse source.
 
67
            self.source.unlock()
 
68
            raise
 
69
 
50
70
    @classmethod
51
71
    def get(klass, source, target):
52
72
        """Retrieve a Inter worker object for these objects.
58
78
        If an optimised worker exists it will be used otherwise
59
79
        a default Inter worker instance will be created.
60
80
        """
61
 
        for provider in klass._optimisers:
 
81
        for provider in reversed(klass._optimisers):
62
82
            if provider.is_compatible(source, target):
63
83
                return provider(source, target)
64
84
        return klass(source, target)
65
85
 
 
86
    def lock_read(self):
 
87
        """Take out a logical read lock.
 
88
 
 
89
        This will lock the source branch and the target branch. The source gets
 
90
        a read lock and the target a read lock.
 
91
        """
 
92
        self._double_lock(self.source.lock_read, self.target.lock_read)
 
93
 
 
94
    def lock_write(self):
 
95
        """Take out a logical write lock.
 
96
 
 
97
        This will lock the source branch and the target branch. The source gets
 
98
        a read lock and the target a write lock.
 
99
        """
 
100
        self._double_lock(self.source.lock_read, self.target.lock_write)
 
101
 
66
102
    @classmethod
67
103
    def register_optimiser(klass, optimiser):
68
104
        """Register an InterObject optimiser."""
69
 
        klass._optimisers.add(optimiser)
 
105
        klass._optimisers.append(optimiser)
 
106
 
 
107
    def unlock(self):
 
108
        """Release the locks on source and target."""
 
109
        try:
 
110
            self.target.unlock()
 
111
        finally:
 
112
            self.source.unlock()
70
113
 
71
114
    @classmethod
72
115
    def unregister_optimiser(klass, optimiser):