python的rpc服务实现
最近工作中需要实现远程机器的自动化执行，调研了下实现机制，大体如下：
内置库：xmlrpc(使用http协议作为传输协议的rpc机制，使用xml文本的方式传输命令和数据，数据包大，速度慢)
第三方：ZeroRPC(底层使用ZeroMQ和MessagePack，基于TCP协议，轻量级的，速度快，响应时间短，并发高)
1. zerorpc
安装：
pip3 install zerorpc
服务端：
# remote_rpc_server.py
import sys
from zerorpc import Server
class RemoteMethod:
@staticmethod
def print_info(name):
# flush为了及时写入stdout文件
print('name is ', name, flush=True)
if __name__ == '__main__':
# 打开一个txt文件，将标准输出写入，因为远程的代码执行不会在客户端打印
sys.stdout = open('log.txt', 'w+')
server = Server(RemoteMethod())
server.bind('tcp://0.0.0.0:4242')
server.run()
客户端：
# local_rpc_client.py
from zerorpc import Client
def rpc_client():
client = Client(timeout=20)
# 连接远程ip和端口
client.connect('tcp://10.x.x.x:4242')
return client
测试代码：
import subprocess
from time import sleep
from local_rpc_client import rpc_client
# 拷贝项目文件到远程机器
cmd = ('sshpass -p 1 rsync -av --exclude=".idea" --exclude="__pycache__" /home/test/rpc_project test@10.x.x.x:~/Desktop')
with subprocess.Popen(cmd, shell=True) as proc:
pass
# 启动远程rpc服务
cmd = 'sshpass -p 1 ssh test@10.x.x.x "cd ~/Desktop/rpc_project; python3 remote_rpc_server.py"'
# with subprocess.Popen(cmd, shell=True) as proc1:
# pass
# 如果照上面这种写法，启动子进程后，代码执行会阻塞，不接着执行后续的代码。所以只能是不要等待子进程结束。。。
subprocess.Popen(cmd, shell=True)
sleep(2)
print('远程rpc服务已启动')
# 客户端调用远程方法
client = rpc_client()
client.print_info('hope succ')
运行测试代码后，可以在远程机器上查看log.txt文件，确定打印的内容写入，代表远程执行成功。以上代码只是个简单的示例，实际上还需要很多优化，比如要确认下远程rpc服务已启动成功再调用、远程机器的python环境要配置、远程rpc服务的启停设置等。
zerorpc依赖于MessagePack进行数据的序列化和反序列化，而MessagePack可能不支持某些复杂对象的序列化。如果说上面的print_info方法return obj（一个类实例），则可能会报错，这里需要注意，尽量return 基本数据类型。
2. xmlrpc
# 服务端
import inspect
from socketserver import ThreadingMixIn
from xmlrpc.server import SimpleXMLRPCServer
class ThreadXMLRPCServer(ThreadingMixIn, SimpleXMLRPCServer):
pass
# ThreadingMixIn 多线程处理请求
class RemoteMethod:
@staticmethod
def print_info(name):
# flush为了及时写入stdout文件
print('name is ', name, flush=True)
if __name__ == '__main__':
server = ThreadXMLRPCServer(("0.0.0.0", 4243), allow_none=True)
members = inspect.getmembers(RemoteMethod, predicate=inspect.isfunction)
for func_name, func in members:
if not func_name.startswith("_"):
server.register_function(func, func_name)
server.serve_forever()# 客户端和调用方法
from xmlrpc import client
proxy = client.ServerProxy("http://10.x.x.x:4243/", allow_none=True)
proxy.print_info('succ')allow_none=True，允许方法返回空，Python 常量 None 将被转写至 XML；默认行为是针对 None 引发 TypeError。
原文地址：https://blog.csdn.net/weixin_50652063/article/details/144395214
免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！