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