def __init__(self, servers):
threading.Thread.__init__(self)
self.running = True
self.ip_addr = utils.getip()
self.servers = servers
self.traffic_db = db_helper.DBPool(DBConf.COMMONDB, async=True)
self.traffic_db.execute("SET AUTOCOMMIT = 1")
def __init__(self, servers):
threading.Thread.__init__(self)
self.running = True
self.ip_addr = utils.getip()
self.servers = servers
self.traffic_db = db_helper.DBPool(DBConf.COMMONDB, async=True)
self.traffic_db.execute("SET AUTOCOMMIT = 1")
def _init_zk():
# zk_helper
self.zk = zk_helper.ZooKeeper("db_proxy", self.register_watches, "register_service", "proxy", self.port)
self.zk_dbinfo = self.zk.get_dict()
self.register_watches()
self.servers.start()
# mq_helper blocking thread
self.mq = mq_helper.PikaRecvThread(servers=MQConf.SERVERS,
exchange = MQConf.EXCHANGE_FORBID,
queue = utils.getip() + MQConf.PROXY_SUFFIX,
callback = self.mqCallback)
self.mq.start()
self.monitor = threads.ForbidMonitor(self)
self.monitor.start()
self.proxy_stats_log = threads.ProxyStatsLog(self.servers)
self.proxy_stats_log.start()
def init_znode():
# 1. /database/db_type
DBType().write(Quota.Keys)
# 2. /database/db_info
DBInfo().start()
# 3. /database/authed_ips
zk = zk_helper.ZooKeeper("init_znode")
# zk.connect()
# zk.wait_until_connected()
zk.mknode(ZKConf.ZK_PATH_IPS, cjson.encode([utils.getip()]))
# 3. /database/forbid
zk.mknode(ZKConf.ZK_PATH_FORBID, "")
# 4. /database/temp_master
zk.mknode(ZKConf.ZK_PATH_TEMP_MASTER, "")
# 5. /database/temp_stats
zk.mknode(ZKConf.ZK_PATH_TEMP_STATS, "")
# 6. /database/temp_proxy
zk.mknode(ZKConf.ZK_PATH_TEMP_PROXY, "")
# 7. /database/leader/db_stats
zk.mknode(ZKConf.ZK_PATH_LEADER_DB_STATS, "")
# 8. /database/leader/db_master
zk.mknode(ZKConf.ZK_PATH_LEADER_DB_MASTER, "")
def _init_zk():
# zk_helper
self.zk = zk_helper.ZooKeeper("db_proxy", self.register_watches, "register_service", "proxy", self.port)
self.zk_dbinfo = self.zk.get_dict()
self.register_watches()
self.servers.start()
# mq_helper blocking thread
self.mq = mq_helper.PikaRecvThread(
servers=MQConf.SERVERS,
exchange=MQConf.EXCHANGE_FORBID,
queue=utils.getip() + MQConf.PROXY_SUFFIX,
callback=self.mqCallback,
)
self.mq.start()
self.monitor = threads.ForbidMonitor(self)
self.monitor.start()
self.proxy_stats_log = threads.ProxyStatsLog(self.servers)
self.proxy_stats_log.start()
def init_znode():
# 1. /database/db_type
DBType().write(Quota.Keys)
# 2. /database/db_info
DBInfo().start()
# 3. /database/authed_ips
zk = zk_helper.ZooKeeper("init_znode")
# zk.connect()
# zk.wait_until_connected()
zk.mknode(ZKConf.ZK_PATH_IPS, cjson.encode([utils.getip()]))
# 3. /database/forbid
zk.mknode(ZKConf.ZK_PATH_FORBID, "")
# 4. /database/temp_master
zk.mknode(ZKConf.ZK_PATH_TEMP_MASTER, "")
# 5. /database/temp_stats
zk.mknode(ZKConf.ZK_PATH_TEMP_STATS, "")
# 6. /database/temp_proxy
zk.mknode(ZKConf.ZK_PATH_TEMP_PROXY, "")
# 7. /database/leader/db_stats
zk.mknode(ZKConf.ZK_PATH_LEADER_DB_STATS, "")
# 8. /database/leader/db_master
zk.mknode(ZKConf.ZK_PATH_LEADER_DB_MASTER, "")
