師父給了我一個(gè) .proto 文件，我應(yīng)該怎么使用？
回想我年輕的時(shí)候，在做一個(gè)項(xiàng)目時(shí)，需要計(jì)算斐波那契數(shù)列第 n 項(xiàng)的值。但是我只會(huì)使用遞歸來(lái)實(shí)現(xiàn)。眾所周知，遞歸算法計(jì)算斐波那契數(shù)列的效率極差，速度極慢。
于是我求助于當(dāng)時(shí)我的師父，問(wèn)他有沒有辦法幫我解決這個(gè)問(wèn)題。
我?guī)煾刚f(shuō)：“有啊，我寫過(guò)，但是代碼是用C++ 寫的，你估計(jì)看不懂。不過(guò)沒關(guān)系，你用 Python 直接調(diào)用就可以了。”
我很驚訝：“用 Python 直接調(diào)用C++代碼嗎?看起來(lái)似乎很麻煩啊。”
師父說(shuō)：“一點(diǎn)也不麻煩。我給你一個(gè).proto 文件和一個(gè)地址，你拿去自動(dòng)生成代碼就能調(diào)用了。”
于是，我拿到了一個(gè)mentors_secret.proto文件，里面的內(nèi)容非常簡(jiǎn)單：
- syntax = "proto3";
- message NumToCalc {
- int32 num = 1;
- }
- message Result {
- int32 result = 1;
- string msg = 2;
- bool success = 3;
- }
- service MentorsSecret {
- rpc CalcFib(NumToCalc) returns (Result) {}
- }
還有一個(gè)地址：122.51.39.219:8766。
原來(lái)是使用 gRPC 啊。這樣我就知道怎么做了。
安裝環(huán)境
首先，我們先安裝Python 版本的gRPC：
- pip install grpcio grpcio-tools
生成代碼
接下來(lái)，基于這個(gè)mentors_secret.proto文件，自動(dòng)生成代碼。cd進(jìn)入mentors_secret.proto文件所在的文件，執(zhí)行下面的命令：
- python3 -m grpc_tools.protoc -I . --python_out=. --grpc_python_out=. ./mentors_secret.proto
此時(shí)，可以看到當(dāng)前文件夾中生成了兩個(gè)文件：mentors_secret_pb2.py和mentors_secret_pb2_grpc.py。
如下圖所示：
這兩個(gè)文件的內(nèi)容，不需要看。
調(diào)用遠(yuǎn)程服務(wù)
接下來(lái)，創(chuàng)建一個(gè)新文件，叫做client.py，用來(lái)調(diào)用遠(yuǎn)程的函數(shù)：
- import grpc
- from mentors_secret_pb2 import NumToCalc
- from mentors_secret_pb2_grpc import MentorsSecretStub
- channel = grpc.insecure_channel('122.51.39.219:8766')
- stub = MentorsSecretStub(channel)
- result = stub.CalcFib(NumToCalc(num=36))
- print('斐波那契數(shù)列第36像是：', result.result)
代碼加上空行總共只有10行。1-3行導(dǎo)入模塊，6，7行創(chuàng)建遠(yuǎn)程鏈接。第9行調(diào)用遠(yuǎn)程的函數(shù)。第10行打印結(jié)果。
我們來(lái)看看運(yùn)行效果：
運(yùn)行結(jié)果秒出。
總結(jié)
當(dāng)我們拿到一個(gè).proto文件，需要去調(diào)用gRPC 服務(wù)時(shí)，我們一般有如下步驟：
(1) 自動(dòng)生成mentors_secret_pb2.py和mentors_secret_pb2_grpc.py文件。
(2) 查看.proto文件里面，service后面的名字MentorsSecret，如下圖所示：
(3) 編寫固定的代碼：
- import grpc
- from mentors_secret_pb2_grpc import MentorsSecretStub # service 名字后面加上 Stub
- channel = grpc.insecure_channel('遠(yuǎn)程服務(wù)地址和 端口')
- stub = MentorsSecretStub(channel)
(4) 調(diào)用遠(yuǎn)程函數(shù)。通過(guò).proto文件，可以知道遠(yuǎn)程函數(shù)CalcFib接收一個(gè)參數(shù)NumToCalc，所以把它導(dǎo)入進(jìn)來(lái)：from mentors_secret_pb2 import NumToCalc，它里面的參數(shù)為num，所以調(diào)用遠(yuǎn)程服務(wù)時(shí)賦值：
- para = NumToCalc(num=36)
- calc_result = stub.CalcFib(para)
(5) 從.proto知道返回的結(jié)果是Result里面的result屬性。于是打印calc_result.result獲得結(jié)果。
