Python和Go实现简单Grpc服务

一、简介

Grpc 使用了 Google 的 Protocol Buffers 作为接口定义语言(IDL),并使用 HTTP/2 作为传输协议。它支持多种编程语言,包括 C++、Java、Python、Go、Node.js 等。Grpc 提供了强大的功能,如双向流、流式处理、身份验证和拦截器等。

Grpc 的核心概念是服务和消息。服务定义了一组方法,客户端可以通过这些方法与服务端进行交互。消息定义了数据的结构,用于在服务和客户端之间传递。

二、安装proto

下载proto文件:

Python和Go实现简单Grpc服务

安装后设置系统环境变量,然后打开控制台,执行protoc -h命令验证是否安装成功。

三、初始化工程

Python中使用Grpc

初始化工程

mkdir grpc_demo
cd grpc_demo
pdm init

安装Python工具库

pdm add grpcio
pdm add grpcio-tools

Go中使用Grpc

初始化工程

mkdir grpc_demo
cd grpc_demo
go mod init grpc_demo

安装Go工具库

go install github.com/golang/protobuf/protoc-gen-go@latest
go install google.golang.org/grpc/cmd/protoc-gen-go-grpc@latest

四、定义服务和消息

在 Grpc 中,使用 Protocol Buffers(简称 Protobuf)来定义服务和消息的结构。Protobuf 是一种轻量级的数据交换格式,它可以定义结构化数据的模式,并生成相应的代码用于序列化和反序列化。

定义服务和消息的步骤如下:

  1. 创建proto文件夹:proto,将对应的proto文件创建在里面
syntax = "proto3";

option go_package = ".;proto";

service Greeter {
  rpc SayHello (HelloRequest) returns (HelloReply) {}
}

message HelloRequest {
  string name = 1;
}

message HelloReply {
  string message = 1;
}
  1. 使用 Protobuf 编译器将 .proto 文件编译成所需语言的代码。
  • 在Python中运行以下命令:
pdm run python -m grpc_tools.protoc --python_out=. --grpc_python_out=. -I . hello.proto
- —python_out:生成的Python文件存储目录。
- —grpc_python_out:生成的Grpc使用的Python文件存储目录。
- -I:proto文件的目录,这里的目录地址将影响proto文件相互引用的路径。

执行完以上命令后,会在指定目录下生成:`hello_pb2.py`和`hello_pb2_grpc.py` 两个文件。
  • 在Go中运行以下命令:
protoc --go_out=. --go-grpc_out=. -I . hello.proto
- —go_out:生成的Go文件存储目录。
- —go-grpc_out:生成的Grpc使用的Go文件存储目录。
- -I:proto文件的目录。

执行完以上命令后,会在指定目录下生成:`hello_pb.g`o和`hello_grpc.pb.go` 两个文件。

五、编写服务端

在 Grpc 中,编写服务端需要执行以下几个步骤:

  1. 导入所需的库和生成的消息定义文件:

Go实现:

package main

import (
  "context"
  "go_demo/proto"
  "google.golang.org/grpc"
  "net"
)

Python实现:

import grpc
from concurrent import futures
from proto import hello_pb2_grpc, hello_pb2
  1. 创建一个类继承自自动生成的服务定义类,并实现其中定义的方法:

Go实现:

type Server struct {
  proto.UnimplementedGreeterServer
}

func (s *Server) SayHello(ctx context.Context, in *proto.HelloRequest) (*proto.HelloReply, error) {
  return &proto.HelloReply{Message: "Hello Go " + in.Name}, nil
}

Python实现:

class GreeterServicer(hello_pb2_grpc.GreeterServicer):
    def SayHello(self, request, context):
        return hello_pb2.HelloReply(message='Hello, %s!' % request.name)
  1. 创建一个 gRPC 服务器并将服务实现添加到服务器中:

Go实现:

func serve() {
    g := grpc.NewServer()
    proto.RegisterGreeterServer(g, &Server{})
    lis, err := net.Listen("tcp", ":50051")
    if err != nil {
      panic("failed to listen: " + err.Error())
    }
    err = g.Serve(lis)
    if err != nil {
      panic("failed to start serve: " + err.Error())
    }
}

Python实现:

def serve():
    server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
    hello_pb2_grpc.add_GreeterServicer_to_server(Greeter(), server)
    server.add_insecure_port('0.0.0.0:50051')
    server.start()
    server.wait_for_termination()
  1. 启动服务器:

Go实现:

func main() {
    serve()
}

Python实现:

if __name__ == '__main__':
    serve()

以上代码实现一个简单的Grpc服务端,其中SayHello是在服务定义中声明的方法,我们可以根据需求添加更多的方法和逻辑。

六、编写客户端

编写 Grpc 客户端的步骤如下:

  1. 导入所需的库和生成的消息定义文件:

Go实现:

package main

import (
  "context"
  "fmt"
  "go_demo/proto"
  "google.golang.org/grpc"
  "google.golang.org/grpc/credentials/insecure"
)

Python实现:

import grpc
from proto import hello_pb2_grpc, hello_pb2
  1. 创建一个Grpc通道:

Go实现:

conn, err := grpc.Dial("127.0.0.1:50051", grpc.WithTransportCredentials(insecure.NewCredentials()))
if err != nil {
  panic("failed to dial: " + err.Error())
}
defer conn.Close()

Python实现:

channel = grpc.insecure_channel('localhost:50051')
  1. 创建一个 Stub 对象,用于调用服务端的方法:

Go实现:

stub := proto.NewGreeterClient(conn)

Python实现:

stub = hello_pb2_grpc.GreeterStub(channel)
  1. 调用服务端的方法:

Go实现:

response, err := stub.SayHello(context.Background(), &proto.HelloRequest{Name: "world"})
if err != nil {
  panic("failed to say hello: " + err.Error())
}
fmt.Println(response.Message)

Python实现:

response = stub.SayHello(hello_pb2.HelloRequest(name='tom'))
print(response.message)

以上分别用Python和Go实现了简单的Grpc服务端和客户端。

文章版权声明

 1 原创文章作者:﹏ ノ stop",如若转载,请注明出处: https://www.52hwl.com/30453.html

 2 温馨提示:软件侵权请联系469472785#qq.com(三天内删除相关链接)资源失效请留言反馈

 3 下载提示:如遇蓝奏云无法访问,请修改lanzous(把s修改成x)

 免责声明:本站为个人博客,所有软件信息均来自网络 修改版软件,加群广告提示为修改者自留,非本站信息,注意鉴别

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2023年7月14日 上午12:00
下一篇 2023年7月15日