"""
@Time ： 2022/9/30 17:09
@Auth ： 东
@File ：RpcClient.py
@IDE ：PyCharm
@Motto：ABC(Always Be Coding)
@Desc：RPC客户端

"""

import json
import socket
import time


class TCPClient(object):
    def __init__(self):
        self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

    def connect(self, host, port):
        """链接Server端"""
        self.sock.connect((host, port))

    def send(self, data):
        """将数据发送到Server端"""
        self.sock.send(data)

    def recv(self, length):
        """接受Server端回传的数据"""
        return self.sock.recv(length)


class RPCStub(object):

    def __getattr__(self, function):
        def _func(*args, **kwargs):
            d = {'method_name': function, 'method_args': args, 'method_kwargs': kwargs}
            self.send(json.dumps(d).encode('utf-8'))  # 发送数据
            data = self.recv(1024)  # 接收方法执行后返回的结果
            return data.decode('utf-8')

        setattr(self, function, _func)
        return _func


class RPCClient(TCPClient, RPCStub):
    pass


# c = RPCClient()
# c.connect('127.0.0.1', 5003)
# print(c.start('1'))
# print(c.add(1, 2, 3))
# print(c.setData({"sss": "ssss", "list": [1, 2, 3, 4]}))