def get_all_buckets_stats(digit=2):
partial_get_buckets_info = partial(get_buckets_info, digit=digit)
pool = ThreadPool(8)
buckets = pool.map(partial_get_buckets_info, get_servers()[0])
pool.close()
pool.join()
return [b for b in buckets if b is not None]
def get_all_server_stats():
pool = ThreadPool(8)
results = pool.map(gen_server_info, get_servers()[0])
pool.close()
pool.join()
results.sort(key=lambda x: (x.err is None, x.host))
return results
def get_all_buckets_key_counts(n):
buckets = [Bucket(n, i) for i in range(n)]
primaries, backups = get_servers()
for h in primaries:
d = get_buckets_key_counts(h, n)
for bkt, count in d.items():
buckets[bkt].servers.append((h, count))
for h in backups:
d = get_buckets_key_counts(h, n)
for bkt, count in d.items():
buckets[bkt].backups.append((h, count))
for bkt in buckets:
bkt.compute()
buckets.sort(key=lambda x: x.cmpkey, reverse=True)
for i, bkt in enumerate(buckets):
bkt.rank = i
return buckets
def get_servers(exclude):
exclude.extend(["chubb2", "chubb3"])
servers = config.get_servers()[0]
servers = [s for s in servers if s not in exclude]
return servers