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)