34
34
from bzrlib.trace import mutter
35
35
from bzrlib.errors import BzrCheckError, NoSuchRevision
36
36
from bzrlib.inventory import ROOT_ID
37
from bzrlib.branch import gen_root_id
39
40
class Check(object):
50
51
self.missing_revision_cnt = 0
51
52
# maps (file-id, version) -> sha1
52
53
self.checked_texts = {}
53
self.checked_weaves = {}
56
56
self.branch.lock_read()
57
self.progress = bzrlib.ui.ui_factory.progress_bar()
59
self.progress.update('retrieving inventory', 0, 0)
60
# do not put in init, as it should be done with progess,
61
# and inside the lock.
62
self.inventory_weave = self.branch._get_inventory_weave()
63
58
self.history = self.branch.revision_history()
64
59
if not len(self.history):
65
60
# nothing to see here
62
self.planned_revisions = self.branch.get_ancestry(self.history[-1])
63
self.planned_revisions.remove(None)
66
self.progress = bzrlib.ui.ui_factory.progress_bar()
70
67
while revno < len(self.planned_revisions):
71
68
rev_id = self.planned_revisions[revno]
72
69
self.progress.update('checking revision', revno,
73
70
len(self.planned_revisions))
75
72
self.check_one_rev(rev_id)
78
75
self.branch.unlock()
80
def plan_revisions(self):
81
if not self.branch.revision_store.listable():
82
self.planned_revisions = self.branch.get_ancestry(self.history[-1])
83
self.planned_revisions.remove(None)
84
# FIXME progress bars should support this more nicely.
86
print ("Checking reachable history -"
87
" for a complete check use a local branch.")
90
self.planned_revisions = set(self.branch.revision_store)
91
inventoried = set(self.inventory_weave.names())
92
awol = self.planned_revisions - inventoried
94
raise BzrCheckError('Stored revisions missing from inventory'
95
'{%s}' % ','.join([f for f in awol]))
96
self.planned_revisions = list(self.planned_revisions)
98
77
def report_results(self, verbose):
99
78
note('checked branch %s format %d',
103
82
note('%6d revisions', self.checked_rev_cnt)
104
83
note('%6d unique file texts', self.checked_text_cnt)
105
84
note('%6d repeated file texts', self.repeated_text_cnt)
106
note('%6d weaves', len(self.checked_weaves))
107
85
if self.missing_inventory_sha_cnt:
108
86
note('%6d revisions are missing inventory_sha1',
109
87
self.missing_inventory_sha_cnt)
132
110
last_rev_id - the previous one on the mainline, if any.
135
# mutter(' revision {%s}', rev_id)
113
# mutter(' revision {%s}' % rev_id)
136
114
branch = self.branch
138
116
rev_history_position = self.history.index(rev_id)
165
143
missing_links = self.missing_parent_links.get(parent, [])
166
144
missing_links.append(rev_id)
167
145
self.missing_parent_links[parent] = missing_links
168
# list based so somewhat slow,
169
# TODO have a planned_revisions list and set.
146
# list based so slow, TODO have a planned_revisions list and set.
170
147
if self.branch.has_revision(parent):
171
148
missing_ancestry = self.branch.get_ancestry(parent)
172
149
for missing in missing_ancestry:
187
164
' value in revision {%s}' % rev_id)
189
166
missing_inventory_sha_cnt += 1
190
mutter("no inventory_sha1 on revision {%s}", rev_id)
167
mutter("no inventory_sha1 on revision {%s}" % rev_id)
191
168
self._check_revision_tree(rev_id)
192
169
self.checked_rev_cnt += 1
194
def check_weaves(self):
195
"""Check all the weaves we can get our hands on.
199
if self.branch.weave_store.listable():
200
weave_ids = list(self.branch.weave_store)
201
n_weaves = len(weave_ids)
202
self.progress.update('checking weave', 0, n_weaves)
203
self.inventory_weave.check(progress_bar=self.progress)
204
for i, weave_id in enumerate(weave_ids):
205
self.progress.update('checking weave', i, n_weaves)
206
w = self.branch.weave_store.get_weave(weave_id,
207
self.branch.get_transaction())
208
# No progress here, because it looks ugly.
210
self.checked_weaves[weave_id] = True
212
171
def _check_revision_tree(self, rev_id):
213
172
tree = self.branch.revision_tree(rev_id)
214
173
inv = tree.inventory