import msgpack, json import balancer import balancelogicadapter import node_info_updater logging = Log() with open(manifest()["config"], "r") as config_file: config = json.load(config_file) log = elliptics.Logger(str(config["dnet_log"]), config["dnet_log_mask"]) n = elliptics.Node(log) for host in config["elliptics_nodes"]: try: logging.info("host: " + str(host)) n.add_remote(str(host[0]), host[1]) except Exception as e: logging.error("Error: " + str(e) + "\n" + traceback.format_exc()) meta_node = elliptics.Node(log) for host in config["metadata"]["nodes"]: try: logging.info("host: " + str(host)) meta_node.add_remote(str(host[0]), host[1]) except Exception as e: logging.error("Error: " + str(e) + "\n" + traceback.format_exc()) meta_session = elliptics.Session(meta_node) meta_session.add_groups(list(config["metadata"]["groups"])) n.meta_session = meta_session
# encoding: utf-8 from cocaine.context import Log, manifest import msgpack import traceback import sys import copy import elliptics import balancelogicadapter as bla import balancelogic logging = Log() logging.info("balancer.py") symmetric_groups_key = "metabalancer\0symmetric_groups" def get_groups(n): return bla.all_group_ids() def get_symmetric_groups(n): if manifest().get("symmetric_groups", False): (good_symm_groups, bad_symm_groups) = bla.filter_symm_groups() result = list(good_symm_groups) logging.info("good_symm_groups: " + str(result)) return result else: return None