1
# Copyright (C) 2005 Canonical Ltd
2
# Authors: Robert Collins <robert.collins@canonical.com>
4
# This program is free software; you can redistribute it and/or modify
5
# it under the terms of the GNU General Public License as published by
6
# the Free Software Foundation; either version 2 of the License, or
7
# (at your option) any later version.
9
# This program is distributed in the hope that it will be useful,
10
# but WITHOUT ANY WARRANTY; without even the implied warranty of
11
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12
# GNU General Public License for more details.
14
# You should have received a copy of the GNU General Public License
15
# along with this program; if not, write to the Free Software
16
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
18
"""This module provides a transactional facility.
20
Transactions provide hooks to allow data objects (i.e. inventory weaves or
21
the revision-history file) to be placed in a registry and retrieved later
22
during the same transaction. Transactions in bzr are not atomic - they
23
depend on data ordering of writes, so we do not have commit or rollback
24
facilities at the transaction level.
26
Read only transactions raise an assert when objects are listed as dirty
27
against them - preventing unintended writes.
29
Write transactions preserve dirty objects in the cache, though due to the
30
write ordering approach we use for consistency 'dirty' is a misleading term.
31
A dirty object is one we have modified.
33
Both read and write transactions *may* flush unchanged objects out of
34
memory, unless they are marked as 'precious' which indicates that
35
repeated reads cannot be obtained if the object is ejected, or that
36
the object is an expensive one for obtaining.
41
import bzrlib.errors as errors
42
from bzrlib.identitymap import IdentityMap, NullIdentityMap
43
from bzrlib.trace import mutter
46
class ReadOnlyTransaction(object):
47
"""A read only unit of work for data objects."""
50
"""Clean up this transaction."""
53
super(ReadOnlyTransaction, self).__init__()
54
self.map = IdentityMap()
55
self._clean_objects = set()
56
self._clean_queue = []
58
self._precious_objects = set()
60
def is_clean(self, an_object):
61
"""Return True if an_object is clean."""
62
return (an_object in self._clean_objects)
64
def register_clean(self, an_object, precious=False):
65
"""Register an_object as being clean.
67
If the precious hint is True, the object will not
68
be ejected from the object identity map ever.
70
self._clean_objects.add(an_object)
71
self._clean_queue.append(an_object)
73
self._precious_objects.add(an_object)
76
def register_dirty(self, an_object):
77
"""Register an_object as being dirty."""
78
raise errors.ReadOnlyObjectDirtiedError(an_object)
80
def set_cache_size(self, size):
81
"""Set a new cache size."""
87
"""Trim the cache back if needed."""
88
if self._limit < 0 or self._limit - len(self._clean_objects) > 0:
90
needed = len(self._clean_objects) - self._limit
92
while needed and offset < len(self._clean_objects):
93
# references we know of:
94
# temp passed to getrefcount in our frame
95
# temp in getrefcount's frame
101
if (sys.getrefcount(self._clean_queue[offset]) <= 7 and
102
not self._clean_queue[offset] in self._precious_objects):
103
removed = self._clean_queue[offset]
104
self._clean_objects.remove(removed)
105
del self._clean_queue[offset]
106
self.map.remove_object(removed)
107
mutter('removed object %r', removed)
113
"""Read only transactions do not allow writes."""
116
class WriteTransaction(ReadOnlyTransaction):
117
"""A write transaction
119
- caches domain objects
120
- clean objects can be removed from the cache
121
- dirty objects are retained.
125
"""Clean up this transaction."""
126
for thing in self._dirty_objects:
127
callback = getattr(thing, 'transaction_finished', None)
128
if callback is not None:
132
super(WriteTransaction, self).__init__()
133
self._dirty_objects = set()
135
def is_dirty(self, an_object):
136
"""Return True if an_object is dirty."""
137
return (an_object in self._dirty_objects)
139
def register_dirty(self, an_object):
140
"""Register an_object as being dirty.
142
Dirty objects are not ejected from the identity map
143
until the transaction finishes and get informed
144
when the transaction finishes.
146
self._dirty_objects.add(an_object)
147
if self.is_clean(an_object):
148
self._clean_objects.remove(an_object)
149
del self._clean_queue[self._clean_queue.index(an_object)]
153
"""Write transactions allow writes."""
157
class PassThroughTransaction(object):
158
"""A pass through transaction
161
- nothing ever gets into the identity map.
165
"""Clean up this transaction."""
166
for thing in self._dirty_objects:
167
callback = getattr(thing, 'transaction_finished', None)
168
if callback is not None:
172
super(PassThroughTransaction, self).__init__()
173
self.map = NullIdentityMap()
174
self._dirty_objects = set()
176
def register_clean(self, an_object, precious=False):
177
"""Register an_object as being clean.
179
Note that precious is only a hint, and PassThroughTransaction
183
def register_dirty(self, an_object):
184
"""Register an_object as being dirty.
186
Dirty objects get informed
187
when the transaction finishes.
189
self._dirty_objects.add(an_object)
191
def set_cache_size(self, ignored):
192
"""Do nothing, we are passing through."""
195
"""Pass through transactions allow writes."""
added
removed
bzrlib/transactions.py
