Golang gRPC的四种数据流

四种数据流

1. 简单模式
   这种模式最为传统，即客户端发起一次请求，服务端响应一个数据
2. 服务器数据流
   这种模式是客户端发起一次请求，服务端返回一段连续的数据流。典型的例子是客户端向服务端发送一个股票代码，服务端就把该股票的实时数据源源不断的返回给客户端。
3. 客户端数据流
   与服务端数据流模式相反，这次是客户端源源不断的向服务端发送数据流，而在发送结束后，由服务端返回一个响应。典型的例子是物联网终端向服务器报送数据。
4. 双向数据流
   顾名思义，这是客户端和服务端都可以向对方发送数据流，这个时候双方的数据可以同时互相发送，也就是可以实现实时交互。典型的例子是聊天机器人。

这里只会讲 grpc 中的 * stream*，srteam 顾名思义 就是 一种 流，可以源源不断的推送 数据，很适合 传输一些大数据，或者 服务端 和 客户端 长时间 数据交互，比如 客户端 可以向 服务端 订阅 一个数据，服务端 就 可以利用 stream ，源源不断地 推送数据。

proto

代码

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

service Greeter {
  // 服务端流模式
  rpc GetStream(StreamRequestData) returns (stream StreamResponseData);
  // 客户端流模式
  rpc PutStream(stream StreamRequestData) returns (StreamResponseData);
  // 双向流模式
  rpc AllStream(stream StreamRequestData) returns (stream StreamResponseData);
}

message StreamRequestData {
  string data = 1;
}

message StreamResponseData{
  string data = 1;
}

生成代码

• go

protoc -I . stream.proto --go_out=plugins=grpc:.

• python

python -m grpc_tools.protoc --python_out=. --grpc_python_out=. -I.  stream.proto

服务端

package main

import (
	"demo1/grpc_stream/proto"
	"fmt"
	"google.golang.org/grpc"
	"io"
	"net"
	"sync"
	"time"
)

const ADDRESS = "localhost:50051"

// 必须实现 proto 中定义的所有方法
type Server struct{}

// 服务器数据流模式 服务端不断的发送数据给客户端
func (s *Server) GetStream(request *proto.StreamRequestData, stream proto.Greeter_GetStreamServer) error {
	for i := 0; i < 10; i++ {
		// 不断的返回时间戳
		_ = stream.Send(&proto.StreamResponseData{
			Data: fmt.Sprintf(request.Data+": %v", time.Now().Unix()),
		})
		time.Sleep(time.Second)
	}
	return nil
}

// 客户端流模式 客户端不断给服务端发送数据
func (s *Server) PutStream(stream proto.Greeter_PutStreamServer) error {
	for {
		r, err := stream.Recv()
		if err == io.EOF {
			fmt.Println("EOF")
			break
		} else if err != nil {
			panic(err)
		}
		fmt.Println("客户端流模式:" + r.Data)
	}
	return nil
}

func (s *Server) AllStream(stream proto.Greeter_AllStreamServer) error {
	// 因为发送和接收数据是并行的，不能因为接受或发送数据阻塞，所以用到协程
	wg := sync.WaitGroup{}
	wg.Add(2)
	//接受数据
	go func() {
		for {
			data, err := stream.Recv()
			if err == io.EOF {
				fmt.Println("EOF")
				break
			} else if err != nil {
				panic(err)
			}
			fmt.Println("接收到客户端数据：" + data.Data)
		}
		wg.Done()
	}()
	// 发送数据
	go func() {
		for i := 0; i < 10; i++ {
			err := stream.Send(&proto.StreamResponseData{
				Data: fmt.Sprintf("我是服务端%d！", i),
			})
			if err != nil {
				panic(err)
			}
			time.Sleep(time.Second)
		}
		wg.Done()
	}()
	wg.Wait()
	return nil
}

func main() {
	listener, err := net.Listen("tcp", ADDRESS)
	if err != nil {
		panic("failed Listener:" + err.Error())
	}
	g := grpc.NewServer()
	proto.RegisterGreeterServer(g, &Server{})
	err = g.Serve(listener)
	if err != nil {
		panic(err)
	}
}

客户端

package main

import (
	"context"
	"demo1/grpc_stream/proto"
	"fmt"
	"google.golang.org/grpc"
	"google.golang.org/grpc/credentials/insecure"
	"io"
	"sync"
	"time"
)

const ADDRESS = "localhost:50051"

// 服务端流模式
func GetStream(c proto.GreeterClient) {
	rsp, _ := c.GetStream(context.Background(), &proto.StreamRequestData{Data: "lzscxb"})
	for {
		// 不断的接收数据
		r, err := rsp.Recv()
		// 当没有数据时结束
		if err == io.EOF {
			fmt.Println("EOF")
			break
		} else if err != nil {
			panic(err)
		}
		fmt.Println(r.Data)
	}
}

// 客户端流模式
func PutStream(c proto.GreeterClient) {
	rsp, _ := c.PutStream(context.Background())
	for i := 0; i < 10; i++ {
		_ = rsp.Send(&proto.StreamRequestData{
			Data: fmt.Sprintf("%v", time.Now().Unix()),
		})
		time.Sleep(time.Second)
	}
	// 关闭发送
	_ = rsp.CloseSend()
}

// 双向数据流模式
func AllStream(c proto.GreeterClient) {
	conn, err := c.AllStream(context.Background())
	if err != nil {
		panic("the connect failed! " + err.Error())
	}
	wg := sync.WaitGroup{}
	wg.Add(2)
	// 接受数据
	go func() {
		for {
			r, err := conn.Recv()
			if err == io.EOF {
				fmt.Println("EOF")
				break
			} else if err != nil {
				panic(err)
			}
			fmt.Println("收到服务器发送数据：" + r.Data)
		}
		wg.Done()
	}()
	// 发送数据
	go func() {
		for i := 0; i < 10; i++ {
			err = conn.Send(&proto.StreamRequestData{
				Data: fmt.Sprintf("我是客户端%v！", i),
			})
			if err != nil {
				panic(err)
			}
			time.Sleep(time.Second)
		}
		_ = conn.CloseSend()
		wg.Done()
	}()
	wg.Wait()
}

func main() {
	conn, err := grpc.Dial(ADDRESS, grpc.WithTransportCredentials(insecure.NewCredentials()))
	if err != nil {
		panic("the connect failed! " + err.Error())
	}
	defer conn.Close()
	c := proto.NewGreeterClient(conn)
	GetStream(c)
	PutStream(c)
	AllStream(c)
}