94
95
pb.update("Finding revisions", 0, 2)
95
96
search = self._revids_to_fetch()
97
mutter('fetching: %s', search)
98
100
pb.update("Fetching revisions", 1, 2)
99
101
self._fetch_everything_for_search(search)
151
150
"""Determines the exact revisions needed from self.from_repository to
152
151
install self._last_revision in self.to_repository.
154
If no revisions need to be fetched, then this just returns None.
153
:returns: A SearchResult of some sort. (Possibly a
154
PendingAncestryResult, EmptySearchResult, etc.)
156
if self._fetch_spec is not None:
156
mutter("self._fetch_spec, self._last_revision: %r, %r",
157
self._fetch_spec, self._last_revision)
158
get_search_result = getattr(self._fetch_spec, 'get_search_result', None)
159
if get_search_result is not None:
161
'resolving fetch_spec into search result: %s', self._fetch_spec)
162
# This is EverythingNotInOther or a similar kind of fetch_spec.
163
# Turn it into a search result.
164
return get_search_result()
165
elif self._fetch_spec is not None:
166
# The fetch spec is already a concrete search result.
157
167
return self._fetch_spec
158
mutter('fetch up to rev {%s}', self._last_revision)
159
if self._last_revision is NULL_REVISION:
168
elif self._last_revision == NULL_REVISION:
169
# fetch_spec is None + last_revision is null => empty fetch.
160
170
# explicit limit of no revisions needed
162
return self.to_repository.search_missing_revision_ids(
163
self.from_repository, self._last_revision,
164
find_ghosts=self.find_ghosts)
166
def _parent_inventories(self, revision_ids):
167
# Find all the parent revisions referenced by the stream, but
168
# not present in the stream, and make sure we send their
170
parent_maps = self.to_repository.get_parent_map(revision_ids)
172
map(parents.update, parent_maps.itervalues())
173
parents.discard(NULL_REVISION)
174
parents.difference_update(revision_ids)
175
missing_keys = set(('inventories', rev_id) for rev_id in parents)
171
return graph.EmptySearchResult()
172
elif self._last_revision is not None:
173
return graph.NotInOtherForRevs(self.to_repository,
174
self.from_repository, [self._last_revision],
175
find_ghosts=self.find_ghosts).get_search_result()
176
else: # self._last_revision is None:
177
return graph.EverythingNotInOther(self.to_repository,
178
self.from_repository,
179
find_ghosts=self.find_ghosts).get_search_result()
179
182
class Inter1and2Helper(object):
243
249
# yet, and are unlikely to in non-rich-root environments anyway.
244
250
root_id_order.sort(key=operator.itemgetter(0))
245
251
# Create a record stream containing the roots to create.
247
# XXX: not covered by tests, should have a flag to always run
248
# this. -- mbp 20100129
249
graph = self.source_repo.get_known_graph_ancestry(revs)
252
if len(revs) > self.known_graph_threshold:
253
graph = self.source.get_known_graph_ancestry(revs)
250
254
new_roots_stream = _new_root_data_stream(
251
255
root_id_order, rev_id_to_root_id, parent_map, self.source, graph)
252
256
return [('texts', new_roots_stream)]
255
def _get_rich_root_heads_graph(source_repo, revision_ids):
256
"""Get a Graph object suitable for asking heads() for new rich roots."""
260
259
def _new_root_data_stream(
261
260
root_keys_to_create, rev_id_to_root_id_map, parent_map, repo, graph=None):
262
261
"""Generate a texts substream of synthesised root entries.