2 from zope.interface import implements
3 from allmydata.interfaces import ICheckResults, ICheckAndRepairResults, \
4 IDeepCheckResults, IDeepCheckAndRepairResults, IURI, IDisplayableServer
5 from allmydata.util import base32
8 implements(ICheckResults)
10 def __init__(self, uri, storage_index,
11 healthy, recoverable, needs_rebalancing,
12 count_shares_needed, count_shares_expected,
13 count_shares_good, count_good_share_hosts,
14 count_recoverable_versions, count_unrecoverable_versions,
15 servers_responding, sharemap,
16 count_wrong_shares, list_corrupt_shares, count_corrupt_shares,
17 list_incompatible_shares, count_incompatible_shares,
18 summary, report, share_problems, servermap):
19 assert IURI.providedBy(uri), uri
21 self._storage_index = storage_index
23 self._healthy = bool(healthy)
30 summary = "not healthy"
31 self._recoverable = recoverable
32 if not self._recoverable:
33 assert not self._healthy
34 self._needs_rebalancing_p = bool(needs_rebalancing)
36 self._count_shares_needed = count_shares_needed
37 self._count_shares_expected = count_shares_expected
38 self._count_shares_good = count_shares_good
39 self._count_good_share_hosts = count_good_share_hosts
40 self._count_recoverable_versions = count_recoverable_versions
41 self._count_unrecoverable_versions = count_unrecoverable_versions
42 for server in servers_responding:
43 assert IDisplayableServer.providedBy(server), server
44 self._servers_responding = servers_responding
45 for shnum, servers in sharemap.items():
46 for server in servers:
47 assert IDisplayableServer.providedBy(server), server
48 self._sharemap = sharemap
49 self._count_wrong_shares = count_wrong_shares
50 for (server, SI, shnum) in list_corrupt_shares:
51 assert IDisplayableServer.providedBy(server), server
52 self._list_corrupt_shares = list_corrupt_shares
53 self._count_corrupt_shares = count_corrupt_shares
54 for (server, SI, shnum) in list_incompatible_shares:
55 assert IDisplayableServer.providedBy(server), server
56 self._list_incompatible_shares = list_incompatible_shares
57 self._count_incompatible_shares = count_incompatible_shares
59 assert isinstance(summary, str) # should be a single string
60 self._summary = summary
61 assert not isinstance(report, str) # should be list of strings
64 from allmydata.mutable.servermap import ServerMap
65 assert isinstance(servermap, ServerMap), servermap
66 self._servermap = servermap # mutable only
67 self._share_problems = share_problems
69 def get_storage_index(self):
70 return self._storage_index
71 def get_storage_index_string(self):
72 return base32.b2a(self._storage_index)
78 def is_recoverable(self):
79 return self._recoverable
81 def needs_rebalancing(self):
82 return self._needs_rebalancing_p
84 def get_encoding_needed(self):
85 return self._count_shares_needed
86 def get_encoding_expected(self):
87 return self._count_shares_expected
89 def get_share_counter_good(self):
90 return self._count_shares_good
91 def get_share_counter_wrong(self):
92 return self._count_wrong_shares
94 def get_new_corrupt_shares(self):
95 return self._list_corrupt_shares
96 def get_corrupt_shares(self):
97 return [(s.get_serverid(), SI, shnum)
98 for (s, SI, shnum) in self._list_corrupt_shares]
100 def get_new_incompatible_shares(self):
101 return self._list_incompatible_shares
102 def get_incompatible_shares(self):
103 return [(s.get_serverid(), SI, shnum)
104 for (s, SI, shnum) in self._list_incompatible_shares]
106 def get_servers_responding(self):
107 return self._servers_responding
109 def get_host_counter_good_shares(self):
110 return self._count_good_share_hosts
112 def get_version_counter_recoverable(self):
113 return self._count_recoverable_versions
114 def get_version_counter_unrecoverable(self):
115 return self._count_unrecoverable_versions
117 def get_sharemap(self):
118 return self._sharemap
122 for shnum, servers in self._sharemap.items():
123 sharemap[shnum] = sorted([s.get_serverid() for s in servers])
124 responding = [s.get_serverid() for s in self._servers_responding]
125 corrupt = [(s.get_serverid(), SI, shnum)
126 for (s, SI, shnum) in self._list_corrupt_shares]
127 incompatible = [(s.get_serverid(), SI, shnum)
128 for (s, SI, shnum) in self._list_incompatible_shares]
129 d = {"count-shares-needed": self._count_shares_needed,
130 "count-shares-expected": self._count_shares_expected,
131 "count-shares-good": self._count_shares_good,
132 "count-good-share-hosts": self._count_good_share_hosts,
133 "count-recoverable-versions": self._count_recoverable_versions,
134 "count-unrecoverable-versions": self._count_unrecoverable_versions,
135 "servers-responding": responding,
136 "sharemap": sharemap,
137 "count-wrong-shares": self._count_wrong_shares,
138 "list-corrupt-shares": corrupt,
139 "count-corrupt-shares": self._count_corrupt_shares,
140 "list-incompatible-shares": incompatible,
141 "count-incompatible-shares": self._count_incompatible_shares,
145 def get_summary(self):
147 def get_report(self):
149 def get_share_problems(self):
150 return self._share_problems
151 def get_servermap(self):
152 return self._servermap
154 class CheckAndRepairResults:
155 implements(ICheckAndRepairResults)
157 def __init__(self, storage_index):
158 self.storage_index = storage_index
159 self.repair_attempted = False
161 def get_storage_index(self):
162 return self.storage_index
163 def get_storage_index_string(self):
164 return base32.b2a(self.storage_index)
165 def get_repair_attempted(self):
166 return self.repair_attempted
167 def get_repair_successful(self):
168 if not self.repair_attempted:
170 return self.repair_successful
171 def get_pre_repair_results(self):
172 return self.pre_repair_results
173 def get_post_repair_results(self):
174 return self.post_repair_results
177 class DeepResultsBase:
179 def __init__(self, root_storage_index):
180 self.root_storage_index = root_storage_index
181 if root_storage_index is None:
182 self.root_storage_index_s = "<none>" # is this correct?
184 self.root_storage_index_s = base32.b2a(root_storage_index)
186 self.objects_checked = 0
187 self.objects_healthy = 0
188 self.objects_unhealthy = 0
189 self.objects_unrecoverable = 0
190 self.corrupt_shares = []
191 self.all_results = {}
192 self.all_results_by_storage_index = {}
195 def update_stats(self, new_stats):
196 self.stats.update(new_stats)
198 def get_root_storage_index_string(self):
199 return self.root_storage_index_s
201 def get_corrupt_shares(self):
202 return self.corrupt_shares
204 def get_all_results(self):
205 return self.all_results
207 def get_results_for_storage_index(self, storage_index):
208 return self.all_results_by_storage_index[storage_index]
214 class DeepCheckResults(DeepResultsBase):
215 implements(IDeepCheckResults)
217 def add_check(self, r, path):
219 return # non-distributed object, i.e. LIT file
221 assert isinstance(path, (list, tuple))
222 self.objects_checked += 1
224 self.objects_healthy += 1
226 self.objects_unhealthy += 1
227 if not r.is_recoverable():
228 self.objects_unrecoverable += 1
229 self.all_results[tuple(path)] = r
230 self.all_results_by_storage_index[r.get_storage_index()] = r
231 self.corrupt_shares.extend(r.get_corrupt_shares())
233 def get_counters(self):
234 return {"count-objects-checked": self.objects_checked,
235 "count-objects-healthy": self.objects_healthy,
236 "count-objects-unhealthy": self.objects_unhealthy,
237 "count-objects-unrecoverable": self.objects_unrecoverable,
238 "count-corrupt-shares": len(self.corrupt_shares),
242 class DeepCheckAndRepairResults(DeepResultsBase):
243 implements(IDeepCheckAndRepairResults)
245 def __init__(self, root_storage_index):
246 DeepResultsBase.__init__(self, root_storage_index)
247 self.objects_healthy_post_repair = 0
248 self.objects_unhealthy_post_repair = 0
249 self.objects_unrecoverable_post_repair = 0
250 self.repairs_attempted = 0
251 self.repairs_successful = 0
252 self.repairs_unsuccessful = 0
253 self.corrupt_shares_post_repair = []
255 def add_check_and_repair(self, r, path):
257 return # non-distributed object, i.e. LIT file
258 r = ICheckAndRepairResults(r)
259 assert isinstance(path, (list, tuple))
260 pre_repair = r.get_pre_repair_results()
261 post_repair = r.get_post_repair_results()
262 self.objects_checked += 1
263 if pre_repair.is_healthy():
264 self.objects_healthy += 1
266 self.objects_unhealthy += 1
267 if not pre_repair.is_recoverable():
268 self.objects_unrecoverable += 1
269 self.corrupt_shares.extend(pre_repair.get_corrupt_shares())
270 if r.get_repair_attempted():
271 self.repairs_attempted += 1
272 if r.get_repair_successful():
273 self.repairs_successful += 1
275 self.repairs_unsuccessful += 1
276 if post_repair.is_healthy():
277 self.objects_healthy_post_repair += 1
279 self.objects_unhealthy_post_repair += 1
280 if not post_repair.is_recoverable():
281 self.objects_unrecoverable_post_repair += 1
282 self.all_results[tuple(path)] = r
283 self.all_results_by_storage_index[r.get_storage_index()] = r
284 self.corrupt_shares_post_repair.extend(post_repair.get_corrupt_shares())
286 def get_counters(self):
287 return {"count-objects-checked": self.objects_checked,
288 "count-objects-healthy-pre-repair": self.objects_healthy,
289 "count-objects-unhealthy-pre-repair": self.objects_unhealthy,
290 "count-objects-unrecoverable-pre-repair": self.objects_unrecoverable,
291 "count-objects-healthy-post-repair": self.objects_healthy_post_repair,
292 "count-objects-unhealthy-post-repair": self.objects_unhealthy_post_repair,
293 "count-objects-unrecoverable-post-repair": self.objects_unrecoverable_post_repair,
294 "count-repairs-attempted": self.repairs_attempted,
295 "count-repairs-successful": self.repairs_successful,
296 "count-repairs-unsuccessful": self.repairs_unsuccessful,
297 "count-corrupt-shares-pre-repair": len(self.corrupt_shares),
298 "count-corrupt-shares-post-repair": len(self.corrupt_shares_post_repair),
301 def get_remaining_corrupt_shares(self):
302 return self.corrupt_shares_post_repair