← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/lockable_files.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2008-09-03 22:30:56 UTC
  • mfrom: (3644.2.13 index_builder_cleanup)
  • Revision ID: pqm@pqm.ubuntu.com-20080903223056-b108iytb38xkznci
(jam) Streamline BTreeBuilder.add_node et al to make btree creation
        faster.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/lockable_files.py
1
 
# Copyright (C) 2005-2011 Canonical Ltd
 
1
# Copyright (C) 2005, 2006, 2008 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
12
12
#
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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
16
 
17
 
from __future__ import absolute_import
 
17
from cStringIO import StringIO
18
18
 
19
19
from bzrlib.lazy_import import lazy_import
20
20
lazy_import(globals(), """
 
21
import codecs
21
22
import warnings
22
23
 
23
24
from bzrlib import (
24
 
    counted_lock,
25
25
    errors,
26
 
    lock,
27
26
    osutils,
28
27
    transactions,
29
28
    urlutils,
31
30
""")
32
31
 
33
32
from bzrlib.decorators import (
34
 
    only_raises,
35
 
    )
36
 
 
 
33
    needs_read_lock,
 
34
    needs_write_lock,
 
35
    )
 
36
from bzrlib.symbol_versioning import (
 
37
    deprecated_in,
 
38
    deprecated_method,
 
39
    )
 
40
 
 
41
 
 
42
# XXX: The tracking here of lock counts and whether the lock is held is
 
43
# somewhat redundant with what's done in LockDir; the main difference is that
 
44
# LockableFiles permits reentrancy.
37
45
 
38
46
class LockableFiles(object):
39
47
    """Object representing a set of related files locked within the same scope.
40
48
 
41
 
    This coordinates access to the lock along with providing a transaction.
 
49
    These files are used by a WorkingTree, Repository or Branch, and should
 
50
    generally only be touched by that object.
 
51
 
 
52
    LockableFiles also provides some policy on top of Transport for encoding
 
53
    control files as utf-8.
42
54
 
43
55
    LockableFiles manage a lock count and can be locked repeatedly by
44
56
    a single caller.  (The underlying lock implementation generally does not
45
57
    support this.)
46
58
 
47
59
    Instances of this class are often called control_files.
 
60
    
 
61
    This object builds on top of a Transport, which is used to actually write
 
62
    the files to disk, and an OSLock or LockDir, which controls how access to
 
63
    the files is controlled.  The particular type of locking used is set when
 
64
    the object is constructed.  In older formats OSLocks are used everywhere.
 
65
    in newer formats a LockDir is used for Repositories and Branches, and 
 
66
    OSLocks for the local filesystem.
48
67
 
49
 
    This class is now deprecated; code should move to using the Transport
50
 
    directly for file operations and using the lock or CountedLock for
 
68
    This class is now deprecated; code should move to using the Transport 
 
69
    directly for file operations and using the lock or CountedLock for 
51
70
    locking.
52
 
 
53
 
    :ivar _lock: The real underlying lock (e.g. a LockDir)
54
 
    :ivar _lock_count: If _lock_mode is true, a positive count of the number
55
 
        of times the lock has been taken (and not yet released) *by this
56
 
        process*, through this particular object instance.
57
 
    :ivar _lock_mode: None, or 'r' or 'w'
58
71
    """
59
72
 
 
73
    # _lock_mode: None, or 'r' or 'w'
 
74
 
 
75
    # _lock_count: If _lock_mode is true, a positive count of the number of
 
76
    # times the lock has been taken *by this process*.   
 
77
    
60
78
    def __init__(self, transport, lock_name, lock_class):
61
79
        """Create a LockableFiles group
62
80
 
63
 
        :param transport: Transport pointing to the directory holding the
 
81
        :param transport: Transport pointing to the directory holding the 
64
82
            control files and lock.
65
83
        :param lock_name: Name of the lock guarding these files.
66
84
        :param lock_class: Class of lock strategy to use: typically
76
94
        self._lock = lock_class(transport, esc_name,
77
95
                                file_modebits=self._file_mode,
78
96
                                dir_modebits=self._dir_mode)
79
 
        self._counted_lock = counted_lock.CountedLock(self._lock)
80
97
 
81
98
    def create_lock(self):
82
99
        """Create the lock.
89
106
    def __repr__(self):
90
107
        return '%s(%r)' % (self.__class__.__name__,
91
108
                           self._transport)
92
 
 
93
109
    def __str__(self):
94
110
        return 'LockableFiles(%s, %s)' % (self.lock_name, self._transport.base)
95
111
 
 
112
    def __del__(self):
 
113
        if self.is_locked():
 
114
            # do not automatically unlock; there should have been a
 
115
            # try/finally to unlock this.
 
116
            warnings.warn("%r was gc'd while locked" % self)
 
117
 
96
118
    def break_lock(self):
97
119
        """Break the lock of this lockable files group if it is held.
98
120
 
101
123
        self._lock.break_lock()
102
124
 
103
125
    def _escape(self, file_or_path):
104
 
        """DEPRECATED: Do not use outside this class"""
105
126
        if not isinstance(file_or_path, basestring):
106
127
            file_or_path = '/'.join(file_or_path)
107
128
        if file_or_path == '':
110
131
 
111
132
    def _find_modes(self):
112
133
        """Determine the appropriate modes for files and directories.
113
 
 
114
 
        :deprecated: Replaced by BzrDir._find_creation_modes.
 
134
        
 
135
        :deprecated: Replaced by BzrDir._find_modes.
115
136
        """
116
 
        # XXX: The properties created by this can be removed or deprecated
117
 
        # once all the _get_text_store methods etc no longer use them.
118
 
        # -- mbp 20080512
119
137
        try:
120
138
            st = self._transport.stat('.')
121
139
        except errors.TransportNotPossible:
130
148
            # Remove the sticky and execute bits for files
131
149
            self._file_mode = self._dir_mode & ~07111
132
150
 
 
151
    @deprecated_method(deprecated_in((1, 6, 0)))
 
152
    def controlfilename(self, file_or_path):
 
153
        """Return location relative to branch.
 
154
        
 
155
        :deprecated: Use Transport methods instead.
 
156
        """
 
157
        return self._transport.abspath(self._escape(file_or_path))
 
158
 
 
159
    @needs_read_lock
 
160
    @deprecated_method(deprecated_in((1, 5, 0)))
 
161
    def get(self, relpath):
 
162
        """Get a file as a bytestream.
 
163
        
 
164
        :deprecated: Use a Transport instead of LockableFiles.
 
165
        """
 
166
        relpath = self._escape(relpath)
 
167
        return self._transport.get(relpath)
 
168
 
 
169
    @needs_read_lock
 
170
    @deprecated_method(deprecated_in((1, 5, 0)))
 
171
    def get_utf8(self, relpath):
 
172
        """Get a file as a unicode stream.
 
173
        
 
174
        :deprecated: Use a Transport instead of LockableFiles.
 
175
        """
 
176
        relpath = self._escape(relpath)
 
177
        # DO NOT introduce an errors=replace here.
 
178
        return codecs.getreader('utf-8')(self._transport.get(relpath))
 
179
 
 
180
    @needs_write_lock
 
181
    @deprecated_method(deprecated_in((1, 6, 0)))
 
182
    def put(self, path, file):
 
183
        """Write a file.
 
184
        
 
185
        :param path: The path to put the file, relative to the .bzr control
 
186
                     directory
 
187
        :param file: A file-like or string object whose contents should be copied.
 
188
 
 
189
        :deprecated: Use Transport methods instead.
 
190
        """
 
191
        self._transport.put_file(self._escape(path), file, mode=self._file_mode)
 
192
 
 
193
    @needs_write_lock
 
194
    @deprecated_method(deprecated_in((1, 6, 0)))
 
195
    def put_bytes(self, path, a_string):
 
196
        """Write a string of bytes.
 
197
 
 
198
        :param path: The path to put the bytes, relative to the transport root.
 
199
        :param a_string: A string object, whose exact bytes are to be copied.
 
200
 
 
201
        :deprecated: Use Transport methods instead.
 
202
        """
 
203
        self._transport.put_bytes(self._escape(path), a_string,
 
204
                                  mode=self._file_mode)
 
205
 
 
206
    @needs_write_lock
 
207
    @deprecated_method(deprecated_in((1, 6, 0)))
 
208
    def put_utf8(self, path, a_string):
 
209
        """Write a string, encoding as utf-8.
 
210
 
 
211
        :param path: The path to put the string, relative to the transport root.
 
212
        :param string: A string or unicode object whose contents should be copied.
 
213
 
 
214
        :deprecated: Use Transport methods instead.
 
215
        """
 
216
        # IterableFile would not be needed if Transport.put took iterables
 
217
        # instead of files.  ADHB 2005-12-25
 
218
        # RBC 20060103 surely its not needed anyway, with codecs transcode
 
219
        # file support ?
 
220
        # JAM 20060103 We definitely don't want encode(..., 'replace')
 
221
        # these are valuable files which should have exact contents.
 
222
        if not isinstance(a_string, basestring):
 
223
            raise errors.BzrBadParameterNotString(a_string)
 
224
        self.put_bytes(path, a_string.encode('utf-8'))
 
225
 
133
226
    def leave_in_place(self):
134
227
        """Set this LockableFiles to not clear the physical lock on unlock."""
135
228
        self._lock.leave_in_place()
140
233
 
141
234
    def lock_write(self, token=None):
142
235
        """Lock this group of files for writing.
143
 
 
 
236
        
144
237
        :param token: if this is already locked, then lock_write will fail
145
238
            unless the token matches the existing lock.
146
239
        :returns: a token if this instance supports tokens, otherwise None.
153
246
        some other way, and need to synchronise this object's state with that
154
247
        fact.
155
248
        """
 
249
        # TODO: Upgrade locking to support using a Transport,
 
250
        # and potentially a remote locking protocol
156
251
        if self._lock_mode:
157
 
            if (self._lock_mode != 'w'
158
 
                or not self.get_transaction().writeable()):
 
252
            if self._lock_mode != 'w' or not self.get_transaction().writeable():
159
253
                raise errors.ReadOnlyError(self)
160
254
            self._lock.validate_token(token)
161
255
            self._lock_count += 1
165
259
            #traceback.print_stack()
166
260
            self._lock_mode = 'w'
167
261
            self._lock_count = 1
168
 
            self._set_write_transaction()
 
262
            self._set_transaction(transactions.WriteTransaction())
169
263
            self._token_from_lock = token_from_lock
170
264
            return token_from_lock
171
265
 
179
273
            #traceback.print_stack()
180
274
            self._lock_mode = 'r'
181
275
            self._lock_count = 1
182
 
            self._set_read_transaction()
183
 
 
184
 
    def _set_read_transaction(self):
185
 
        """Setup a read transaction."""
186
 
        self._set_transaction(transactions.ReadOnlyTransaction())
187
 
        # 5K may be excessive, but hey, its a knob.
188
 
        self.get_transaction().set_cache_size(5000)
189
 
 
190
 
    def _set_write_transaction(self):
191
 
        """Setup a write transaction."""
192
 
        self._set_transaction(transactions.WriteTransaction())
193
 
 
194
 
    @only_raises(errors.LockNotHeld, errors.LockBroken)
 
276
            self._set_transaction(transactions.ReadOnlyTransaction())
 
277
            # 5K may be excessive, but hey, its a knob.
 
278
            self.get_transaction().set_cache_size(5000)
 
279
                        
195
280
    def unlock(self):
196
281
        if not self._lock_mode:
197
 
            return lock.cant_unlock_not_held(self)
 
282
            raise errors.LockNotHeld(self)
198
283
        if self._lock_count > 1:
199
284
            self._lock_count -= 1
200
285
        else:
211
296
 
212
297
    def get_physical_lock_status(self):
213
298
        """Return physical lock status.
214
 
 
 
299
        
215
300
        Returns true if a lock is held on the transport. If no lock is held, or
216
301
        the underlying locking mechanism does not support querying lock
217
302
        status, false is returned.
299
384
    def validate_token(self, token):
300
385
        if token is not None:
301
386
            raise errors.TokenLockingNotSupported(self)
 
387