通常情况下,Zerorpc 只能初始化一个类,要是有一些任务场景,需要按资源或多个类初始化呢?
从zerorpc的源码core.py 74-82行可以看出, 如果methods是dict对象,直接返回;如果是类名,则转化成dict。 所以我们可以直接初始化成dict,而dict是支持n个。
我们可以依葫芦画瓢,在server端可以这样现实:
import zerorpc
_method = {}
def register(cls):
obj = cls()
prefix = cls.__name__
_method.update({'{}:{}'.format(prefix, k): getattr(obj, k)
for k in dir(obj)
if not k.startswith('_') and callable(getattr(obj, k))})
return cls
@register
class RpcServer1:
def add(self, x, y):
return x+y
@register
class RpcServer2:
def add(self, x, y):
return x*y
server = zerorpc.Server(_method, heartbeat=20)
server.bind("tcp://{0}:{1}".format('127.0.0.1', 4242))
server.run()
我们定义了两个类,分别使用了register装饰器(把每个类存入字典),然后在实例化rpc server的时候传入定义好的字典即可。
下面是客户端调用:
import zerorpc
class Client(zerorpc.Client):
def __init__(self, *arg, **kwargs):
self._prefix = kwargs.pop("prefix", "")
super(Client, self).__init__(*arg, **kwargs)
def set_prefix(self, prefix):
self._prefix = prefix
def __getattr__(self, method):
method = ":".join([self._prefix, method]) if self._prefix else method
return lambda *args, **kwargs: self(method, *args, **kwargs)
if __name__ == "__main__":
c1 = Client(prefix="RpcServer1")
c1.connect('tcp://127.0.0.1:4242')
c2 = Client(prefix="RpcServer2")
c2.connect('tcp://127.0.0.1:4242')
print(c1.add(5,5))
print(c2.add(5,5))