grpc学习golang版（七、客户端流式传输）

原创

Donkor 2024-07-09 10:48:33 博主文章分类：grpc学习golang版 ©著作权

©著作权归作者所有：来自51CTO博客作者Donkor的原创作品，请联系作者获取转载授权，否则将追究法律责任

一、前言

在上一章中我们讲到了服务器流式传输，与应用场景。接下来本文介绍的是客户端流式传输。主要应用场景就是客户端向服务端源源不断的发送数据，服务端保存数据。比如上传视频、文件、图片等等。

二、定义proto文件

新建stream.proto文件

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

//定义个流服务，叫什么名字无所谓
service ClientStream {
  //上传载文件，关键字stream
  rpc UploadFile(stream FileRequest)returns(Response){}
}
//请求参数
message FileRequest{
  //字节数据类型，即文件内容、数据
  bytes content = 1;
}

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

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

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

目录结构变更后为

grpc学习golang版（七、客户端流式传输）_服务端

具体步骤如下:

1）定义回调message结构体FileRequest，使用bytes数据类型
2）定义ClientStream服务
3）在服务里面，定义rpc方法UploadFile，使用关键词stream用于FileRequest结构体

三、拷贝任意文件进项目

任意文件，可以是视频、音频、图片、文档等等，不做限制。这里以一张png图片为例。新建static目录，拷贝一张grpc.png到static目录图片如下:

grpc学习golang版（七、客户端流式传输）_服务端_02

目录结构如下

grpc学习golang版（七、客户端流式传输）_grpc_03

四、编写server服务端

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

grpc学习golang版（七、客户端流式传输）_golang_04

编写server/main.go文件

package main

import (
	"bufio"
	"fmt"
	"go_grpc_study/example_5/grpc_proto/stream_proto"
	"google.golang.org/grpc"
	"io"
	"log"
	"net"
	"os"
)

// 新版本 gRPC 要求必须嵌入 UnimplementedGreeterServer 结构体
type ClientStream struct {
	stream_proto.UnimplementedClientStreamServer
}

func (ClientStream) UploadFile(stream stream_proto.ClientStream_UploadFileServer) error {
	//os.O_CREATE ： 创建并打开一个新文件
	//os.O_TRUNC ：打开一个文件并截断它的长度为零（必须有写权限）
	//os.O_WRONLY ：以只写的方式打开
	file, err := os.OpenFile("../static/grpc_x.png", os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0666)
	if err != nil {
		log.Fatalln(err)
	}
	defer file.Close()

	writer := bufio.NewWriter(file)
	var index int
	for {
		index++
		response, errRecv := stream.Recv()
		if errRecv == io.EOF {
			break
		}
		writer.Write(response.Content)
		fmt.Printf("第 %d 次写入数据\n", index)
	}
	writer.Flush()
	stream.SendAndClose(&stream_proto.Response{Message: "服务端接接收到上传的文件了"})
	return nil
}

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

具体步骤如下:

1）定义1个结构体，结构体名称无所谓，必须包含stream_proto.UnimplementedClientStreamServer 对象
2）实现 .proto文件中定义的API，即UploadFile 上传文件方法
3）通过stream对象的Recv()方法得到[]byte数据，循环读取
4）通过缓冲写的方式另存为新的图片grpc_x.png
5）将服务描述及其具体实现注册到 gRPC 中

五、编写client客户端

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

grpc学习golang版（七、客户端流式传输）_grpc_05

编写clinet/main.go文件

package main

import (
	"context"
	"fmt"
	"go_grpc_study/example_5/grpc_proto/stream_proto"
	"google.golang.org/grpc"
	"google.golang.org/grpc/credentials/insecure"
	"io"
	"log"
	"os"
)

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 := stream_proto.NewClientStreamClient(conn)
	stream, err := client.UploadFile(context.Background())

	//分片读的方式读取图片
	file, err := os.Open("../static/grpc.png")
	if err != nil {
		log.Fatalf(fmt.Sprintf("open file err [%s]", err))
	}

	defer file.Close()
	for {
		buf := make([]byte, 1024)
		_, err = file.Read(buf)
		if err == io.EOF {
			break
		}
		if err != nil {
			break
		}
		stream.Send(&stream_proto.FileRequest{
			Content: buf,
		})
	}

	response, err := stream.CloseAndRecv()
	fmt.Println(response, err)
}

具体步骤如下：