reco.proto
syntax = "proto3";
package rpc_package;
service HelloWorldService {
rpc SayHello (HelloRequest) returns (HelloReply) {}
}
message HelloRequest {
repeated Student student=1;
}
message Student{
int32 age=1;
string name=2;
}
message HelloReply {
string message = 1;
repeated int32 s_list=2;
}

python -m grpc_tools.protoc -I. --python_out=. --grpc_python_out=. reco.proto
創建出 reco_pb2.py 和 reco_pb2_grpc.py 兩個文件
然后寫服務端代碼
test_grpc_server.py
from concurrent import futures
import grpc
import logging
import time
from reco_pb2_grpc import add_HelloWorldServiceServicer_to_server, \
HelloWorldServiceServicer
from reco_pb2 import HelloRequest, HelloReply
class Hello(HelloWorldServiceServicer):
# 這里實現我們定義的接口
def SayHello(self, request, context):
s_list = [0,1,2,4,3]
return HelloReply(message='Hello, {}! '.format(request.student),s_list=s_list)
def serve():
# 這里通過thread pool來並發處理server的任務
server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
# 將對應的任務處理函數添加到rpc server中
add_HelloWorldServiceServicer_to_server(Hello(), server)
# 這里使用的非安全接口,世界gRPC支持TLS/SSL安全連接,以及各種鑒權機制
server.add_insecure_port('[::]:50000')
server.start()
try:
while True:
time.sleep(60 * 60 * 24)
except KeyboardInterrupt:
server.stop(0)
if __name__ == "__main__":
logging.basicConfig()
serve()
test_grpc_client.py
from __future__ import print_function
import logging
import grpc
from reco_pb2 import HelloRequest, HelloReply
from reco_pb2_grpc import HelloWorldServiceStub
def run():
# 使用with語法保證channel自動close
with grpc.insecure_channel('localhost:50000') as channel:
# 客戶端通過stub來實現rpc通信
stub = HelloWorldServiceStub(channel)
# 客戶端必須使用定義好的類型,這里是HelloRequest類型
hellorequest = HelloRequest()
for i in range(5):
request = hellorequest.student.add()
request.age = 18
request.name = "xinyuuliu"
response = stub.SayHello(hellorequest)
print ("hello client received: " + response.message)
print(response.s_list)
if __name__ == "__main__":
logging.basicConfig()
run()
結果:
hello client received: Hello, [age: 18 name: "xinyuuliu" , age: 18 name: "xinyuuliu" , age: 18 name: "xinyuuliu" , age: 18 name: "xinyuuliu" , age: 18 name: "xinyuuliu" ]! [0, 1, 2, 4, 3]
