用一个例子说说gRPC的四种服务方法

时间:2024-01-26 16:58:42

本文通过一个简单的例子来演示这4种类型的使用方法

案例代码:https://github.com/codeAB/grpc-sample-example

目录结构说明
├── calculator.proto # 定义 protobuf 
├── client 
│   ├── client.go # 客户端
│   ├── gencode
│   │   └── calculator.pb.go # protoc-gen-go中间件生成的go中间代码
│   ├── go.mod
│   └── go.sum
├── README.md
└── server
    ├── gencode
    │   └── calculator.pb.go # 和上面一样
    ├── go.mod
    ├── go.sum
    └── server.go # 服务端

client和server是两个完全独立的项目,所以protoc-gen-go生成的中间代码并没有公用,
而是放到各自单独的项目中,实际上两者内容是完全一样的

首先来看 calculator.proto

syntax = "proto3";

package rpc.calculator.test;

service Calculator {
    //简单模式
    rpc Add (TwoNum) returns (Response) {}

    //服务端流
    rpc GetStream (TwoNum) returns (stream Response) {}

    //客户端流
    rpc PutStream (stream OneNum) returns (Response) {}

    //双向流
    rpc DoubleStream (stream TwoNum) returns (stream Response) {}
}

message TwoNum {
    int32 a = 1;
    int32 b = 2;
}
message Response {
    int32 c = 1;
}
message OneNum{
    int32 a = 1;
}

这里定义了一个计算器服务,包含四个方法,这个四个方法分别演示了个RPC的四种服务类型
下面分别介绍这个四个方法是做什么的:
1. 客户端发送一个请求,包含两个数字,服务端是返回两个数字的和,这种最基本也是最常用的叫简单模式,
客户端发送一次,服务端返回一次,类似于接口请求

2. 客户端发送一个请求包含两个数字,服务端返回多次,第一次返回两数子和,第二次返回两数字乘,
这种叫服务端流模式,形象点说就是服务端向客户端搭了一根单向水管,可以不停的往客户端发送数据,
客户端不停的接收,直到接收到服务端发送的结束标记后停止接收
使用场景:
客户端请求一个数据列表,但是这个数据太多了,不可能一次返回,就可以利用这种模式,
服务端一次返回100条数据,前端一直接收处理

3. 客户端发送了很多次数据,数据是单个数字,服务端不停的接收数据,客户端发送结束后服务端返回所有数据的总和,
这就是客户端流模式,形象点说就是客户端往服务端搭了一个单向的水管,可以不停的往服务端发送数据,
服务端不停的接收,直到接收到客户端发送的结束标记后停止接收,处理完数据后一次性返回给客户端,

4. 客户端分多次发送数据给服务端,每次数据是两个数,服务端收到数据后多次返回给服务端,每次返回两个数之和,
客户端可以多次给服务端发送数据,服务端可以多次返回数据,这就是双向流模式,双方都需要同时处理发送数据和接收数据,
直到双方通道流关闭

利用calculator.proto生成go中间代码
在项目跟目录执行:

protoc -I ./ ./calculator.proto  --go_out=plugins=grpc:./server/gencode
protoc -I ./ ./calculator.proto  --go_out=plugins=grpc:./client/gencode

分别在 ./server/gencode 和 ./client/gencode 两个目录生成了相同的go中间代码

有了中间代码后就可以编写服务端server.go了,由于篇幅限制,这里就不贴完整代码了,只对几个关键点做分析
导入中间代码

import pb "rpcserver/gencode"

定义了一个server,继承自 UnimplementedCalculatorServer

type server struct {
    pb.UnimplementedCalculatorServer
}

这里有个技巧,打开calculator.pb.go你会发现,
UnimplementedCalculatorServer类已经"实现"了我们在calculator.proto中定义的所有个方法

type UnimplementedCalculatorServer struct {
}
func (*UnimplementedCalculatorServer) Add(ctx context.Context, req *TwoNum) (*Response, error) {
    return nil, status.Errorf(codes.Unimplemented, "method Add not implemented")
}
....

如果你是新手,可以直接粘贴到server,稍作修改就可以了,函数体还是得自己写

func (s *server) Add(ctx context.Context, in *pb.TwoNum) (*pb.Response, error) {
    return &pb.Response{C: in.A + in.B}, nil
}

其实gRPC的函数不多,定义也非常清晰,比如在写下面这个方法的时候

func (s *server) GetStream(in *pb.TwoNum, pipe pb.Calculator_GetStreamServer) error {
    _ = pipe.Send(&pb.Response{C: in.A + in.B})
    time.Sleep(time.Second * 2)
    _ = pipe.Send(&pb.Response{C: in.A * in.B})
    return nil
}

我也不知道有Send方法,如果用的GoLand,pipe点一下,能用的函数就出来了,就几个,一看就知道什么作用,
全程都没有用到godoc,当然我还是推荐你仔细看一下文档 https://godoc.org/google.golang.org/grpc

在来看一下客户端client.go
导入中间代码

import pb "rpcclient/gencode"

客户端感觉没什么重点可以说,直接看代码吧,每个函数都有相应的备注