1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
|
# Copyright (C) 2006 Canonical Ltd
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
"""Server-side branch related request implmentations."""
from bzrlib import errors
from bzrlib.bzrdir import BzrDir
from bzrlib.smart.request import (
FailedSmartServerResponse,
SmartServerRequest,
SuccessfulSmartServerResponse,
)
class SmartServerBranchRequest(SmartServerRequest):
"""Base class for handling common branch request logic.
"""
def do(self, path, *args):
"""Execute a request for a branch at path.
All Branch requests take a path to the branch as their first argument.
If the branch is a branch reference, NotBranchError is raised.
:param path: The path for the repository as received from the
client.
:return: A SmartServerResponse from self.do_with_branch().
"""
transport = self.transport_from_client_path(path)
bzrdir = BzrDir.open_from_transport(transport)
if bzrdir.get_branch_reference() is not None:
raise errors.NotBranchError(transport.base)
branch = bzrdir.open_branch()
return self.do_with_branch(branch, *args)
class SmartServerLockedBranchRequest(SmartServerBranchRequest):
"""Base class for handling common branch request logic for requests that
need a write lock.
"""
def do_with_branch(self, branch, branch_token, repo_token, *args):
"""Execute a request for a branch.
A write lock will be acquired with the given tokens for the branch and
repository locks. The lock will be released once the request is
processed. The physical lock state won't be changed.
"""
# XXX: write a test for LockContention
branch.repository.lock_write(token=repo_token)
try:
branch.lock_write(token=branch_token)
try:
return self.do_with_locked_branch(branch, *args)
finally:
branch.unlock()
finally:
branch.repository.unlock()
class SmartServerBranchGetConfigFile(SmartServerBranchRequest):
def do_with_branch(self, branch):
"""Return the content of branch.control_files.get('branch.conf').
The body is not utf8 decoded - its the literal bytestream from disk.
"""
try:
content = branch.control_files.get('branch.conf').read()
except errors.NoSuchFile:
content = ''
return SuccessfulSmartServerResponse( ('ok', ), content)
class SmartServerRequestRevisionHistory(SmartServerBranchRequest):
def do_with_branch(self, branch):
"""Get the revision history for the branch.
The revision list is returned as the body content,
with each revision utf8 encoded and \x00 joined.
"""
return SuccessfulSmartServerResponse(
('ok', ), ('\x00'.join(branch.revision_history())))
class SmartServerBranchRequestLastRevisionInfo(SmartServerBranchRequest):
def do_with_branch(self, branch):
"""Return branch.last_revision_info().
The revno is encoded in decimal, the revision_id is encoded as utf8.
"""
revno, last_revision = branch.last_revision_info()
return SuccessfulSmartServerResponse(('ok', str(revno), last_revision))
class SmartServerBranchRequestSetLastRevision(SmartServerLockedBranchRequest):
def do_with_locked_branch(self, branch, new_last_revision_id):
if new_last_revision_id == 'null:':
branch.set_revision_history([])
else:
if not branch.repository.has_revision(new_last_revision_id):
return FailedSmartServerResponse(
('NoSuchRevision', new_last_revision_id))
branch.generate_revision_history(new_last_revision_id)
return SuccessfulSmartServerResponse(('ok',))
class SmartServerBranchRequestSetLastRevisionInfo(
SmartServerLockedBranchRequest):
"""Branch.set_last_revision_info. Sets the revno and the revision ID of
the specified branch.
New in bzrlib 1.4.
"""
def do_with_locked_branch(self, branch, new_revno, new_last_revision_id):
try:
branch.set_last_revision_info(int(new_revno), new_last_revision_id)
except errors.NoSuchRevision:
return FailedSmartServerResponse(
('NoSuchRevision', new_last_revision_id))
return SuccessfulSmartServerResponse(('ok',))
class SmartServerBranchRequestLockWrite(SmartServerBranchRequest):
def do_with_branch(self, branch, branch_token='', repo_token=''):
if branch_token == '':
branch_token = None
if repo_token == '':
repo_token = None
try:
repo_token = branch.repository.lock_write(token=repo_token)
try:
branch_token = branch.lock_write(token=branch_token)
finally:
# this leaves the repository with 1 lock
branch.repository.unlock()
except errors.LockContention:
return FailedSmartServerResponse(('LockContention',))
except errors.TokenMismatch:
return FailedSmartServerResponse(('TokenMismatch',))
except errors.UnlockableTransport:
return FailedSmartServerResponse(('UnlockableTransport',))
except errors.LockFailed, e:
return FailedSmartServerResponse(('LockFailed', str(e.lock), str(e.why)))
if repo_token is None:
repo_token = ''
else:
branch.repository.leave_lock_in_place()
branch.leave_lock_in_place()
branch.unlock()
return SuccessfulSmartServerResponse(('ok', branch_token, repo_token))
class SmartServerBranchRequestUnlock(SmartServerBranchRequest):
def do_with_branch(self, branch, branch_token, repo_token):
try:
branch.repository.lock_write(token=repo_token)
try:
branch.lock_write(token=branch_token)
finally:
branch.repository.unlock()
except errors.TokenMismatch:
return FailedSmartServerResponse(('TokenMismatch',))
if repo_token:
branch.repository.dont_leave_lock_in_place()
branch.dont_leave_lock_in_place()
branch.unlock()
return SuccessfulSmartServerResponse(('ok',))
|