用python搭建gRPC服務(wù)
gRPC是一個高性能、通用的開源RPC框架,其由Google主要面向移動應(yīng)用開發(fā)并基于HTTP/2協(xié)議標準而設(shè)計,基于ProtoBuf序列化協(xié)議開發(fā),且支持眾多開發(fā)語言。一個gRPC服務(wù)的大體結(jié)構(gòu)圖為:

圖一表明,grpc的服務(wù)是跨語言的,但需要遵循相同的協(xié)議(proto)。相比于REST服務(wù),gPRC 的一個很明顯的優(yōu)勢是它使用了二進制編碼,所以它比 JSON/HTTP 更快,且有清晰的接口規(guī)范以及支持流式傳輸,但它的實現(xiàn)相比rest服務(wù)要稍微要復雜一些,下面簡單介紹搭建gRPC服務(wù)的步驟。
安裝Python需要的庫
pip install grpcio pip install grpcio-tools pip install protobuf
定義gRPC的接口
創(chuàng)建 gRPC 服務(wù)的第一步是在.proto 文件中定義好接口,proto是一個協(xié)議文件,客戶端和服務(wù)器的通信接口正是通過proto文件協(xié)定的,可以根據(jù)不同語言生成對應(yīng)語言的代碼文件。這個協(xié)議文件主要就是定義好服務(wù)(service)接口,以及請求參數(shù)和相應(yīng)結(jié)果的數(shù)據(jù)結(jié)構(gòu),具體的proto語法參見如下鏈接(https://www.jianshu.com/p/da7ed5914088),關(guān)于二維數(shù)組、字典等Python中常用的數(shù)據(jù)類型,proto語法的表達見鏈接(https://blog.csdn.net/xiaoxiaojie521/article/details/106938519),下面是一個簡單的例子。
syntax = "proto3"; option cc_generic_services = true; //定義服務(wù)接口 service GrpcService { rpc hello (HelloRequest) returns (HelloResponse) {} //一個服務(wù)中可以定義多個接口,也就是多個函數(shù)功能 } //請求的參數(shù) message HelloRequest { string data = 1; //數(shù)字1,2是參數(shù)的位置順序,并不是對參數(shù)賦值 Skill skill = 2; //支持自定義的數(shù)據(jù)格式,非常靈活 }; //返回的對象 message HelloResponse { string result = 1; map
使用 protoc 和相應(yīng)的插件編譯生成對應(yīng)語言的代碼
python -m grpc_tools.protoc -I ./ --python_out=./ --grpc_python_out=. ./hello.proto
利用編譯工具把proto文件轉(zhuǎn)化成py文件,直接在當前文件目錄下運行上述代碼即可。
-I 指定proto所在目錄
-m 指定通過protoc生成py文件
--python_out指定生成py文件的輸出路徑
hello.proto 輸入的proto文件
執(zhí)行上述命令后,生成hello_pb2.py 和hello_pb2_grpc.py這兩個文件。
編寫grpc的服務(wù)端代碼
#! /usr/bin/env python # coding=utf8 import time from concurrent import futures import grpc from gRPC_example import hello_pb2_grpc, hello_pb2 _ONE_DAY_IN_SECONDS = 60 * 60 * 24 class TestService(hello_pb2_grpc.GrpcServiceServicer): ''' 繼承GrpcServiceServicer,實現(xiàn)hello方法 ''' def __init__(self): pass def hello(self, request, context): ''' 具體實現(xiàn)hello的方法,并按照pb的返回對象構(gòu)造HelloResponse返回 :param request: :param context: :return: ''' result = request.data + request.skill.name + " this is gprc test service" list_result = {"12": 1232} return hello_pb2.HelloResponse(result=str(result), map_result=list_result) def run(): ''' 模擬服務(wù)啟動 :return: ''' server = grpc.server(futures.ThreadPoolExecutor(max_workers=10)) hello_pb2_grpc.add_GrpcServiceServicer_to_server(TestService(),server) server.add_insecure_port('[::]:50052') server.start() print("start service...") try: while True: time.sleep(_ONE_DAY_IN_SECONDS) except KeyboardInterrupt: server.stop(0) if __name__ == '__main__': run()
在服務(wù)端側(cè),需要實現(xiàn)hello的方法來滿足proto文件中GrpcService的接口需求,hello方法的傳入?yún)?shù),是在proto文件中定義的HelloRequest,context是保留字段,不用管,返回參數(shù)則是在proto中定義的HelloResponse,服務(wù)啟動的代碼是標準的,可以根據(jù)需求修改提供服務(wù)的ip地址以及端口號。
編寫gRPC客戶端的代碼
#! /usr/bin/env python # coding=utf8 import grpc from gRPC_example import #! /usr/bin/env python # coding=utf8 import grpc from gRPC_example import hello_pb2_grpc, hello_pb2 def run(): ''' 模擬請求服務(wù)方法信息 :return: ''' conn=grpc.insecure_channel('localhost:50052') client = hello_pb2_grpc.GrpcServiceStub(channel=conn) skill = hello_pb2.Skill(name="engineer") request = hello_pb2.HelloRequest(data="xiao gang", skill=skill) respnse = client.hello(request) print("received:",respnse.result) if __name__ == '__main__': run() def run(): ''' 模擬請求服務(wù)方法信息 :return: ''' conn=grpc.insecure_channel('localhost:50052') client = hello_pb2_grpc.GrpcServiceStub(channel=conn) skill = hello_pb2.Skill(name="engineer") request = hello_pb2.HelloRequest(data="xiao gang", skill=skill) response = client.hello(request) print("received:",response.result) if __name__ == '__main__': run()
客戶端側(cè)代碼的實現(xiàn)比較簡單,首先定義好訪問ip和端口號,然后定義好HelloRequest數(shù)據(jù)結(jié)構(gòu),遠程調(diào)用hello即可。需要強調(diào)的是,客戶端和服務(wù)端一定要import相同proto文件編譯生成的hello_pb2_grpc, hello_pb2模塊,即使服務(wù)端和客戶端使用的語言不一樣,這也是grpc接口規(guī)范一致的體現(xiàn)。
調(diào)用測試
先啟動運行服務(wù)端的代碼,再啟動運行客戶端的代碼即可。
gRPC的使用總結(jié)
定義好接口文檔
工具生成服務(wù)端/客戶端代碼
服務(wù)端補充業(yè)務(wù)代碼
客戶端建立 gRPC 連接后,使用自動生成的代碼調(diào)用函數(shù)
編譯、運行
Python
版權(quán)聲明:本文內(nèi)容由網(wǎng)絡(luò)用戶投稿,版權(quán)歸原作者所有,本站不擁有其著作權(quán),亦不承擔相應(yīng)法律責任。如果您發(fā)現(xiàn)本站中有涉嫌抄襲或描述失實的內(nèi)容,請聯(lián)系我們jiasou666@gmail.com 處理,核實后本網(wǎng)站將在24小時內(nèi)刪除侵權(quán)內(nèi)容。
版權(quán)聲明:本文內(nèi)容由網(wǎng)絡(luò)用戶投稿,版權(quán)歸原作者所有,本站不擁有其著作權(quán),亦不承擔相應(yīng)法律責任。如果您發(fā)現(xiàn)本站中有涉嫌抄襲或描述失實的內(nèi)容,請聯(lián)系我們jiasou666@gmail.com 處理,核實后本網(wǎng)站將在24小時內(nèi)刪除侵權(quán)內(nèi)容。