go gRPC 服务端推送给客户端流demo

最新推荐文章于 2022-09-03 12:11:58 发布

白小白的小白

最新推荐文章于 2022-09-03 12:11:58 发布

阅读量1.7k

点赞数

CC 4.0 BY-SA版权

分类专栏： go 文章标签： golang 开发语言后端

本文链接：https://blog.csdn.net/qq_42874635/article/details/126675605

go 专栏收录该内容

36 篇文章

订阅专栏

该博客展示了如何使用Go语言实现gRPC服务，特别是服务端如何以流的形式向客户端推送数据。在`hello.proto`中定义了`Greeter`服务和`GetStream`流式RPC方法。客户端通过`grpc.Dial`建立连接，调用`GetStream`获取服务端推送的数据流，并使用循环接收并打印。服务端则实现了`GetStream`方法，循环发送当前时间戳到客户端，当发送超过10次后停止。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

具体文件目录看上一篇的grpc，这个demo演示的是服务端以流的形式推送给客户端
pb/hello.proto:

syntax = "proto3";

option go_package = "./;pb";

package pb;

service Greeter{
    rpc GetStream (StreamReqData) returns (stream StreamResData) {}
}

// 请求数据格式
message StreamReqData{
    string data = 1;
}

// 返回数据格式
message StreamResData{
    string data = 1;
}

client/main.go:

package main

import (
	"context"
	pb "demo01/pb"
	"log"
	"google.golang.org/grpc"
	_ "google.golang.org/grpc/balancer/grpclb"
)
const (
	ADDRESS = "localhost:50051"
)

func main(){
	// 建立连接
	conn, err := grpc.Dial(ADDRESS, grpc.WithInsecure())
	if err != nil {
		return
	}
	defer conn.Close()
	// 通过刚才的链接 生成一个client对象
	c := pb.NewGreeterClient(conn)

	// 通过服务端推送流
	reqstreamData := &pb.StreamReqData{Data:"aaa"}

	res, _ := c.GetStream(context.Background(),reqstreamData)
	for {
		aa, err := res.Recv()
		if err != nil {
			log.Println(err)
			return
		}
		log.Println(aa)
	}
}

server/main.go:

package main

import (
	pb "demo01/pb"
	"fmt"
	"net"
	"time"
	"google.golang.org/grpc"
)

const (
	port = ":50051"
)

// 服务对象
type server struct {
	pb.UnimplementedGreeterServer
}

func (s *server) GetStream(req *pb.StreamReqData,res pb.Greeter_GetStreamServer) error{
	i := 0
	for {
		i++
		res.Send(&pb.StreamResData{Data: fmt.Sprintf("%v",time.Now().Unix())})
		time.Sleep(time.Second)
		if i > 10 {
			break
		}
	}
	return nil

}

func main(){
	lis, err := net.Listen("tcp", port)	
	if err != nil {
		return 
	}
	// 创建一个grpc 服务器
	s := grpc.NewServer()
	// 注册事件
	pb.RegisterGreeterServer(s, &server{})
	// 处理链接
	s.Serve(lis)
}