~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/reconcile.py

  • Committer: Martin Pool
  • Date: 2008-06-17 01:13:49 UTC
  • mto: (3495.3.3 trivial)
  • mto: This revision was merged to the branch mainline in revision 3501.
  • Revision ID: mbp@sourcefrog.net-20080617011349-jsqy1id9wgk2pqzs
Show short error for missing libraries

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2006-2010 Canonical Ltd
 
1
# Copyright (C) 2005, 2006 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
17
"""Reconcilers are able to fix some potential data errors in a branch."""
18
18
 
27
27
 
28
28
 
29
29
from bzrlib import (
30
 
    cleanup,
31
30
    errors,
32
31
    ui,
 
32
    repository,
 
33
    repofmt,
33
34
    )
34
 
from bzrlib.trace import mutter
35
 
from bzrlib.tsort import topo_sort
36
 
from bzrlib.versionedfile import AdapterFactory, FulltextContentFactory
 
35
from bzrlib.trace import mutter, note
 
36
from bzrlib.tsort import TopoSorter
37
37
 
38
38
 
39
39
def reconcile(dir, other=None):
61
61
 
62
62
    def reconcile(self):
63
63
        """Perform reconciliation.
64
 
 
 
64
        
65
65
        After reconciliation the following attributes document found issues:
66
66
        inconsistent_parents: The number of revisions in the repository whose
67
67
                              ancestry was being reported incorrectly.
89
89
            # Nothing to check here
90
90
            self.fixed_branch_history = None
91
91
            return
92
 
        ui.ui_factory.note('Reconciling branch %s' % self.branch.base)
 
92
        self.pb.note('Reconciling branch %s',
 
93
                     self.branch.base)
93
94
        branch_reconciler = self.branch.reconcile(thorough=True)
94
95
        self.fixed_branch_history = branch_reconciler.fixed_history
95
96
 
96
97
    def _reconcile_repository(self):
97
98
        self.repo = self.bzrdir.find_repository()
98
 
        ui.ui_factory.note('Reconciling repository %s' %
99
 
            self.repo.user_url)
 
99
        self.pb.note('Reconciling repository %s',
 
100
                     self.repo.bzrdir.root_transport.base)
100
101
        self.pb.update("Reconciling repository", 0, 1)
101
102
        repo_reconciler = self.repo.reconcile(thorough=True)
102
103
        self.inconsistent_parents = repo_reconciler.inconsistent_parents
103
104
        self.garbage_inventories = repo_reconciler.garbage_inventories
104
105
        if repo_reconciler.aborted:
105
 
            ui.ui_factory.note(
 
106
            self.pb.note(
106
107
                'Reconcile aborted: revision index has inconsistent parents.')
107
 
            ui.ui_factory.note(
 
108
            self.pb.note(
108
109
                'Run "bzr check" for more details.')
109
110
        else:
110
 
            ui.ui_factory.note('Reconciliation complete.')
 
111
            self.pb.note('Reconciliation complete.')
111
112
 
112
113
 
113
114
class BranchReconciler(object):
119
120
        self.branch = a_branch
120
121
 
121
122
    def reconcile(self):
122
 
        operation = cleanup.OperationWithCleanups(self._reconcile)
123
 
        self.add_cleanup = operation.add_cleanup
124
 
        operation.run_simple()
125
 
 
126
 
    def _reconcile(self):
127
123
        self.branch.lock_write()
128
 
        self.add_cleanup(self.branch.unlock)
129
 
        self.pb = ui.ui_factory.nested_progress_bar()
130
 
        self.add_cleanup(self.pb.finished)
131
 
        self._reconcile_steps()
 
124
        try:
 
125
            self.pb = ui.ui_factory.nested_progress_bar()
 
126
            try:
 
127
                self._reconcile_steps()
 
128
            finally:
 
129
                self.pb.finished()
 
130
        finally:
 
131
            self.branch.unlock()
132
132
 
133
133
    def _reconcile_steps(self):
134
134
        self._reconcile_revision_history()
136
136
    def _reconcile_revision_history(self):
137
137
        repo = self.branch.repository
138
138
        last_revno, last_revision_id = self.branch.last_revision_info()
139
 
        real_history = []
140
 
        try:
141
 
            for revid in repo.iter_reverse_revision_history(
142
 
                    last_revision_id):
143
 
                real_history.append(revid)
144
 
        except errors.RevisionNotPresent:
145
 
            pass # Hit a ghost left hand parent
 
139
        real_history = list(repo.iter_reverse_revision_history(
 
140
                                last_revision_id))
146
141
        real_history.reverse()
147
142
        if last_revno != len(real_history):
148
143
            self.fixed_history = True
150
145
            # set_revision_history, as this will regenerate it again.
151
146
            # Not really worth a whole BranchReconciler class just for this,
152
147
            # though.
153
 
            ui.ui_factory.note('Fixing last revision info %s => %s' % (
154
 
                 last_revno, len(real_history)))
 
148
            self.pb.note('Fixing last revision info %s => %s',
 
149
                         last_revno, len(real_history))
155
150
            self.branch.set_last_revision_info(len(real_history),
156
151
                                               last_revision_id)
157
152
        else:
158
153
            self.fixed_history = False
159
 
            ui.ui_factory.note('revision_history ok.')
 
154
            self.pb.note('revision_history ok.')
160
155
 
161
156
 
162
157
class RepoReconciler(object):
163
158
    """Reconciler that reconciles a repository.
164
159
 
165
160
    The goal of repository reconciliation is to make any derived data
166
 
    consistent with the core data committed by a user. This can involve
 
161
    consistent with the core data committed by a user. This can involve 
167
162
    reindexing, or removing unreferenced data if that can interfere with
168
163
    queries in a given repository.
169
164
 
185
180
 
186
181
    def reconcile(self):
187
182
        """Perform reconciliation.
188
 
 
 
183
        
189
184
        After reconciliation the following attributes document found issues:
190
185
        inconsistent_parents: The number of revisions in the repository whose
191
186
                              ancestry was being reported incorrectly.
192
187
        garbage_inventories: The number of inventory objects without revisions
193
188
                             that were garbage collected.
194
189
        """
195
 
        operation = cleanup.OperationWithCleanups(self._reconcile)
196
 
        self.add_cleanup = operation.add_cleanup
197
 
        operation.run_simple()
198
 
 
199
 
    def _reconcile(self):
200
190
        self.repo.lock_write()
201
 
        self.add_cleanup(self.repo.unlock)
202
 
        self.pb = ui.ui_factory.nested_progress_bar()
203
 
        self.add_cleanup(self.pb.finished)
204
 
        self._reconcile_steps()
 
191
        try:
 
192
            self.pb = ui.ui_factory.nested_progress_bar()
 
193
            try:
 
194
                self._reconcile_steps()
 
195
            finally:
 
196
                self.pb.finished()
 
197
        finally:
 
198
            self.repo.unlock()
205
199
 
206
200
    def _reconcile_steps(self):
207
201
        """Perform the steps to reconcile this repository."""
209
203
 
210
204
    def _reweave_inventory(self):
211
205
        """Regenerate the inventory weave for the repository from scratch.
212
 
 
213
 
        This is a smart function: it will only do the reweave if doing it
 
206
        
 
207
        This is a smart function: it will only do the reweave if doing it 
214
208
        will correct data issues. The self.thorough flag controls whether
215
209
        only data-loss causing issues (!self.thorough) or all issues
216
210
        (self.thorough) are treated as requiring the reweave.
218
212
        # local because needing to know about WeaveFile is a wart we want to hide
219
213
        from bzrlib.weave import WeaveFile, Weave
220
214
        transaction = self.repo.get_transaction()
221
 
        self.pb.update('Reading inventory data')
222
 
        self.inventory = self.repo.inventories
223
 
        self.revisions = self.repo.revisions
 
215
        self.pb.update('Reading inventory data.')
 
216
        self.inventory = self.repo.get_inventory_weave()
224
217
        # the total set of revisions to process
225
 
        self.pending = set([key[-1] for key in self.revisions.keys()])
 
218
        self.pending = set([rev_id for rev_id in self.repo._revision_store.all_revision_ids(transaction)])
226
219
 
227
220
        # mapping from revision_id to parents
228
221
        self._rev_graph = {}
234
227
            # put a revision into the graph.
235
228
            self._graph_revision(rev_id)
236
229
        self._check_garbage_inventories()
237
 
        # if there are no inconsistent_parents and
 
230
        # if there are no inconsistent_parents and 
238
231
        # (no garbage inventories or we are not doing a thorough check)
239
 
        if (not self.inconsistent_parents and
 
232
        if (not self.inconsistent_parents and 
240
233
            (not self.garbage_inventories or not self.thorough)):
241
 
            ui.ui_factory.note('Inventory ok.')
 
234
            self.pb.note('Inventory ok.')
242
235
            return
243
 
        self.pb.update('Backing up inventory', 0, 0)
244
 
        self.repo._backup_inventory()
245
 
        ui.ui_factory.note('Backup inventory created.')
246
 
        new_inventories = self.repo._temp_inventories()
 
236
        self.pb.update('Backing up inventory...', 0, 0)
 
237
        self.repo.control_weaves.copy(self.inventory, 'inventory.backup', self.repo.get_transaction())
 
238
        self.pb.note('Backup Inventory created.')
 
239
        # asking for '' should never return a non-empty weave
 
240
        new_inventory_vf = self.repo.control_weaves.get_empty('inventory.new',
 
241
            self.repo.get_transaction())
247
242
 
248
243
        # we have topological order of revisions and non ghost parents ready.
249
244
        self._setup_steps(len(self._rev_graph))
250
 
        revision_keys = [(rev_id,) for rev_id in topo_sort(self._rev_graph)]
251
 
        stream = self._change_inv_parents(
252
 
            self.inventory.get_record_stream(revision_keys, 'unordered', True),
253
 
            self._new_inv_parents,
254
 
            set(revision_keys))
255
 
        new_inventories.insert_record_stream(stream)
256
 
        # if this worked, the set of new_inventories.keys should equal
 
245
        for rev_id in TopoSorter(self._rev_graph.items()).iter_topo_order():
 
246
            parents = self._rev_graph[rev_id]
 
247
            # double check this really is in topological order.
 
248
            unavailable = [p for p in parents if p not in new_inventory_vf]
 
249
            if unavailable:
 
250
                raise AssertionError('unavailable parents: %r'
 
251
                    % unavailable)
 
252
            # this entry has all the non ghost parents in the inventory
 
253
            # file already.
 
254
            self._reweave_step('adding inventories')
 
255
            if isinstance(new_inventory_vf, WeaveFile):
 
256
                # It's really a WeaveFile, but we call straight into the
 
257
                # Weave's add method to disable the auto-write-out behaviour.
 
258
                # This is done to avoid a revision_count * time-to-write additional overhead on 
 
259
                # reconcile.
 
260
                new_inventory_vf._check_write_ok()
 
261
                Weave._add_lines(new_inventory_vf, rev_id, parents,
 
262
                    self.inventory.get_lines(rev_id), None, None, None, False, True)
 
263
            else:
 
264
                new_inventory_vf.add_lines(rev_id, parents, self.inventory.get_lines(rev_id))
 
265
 
 
266
        if isinstance(new_inventory_vf, WeaveFile):
 
267
            new_inventory_vf._save()
 
268
        # if this worked, the set of new_inventory_vf.names should equal
257
269
        # self.pending
258
 
        if not (set(new_inventories.keys()) ==
259
 
            set([(revid,) for revid in self.pending])):
 
270
        if not (set(new_inventory_vf.versions()) == self.pending):
260
271
            raise AssertionError()
261
272
        self.pb.update('Writing weave')
262
 
        self.repo._activate_new_inventory()
 
273
        self.repo.control_weaves.copy(new_inventory_vf, 'inventory', self.repo.get_transaction())
 
274
        self.repo.control_weaves.delete('inventory.new', self.repo.get_transaction())
263
275
        self.inventory = None
264
 
        ui.ui_factory.note('Inventory regenerated.')
265
 
 
266
 
    def _new_inv_parents(self, revision_key):
267
 
        """Lookup ghost-filtered parents for revision_key."""
268
 
        # Use the filtered ghostless parents list:
269
 
        return tuple([(revid,) for revid in self._rev_graph[revision_key[-1]]])
270
 
 
271
 
    def _change_inv_parents(self, stream, get_parents, all_revision_keys):
272
 
        """Adapt a record stream to reconcile the parents."""
273
 
        for record in stream:
274
 
            wanted_parents = get_parents(record.key)
275
 
            if wanted_parents and wanted_parents[0] not in all_revision_keys:
276
 
                # The check for the left most parent only handles knit
277
 
                # compressors, but this code only applies to knit and weave
278
 
                # repositories anyway.
279
 
                bytes = record.get_bytes_as('fulltext')
280
 
                yield FulltextContentFactory(record.key, wanted_parents, record.sha1, bytes)
281
 
            else:
282
 
                adapted_record = AdapterFactory(record.key, wanted_parents, record)
283
 
                yield adapted_record
284
 
            self._reweave_step('adding inventories')
 
276
        self.pb.note('Inventory regenerated.')
285
277
 
286
278
    def _setup_steps(self, new_total):
287
279
        """Setup the markers we need to control the progress bar."""
301
293
            else:
302
294
                mutter('found ghost %s', parent)
303
295
        self._rev_graph[rev_id] = parents
 
296
        if self._parents_are_inconsistent(rev_id, parents):
 
297
            self.inconsistent_parents += 1
 
298
            mutter('Inconsistent inventory parents: id {%s} '
 
299
                   'inventory claims %r, '
 
300
                   'available parents are %r, '
 
301
                   'unavailable parents are %r',
 
302
                   rev_id,
 
303
                   set(self.inventory.get_parent_map([rev_id])[rev_id]),
 
304
                   set(parents),
 
305
                   set(rev.parent_ids).difference(set(parents)))
 
306
 
 
307
    def _parents_are_inconsistent(self, rev_id, parents):
 
308
        """Return True if the parents list of rev_id does not match the weave.
 
309
 
 
310
        This detects inconsistencies based on the self.thorough value:
 
311
        if thorough is on, the first parent value is checked as well as ghost
 
312
        differences.
 
313
        Otherwise only the ghost differences are evaluated.
 
314
        """
 
315
        weave_parents = self.inventory.get_parent_map([rev_id])[rev_id]
 
316
        weave_missing_old_ghosts = set(weave_parents) != set(parents)
 
317
        first_parent_is_wrong = (
 
318
            len(weave_parents) and len(parents) and
 
319
            parents[0] != weave_parents[0])
 
320
        if self.thorough:
 
321
            return weave_missing_old_ghosts or first_parent_is_wrong
 
322
        else:
 
323
            return weave_missing_old_ghosts
304
324
 
305
325
    def _check_garbage_inventories(self):
306
326
        """Check for garbage inventories which we cannot trust
310
330
        """
311
331
        if not self.thorough:
312
332
            return
313
 
        inventories = set(self.inventory.keys())
314
 
        revisions = set(self.revisions.keys())
 
333
        inventories = set(self.inventory.versions())
 
334
        revisions = set(self._rev_graph.keys())
315
335
        garbage = inventories.difference(revisions)
316
336
        self.garbage_inventories = len(garbage)
317
 
        for revision_key in garbage:
318
 
            mutter('Garbage inventory {%s} found.', revision_key[-1])
 
337
        for revision_id in garbage:
 
338
            mutter('Garbage inventory {%s} found.', revision_id)
319
339
 
320
340
    def _parent_is_available(self, parent):
321
341
        """True if parent is a fully available revision
323
343
        A fully available revision has a inventory and a revision object in the
324
344
        repository.
325
345
        """
326
 
        if parent in self._rev_graph:
327
 
            return True
328
 
        inv_present = (1 == len(self.inventory.get_parent_map([(parent,)])))
329
 
        return (inv_present and self.repo.has_revision(parent))
 
346
        return (parent in self._rev_graph or 
 
347
                (parent in self.inventory and self.repo.has_revision(parent)))
330
348
 
331
349
    def _reweave_step(self, message):
332
350
        """Mark a single step of regeneration complete."""
355
373
    def _load_indexes(self):
356
374
        """Load indexes for the reconciliation."""
357
375
        self.transaction = self.repo.get_transaction()
358
 
        self.pb.update('Reading indexes', 0, 2)
359
 
        self.inventory = self.repo.inventories
360
 
        self.pb.update('Reading indexes', 1, 2)
 
376
        self.pb.update('Reading indexes.', 0, 2)
 
377
        self.inventory = self.repo.get_inventory_weave()
 
378
        self.pb.update('Reading indexes.', 1, 2)
361
379
        self.repo._check_for_inconsistent_revision_parents()
362
 
        self.revisions = self.repo.revisions
363
 
        self.pb.update('Reading indexes', 2, 2)
 
380
        self.revisions = self.repo._revision_store.get_revision_file(self.transaction)
 
381
        self.pb.update('Reading indexes.', 2, 2)
364
382
 
365
383
    def _gc_inventory(self):
366
384
        """Remove inventories that are not referenced from the revision store."""
367
 
        self.pb.update('Checking unused inventories', 0, 1)
 
385
        self.pb.update('Checking unused inventories.', 0, 1)
368
386
        self._check_garbage_inventories()
369
 
        self.pb.update('Checking unused inventories', 1, 3)
 
387
        self.pb.update('Checking unused inventories.', 1, 3)
370
388
        if not self.garbage_inventories:
371
 
            ui.ui_factory.note('Inventory ok.')
 
389
            self.pb.note('Inventory ok.')
372
390
            return
373
 
        self.pb.update('Backing up inventory', 0, 0)
374
 
        self.repo._backup_inventory()
375
 
        ui.ui_factory.note('Backup Inventory created')
 
391
        self.pb.update('Backing up inventory...', 0, 0)
 
392
        self.repo.control_weaves.copy(self.inventory, 'inventory.backup', self.transaction)
 
393
        self.pb.note('Backup Inventory created.')
376
394
        # asking for '' should never return a non-empty weave
377
 
        new_inventories = self.repo._temp_inventories()
 
395
        new_inventory_vf = self.repo.control_weaves.get_empty('inventory.new',
 
396
            self.transaction)
 
397
 
378
398
        # we have topological order of revisions and non ghost parents ready.
379
 
        graph = self.revisions.get_parent_map(self.revisions.keys())
380
 
        revision_keys = topo_sort(graph)
381
 
        revision_ids = [key[-1] for key in revision_keys]
382
 
        self._setup_steps(len(revision_keys))
383
 
        stream = self._change_inv_parents(
384
 
            self.inventory.get_record_stream(revision_keys, 'unordered', True),
385
 
            graph.__getitem__,
386
 
            set(revision_keys))
387
 
        new_inventories.insert_record_stream(stream)
 
399
        self._setup_steps(len(self.revisions))
 
400
        revision_ids = self.revisions.versions()
 
401
        graph = self.revisions.get_parent_map(revision_ids)
 
402
        for rev_id in TopoSorter(graph.items()).iter_topo_order():
 
403
            parents = graph[rev_id]
 
404
            # double check this really is in topological order, ignoring existing ghosts.
 
405
            unavailable = [p for p in parents if p not in new_inventory_vf and
 
406
                p in self.revisions]
 
407
            if unavailable:
 
408
                raise AssertionError(
 
409
                    'unavailable parents: %r' % (unavailable,))
 
410
            # this entry has all the non ghost parents in the inventory
 
411
            # file already.
 
412
            self._reweave_step('adding inventories')
 
413
            # ugly but needed, weaves are just way tooooo slow else.
 
414
            new_inventory_vf.add_lines_with_ghosts(rev_id, parents,
 
415
                self.inventory.get_lines(rev_id))
 
416
 
388
417
        # if this worked, the set of new_inventory_vf.names should equal
389
 
        # the revisionds list
390
 
        if not(set(new_inventories.keys()) == set(revision_keys)):
 
418
        # self.pending
 
419
        if not(set(new_inventory_vf.versions()) == set(self.revisions.versions())):
391
420
            raise AssertionError()
392
421
        self.pb.update('Writing weave')
393
 
        self.repo._activate_new_inventory()
 
422
        self.repo.control_weaves.copy(new_inventory_vf, 'inventory', self.transaction)
 
423
        self.repo.control_weaves.delete('inventory.new', self.transaction)
394
424
        self.inventory = None
395
 
        ui.ui_factory.note('Inventory regenerated.')
 
425
        self.pb.note('Inventory regenerated.')
 
426
 
 
427
    def _check_garbage_inventories(self):
 
428
        """Check for garbage inventories which we cannot trust
 
429
 
 
430
        We cant trust them because their pre-requisite file data may not
 
431
        be present - all we know is that their revision was not installed.
 
432
        """
 
433
        inventories = set(self.inventory.versions())
 
434
        revisions = set(self.revisions.versions())
 
435
        garbage = inventories.difference(revisions)
 
436
        self.garbage_inventories = len(garbage)
 
437
        for revision_id in garbage:
 
438
            mutter('Garbage inventory {%s} found.', revision_id)
396
439
 
397
440
    def _fix_text_parents(self):
398
441
        """Fix bad versionedfile parent entries.
404
447
        parent lists, and replaces the versionedfile with a corrected version.
405
448
        """
406
449
        transaction = self.repo.get_transaction()
407
 
        versions = [key[-1] for key in self.revisions.keys()]
 
450
        versions = self.revisions.versions()
408
451
        mutter('Prepopulating revision text cache with %d revisions',
409
452
                len(versions))
410
453
        vf_checker = self.repo._get_versioned_file_checker()
411
 
        bad_parents, unused_versions = vf_checker.check_file_version_parents(
412
 
            self.repo.texts, self.pb)
413
 
        text_index = vf_checker.text_index
414
 
        per_id_bad_parents = {}
415
 
        for key in unused_versions:
416
 
            # Ensure that every file with unused versions gets rewritten.
417
 
            # NB: This is really not needed, reconcile != pack.
418
 
            per_id_bad_parents[key[0]] = {}
419
 
        # Generate per-knit/weave data.
420
 
        for key, details in bad_parents.iteritems():
421
 
            file_id = key[0]
422
 
            rev_id = key[1]
423
 
            knit_parents = tuple([parent[-1] for parent in details[0]])
424
 
            correct_parents = tuple([parent[-1] for parent in details[1]])
425
 
            file_details = per_id_bad_parents.setdefault(file_id, {})
426
 
            file_details[rev_id] = (knit_parents, correct_parents)
427
 
        file_id_versions = {}
428
 
        for text_key in text_index:
429
 
            versions_list = file_id_versions.setdefault(text_key[0], [])
430
 
            versions_list.append(text_key[1])
431
 
        # Do the reconcile of individual weaves.
432
 
        for num, file_id in enumerate(per_id_bad_parents):
 
454
        # List all weaves before altering, to avoid race conditions when we
 
455
        # delete unused weaves.
 
456
        weaves = list(enumerate(self.repo.weave_store))
 
457
        for num, file_id in weaves:
433
458
            self.pb.update('Fixing text parents', num,
434
 
                           len(per_id_bad_parents))
435
 
            versions_with_bad_parents = per_id_bad_parents[file_id]
436
 
            id_unused_versions = set(key[-1] for key in unused_versions
437
 
                if key[0] == file_id)
438
 
            if file_id in file_id_versions:
439
 
                file_versions = file_id_versions[file_id]
440
 
            else:
441
 
                # This id was present in the disk store but is not referenced
442
 
                # by any revision at all.
443
 
                file_versions = []
444
 
            self._fix_text_parent(file_id, versions_with_bad_parents,
445
 
                 id_unused_versions, file_versions)
 
459
                           len(self.repo.weave_store))
 
460
            vf = self.repo.weave_store.get_weave(file_id, transaction)
 
461
            versions_with_bad_parents, unused_versions = \
 
462
                vf_checker.check_file_version_parents(vf, file_id)
 
463
            if (len(versions_with_bad_parents) == 0 and
 
464
                len(unused_versions) == 0):
 
465
                continue
 
466
            full_text_versions = set()
 
467
            self._fix_text_parent(file_id, vf, versions_with_bad_parents,
 
468
                full_text_versions, unused_versions)
446
469
 
447
 
    def _fix_text_parent(self, file_id, versions_with_bad_parents,
448
 
            unused_versions, all_versions):
 
470
    def _fix_text_parent(self, file_id, vf, versions_with_bad_parents,
 
471
            full_text_versions, unused_versions):
449
472
        """Fix bad versionedfile entries in a single versioned file."""
450
473
        mutter('fixing text parent: %r (%d versions)', file_id,
451
474
                len(versions_with_bad_parents))
452
 
        mutter('(%d are unused)', len(unused_versions))
453
 
        new_file_id = 'temp:%s' % file_id
 
475
        mutter('(%d need to be full texts, %d are unused)',
 
476
                len(full_text_versions), len(unused_versions))
 
477
        new_vf = self.repo.weave_store.get_empty('temp:%s' % file_id,
 
478
            self.transaction)
454
479
        new_parents = {}
455
 
        needed_keys = set()
456
 
        for version in all_versions:
 
480
        for version in vf.versions():
457
481
            if version in unused_versions:
458
482
                continue
459
483
            elif version in versions_with_bad_parents:
460
484
                parents = versions_with_bad_parents[version][1]
461
485
            else:
462
 
                pmap = self.repo.texts.get_parent_map([(file_id, version)])
463
 
                parents = [key[-1] for key in pmap[(file_id, version)]]
464
 
            new_parents[(new_file_id, version)] = [
465
 
                (new_file_id, parent) for parent in parents]
466
 
            needed_keys.add((file_id, version))
467
 
        def fix_parents(stream):
468
 
            for record in stream:
469
 
                bytes = record.get_bytes_as('fulltext')
470
 
                new_key = (new_file_id, record.key[-1])
471
 
                parents = new_parents[new_key]
472
 
                yield FulltextContentFactory(new_key, parents, record.sha1, bytes)
473
 
        stream = self.repo.texts.get_record_stream(needed_keys, 'topological', True)
474
 
        self.repo._remove_file_id(new_file_id)
475
 
        self.repo.texts.insert_record_stream(fix_parents(stream))
476
 
        self.repo._remove_file_id(file_id)
477
 
        if len(new_parents):
478
 
            self.repo._move_file_id(new_file_id, file_id)
 
486
                parents = vf.get_parent_map([version])[version]
 
487
            new_parents[version] = parents
 
488
        if not len(new_parents):
 
489
            # No used versions, remove the VF.
 
490
            self.repo.weave_store.delete(file_id, self.transaction)
 
491
            return
 
492
        for version in TopoSorter(new_parents.items()).iter_topo_order():
 
493
            lines = vf.get_lines(version)
 
494
            parents = new_parents[version]
 
495
            if parents and (parents[0] in full_text_versions):
 
496
                # Force this record to be a fulltext, not a delta.
 
497
                new_vf._add(version, lines, parents, False,
 
498
                    None, None, None, False)
 
499
            else:
 
500
                new_vf.add_lines(version, parents, lines)
 
501
        self.repo.weave_store.copy(new_vf, file_id, self.transaction)
 
502
        self.repo.weave_store.delete('temp:%s' % file_id, self.transaction)
479
503
 
480
504
 
481
505
class PackReconciler(RepoReconciler):
503
527
        collection = self.repo._pack_collection
504
528
        collection.ensure_loaded()
505
529
        collection.lock_names()
506
 
        self.add_cleanup(collection._unlock_names)
507
 
        packs = collection.all_packs()
508
 
        all_revisions = self.repo.all_revision_ids()
509
 
        total_inventories = len(list(
510
 
            collection.inventory_index.combined_index.iter_all_entries()))
511
 
        if len(all_revisions):
512
 
            new_pack =  self.repo._reconcile_pack(collection, packs,
513
 
                ".reconcile", all_revisions, self.pb)
514
 
            if new_pack is not None:
 
530
        try:
 
531
            packs = collection.all_packs()
 
532
            all_revisions = self.repo.all_revision_ids()
 
533
            total_inventories = len(list(
 
534
                collection.inventory_index.combined_index.iter_all_entries()))
 
535
            if len(all_revisions):
 
536
                self._packer = repofmt.pack_repo.ReconcilePacker(
 
537
                    collection, packs, ".reconcile", all_revisions)
 
538
                new_pack = self._packer.pack(pb=self.pb)
 
539
                if new_pack is not None:
 
540
                    self._discard_and_save(packs)
 
541
            else:
 
542
                # only make a new pack when there is data to copy.
515
543
                self._discard_and_save(packs)
516
 
        else:
517
 
            # only make a new pack when there is data to copy.
518
 
            self._discard_and_save(packs)
519
 
        self.garbage_inventories = total_inventories - len(list(
520
 
            collection.inventory_index.combined_index.iter_all_entries()))
 
544
            self.garbage_inventories = total_inventories - len(list(
 
545
                collection.inventory_index.combined_index.iter_all_entries()))
 
546
        finally:
 
547
            collection._unlock_names()
521
548
 
522
549
    def _discard_and_save(self, packs):
523
550
        """Discard some packs from the repository.