grpc学习golang版（八、双向流示例）

news2024/11/24 16:03:18

系列文章目录
第一章 grpc基本概念与安装
第二章 grpc入门示例
第三章 proto文件数据类型
第四章多服务示例
第五章多proto文件示例
第六章服务器流式传输
第七章客户端流式传输
第八章双向流示例

文章目录

一、前言
二、定义proto文件
三、编写server服务端
四、编写client客户端
五、测试
六、示例代码

一、前言

了解了rpc服务器流式传输，客户端流式传输，也知道了对应的使用场景。本文开始介绍双向流，双向流其实就是客户端源源不断的发数据、内容给服务端，服务端源源不断的发数据、内容给客户端。其应用场景，最常见的就是即时通讯。

二、定义proto文件

新建stream.proto文件

// 指定proto版本
syntax = "proto3";
// 指定默认包名
package each_proto;
// 指定golang包名
option go_package = "/each_proto";

//定义个流服务，叫什么名字无所谓
service EachStream {
  //关键字stream
  rpc Chat(stream Request)returns(stream Response){}
}
//请求参数
message Request{
  string content = 1;
}

//回调参数
message Response{
  string message =1;
}

在go_grpc_study/example_6/grpc_proto目录下新建Terminal，执行生成文件，命令如下

protoc --go_out=. --go-grpc_out=. ./stream.proto

目录结构变更后为

具体步骤如下:

1）定义请求message结构体Request，回调message结构体Response，使用string数据类型
2）定义EachStream服务
3）在服务里面，定义rpc方法Chat，使用关键词stream用于Request结构体、Response结构体

三、编写server服务端

新建server目录，新建main.go文件
目录结构如下

编写server/main.go文件

package main

import (
	"fmt"
	"go_grpc_study/example_6/grpc_proto/each_proto"
	"google.golang.org/grpc"
	"log"
	"net"
)

// 新版本 gRPC 要求必须嵌入 UnimplementedGreeterServer 结构体
type EachStream struct {
	each_proto.UnimplementedEachStreamServer
}

func (EachStream) Chat(stream each_proto.EachStream_ChatServer) error {
	for i := 0; i < 5; i++ {
		request, _ := stream.Recv()
		fmt.Println(request)
		stream.Send(&each_proto.Response{
			Message: fmt.Sprintf("第 %d 次回应你好", i+1),
		})
	}
	return nil
}

func main() {
	// 监听端口
	listen, err := net.Listen("tcp", ":8080")
	if err != nil {
		log.Fatal(err)
	}
	// 创建一个gRPC服务器实例。
	server := grpc.NewServer()
	// 将server结构体注册为gRPC服务。
	each_proto.RegisterEachStreamServer(server, &EachStream{})
	fmt.Println("grpc server running :8080")
	// 开始处理客户端请求。
	server.Serve(listen)
}

具体步骤如下:

1）定义1个结构体，结构体名称无所谓，必须包含each_proto.UnimplementedEachStreamServer 对象
2）实现 .proto文件中定义的API，即Chat 聊天方法
3）通过stream对象的Recv()方法得到客户端发送的数据，循环5次读取，并依次回应给客户端
4）将服务描述及其具体实现注册到 gRPC 中

四、编写client客户端

新建client目录，新建main.go文件
目录结构如下

编写clinet/main.go文件

package main

import (
	"context"
	"fmt"
	"go_grpc_study/example_6/grpc_proto/each_proto"
	"google.golang.org/grpc"
	"google.golang.org/grpc/credentials/insecure"
	"log"
)

func main() {
	addr := ":8080"
	// 使用 grpc.Dial 创建一个到指定地址的 gRPC 连接。
	// 此处使用不安全的证书来实现 SSL/TLS 连接
	conn, err := grpc.Dial(addr, grpc.WithTransportCredentials(insecure.NewCredentials()))
	if err != nil {
		log.Fatalf(fmt.Sprintf("grpc connect addr [%s] 连接失败 %s", addr, err))
	}
	defer conn.Close()
	// 初始化客户端
	client := each_proto.NewEachStreamClient(conn)
	stream, err := client.Chat(context.Background())

	for i := 0; i < 5; i++ {
		stream.Send(&each_proto.Request{
			Content: fmt.Sprintf("第% d 次 打招呼", i),
		})
		response, err := stream.Recv()
		fmt.Println(response, err)
	}
}