python的rpc服务实现

最近工作中需要实现远程机器的自动化执行,调研了下实现机制,大体如下:

内置库:xmlrpc(使用http协议作为传输协议的rpc机制,使用xml文本的方式传输命令和数据,数据包大,速度慢)

第三方:ZeroRPC(底层使用ZeroMQ和MessagePack,基于TCP协议,轻量级的,速度快,响应时间短,并发高)

1. zerorpc

安装:

pip3 install zerorpc

服务端:

python 复制代码
# 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()

客户端:

python 复制代码
# 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

测试代码:

python 复制代码
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

官方文档戳这里

服务端

python 复制代码
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()

客户端和调用方法

python 复制代码
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

相关推荐
白帽黑客沐瑶2 天前
【网络安全就业】信息安全专业的就业前景(非常详细)零基础入门到精通,收藏这篇就够了
网络·安全·web安全·计算机·程序员·编程·网络安全就业
树码小子2 天前
Java网络编程:(socket API编程:TCP协议的 socket API -- 回显程序的服务器端程序的编写)
java·网络·tcp/ip
绿箭柠檬茶2 天前
Ubuntu 服务器配置转发网络访问
服务器·网络·ubuntu
FPGA_Linuxer2 天前
FPGA 40 DAC线缆和光模块带光纤实现40G UDP差异
网络协议·fpga开发·udp
real 12 天前
传输层协议UDP
网络·网络协议·udp
路由侠内网穿透2 天前
本地部署 GPS 跟踪系统 Traccar 并实现外部访问
运维·服务器·网络·windows·tcp/ip
喵手2 天前
玩转Java网络编程:基于Socket的服务器和客户端开发!
java·服务器·网络
徐子元竟然被占了!!2 天前
实验-基本ACL
网络
ftpeak2 天前
从零开始使用 axum-server 构建 HTTP/HTTPS 服务
网络·http·https·rust·web·web app
LabVIEW开发2 天前
LabVIEW气体污染无线监测
网络·labview·labview知识·labview功能·labview程序