13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
18
17
"""Inter-object utility class."""
19
from __future__ import absolute_import
21
from bzrlib.errors import NoCompatibleInter
21
24
class InterObject(object):
22
25
"""This class represents operations taking place between two objects.
24
27
Its instances have methods like join or copy_content or fetch, and contain
25
references to the source and target objects these operations can be
28
references to the source and target objects these operations can be
26
29
carried out between.
28
31
Often we will provide convenience methods on the objects which carry out
29
32
operations with another of similar type - they will always forward to
30
a subclass of InterObject - i.e.
33
a subclass of InterObject - i.e.
31
34
InterVersionedFile.get(other).method_name(parameters).
36
If the source and target objects implement the locking protocol -
37
lock_read, lock_write, unlock, then the InterObject's lock_read,
38
lock_write and unlock methods may be used (optionally in conjunction with
39
the needs_read_lock and needs_write_lock decorators.)
41
When looking for an inter, the most recently registered types are tested
42
first. So typically the most generic and slowest InterObjects should be
35
# Each concrete InterObject type should have its own optimisers set.
46
# _optimisers = list()
47
# Each concrete InterObject type should have its own optimisers list.
37
49
def __init__(self, source, target):
38
50
"""Construct a default InterObject instance. Please use 'get'.
40
Only subclasses of InterObject should call
52
Only subclasses of InterObject should call
41
53
InterObject.__init__ - clients should call InterFOO.get where FOO
42
54
is the base type of the objects they are interacting between. I.e.
43
55
InterVersionedFile or InterRepository.
47
59
self.source = source
48
60
self.target = target
62
def _double_lock(self, lock_source, lock_target):
63
"""Take out two locks, rolling back the first if the second throws."""
68
# we want to ensure that we don't leave source locked by mistake.
69
# and any error on target should not confuse source.
51
74
def get(klass, source, target):
52
75
"""Retrieve a Inter worker object for these objects.
55
78
the InterObject instance.
56
79
:param target: the object to be the 'target' member of
57
80
the InterObject instance.
58
82
If an optimised worker exists it will be used otherwise
59
83
a default Inter worker instance will be created.
61
for provider in klass._optimisers:
85
for provider in reversed(klass._optimisers):
62
86
if provider.is_compatible(source, target):
63
87
return provider(source, target)
64
return klass(source, target)
88
raise NoCompatibleInter(source, target)
91
"""Take out a logical read lock.
93
This will lock the source branch and the target branch. The source gets
94
a read lock and the target a read lock.
96
self._double_lock(self.source.lock_read, self.target.lock_read)
99
"""Take out a logical write lock.
101
This will lock the source branch and the target branch. The source gets
102
a read lock and the target a write lock.
104
self._double_lock(self.source.lock_read, self.target.lock_write)
67
107
def register_optimiser(klass, optimiser):
68
108
"""Register an InterObject optimiser."""
69
klass._optimisers.add(optimiser)
109
klass._optimisers.append(optimiser)
112
"""Release the locks on source and target."""
72
119
def unregister_optimiser(klass, optimiser):