在Python中使用gRPC的方法示例


Posted in Python onAugust 08, 2018

本文介绍了在Python中使用gRPC的方法示例,分享给大家,具体如下:

在Python中使用gRPC的方法示例

使用Protocol Buffers的跨平台RPC系统。

安装

使用 pip

pip install grpcio
pip install grpcio-tools googleapis-common-protos

gRPC由两个部分构成,grpcio 和 gRPC 工具, 后者是编译 protocol buffer 以及提供生成代码的插件。

使用

编写protocol buffer

使用 gRPC 首先需要做的是设计 protocol buffer。新建一个 msg.proto 文件。

syntax = "proto3";

service MsgService {
 rpc GetMsg (MsgRequest) returns (MsgResponse){}
}

message MsgRequest {
  string name = 1;
}

message MsgResponse {
  string msg = 1;
}

以上面的这个消息服务为例,首先是规定语法,这里使用的是 proto3 的语法。接着使用 service 关键字定义服务,gRPC 提供4种 RPC 类型的服务,这里定义的是第一种单一请求单一回应,类似普通的函数调用,其他的使用到了 stream 关键字,将其放在括号里,代表这个数据是流数据。这个以后再来研究,本次先设计一个简单的RPC。

之后定义两个 message ,一个是请求的结构,一个是回应的结果。 这里表示这个数据结构是字符串,protocol buffer 还可以定义为 int32,int64,double,float 等等。这里赋予的初值可以随便填写,实际使用中,会被赋予新的值。

生成接口代码

因为之前安装好了一些辅助插件,使用这里直接可以生成。

python -m grpc_tools.protoc -I . --pythoout=. --grpc_python_out=. msg.proto

这里会生成两个文件, msg_pb2.pymsg_pb2_grpc.py 。这两个文件是为后续的服务端和客户端所用。前者是定义了一些变量,例如 _MSGREQUEST 中就包含了请求函数的名字,可接受的变量,实际上还是 msg.proto 里定义的东西。

创建服务端

首先需要导入 RPC 必备的包,以及刚才生成的两个文件。

import grpc
import msg_pb2
import msg_pb2_grpc

因为 RPC 应该长时间运行,考虑到性能,还需要用到并发的库。

from concurrent import futures
import time

_ONE_DAY_IN_SECONDS = 60 * 60 * 24

在 Server 中,主要是实现服务,按照 msg.proto 定义的,这里需要写一个服务类 MsgServicer ,这个类需要实现之前定义的 GetMsg

class MsgServicer(msg_pb2_grpc.MsgServiceServicer):

  def GetMsg(self, request, context):
    print("Received name: %s" % request.name)
    return msg_pb2.MsgResponse(msg='Hello, %s!' % request.name)

GetMsg 接收到的请求是在 request 中, msg.proto 中定义的 name 就是 request.name ,接着在 GetMsg 中设计 msg.proto 中定义的 MsgResponse

之后实现启动服务的部分即可。

def serve():
  server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
  msg_pb2_grpc.add_MsgServiceServicer_to_server(MsgServicer(), server)
  server.add_insecure_port('[::]:50051')
  server.start()
  try:
    while True:
      time.sleep(_ONE_DAY_IN_SECONDS)
  except KeyboardInterrupt:
    server.stop(0)

通过并发库,将服务端放到多进程里运行。

完整 msg_server.py 代码如下

import grpc
import msg_pb2
import msg_pb2_grpc

from concurrent import futures
import time

_ONE_DAY_IN_SECONDS = 60 * 60 * 24


class MsgServicer(msg_pb2_grpc.MsgServiceServicer):

  def GetMsg(self, request, context):
    print("Received name: %s" % request.name)
    return msg_pb2.MsgResponse(msg='Hello, %s!' % request.name)


def serve():
  server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
  msg_pb2_grpc.add_MsgServiceServicer_to_server(MsgServicer(), server)
  server.add_insecure_port('[::]:50051')
  server.start()
  try:
    while True:
      time.sleep(_ONE_DAY_IN_SECONDS)
  except KeyboardInterrupt:
    server.stop(0)

if __name__ == '__main__':
  serve()

创建客户端

客户端相对简单一些,这里我写了一个简单的客户端。

import grpc

import msg_pb2
import msg_pb2_grpc

def run():
  # NOTE(gRPC Python Team): .close() is possible on a channel and should be
  # used in circumstances in which the with statement does not fit the needs
  # of the code.
  with grpc.insecure_channel('localhost:50051') as channel:
    stub = msg_pb2_grpc.MsgServiceStub(channel)
    response = stub.GetMsg(msg_pb2.MsgRequest(name='world'))
  print("Client received: " + response.msg)


if __name__ == '__main__':
  run()

使用 grpc.insecure_channel('localhost:50051') 进行连接 服务端, 接着在这个 channel 上创建 stub , 在 msg_pb2_grpc 里可以找到 MsgServiceStub 这个类相关信息。这个 stub 可以调用远程的 GetMsg 函数。 MsgRequest 中的 namemsg.proto 中定义的数据。在回应里可以得到 msg.proto 中定义的 msg

运行

首先运行 python msg_server.py 启动服务端,接着运行 python msg_client.py 机会看到客户端接收到了服务端传来的消息。以上就是一个简单的 RPC 的使用。

总结

这里只是简单的用了一下 gRPC,关于另外三种模式,还在摸索。比起gRPC,我感觉简单 RestFul 更讨我喜欢。

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python读写ini文件的方法
May 28 Python
总结网络IO模型与select模型的Python实例讲解
Jun 27 Python
python中nan与inf转为特定数字方法示例
May 11 Python
django实现前后台交互实例
Aug 07 Python
DataFrame 将某列数据转为数组的方法
Apr 13 Python
Python2和Python3中urllib库中urlencode的使用注意事项
Nov 26 Python
Python使用字典的嵌套功能详解
Feb 27 Python
Python变量访问权限控制详解
Jun 29 Python
python实现批量命名照片
Jun 18 Python
python matplotlib库的基本使用
Sep 23 Python
Python爬虫基础初探selenium
May 31 Python
python神经网络学习 使用Keras进行简单分类
May 04 Python
Python实现购物评论文本情感分析操作【基于中文文本挖掘库snownlp】
Aug 07 #Python
python实现彩票系统
Jun 28 #Python
django框架自定义用户表操作示例
Aug 07 #Python
Python实现基于POS算法的区块链
Aug 07 #Python
tensorflow学习教程之文本分类详析
Aug 07 #Python
Django添加feeds功能的示例
Aug 07 #Python
Python爬虫实现抓取京东店铺信息及下载图片功能示例
Aug 07 #Python
You might like
php常用数组函数实例小结
2016/12/29 PHP
JavaScript replace(rgExp,fn)正则替换的用法
2010/03/04 Javascript
JS复制到剪贴板示例代码
2013/10/30 Javascript
js获取某元素的class里面的css属性值代码
2014/01/16 Javascript
JS禁用页面上所有控件的实现方法(附demo源码下载)
2015/12/17 Javascript
老生常谈JavaScript中的this关键字
2016/10/01 Javascript
Angular2表单自定义验证器的实现
2016/10/19 Javascript
JQ图片文件上传之前预览功能的简单实例(分享)
2017/11/12 Javascript
JS实现的类似微信聊天效果示例
2019/01/29 Javascript
详解Vue3.0 前的 TypeScript 最佳入门实践
2019/06/18 Javascript
mpvue实现左侧导航与右侧内容的联动
2019/10/21 Javascript
使用node-media-server搭建一个简易的流媒体服务器
2021/01/20 Javascript
[01:22:42]2014 DOTA2华西杯精英邀请赛 5 24 DK VS LGD
2014/05/26 DOTA
[01:20:38]完美世界DOTA2联赛 GXR vs IO 第一场 11.07
2020/11/09 DOTA
Python中的字典与成员运算符初步探究
2015/10/13 Python
使用Python3 编写简单信用卡管理程序
2016/12/21 Python
Python开发中爬虫使用代理proxy抓取网页的方法示例
2017/09/26 Python
Python字典,函数,全局变量代码解析
2017/12/18 Python
python处理两种分隔符的数据集方法
2018/12/12 Python
详解Python函数式编程—高阶函数
2019/03/29 Python
实例详解Matlab 与 Python 的区别
2019/04/26 Python
详解python中的线程与线程池
2019/05/10 Python
Python使用os.listdir和os.walk获取文件路径
2020/05/21 Python
python字典的值可以修改吗
2020/06/29 Python
html5构建触屏网站之网站尺寸探讨
2013/01/07 HTML / CSS
倩碧香港官方网站:Clinique香港
2017/11/13 全球购物
酒店行政人事部经理职务说明书
2014/02/26 职场文书
小学优秀班主任事迹材料
2014/05/17 职场文书
公安学专业求职信
2014/07/27 职场文书
安全保卫工作竞聘材料
2014/08/25 职场文书
工程索赔意向书
2014/08/30 职场文书
2014年财务个人工作总结
2014/12/08 职场文书
2014年团支部年度工作总结
2014/12/24 职场文书
停电通知范文
2015/04/16 职场文书
2015年学校教务处工作总结
2015/05/11 职场文书
少先队入队仪式主持词
2015/07/04 职场文书