コード例 #1
0
def wrap_grpc_packet(_json_body,
                     _method,
                     _url,
                     _src_party_id,
                     _dst_party_id,
                     job_id=None,
                     overall_timeout=DEFAULT_GRPC_OVERALL_TIMEOUT):
    _src_end_point = fate_meta_pb2.Endpoint(ip=IP, port=GRPC_PORT)
    _src = proxy_pb2.Topic(name=job_id,
                           partyId="{}".format(_src_party_id),
                           role=ROLE,
                           callback=_src_end_point)
    _dst = proxy_pb2.Topic(name=job_id,
                           partyId="{}".format(_dst_party_id),
                           role=ROLE,
                           callback=None)
    _task = proxy_pb2.Task(taskId=job_id)
    _command = proxy_pb2.Command(name=ROLE)
    _conf = proxy_pb2.Conf(overallTimeout=overall_timeout)
    _meta = proxy_pb2.Metadata(src=_src,
                               dst=_dst,
                               task=_task,
                               command=_command,
                               operator=_method,
                               conf=_conf)
    _data = proxy_pb2.Data(key=_url,
                           value=bytes(json.dumps(_json_body), 'utf-8'))
    return proxy_pb2.Packet(header=_meta, body=_data)
コード例 #2
0
ファイル: grpc_utils.py プロジェクト: drougon/FATE
def wrap_grpc_packet(_json_body,
                     _method,
                     _url,
                     _dst_party_id=None,
                     job_id=None):
    _src_end_point = basic_meta_pb2.Endpoint(ip=IP, port=GRPC_PORT)
    _src = proxy_pb2.Topic(name=job_id,
                           partyId="{}".format(PARTY_ID),
                           role=ROLE,
                           callback=_src_end_point)
    _dst = proxy_pb2.Topic(name=job_id,
                           partyId="{}".format(_dst_party_id),
                           role=ROLE,
                           callback=None)
    _task = proxy_pb2.Task(taskId=job_id)
    _command = proxy_pb2.Command(name=ROLE)
    _meta = proxy_pb2.Metadata(src=_src,
                               dst=_dst,
                               task=_task,
                               command=_command,
                               operator=_method)
    _data = proxy_pb2.Data(key=_url,
                           value=bytes(json.dumps(_json_body), 'utf-8'))
    return proxy_pb2.Packet(header=_meta, body=_data)