44
38
def do(self, path, *args):
45
39
"""Execute a repository request.
47
All Repository requests take a path to the repository as their first
48
argument. The repository must be at the exact path given by the
49
client - no searching is done.
41
The repository must be at the exact path - no searching is done.
51
43
The actual logic is delegated to self.do_repository_request.
53
:param client_path: The path for the repository as received from the
55
:return: A SmartServerResponse from self.do_repository_request().
45
:param path: The path for the repository.
46
:return: A smart server from self.do_repository_request().
57
transport = self.transport_from_client_path(path)
48
transport = self._backing_transport.clone(path)
58
49
bzrdir = BzrDir.open_from_transport(transport)
59
# Save the repository for use with do_body.
60
self._repository = bzrdir.open_repository()
61
return self.do_repository_request(self._repository, *args)
63
def do_repository_request(self, repository, *args):
64
"""Override to provide an implementation for a verb."""
65
# No-op for verbs that take bodies (None as a result indicates a body
69
def recreate_search(self, repository, recipe_bytes):
70
lines = recipe_bytes.split('\n')
71
start_keys = set(lines[0].split(' '))
72
exclude_keys = set(lines[1].split(' '))
73
revision_count = int(lines[2])
74
repository.lock_read()
76
search = repository.get_graph()._make_breadth_first_searcher(
80
next_revs = search.next()
83
search.stop_searching_any(exclude_keys.intersection(next_revs))
84
search_result = search.get_result()
85
if search_result.get_recipe()[2] != revision_count:
86
# we got back a different amount of data than expected, this
87
# gets reported as NoSuchRevision, because less revisions
88
# indicates missing revisions, and more should never happen as
89
# the excludes list considers ghosts and ensures that ghost
90
# filling races are not a problem.
91
return (None, FailedSmartServerResponse(('NoSuchRevision',)))
97
class SmartServerRepositoryReadLocked(SmartServerRepositoryRequest):
98
"""Calls self.do_readlocked_repository_request."""
100
def do_repository_request(self, repository, *args):
101
"""Read lock a repository for do_readlocked_repository_request."""
102
repository.lock_read()
104
return self.do_readlocked_repository_request(repository, *args)
109
class SmartServerRepositoryGetParentMap(SmartServerRepositoryRequest):
110
"""Bzr 1.2+ - get parent data for revisions during a graph search."""
112
def do_repository_request(self, repository, *revision_ids):
113
"""Get parent details for some revisions.
115
All the parents for revision_ids are returned. Additionally up to 64KB
116
of additional parent data found by performing a breadth first search
117
from revision_ids is returned. The verb takes a body containing the
118
current search state, see do_body for details.
120
:param repository: The repository to query in.
121
:param revision_ids: The utf8 encoded revision_id to answer for.
123
self._revision_ids = revision_ids
124
return None # Signal that we want a body.
126
def do_body(self, body_bytes):
127
"""Process the current search state and perform the parent lookup.
129
:return: A smart server response where the body contains an utf8
130
encoded flattened list of the parents of the revisions (the same
131
format as Repository.get_revision_graph) which has been bz2
134
repository = self._repository
135
repository.lock_read()
137
return self._do_repository_request(body_bytes)
141
def _do_repository_request(self, body_bytes):
142
repository = self._repository
143
revision_ids = set(self._revision_ids)
144
search, error = self.recreate_search(repository, body_bytes)
145
if error is not None:
147
# TODO might be nice to start up the search again; but thats not
148
# written or tested yet.
149
client_seen_revs = set(search.get_result().get_keys())
150
# Always include the requested ids.
151
client_seen_revs.difference_update(revision_ids)
153
repo_graph = repository.get_graph()
157
next_revs = revision_ids
158
first_loop_done = False
160
queried_revs.update(next_revs)
161
parent_map = repo_graph.get_parent_map(next_revs)
163
for revision_id, parents in parent_map.iteritems():
164
# adjust for the wire
165
if parents == (_mod_revision.NULL_REVISION,):
167
# prepare the next query
168
next_revs.update(parents)
169
if revision_id not in client_seen_revs:
170
# Client does not have this revision, give it to it.
171
# add parents to the result
172
result[revision_id] = parents
173
# Approximate the serialized cost of this revision_id.
174
size_so_far += 2 + len(revision_id) + sum(map(len, parents))
175
# get all the directly asked for parents, and then flesh out to
176
# 64K (compressed) or so. We do one level of depth at a time to
177
# stay in sync with the client. The 250000 magic number is
178
# estimated compression ratio taken from bzr.dev itself.
179
if first_loop_done and size_so_far > 250000:
182
# don't query things we've already queried
183
next_revs.difference_update(queried_revs)
184
first_loop_done = True
186
# sorting trivially puts lexographically similar revision ids together.
188
for revision, parents in sorted(result.items()):
189
lines.append(' '.join((revision, ) + tuple(parents)))
191
return SuccessfulSmartServerResponse(
192
('ok', ), bz2.compress('\n'.join(lines)))
195
class SmartServerRepositoryGetRevisionGraph(SmartServerRepositoryReadLocked):
197
def do_readlocked_repository_request(self, repository, revision_id):
50
repository = bzrdir.open_repository()
51
return self.do_repository_request(repository, *args)
54
class SmartServerRepositoryGetRevisionGraph(SmartServerRepositoryRequest):
56
def do_repository_request(self, repository, revision_id):
198
57
"""Return the result of repository.get_revision_graph(revision_id).
200
Deprecated as of bzr 1.4, but supported for older clients.
202
59
:param repository: The repository to query in.
203
60
:param revision_id: The utf8 encoded revision_id to get a graph from.
added
removed
bzrlib/smart/repository.py
