rpc_client.py 文件源码

def build_transformer(self):
        logger.debug("client: build_transformer, server address: %s", self.server_address)
        if self.is_trans_built:
            logger.debug("client: build_transformer: transformer is already built")
            return
        options = [('grpc.max_send_message_length', -1), ('grpc.max_receive_message_length', -1)]
        channel = grpc.insecure_channel(self.server_address, options=options)
        if not is_channel_ready(channel):
            raise RuntimeError("gRPC channel is not ready...")
        self.RPC = hetr_pb2_grpc.HetrStub(channel)

        if self.close_transformer_response_future is not None:
            response = self.close_transformer_response_future.result()
            if not response.status:
                raise RuntimeError("RPC close_transformer request failed: {}"
                                   .format(response.message))
            self.is_trans_built = False
            self.close_transformer_response_future = None

        response = self.RPC.BuildTransformer(
            hetr_pb2.BuildTransformerRequest(transformer_type=self.transformer_type),
            _TIMEOUT_SECONDS)
        if response.status:
            self.is_trans_built = True
        else:
            self.is_trans_built = False
            raise RuntimeError("RPC build_transformer request failed: {}".format(response.message))