python 使用gRPC

Python gRPC

• 简介
  grpc 是google 开源的一款rpc服务框架，可以轻松的实现跨语言的微服务，将项目中的各个模块独立出来，单独部署，独立升级，也可以根据模块的情况进行不同语言的变成。
  gRPC也是一个C/S框架，使用的是HTTP2 协议进行通信。

• 准备.proto文件

  syntax = "proto3";
  
  package order;
  message OrderRequest {
      string phone = 1;
      string price = 2;
      map<string, string> request_arg = 3;//便于字段扩展
  }

 message JSONResponse{
     string rst_string = 1; //统一返回json字符串作处理
     }

  service OrderHandler {
    // format a list of events.
    rpc create_order (OrderRequest) returns (JSONResponse) {}
 }

其中:

message: 定义数据结构

service: 定义接口的名字，参数，

• 生成所需文件(服务器和客户端均需要)

python -m grpc_tools.protoc -I./ --python_out=./ --grpc_python_out=./ ./*.proto

• 编写server端代码

 import time
 import test_pb2_grpc
 import grpc

 from concurrent import futures
 from order.views import test
 
 import os

 class OrderHandler(test_pb2_grpc.OrderHandlerServicer):
     def create_order(self, request, context):
         return test(request, context)

 def serve():
     server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
     test_pb2_grpc.add_OrderHandlerServicer_to_server(
     										OrderHandler(), server)
     server.add_insecure_port('[::]:{}'.format(12006))
     server.start()
     try:
         while True:
             time.sleep(186400)
     except KeyboardInterrupt:
         server.stop(0)
serve()

• 对应的views.py 文件

 from django.shortcuts import render
 from django.http import JsonResponse
 # Create your views here.
 import test_pb2
 import json
 from django.http import HttpRequest

 def grpc_request(func):
     “”“
     将grpc请求重新构造成django requst， 
     并封装相应返回值
     ”“”
     def f(request, context):
         f = lambda x:{k:v for k,v in x.items()} if hasattr(x, 'items') else x
         args = {i[0].name:f(i[1]) for i in request.ListFields() }
         
         # 构造django request 对象，并添加参数信息
         dj_request = HttpRequest()
         dj_request.GET = args
         # dj_request.POST = args
         # dj_request._body = json.dumps(args)
         dj_request.META = args

         ret = func(dj_request)
         
         # 处理django的response 对象， 转换为grpc的对象
         json_response = test_pb2.JSONResponse()
         json_response.rst_string = ret.getvalue()
         
         return json_response
     return f

 def check_inenty(func):
     def f(request):
         if "identy" not in request.META:
             return JsonResponse(dict(status=403))
         else:
             return func(request)
     return f

 @grpc_request
 @check_inenty
 def test(request):
     return JsonResponse(dict(test=1, name="333"))

• 编写客户端代码进行 测试 client.py

 import grpc
 import test_pb2_grpc
 import test_pb2

 channel = grpc.insecure_channel("127.0.0.1:12006")
 stub = test_pb2_grpc.OrderHandlerStub(channel)
 ret = stub.create_order(test_pb2.OrderRequest(phone="12990", price="50"))
 
 print(ret.rst_string)