python grpc_记python使用grpc

using grpc in Python

gRPC是基于http/2的RPC框架，使用ProtoBuf作为底层数据序列化。Nginx服务器2018年3月17日引入gRPC支持。

gRPC 是用来实现跨语言通信的。比如在你的某个功能里需要用的同事写的接口，而你们俩又不是同一种语言。此时有两种方案，一是使用.so 文件；另一种则是使用 RPC 框架。

创建一个grpc_demo项目，结构如下

grpc_demo/

├── client

│ └── client.py

├── example

│ └── data.proto

└── server

​ └── server.py

client目录下的client.py实现了客户端用于发送数据并打印接收到server端处理后的数据

server目录下的server.py实现了server端用于接收客户端发送的数据，并对数据进行简单处理后返回给客户端

example包用于编写proto文件并生成data接口

grpc安装

安装gRPC

pip install grpcio

安装 ProtoBuf 相关的 python 依赖库

pip install protobuf

安装 python grpc 的 protobuf 编译工具

pip install grpcio-tools

定义gRPC接口

data.proto文件，service关键字就是定义接口，message定义数据结构

syntax = "proto3";

service Greeter {

rpc SayHello (HelloRequest) returns (HelloReply) {}

}

message HelloRequest {

string name = 1;

string message = 2;

string location = 3;

string ip = 4;

}

message HelloReply {

string name = 1;

string message = 2;

string location = 3;

string ip = 4;

}

编译protobuf生成所需文件(服务端和客户端都需要)

# mkdir gen-py

# python -m grpc_tools.protoc --proto_path=./ --python_out=./gen_py --grpc_python_out=./gen_py ./data.proto

在gen-py目录下生成data_pb2_grpc.py和data_pb2.py两个文件。

生成后完整目录结构：

grpc_demo

├── client

│ └── client.py

├── example

│ ├── data.proto

│ └── gen_py

│ ├── data_pb2_grpc.py

│ └── data_pb2.py

└── server

​ └── server.py

服务端实现

#!/usr/bin/env python

# -*- coding: utf-8 -*-

import sys

sys.path.append('../example/gen_py')

import grpc

import time

from concurrent import futures

import data_pb2

import data_pb2_grpc

_ONE_DAY_IN_SECONDS = 60 * 60 * 24

_HOST = "127.0.0.1"

_PORT = "19999"

class HelloWorld(data_pb2_grpc.GreeterServicer):

def SayHello(self, request, context):

print("request: " + str(request))

return data_pb2.HelloReply(message='%s, %s!' % (request.message, request.name))

def server():

grpcServer = grpc.server(futures.ThreadPoolExecutor(max_workers=4))

data_pb2_grpc.add_GreeterServicer_to_server(HelloWorld(), grpcServer)

grpcServer.add_insecure_port("{0}:{1}".format(_HOST, _PORT))

grpcServer.start()

try:

while True:

time.sleep(_ONE_DAY_IN_SECONDS)

except KeyboardInterrupt:

grpcServer.stop(0)

if __name__ == '__main__':

server()

客户端实现

#!/usr/bin/env python

# -*- coding: utf-8 -*-

import sys

sys.path.append("../example/gen_py")

import grpc

import data_pb2

import data_pb2_grpc

_HOST = '127.0.0.1'

_PORT = '19999'

def run():

with grpc.insecure_channel("{0}:{1}".format(_HOST, _PORT)) as channel:

client = data_pb2_grpc.GreeterStub(channel=channel)

response = client.SayHello(data_pb2.HelloRequest(name='you', message='hey guys'))

print("received: " + response.message)

if __name__ == '__main__':

run()

#channel = grpc.insecure_channel("{0}:{1}".format(_HOST, _PORT))

#client = data_pb2_grpc.GreeterStub(channel=channel)

#response = client.SayHello(data_pb2.HelloRequest(name='you', message='hey guys'))

#print("received: " + response.message)

参考：