跳转至

homo_client

HomoClient(server_addr, party_name, cert_path=None)

Bases: base_client.BaseClient

Implement homogeneous client base on base_client.BaseClient.

Source code in iflearner/communication/homo/homo_client.py 
30
31
32
33
34
35
def __init__(
    self, server_addr: str, party_name: str, cert_path: str = None
) -> None:
    super().__init__(server_addr, cert_path)
    self._party_name = party_name
    self._strategy: strategy_client.StrategyClient = None  # type: ignore

notice()

Receive notifications from the server regularly.

Source code in iflearner/communication/homo/homo_client.py 
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
def notice(self) -> None:
    """Receive notifications from the server regularly."""
    while True:
        start = timeit.default_timer()

        req = base_pb2.BaseRequest(party_name=self._party_name)
        resp = self._callback(req)
        if resp.code != 0:
            raise HomoException(
                code=HomoException.HomoResponseCode(resp.code), message=resp.message
            )

        if resp.type == message_type.MSG_AGGREGATE_RESULT:
            data = homo_pb2.AggregateResult()
            data.ParseFromString(resp.data)
            self._strategy.handler_aggregate_result(data)
        elif resp.type == message_type.MSG_NOTIFY_TRAINING:
            self._strategy.handler_notify_training()
        elif resp.type in self._strategy.custom_handlers:
            self._strategy.custom_handlers[resp.type]()  # type: ignore

        if resp.type != "":
            stop = timeit.default_timer()
            logger.info(
                f"IN: party: message type: {resp.type}, time: {1000 * (stop - start)}ms"
            )

        time.sleep(message_type.MSG_HEARTBEAT_INTERVAL)

transport(type, data=None)

Transport data to server.

Source code in iflearner/communication/homo/homo_client.py 
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
def transport(self, type: str, data: Any = None) -> homo_pb2.RegistrationResponse:
    """Transport data to server."""
    start = timeit.default_timer()

    req = base_pb2.BaseRequest(party_name=self._party_name, type=type)
    if data is not None:
        req.data = data.SerializeToString()

    resp = None
    if type == message_type.MSG_REGISTER:
        resp = self._send(req)
    elif type == message_type.MSG_CLIENT_READY:
        resp = self._send(req)
    elif type == message_type.MSG_UPLOAD_PARAM:
        resp = self._post(req)
    elif type == message_type.MSG_COMPLETE:
        resp = self._send(req)

    stop = timeit.default_timer()
    logger.info(f"OUT: message type: {type}, time: {1000 * (stop - start)}ms")

    if resp.code != 0:  # type: ignore
        raise HomoException(
            code=HomoException.HomoResponseCode(resp.code), message=resp.message  # type: ignore
        )

    if type == message_type.MSG_REGISTER:
        data = homo_pb2.RegistrationResponse()
        data.ParseFromString(resp.data)  # type: ignore
        return data