08.grp数据流模式-双向流

1.proto

syntax = "proto3";

option go_package = ".;proto";

service Greenter {
    rpc PGStream (stream allStreamRequest) returns (stream allStreamResponse);
}

message allStreamRequest {
    string data = 1;
}

message allStreamResponse {
    string data = 1;
}

2.server

package main

import (
	"grpctest2/stream_grpc/all_stream/proto"
	"log"
	"net"
	"sync"

	"google.golang.org/grpc"
)

type Server struct{}

func (s *Server) PGStream(allStr proto.Greenter_PGStreamServer) error {
	wg := sync.WaitGroup{}
	wg.Add(2)
	go func() {
		for {
			data, _ := allStr.Recv()
			log.Println(data)
		}
		wg.Done()
	}()
	go func() {
		for {
			allStr.Send(&proto.AllStreamResponse{Data: "qweqwe"})
		}
		wg.Done()
	}()
	wg.Wait()
	return nil
}

func main() {
	// 创建GRPC服务器
	g := grpc.NewServer()
	// 注册事件
	proto.RegisterGreenterServer(g, &Server{})
	// 监听端口
	conn, _ := net.Listen("tcp", "127.0.0.1:8090")
	g.Serve(conn)
}

3.client

package main

import (
	"context"
	"grpctest2/stream_grpc/all_stream/proto"
	"log"
	"time"

	"google.golang.org/grpc"
)

func main() {
	// 拨号得到连接
	conn, _ := grpc.Dial("127.0.0.1:8090", grpc.WithInsecure())
	defer conn.Close()
	// 通过连接创建client对象
	c := proto.NewGreenterClient(conn)
	// 调用服务推送流
	requestData := &proto.AllStreamRequest{Data: "asd"}
	allstr, _ := c.PGStream(context.Background())
	go func() {
		for {
			allstr.Send(requestData)
			time.Sleep(time.Second)
		}
	}()
	go func() {
		for {
			data, _ := allstr.Recv()
			log.Println(data)
		}
	}()
	select{}
}