Fast-GRPC: 用python轻松开发grpc接口

简介

Fast-GRPC 旨在帮助开发者更加轻松快捷地使用 Python 开发 gRPC 接口。它的特点包括简化步骤、简单上手、快速开发,同时支持异步和同步代码,以及支持 middleware,灵感来自FastAPI。

安装

需要python 3.7+

pip install python-fast-grpc

快速上手

下面是一个简单的 Fast-GRPC 示例,展示如何创建一个 gRPC 服务

from fast_grpc import BaseSchema, FastGRPC
# 创建 FastGRPC 实例并定义服务名
rpc = FastGRPC("Greeter")

class HelloRequest(BaseSchema):
    name: str

class HelloReply(BaseSchema):
    message: str
# 添加 gRPC 方法,使用 add_method 装饰器,指定请求和响应的模型类
@rpc.add_method("SayHello", request_model=HelloRequest, response_model=HelloReply)
async def say_hello(request: HelloRequest) -> HelloReply:
    return HelloReply(message=f"Hello {request.name}")

# 启动 gRPC 服务。无需手动编写 proto 文件,Fast-GRPC 会根据你的 Python 代码自动生成 proto 文件,并编译为 Python gRPC 代码,最后启动 gRPC 服务
rpc.run()

在上面的示例中,我们首先使用 FastGRPC 创建了一个 gRPC 服务。接下来,我们定义了两个数据模型 HelloRequest 和 HelloReply,并使用 add_method 方法将 SayHello 接口添加到 gRPC 服务中。

add_method 方法接受三个参数:rpc方法名、请求模型和响应模型。这里 say_hello 函数可以支持同步和异步代码,同步代码使用线程模拟异步来执行。

最后,我们通过 run 方法启动 gRPC 服务器。Fast-GRPC 会自动根据 Python 代码的接口定义生成 .proto 文件和 Python gRPC 代码,从而大大简化了开发过程。

最后我们用调用客户端来看看效果

import grpc
import greeter_pb2 as pb2
import greeter_pb2_grpc as pb2_grpc

channel = grpc.insecure_channel("127.0.0.1:50051")
stub = pb2_grpc.GreeterStub(channel)
response = stub.SayHello(pb2.HelloRequest(name="fastGRPC"))
print("Greeter client received: ", response)

相关链接

Fast-GRPC GitHub 仓库

下一步计划

目前,Fast-GRPC 支持的功能还比较简单,未来将继续改进和完善。如果您有任何建议或意见,请不要犹豫,欢迎提交 issue 或者 PR。

热门相关:最强狂兵   学霸女神超给力   夫人,你马甲又掉了!   战神   夫人,你马甲又掉了!