/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/fetch.py

  • Committer: Robert Collins
  • Date: 2010-05-06 11:08:10 UTC
  • mto: This revision was merged to the branch mainline in revision 5223.
  • Revision ID: robertc@robertcollins.net-20100506110810-h3j07fh5gmw54s25
Cleaner matcher matching revised unlocking protocol.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/fetch.py
1
 
# Copyright (C) 2005, 2006 Canonical Ltd
 
1
# Copyright (C) 2005-2010 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
17
17
 
18
18
"""Copying of history from one branch to another.
21
21
that has merged into it.  As the first step of a merge, pull, or
22
22
branch operation we copy history from the source into the destination
23
23
branch.
24
 
 
25
 
The copying is done in a slightly complicated order.  We don't want to
26
 
add a revision to the store until everything it refers to is also
27
 
stored, so that if a revision is present we can totally recreate it.
28
 
However, we can't know what files are included in a revision until we
29
 
read its inventory.  So we query the inventory store of the source for
30
 
the ids we need, and then pull those ids and finally actually join
31
 
the inventories.
32
24
"""
33
25
 
 
26
import operator
 
27
 
 
28
from bzrlib.lazy_import import lazy_import
 
29
lazy_import(globals(), """
 
30
from bzrlib import (
 
31
    tsort,
 
32
    versionedfile,
 
33
    )
 
34
""")
34
35
import bzrlib
35
 
import bzrlib.errors as errors
36
 
from bzrlib.errors import (InstallFailed,
37
 
                           )
38
 
from bzrlib.progress import ProgressPhase
 
36
from bzrlib import (
 
37
    errors,
 
38
    ui,
 
39
    )
39
40
from bzrlib.revision import NULL_REVISION
40
 
from bzrlib.symbol_versioning import (deprecated_function,
41
 
        deprecated_method,
42
 
        zero_eight,
43
 
        )
44
41
from bzrlib.trace import mutter
45
 
import bzrlib.ui
46
 
 
47
 
from bzrlib.lazy_import import lazy_import
48
 
 
49
 
# TODO: Avoid repeatedly opening weaves so many times.
50
 
 
51
 
# XXX: This doesn't handle ghost (not present in branch) revisions at
52
 
# all yet.  I'm not sure they really should be supported.
53
 
 
54
 
# NOTE: This doesn't copy revisions which may be present but not
55
 
# merged into the last revision.  I'm not sure we want to do that.
56
 
 
57
 
# - get a list of revisions that need to be pulled in
58
 
# - for each one, pull in that revision file
59
 
#   and get the inventory, and store the inventory with right
60
 
#   parents.
61
 
# - and get the ancestry, and store that with right parents too
62
 
# - and keep a note of all file ids and version seen
63
 
# - then go through all files; for each one get the weave,
64
 
#   and add in all file versions
65
 
 
66
 
 
67
 
@deprecated_function(zero_eight)
68
 
def greedy_fetch(to_branch, from_branch, revision=None, pb=None):
69
 
    """Legacy API, please see branch.fetch(from_branch, last_revision, pb)."""
70
 
    f = Fetcher(to_branch, from_branch, revision, pb)
71
 
    return f.count_copied, f.failed_revisions
72
 
 
73
 
fetch = greedy_fetch
74
42
 
75
43
 
76
44
class RepoFetcher(object):
77
45
    """Pull revisions and texts from one repository to another.
78
46
 
79
 
    last_revision
80
 
        if set, try to limit to the data this revision references.
81
 
 
82
 
    after running:
83
 
    count_copied -- number of revisions copied
84
 
 
85
 
    This should not be used directory, its essential a object to encapsulate
 
47
    This should not be used directly, it's essential a object to encapsulate
86
48
    the logic in InterRepository.fetch().
87
49
    """
88
 
    def __init__(self, to_repository, from_repository, last_revision=None, pb=None):
89
 
        # result variables.
90
 
        self.failed_revisions = []
91
 
        self.count_copied = 0
92
 
        if to_repository.control_files._transport.base == from_repository.control_files._transport.base:
93
 
            # check that last_revision is in 'from' and then return a no-operation.
94
 
            if last_revision not in (None, NULL_REVISION):
95
 
                to_repository.get_revision(last_revision)
96
 
            return
 
50
 
 
51
    def __init__(self, to_repository, from_repository, last_revision=None,
 
52
        find_ghosts=True, fetch_spec=None):
 
53
        """Create a repo fetcher.
 
54
 
 
55
        :param last_revision: If set, try to limit to the data this revision
 
56
            references.
 
57
        :param find_ghosts: If True search the entire history for ghosts.
 
58
        """
 
59
        # repository.fetch has the responsibility for short-circuiting
 
60
        # attempts to copy between a repository and itself.
97
61
        self.to_repository = to_repository
98
62
        self.from_repository = from_repository
 
63
        self.sink = to_repository._get_sink()
99
64
        # must not mutate self._last_revision as its potentially a shared instance
100
65
        self._last_revision = last_revision
101
 
        if pb is None:
102
 
            self.pb = bzrlib.ui.ui_factory.nested_progress_bar()
103
 
            self.nested_pb = self.pb
104
 
        else:
105
 
            self.pb = pb
106
 
            self.nested_pb = None
 
66
        self._fetch_spec = fetch_spec
 
67
        self.find_ghosts = find_ghosts
107
68
        self.from_repository.lock_read()
 
69
        mutter("Using fetch logic to copy between %s(%s) and %s(%s)",
 
70
               self.from_repository, self.from_repository._format,
 
71
               self.to_repository, self.to_repository._format)
108
72
        try:
109
 
            self.to_repository.lock_write()
110
 
            try:
111
 
                self.to_repository.start_write_group()
112
 
                try:
113
 
                    self.__fetch()
114
 
                except:
115
 
                    self.to_repository.abort_write_group()
116
 
                    raise
117
 
                else:
118
 
                    self.to_repository.commit_write_group()
119
 
            finally:
120
 
                if self.nested_pb is not None:
121
 
                    self.nested_pb.finished()
122
 
                self.to_repository.unlock()
 
73
            self.__fetch()
123
74
        finally:
124
75
            self.from_repository.unlock()
125
76
 
126
77
    def __fetch(self):
127
78
        """Primary worker function.
128
79
 
129
 
        This initialises all the needed variables, and then fetches the 
 
80
        This initialises all the needed variables, and then fetches the
130
81
        requested revisions, finally clearing the progress bar.
131
82
        """
132
 
        self.to_weaves = self.to_repository.weave_store
133
 
        self.to_control = self.to_repository.control_weaves
134
 
        self.from_weaves = self.from_repository.weave_store
135
 
        self.from_control = self.from_repository.control_weaves
 
83
        # Roughly this is what we're aiming for fetch to become:
 
84
        #
 
85
        # missing = self.sink.insert_stream(self.source.get_stream(search))
 
86
        # if missing:
 
87
        #     missing = self.sink.insert_stream(self.source.get_items(missing))
 
88
        # assert not missing
136
89
        self.count_total = 0
137
90
        self.file_ids_names = {}
138
 
        pp = ProgressPhase('Fetch phase', 4, self.pb)
139
 
        try:
140
 
            pp.next_phase()
141
 
            revs = self._revids_to_fetch()
142
 
            # something to do ?
143
 
            if revs:
144
 
                pp.next_phase()
145
 
                self._fetch_weave_texts(revs)
146
 
                pp.next_phase()
147
 
                self._fetch_inventory_weave(revs)
148
 
                pp.next_phase()
149
 
                self._fetch_revision_texts(revs)
150
 
                self.count_copied += len(revs)
151
 
        finally:
152
 
            self.pb.clear()
 
91
        pb = ui.ui_factory.nested_progress_bar()
 
92
        pb.show_pct = pb.show_count = False
 
93
        try:
 
94
            pb.update("Finding revisions", 0, 2)
 
95
            search = self._revids_to_fetch()
 
96
            if search is None:
 
97
                return
 
98
            pb.update("Fetching revisions", 1, 2)
 
99
            self._fetch_everything_for_search(search)
 
100
        finally:
 
101
            pb.finished()
 
102
 
 
103
    def _fetch_everything_for_search(self, search):
 
104
        """Fetch all data for the given set of revisions."""
 
105
        # The first phase is "file".  We pass the progress bar for it directly
 
106
        # into item_keys_introduced_by, which has more information about how
 
107
        # that phase is progressing than we do.  Progress updates for the other
 
108
        # phases are taken care of in this function.
 
109
        # XXX: there should be a clear owner of the progress reporting.  Perhaps
 
110
        # item_keys_introduced_by should have a richer API than it does at the
 
111
        # moment, so that it can feed the progress information back to this
 
112
        # function?
 
113
        if (self.from_repository._format.rich_root_data and
 
114
            not self.to_repository._format.rich_root_data):
 
115
            raise errors.IncompatibleRepositories(
 
116
                self.from_repository, self.to_repository,
 
117
                "different rich-root support")
 
118
        pb = ui.ui_factory.nested_progress_bar()
 
119
        try:
 
120
            pb.update("Get stream source")
 
121
            source = self.from_repository._get_source(
 
122
                self.to_repository._format)
 
123
            stream = source.get_stream(search)
 
124
            from_format = self.from_repository._format
 
125
            pb.update("Inserting stream")
 
126
            resume_tokens, missing_keys = self.sink.insert_stream(
 
127
                stream, from_format, [])
 
128
            if self.to_repository._fallback_repositories:
 
129
                missing_keys.update(
 
130
                    self._parent_inventories(search.get_keys()))
 
131
            if missing_keys:
 
132
                pb.update("Missing keys")
 
133
                stream = source.get_stream_for_missing_keys(missing_keys)
 
134
                pb.update("Inserting missing keys")
 
135
                resume_tokens, missing_keys = self.sink.insert_stream(
 
136
                    stream, from_format, resume_tokens)
 
137
            if missing_keys:
 
138
                raise AssertionError(
 
139
                    "second push failed to complete a fetch %r." % (
 
140
                        missing_keys,))
 
141
            if resume_tokens:
 
142
                raise AssertionError(
 
143
                    "second push failed to commit the fetch %r." % (
 
144
                        resume_tokens,))
 
145
            pb.update("Finishing stream")
 
146
            self.sink.finished()
 
147
        finally:
 
148
            pb.finished()
153
149
 
154
150
    def _revids_to_fetch(self):
 
151
        """Determines the exact revisions needed from self.from_repository to
 
152
        install self._last_revision in self.to_repository.
 
153
 
 
154
        If no revisions need to be fetched, then this just returns None.
 
155
        """
 
156
        if self._fetch_spec is not None:
 
157
            return self._fetch_spec
155
158
        mutter('fetch up to rev {%s}', self._last_revision)
156
159
        if self._last_revision is NULL_REVISION:
157
160
            # explicit limit of no revisions needed
158
161
            return None
159
 
        if (self._last_revision is not None and
160
 
            self.to_repository.has_revision(self._last_revision)):
161
 
            return None
162
 
            
163
 
        try:
164
 
            return self.to_repository.missing_revision_ids(self.from_repository,
165
 
                                                           self._last_revision)
166
 
        except errors.NoSuchRevision:
167
 
            raise InstallFailed([self._last_revision])
168
 
 
169
 
    def _fetch_weave_texts(self, revs):
170
 
        texts_pb = bzrlib.ui.ui_factory.nested_progress_bar()
171
 
        try:
172
 
            # fileids_altered_by_revision_ids requires reading the inventory
173
 
            # weave, we will need to read the inventory weave again when
174
 
            # all this is done, so enable caching for that specific weave
175
 
            inv_w = self.from_repository.get_inventory_weave()
176
 
            inv_w.enable_cache()
177
 
            file_ids = self.from_repository.fileids_altered_by_revision_ids(revs)
178
 
            count = 0
179
 
            num_file_ids = len(file_ids)
180
 
            for file_id, required_versions in file_ids.items():
181
 
                texts_pb.update("fetch texts", count, num_file_ids)
182
 
                count +=1
183
 
                to_weave = self.to_weaves.get_weave_or_empty(file_id,
184
 
                    self.to_repository.get_transaction())
185
 
                from_weave = self.from_weaves.get_weave(file_id,
186
 
                    self.from_repository.get_transaction())
187
 
                # we fetch all the texts, because texts do
188
 
                # not reference anything, and its cheap enough
189
 
                to_weave.join(from_weave, version_ids=required_versions)
190
 
                # we don't need *all* of this data anymore, but we dont know
191
 
                # what we do. This cache clearing will result in a new read 
192
 
                # of the knit data when we do the checkout, but probably we
193
 
                # want to emit the needed data on the fly rather than at the
194
 
                # end anyhow.
195
 
                # the from weave should know not to cache data being joined,
196
 
                # but its ok to ask it to clear.
197
 
                from_weave.clear_cache()
198
 
                to_weave.clear_cache()
199
 
        finally:
200
 
            texts_pb.finished()
201
 
 
202
 
    def _fetch_inventory_weave(self, revs):
203
 
        pb = bzrlib.ui.ui_factory.nested_progress_bar()
204
 
        try:
205
 
            pb.update("fetch inventory", 0, 2)
206
 
            to_weave = self.to_control.get_weave('inventory',
207
 
                    self.to_repository.get_transaction())
208
 
    
209
 
            child_pb = bzrlib.ui.ui_factory.nested_progress_bar()
210
 
            try:
211
 
                # just merge, this is optimisable and its means we don't
212
 
                # copy unreferenced data such as not-needed inventories.
213
 
                pb.update("fetch inventory", 1, 3)
214
 
                from_weave = self.from_repository.get_inventory_weave()
215
 
                pb.update("fetch inventory", 2, 3)
216
 
                # we fetch only the referenced inventories because we do not
217
 
                # know for unselected inventories whether all their required
218
 
                # texts are present in the other repository - it could be
219
 
                # corrupt.
220
 
                to_weave.join(from_weave, pb=child_pb, msg='merge inventory',
221
 
                              version_ids=revs)
222
 
                from_weave.clear_cache()
223
 
            finally:
224
 
                child_pb.finished()
225
 
        finally:
226
 
            pb.finished()
227
 
 
228
 
 
229
 
class GenericRepoFetcher(RepoFetcher):
230
 
    """This is a generic repo to repo fetcher.
231
 
 
232
 
    This makes minimal assumptions about repo layout and contents.
233
 
    It triggers a reconciliation after fetching to ensure integrity.
234
 
    """
235
 
 
236
 
    def _fetch_revision_texts(self, revs):
237
 
        """Fetch revision object texts"""
238
 
        rev_pb = bzrlib.ui.ui_factory.nested_progress_bar()
239
 
        try:
240
 
            to_txn = self.to_transaction = self.to_repository.get_transaction()
241
 
            count = 0
242
 
            total = len(revs)
243
 
            to_store = self.to_repository._revision_store
244
 
            for rev in revs:
245
 
                pb = bzrlib.ui.ui_factory.nested_progress_bar()
246
 
                try:
247
 
                    pb.update('copying revisions', count, total)
248
 
                    try:
249
 
                        sig_text = self.from_repository.get_signature_text(rev)
250
 
                        to_store.add_revision_signature_text(rev, sig_text, to_txn)
251
 
                    except errors.NoSuchRevision:
252
 
                        # not signed.
253
 
                        pass
254
 
                    to_store.add_revision(self.from_repository.get_revision(rev),
255
 
                                          to_txn)
256
 
                    count += 1
257
 
                finally:
258
 
                    pb.finished()
259
 
            # fixup inventory if needed: 
260
 
            # this is expensive because we have no inverse index to current ghosts.
261
 
            # but on local disk its a few seconds and sftp push is already insane.
262
 
            # so we just-do-it.
263
 
            # FIXME: repository should inform if this is needed.
264
 
            self.to_repository.reconcile()
265
 
        finally:
266
 
            rev_pb.finished()
267
 
    
268
 
 
269
 
class KnitRepoFetcher(RepoFetcher):
270
 
    """This is a knit format repository specific fetcher.
271
 
 
272
 
    This differs from the GenericRepoFetcher by not doing a 
273
 
    reconciliation after copying, and using knit joining to
274
 
    copy revision texts.
275
 
    """
276
 
 
277
 
    def _fetch_revision_texts(self, revs):
278
 
        # may need to be a InterRevisionStore call here.
279
 
        from_transaction = self.from_repository.get_transaction()
280
 
        to_transaction = self.to_repository.get_transaction()
281
 
        to_sf = self.to_repository._revision_store.get_signature_file(
282
 
            to_transaction)
283
 
        from_sf = self.from_repository._revision_store.get_signature_file(
284
 
            from_transaction)
285
 
        to_sf.join(from_sf, version_ids=revs, ignore_missing=True)
286
 
        to_rf = self.to_repository._revision_store.get_revision_file(
287
 
            to_transaction)
288
 
        from_rf = self.from_repository._revision_store.get_revision_file(
289
 
            from_transaction)
290
 
        to_rf.join(from_rf, version_ids=revs)
 
162
        return self.to_repository.search_missing_revision_ids(
 
163
            self.from_repository, self._last_revision,
 
164
            find_ghosts=self.find_ghosts)
 
165
 
 
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
 
169
        # inventories.
 
170
        parent_maps = self.to_repository.get_parent_map(revision_ids)
 
171
        parents = set()
 
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)
 
176
        return missing_keys
291
177
 
292
178
 
293
179
class Inter1and2Helper(object):
294
180
    """Helper for operations that convert data from model 1 and 2
295
 
    
 
181
 
296
182
    This is for use by fetchers and converters.
297
183
    """
298
184
 
299
 
    def __init__(self, source, target):
 
185
    def __init__(self, source):
300
186
        """Constructor.
301
187
 
302
188
        :param source: The repository data comes from
303
 
        :param target: The repository data goes to
304
189
        """
305
190
        self.source = source
306
 
        self.target = target
307
191
 
308
192
    def iter_rev_trees(self, revs):
309
193
        """Iterate through RevisionTrees efficiently.
315
199
 
316
200
        :param revs: A list of revision ids
317
201
        """
 
202
        # In case that revs is not a list.
 
203
        revs = list(revs)
318
204
        while revs:
319
205
            for tree in self.source.revision_trees(revs[:100]):
320
206
                if tree.inventory.revision_id is None:
322
208
                yield tree
323
209
            revs = revs[100:]
324
210
 
 
211
    def _find_root_ids(self, revs, parent_map, graph):
 
212
        revision_root = {}
 
213
        for tree in self.iter_rev_trees(revs):
 
214
            revision_id = tree.inventory.root.revision
 
215
            root_id = tree.get_root_id()
 
216
            revision_root[revision_id] = root_id
 
217
        # Find out which parents we don't already know root ids for
 
218
        parents = set()
 
219
        for revision_parents in parent_map.itervalues():
 
220
            parents.update(revision_parents)
 
221
        parents.difference_update(revision_root.keys() + [NULL_REVISION])
 
222
        # Limit to revisions present in the versionedfile
 
223
        parents = graph.get_parent_map(parents).keys()
 
224
        for tree in self.iter_rev_trees(parents):
 
225
            root_id = tree.get_root_id()
 
226
            revision_root[tree.get_revision_id()] = root_id
 
227
        return revision_root
 
228
 
325
229
    def generate_root_texts(self, revs):
326
230
        """Generate VersionedFiles for all root ids.
327
 
        
 
231
 
328
232
        :param revs: the revisions to include
329
233
        """
330
 
        inventory_weave = self.source.get_inventory_weave()
331
 
        parent_texts = {}
332
 
        versionedfile = {}
333
 
        to_store = self.target.weave_store
334
 
        for tree in self.iter_rev_trees(revs):
335
 
            revision_id = tree.inventory.root.revision
336
 
            root_id = tree.inventory.root.file_id
337
 
            parents = inventory_weave.get_parents(revision_id)
338
 
            if root_id not in versionedfile:
339
 
                versionedfile[root_id] = to_store.get_weave_or_empty(root_id, 
340
 
                    self.target.get_transaction())
341
 
            parent_texts[root_id] = versionedfile[root_id].add_lines(
342
 
                revision_id, parents, [], parent_texts)
343
 
 
344
 
    def regenerate_inventory(self, revs):
345
 
        """Generate a new inventory versionedfile in target, convertin data.
346
 
        
347
 
        The inventory is retrieved from the source, (deserializing it), and
348
 
        stored in the target (reserializing it in a different format).
349
 
        :param revs: The revisions to include
350
 
        """
351
 
        inventory_weave = self.source.get_inventory_weave()
352
 
        for tree in self.iter_rev_trees(revs):
353
 
            parents = inventory_weave.get_parents(tree.get_revision_id())
354
 
            self.target.add_inventory(tree.get_revision_id(), tree.inventory,
355
 
                                      parents)
356
 
 
357
 
 
358
 
class Model1toKnit2Fetcher(GenericRepoFetcher):
359
 
    """Fetch from a Model1 repository into a Knit2 repository
360
 
    """
361
 
    def __init__(self, to_repository, from_repository, last_revision=None, 
362
 
                 pb=None):
363
 
        self.helper = Inter1and2Helper(from_repository, to_repository)
364
 
        GenericRepoFetcher.__init__(self, to_repository, from_repository,
365
 
                                    last_revision, pb)
366
 
 
367
 
    def _fetch_weave_texts(self, revs):
368
 
        GenericRepoFetcher._fetch_weave_texts(self, revs)
369
 
        # Now generate a weave for the tree root
370
 
        self.helper.generate_root_texts(revs)
371
 
 
372
 
    def _fetch_inventory_weave(self, revs):
373
 
        self.helper.regenerate_inventory(revs)
374
 
 
375
 
 
376
 
class Knit1to2Fetcher(KnitRepoFetcher):
377
 
    """Fetch from a Knit1 repository into a Knit2 repository"""
378
 
 
379
 
    def __init__(self, to_repository, from_repository, last_revision=None, 
380
 
                 pb=None):
381
 
        self.helper = Inter1and2Helper(from_repository, to_repository)
382
 
        KnitRepoFetcher.__init__(self, to_repository, from_repository,
383
 
                                 last_revision, pb)
384
 
 
385
 
    def _fetch_weave_texts(self, revs):
386
 
        KnitRepoFetcher._fetch_weave_texts(self, revs)
387
 
        # Now generate a weave for the tree root
388
 
        self.helper.generate_root_texts(revs)
389
 
 
390
 
    def _fetch_inventory_weave(self, revs):
391
 
        self.helper.regenerate_inventory(revs)
392
 
        
393
 
 
394
 
class Fetcher(object):
395
 
    """Backwards compatibility glue for branch.fetch()."""
396
 
 
397
 
    @deprecated_method(zero_eight)
398
 
    def __init__(self, to_branch, from_branch, last_revision=None, pb=None):
399
 
        """Please see branch.fetch()."""
400
 
        to_branch.fetch(from_branch, last_revision, pb)
 
234
        graph = self.source.get_graph()
 
235
        parent_map = graph.get_parent_map(revs)
 
236
        rev_order = tsort.topo_sort(parent_map)
 
237
        rev_id_to_root_id = self._find_root_ids(revs, parent_map, graph)
 
238
        root_id_order = [(rev_id_to_root_id[rev_id], rev_id) for rev_id in
 
239
            rev_order]
 
240
        # Guaranteed stable, this groups all the file id operations together
 
241
        # retaining topological order within the revisions of a file id.
 
242
        # File id splits and joins would invalidate this, but they don't exist
 
243
        # yet, and are unlikely to in non-rich-root environments anyway.
 
244
        root_id_order.sort(key=operator.itemgetter(0))
 
245
        # Create a record stream containing the roots to create.
 
246
        if len(revs) > 100:
 
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)
 
250
        new_roots_stream = _new_root_data_stream(
 
251
            root_id_order, rev_id_to_root_id, parent_map, self.source, graph)
 
252
        return [('texts', new_roots_stream)]
 
253
 
 
254
 
 
255
def _get_rich_root_heads_graph(source_repo, revision_ids):
 
256
    """Get a Graph object suitable for asking heads() for new rich roots."""
 
257
    return 
 
258
 
 
259
 
 
260
def _new_root_data_stream(
 
261
    root_keys_to_create, rev_id_to_root_id_map, parent_map, repo, graph=None):
 
262
    """Generate a texts substream of synthesised root entries.
 
263
 
 
264
    Used in fetches that do rich-root upgrades.
 
265
    
 
266
    :param root_keys_to_create: iterable of (root_id, rev_id) pairs describing
 
267
        the root entries to create.
 
268
    :param rev_id_to_root_id_map: dict of known rev_id -> root_id mappings for
 
269
        calculating the parents.  If a parent rev_id is not found here then it
 
270
        will be recalculated.
 
271
    :param parent_map: a parent map for all the revisions in
 
272
        root_keys_to_create.
 
273
    :param graph: a graph to use instead of repo.get_graph().
 
274
    """
 
275
    for root_key in root_keys_to_create:
 
276
        root_id, rev_id = root_key
 
277
        parent_keys = _parent_keys_for_root_version(
 
278
            root_id, rev_id, rev_id_to_root_id_map, parent_map, repo, graph)
 
279
        yield versionedfile.FulltextContentFactory(
 
280
            root_key, parent_keys, None, '')
 
281
 
 
282
 
 
283
def _parent_keys_for_root_version(
 
284
    root_id, rev_id, rev_id_to_root_id_map, parent_map, repo, graph=None):
 
285
    """Get the parent keys for a given root id.
 
286
    
 
287
    A helper function for _new_root_data_stream.
 
288
    """
 
289
    # Include direct parents of the revision, but only if they used the same
 
290
    # root_id and are heads.
 
291
    rev_parents = parent_map[rev_id]
 
292
    parent_ids = []
 
293
    for parent_id in rev_parents:
 
294
        if parent_id == NULL_REVISION:
 
295
            continue
 
296
        if parent_id not in rev_id_to_root_id_map:
 
297
            # We probably didn't read this revision, go spend the extra effort
 
298
            # to actually check
 
299
            try:
 
300
                tree = repo.revision_tree(parent_id)
 
301
            except errors.NoSuchRevision:
 
302
                # Ghost, fill out rev_id_to_root_id in case we encounter this
 
303
                # again.
 
304
                # But set parent_root_id to None since we don't really know
 
305
                parent_root_id = None
 
306
            else:
 
307
                parent_root_id = tree.get_root_id()
 
308
            rev_id_to_root_id_map[parent_id] = None
 
309
            # XXX: why not:
 
310
            #   rev_id_to_root_id_map[parent_id] = parent_root_id
 
311
            # memory consumption maybe?
 
312
        else:
 
313
            parent_root_id = rev_id_to_root_id_map[parent_id]
 
314
        if root_id == parent_root_id:
 
315
            # With stacking we _might_ want to refer to a non-local revision,
 
316
            # but this code path only applies when we have the full content
 
317
            # available, so ghosts really are ghosts, not just the edge of
 
318
            # local data.
 
319
            parent_ids.append(parent_id)
 
320
        else:
 
321
            # root_id may be in the parent anyway.
 
322
            try:
 
323
                tree = repo.revision_tree(parent_id)
 
324
            except errors.NoSuchRevision:
 
325
                # ghost, can't refer to it.
 
326
                pass
 
327
            else:
 
328
                try:
 
329
                    parent_ids.append(tree.inventory[root_id].revision)
 
330
                except errors.NoSuchId:
 
331
                    # not in the tree
 
332
                    pass
 
333
    # Drop non-head parents
 
334
    if graph is None:
 
335
        graph = repo.get_graph()
 
336
    heads = graph.heads(parent_ids)
 
337
    selected_ids = []
 
338
    for parent_id in parent_ids:
 
339
        if parent_id in heads and parent_id not in selected_ids:
 
340
            selected_ids.append(parent_id)
 
341
    parent_keys = [(root_id, parent_id) for parent_id in selected_ids]
 
342
    return parent_keys