mprpc 是一个超快速的Python RPC 库,最近一直在看 RPC 的东西,考虑如何应用到公司的项目中,模仿 Celery 的方式二次封装了一下。
项目地址: mprpc_config
安装
pip install mprpc_config
用法
构建一个如下的目录结构
-
config.py 包含RPC配置属性(任意命名)
- INSTALLED_APP 必填参数,list 类型,将每一个包含 RPC 接口的模块名放进去
-
app_module_name RPC 接口模块(任意命名,可有多个)
- implement.py 接口模块(任意命名,可有多个),里面包含 rpc 方法装饰的接口函数
-
server.py RPC 服务器
- 初始化 mprpc_config 的
Configration
类,将config.py
的模块名config
作为字符串传入 - 导入 mprpc_config.rpc_server 的 RPCServer 和 StreamServer,启动 RPC 服务器
- 使用
bind_class
方法将初始化后的所有 RPC 接口绑定给RPCServer
- 使用 StreamServer 启动 RPC 服务器
- 初始化 mprpc_config 的
- client.py RPC 客户端(测试用,实际可分离)
客户端启动方式可参考 mprpc 。
示例
示例代码在
example
目录
server.py
from mprpc_config.rpc_server import RPCServer, StreamServer
from mprpc_config import rpc_config
if __name__ == '__main__':
print('-------start server--------')
config = rpc_config.Configuration("config")
config.bind_class(RPCServer)
server = StreamServer(('127.0.0.1', 6000), RPCServer)
server.serve_forever(stop_timeout=10)
client.py
from mprpc_config.rpc_client import RPCClient, RPCPoolClient
from mprpc_config.rpc_client import RPCPool
print('---------- client ----------')
client = RPCClient('127.0.0.1', 6000)
print('2 add 4: ', client.call('add', 2, 4))
print('2 plus 4: ', client.call('plus', 2, 4))
print('2 minus 4: ', client.call('minus', 2, 4))
print('---------- done ----------')
print('---------- client pool ----------')
client_pool = RPCPool(RPCPoolClient, dict(host='127.0.0.1', port=6000))
with client_pool.connection() as client:
print('2 add 4: ', client.call('add', 2, 4))
print('2 plus 4: ', client.call('plus', 2, 4))
print('2 minus 4: ', client.call('minus', 2, 4))
print('---------- done ----------')
启动 server 和 client
$ python server.py
---------- server ----------
$ python client.py
---------- client ----------
2 add 4: 6
2 plus 4: 8
2 minus 4: -2
---------- done ----------
---------- client pool ----------
2 add 4: 6
2 plus 4: 8
2 minus 4: -2
---------- done ----------