gRPC入门指南 — 双向流式RPC(四)
Go语言精选
共 6190字,需浏览 13分钟
· 2021-07-31
前言
前一篇文章我们学习了客户端流式 RPC,客户端多次向服务端发送数据,发送结束之后,由服务端返回一个响应。与服务端流式 RPC类似,都只支持单项连续发送数据,今天我们要来学习双向流式 RPC 支持通信双方同时多次发送或接收数据。如下如所示:
新建并编译proto文件
新建 bidirectional_stream.proto 文件:
syntax = "proto3";
package proto;
// 定义流式请求信息
message StreamRequest{
// 参数类型 参数名称 标识号
string data = 1;
}
// 定义流响应信息
message StreamResponse{
int32 code = 1;
string value = 2;
}
// 定义我们的服务(可以定义多个服务,每个服务可以定义多个接口)
service StreamService{
// 双向流RPC,需要在请求、响应数据前加stream
rpc Record(stream StreamRequest) returns (stream StreamResponse){};
}
双向流式 RPC,定义方法时需要在请求值和返回值之前加上 stream。
进入 bidirectional_stream.proto 所在的目录,使用如下命令编译文件
protoc --go_out=plugins=grpc:. bidirectional_stream.proto
执行完成之后会生成 bidirectional_stream.pb.go 文件。
创建server端
package main
import (
pb "go-grpc-example/4-bidirectional_stream_rpc/proto"
"google.golang.org/grpc"
"io"
"log"
"net"
"strconv"
"time"
)
const (
Address string = ":8000"
Network string = "tcp"
)
// 定义我们的服务
type StreamService struct{}
// 实现 Record() 方法
func (s *StreamService) Record(srv pb.StreamService_RecordServer) error {
n := 1
for {
// 接收数据
req, err := srv.Recv()
if err == io.EOF {
return nil
}
if err != nil {
log.Fatalf("stream get from client err: %v", err)
return err
}
// 发送数据
err = srv.Send(&pb.StreamResponse{
Code: int32(n),
Value: "This is the " + strconv.Itoa(n) + " message",
})
if err != nil {
log.Fatalf("stream send to client err: %v", err)
return err
}
n++
log.Println("stream get from client: ", req.Data)
time.Sleep(1 * time.Second)
}
return nil
}
func main() {
// 1.监听端口
listener, err := net.Listen(Network, Address)
if err != nil {
log.Fatalf("listener err: %v", err)
}
log.Println(Address + " net.Listing...")
// 2.实例化gRPC实例
grpcServer := grpc.NewServer()
// 3.注册我们的服务
pb.RegisterStreamServiceServer(grpcServer, &StreamService{})
// 4.启动gRPC服务端
err = grpcServer.Serve(listener)
if err != nil {
log.Fatalf("grpc server err: %v", err)
}
}
在实现的 Record() 方法中,for() 循环里面读取客户端发送的消息并返回一个响应数据。
运行服务端:
go run server.go
输出:
:8000 net listening...
创建client端
package main
import (
"context"
pb "go-grpc-example/4-bidirectional_stream_rpc/proto"
"google.golang.org/grpc"
"io"
"log"
"strconv"
"time"
)
const Address = ":8000"
func main() {
// 1.连接服务端
conn, err := grpc.Dial(Address, grpc.WithInsecure())
if err != nil {
log.Fatalf("grpc conn err: %v", err)
}
defer conn.Close()
// 2.创建gRPC客户端
grpcClient := pb.NewStreamServiceClient(conn)
// 3.调用 Record() 方法获取流
stream, err := grpcClient.Record(context.Background())
if err != nil {
log.Fatalf("call record err: %v", err)
}
for i := 0; i < 5; i++ {
// 4.发送数据
err := stream.Send(&pb.StreamRequest{
Data: strconv.Itoa(i),
})
if err != nil {
log.Fatalf("stream send to server err: %v", err)
}
// 5.接收服务端发送过来的数据
resp, err := stream.Recv()
if err == io.EOF {
break
}
if err != nil {
log.Fatalf("stream get from server err: %v", err)
}
log.Printf("stream get from server,code:%v,value:%v", resp.GetCode(), resp.Value)
time.Sleep(1 * time.Second)
}
// 6.关闭流
err = stream.CloseSend()
if err != nil {
log.Fatalf("close stream err:%v", err)
}
}
客户端代码,在 for() 循环里面向服务端发送了 5 次消息,并接收服务端返回的数据,5次数据交互之后调用 CloseSend() 关闭流。
运行客户端:
go run client.go
客户端输出:
stream get from server,code:1,value:This is the 1 message
stream get from server,code:2,value:This is the 2 message
stream get from server,code:3,value:This is the 3 message
stream get from server,code:4,value:This is the 4 message
stream get from server,code:5,value:This is the 5 message
服务端输出:
stream get from client: 0
stream get from client: 1
stream get from client: 2
stream get from client: 3
stream get from client: 4
观察仔细的同学会注意到,客户端和服务端是交替输出的。
总结
这篇文章我们简单介绍了 gRPC 的双向流式 RPC,支持通信双方同时多次发送或接收数据。
推荐阅读
评论
金三银四你都拿到了哪些offer?
本文继续分享最新真实面经,希望对粉丝股东们有帮助,另外文末有直投大厂的机会,别错过。都说行情不好,可是有的同学拿到了很多Offer,各种对比;也有同学连面试都很难约到。“能拿到Offer的大多相似,拿不到的各有各的原因。”金三银四你拿到了哪些offer?也欢迎大家在评论区留言讨论,不要错过文末的企业
高级前端进阶
0
展讯平台手机重启问题分析指南
和你一起终身学习,这里是程序员Android经典好文推荐,通过阅读本文,您将收获以下知识点:一、 User 版本 默认开启 sysdump 方法二、插入SD卡 抓取Sysdump log三、 sysdump log 分析四、展讯平台抓取重启 串口log的方案五、展讯平台判断重启类型六、展讯平台关闭
程序员Android
0
2024跨屏营销指南
下载报告去公众号:硬核刘大 后台回复“ 跨屏营销”,即可下载完整PDF文件。更多报告内容,可加微信:chanpin628 领取。(ps:加过微信:yw5201a1 的不要再加,分享的内容一样,有一个号就行。)申明:报告版权 勾正科技&MMA
产品刘
0
大模型并行训练指南:通俗理解Megatron-DeepSpeed之模型并行与数据并行(下)
文末《大模型项目开发线上营》秒杀倒计时↓↓↓接前文:(上)篇>>>大模型并行训练指南:通俗理解Megatron-DeepSpeed之模型并行与数据并行(上)(中)篇>>>大模型并行训练指南:通俗理解Megatron-DeepSpeed之模型并行与数据并行(中)06
七月在线实验室
10
神经网络调参指南
点击上方“小白学视觉”,选择加"星标"或“置顶”重磅干货,第一时间送达Author:夕小瑶From:夕小瑶的卖萌屋序言虽然现在仅仅靠调参已经在深度学习领域不是香饽饽了,但是如果连参数都不会调,那可能连肉汤都喝不到的。毕竟你有再好的idea,也需要有一个漂亮的实验结果去支撑的对不对,参数调不好,千里马
小白学视觉
10
CPU的入门知识
不管你玩硬件还是做软件,你的世界都少不了计算机最核心的 —— CPU。01CPU是什么?CPU与计算机的关系就相当于大脑和人的关系,它是一种小型的计算机芯片,通常嵌入在电脑的主板上。CPU的构建是通过在单个计算机芯片上放置数十亿个微型晶体管来实现。这些晶体管使它能够执行运行存储在系统内存中的程序所需
机器学习算法与Python实战
10
C#中HttpClient的演进与避坑指南
引子 在互联网时代,与网络通信相关的功能已成为不可或缺的一部分。C# 中的 HttpClient 类是一个强大的工具,用于执行 HTTP 请求和处理响应。虽然HttpClient 类简单易用,但在实际应用中,仍然存在一些需要注意...
llovebo
0
10个python爬虫入门实例
涉及主要知识点:web是如何交互的requests库的get、post函数的应用response对象的相关函数,属性python文件的打开,保存代码中给出了注释,并且可以直接运行哦如何安装requests库(安装好python的朋友可以直接参考,没...
马哥Linux运维
0