07.grp数据流模式-客户端发送-单向流

1.proto

syntax = "proto3";
option go_package = ".;proto";

service Geenter {
    // 客户端推送流
    rpc ClientPutStream (stream putRequest) returns (putResponse);
}

message putRequest {
    string data = 1;
}

message putResponse {
    string data = 1;
}

2.server

package main

import (
	"fmt"
	"grpctest2/stream_grpc/client_put_stream/proto"
	"net"

	"google.golang.org/grpc"
)

type Server struct{}

func (s *Server) ClientPutStream(clistr proto.Geenter_ClientPutStreamServer) error {
	for {
		d, err := clistr.Recv()
		if err != nil {
			return err
		}
		fmt.Println(d)
	}
}
func main() {
	// 创建一个grpc服务器
	g := grpc.NewServer()
	// 注册事件
	proto.RegisterGeenterServer(g, &Server{})
	// 监听端口
	conn, _ := net.Listen("tcp", "127.0.0.1:8080")
	// 处理连接
	g.Serve(conn)
}

3.client

package main

import (
	"context"
	"grpctest2/stream_grpc/client_put_stream/proto"
	"time"

	"google.golang.org/grpc"
)

func main() {
	//grpc 拨号
	conn, _ := grpc.Dial("127.0.0.1:8080", grpc.WithInsecure())
	defer conn.Close()
	//拨号生成的连接创建一个client对象
	c := proto.NewGeenterClient(conn)
	// client端推送流
	putRes, _ := c.ClientPutStream(context.Background())
	i := 0
	for {
		i++
		putRes.Send(&proto.PutRequest{Data: "sss"})
		time.Sleep(time.Second)
		if i > 10 {
			break
		}
	}
}