1.安装
# python3
pip install futures==3.1.1
pip install grpcio protobuf grpcio-tools
# python2
pip install grpcio protobuf grpcio-tools
2.目录结构
.
+-- client
| +-- __init__.py
| +-- main.py
+-- proto
| data.proto
+-- proto_py
| +-- __init_.py
| +-- data_pd2.py
| +-- data_pb2_grpc.py
+-- server
| +-- __init_.py
| +-- main.py
client目录下的 main.py 实现了客户端用于发送数据并打印接收到 server 端处理后的数据
server 目录下的 main.py 实现了 server 端用于接收客户端发送的数据,并对数据进行大写处理后返回给客户端
proto 包用于编写 proto 文件并生成 data 接口
proto_py 存储proto生成的py文件
3.数据
定义 gRPC 接口:
syntax = "proto3";
package example;
service FormatData { //定义服务,用在rpc传输中
rpc DoFormat(Data) returns (Data){}
}
message Data {
string text = 1;
}
编译 protobuf:
$ python -m grpc_tools.protoc -I. --python_out=. --grpc_python_out=. ./data.proto #在 proto 目录中执行编译,会生成:data_pb2.py 与 data_pb2_grpc.py
实现 server 端:
#! /usr/bin/env python
# -*- coding: utf-8 -*-
# 实现了 server 端用于接收客户端发送的数据,并对数据进行大写处理后返回给客户端
import grpc
import time
from concurrent import futures
from proto_py import data_pb2, data_pb2_grpc
_ONE_DAY_IN_SECONDS = 60 * 60 * 24
_HOST = 'localhost'
_PORT = '8080'
# 实现一个派生类,重写rpc中的接口函数.自动生成的grpc文件中比proto中的服务名称多了一个Servicer
class FormatData(data_pb2_grpc.FormatDataServicer):
# 重写接口函数.输入和输出都是proto中定义的Data类型
def DoFormat(self, request, context):
str = request.text
return data_pb2.Data(text=str.upper()) # 返回一个类实例
def serve():
# 定义服务器并设置最大连接数,corcurrent.futures是一个并发库,类似于线程池的概念
grpcServer = grpc.server(futures.ThreadPoolExecutor(max_workers=4)) # 创建一个服务器
data_pb2_grpc.add_FormatDataServicer_to_server(FormatData(), grpcServer) # 在服务器中添加派生的接口服务(自己实现了处理函数)
grpcServer.add_insecure_port(_HOST + ':' + _PORT) # 添加监听端口
grpcServer.start() # 启动服务器
try:
while True:
time.sleep(_ONE_DAY_IN_SECONDS)
except KeyboardInterrupt:
grpcServer.stop(0) # 关闭服务器
if __name__ == '__main__':
serve()
实现 client 端:
#! /usr/bin/env python
# -*- coding: utf-8 -*-
# 实现了客户端用于发送数据并打印接收到 server 端处理后的数据
import grpc
from proto_py import data_pb2, data_pb2_grpc
_HOST = 'localhost'
_PORT = '8080'
def run():
conn = grpc.insecure_channel(_HOST + ':' + _PORT) # 监听频道
client = data_pb2_grpc.FormatDataStub(channel=conn) # 客户端使用Stub类发送请求,参数为频道,为了绑定链接
response = client.DoFormat(data_pb2.Data(text='hello,world!')) # 返回的结果就是proto中定义的类
print("received: " + response.text)
if __name__ == '__main__':
run()
执行验证结果:
先启动 server,之后再执行 client
client 侧控制台如果打印的结果为:“received: HELLO,WORLD!” ,证明 gRPC 接口定义成功
参考: