一、安裝

pip install grpcio # gprc pip install grpcio-tools # 代碼生成工具 pip install protobuf # 協(xié)議

二、創(chuàng)建proto目錄并創(chuàng)建proto文件

# user.proto // 指定版本 syntax="proto3"; // 包 package user;// 創(chuàng)建service service User {// user methodrpc AddUser(UserRequest) returns (UserResponse) {}rpc GetUser(GetUserRequest) returns (GetUserResponse) {} }// 請求參數(shù)消息體 1、2 是指參數(shù)的個數(shù)順序 message UserRequest {string username = 1;int32 age = 2; }// 返回參數(shù)消息體 message UserResponse {string err = 1; }// 請求參數(shù)消息體 message GetUserRequest {string username = 1; }// 返回參數(shù)消息體 message GetUserResponse {string username = 1;int32 age = 2; }

# 生成pb2.py、pb2_grpc.py文件

python3 -m grpc_tools.protoc -I=. --python_out=. --grpc_python_out=. ./proto/user.proto

# ?--python_out= ?pb2.py 文件路徑
# ?--grpc_python_out= ?pb2_grpc.py文件路徑
# ?./proto/user.proto ?proto文件路徑

注: 這里遇到一個問題?
?? ? 使用 python3 -m grpc_tools.protoc -I=./proto --python_out=./proto --grpc_python_out=./proto ./proto/user.proto, 生成的pb2_grpc.py文件 導(dǎo)入 pb2.py 是 import user_pb2 as user__pb2, 這里會報錯，應(yīng)該是 from proto import user_pb2 as user__pb2

?? ? 當(dāng)我使用 ?python3 -m grpc_tools.protoc -I=. --python_out=. --grpc_python_out=. ./proto/user.proto 生成時，不會有這個問題。導(dǎo)入變成 from proto import user_pb2 as proto_dot_user__pb2

三、服務(wù)端代碼

# user.py from proto import user_pb2, user_pb2_grpcclass UserServicer(user_pb2_grpc.UserServicer):def AddUser(self, request, context):print(request.username, request.age)# 具體的業(yè)務(wù)邏輯return user_pb2.UserResponse(err='ok')def GetUser(self, request, context):print(request.username)# 具體的業(yè)務(wù)邏輯return user_pb2.GetUserResponse(username='danni', age=12) # server.py import timeimport grpc from concurrent import futuresfrom proto import user_pb2, user_pb2_grpc from moudle.user import UserServicerdef main():# futures.ThreadPoolExecutor(max_workers=10) 指定最大線程數(shù)，不指定默認(rèn)是 cpu個數(shù) x 5"""if max_workers is None:# Use this number because ThreadPoolExecutor is often# used to overlap I/O instead of CPU work.max_workers = (os.cpu_count() or 1) * 5"""server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))# 注冊User Serviceuser_server = UserServicer()user_pb2_grpc.add_UserServicer_to_server(user_server, server)# 啟動服務(wù)server.add_insecure_port("127.0.0.1:8000")server.start()try:print('start rpc service')while True:time.sleep(60 * 60)except:server.stop(0)if __name__ == '__main__':main()

四、客戶端代碼

# client.py import grpcfrom proto import user_pb2_grpc, user_pb2def main():# 連接RPC 服務(wù)器channel = grpc.insecure_channel("localhost:8000")# 創(chuàng)建 Stubuser_cli = user_pb2_grpc.UserStub(channel)# 調(diào)用 AddUservalue = user_cli.AddUser(user_pb2.UserRequest(username='danni', age=20))print(value)if __name__ == '__main__':main()

一個簡單的python gprc使用，便于理解

總結(jié)

以上是生活随笔為你收集整理的python 使用grpc的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。