def register(self):
req = consumer_register_req()
req.id = self.id
p = pack(command.consumer_register, req.SerializeToString())
self.dispatcher_socket.sendall(p.serialize())
respack = ss.read_pack(self.dispatcher_socket)
if respack is None:
logger.error("no register resp")
sys.exit(1)
if respack.command != command.consumer_register:
logger.info("register failed 1")
sys.exit(1)
resp = consumer_register_resp()
resp.ParseFromString(respack.data)
if resp.id != self.id or resp.res != 0:
logger.info("register failed 2")
def mkpack(self, cmd, msg):
p = pack(cmd, msg.SerializeToString())
return p
def push(self, cmd, msg):
p = pack(cmd, msg.SerializeToString())
return self.dispatcher_stream.write(p)
