- 微信
- 微博
  
  分享文章到微博
- 复制链接
  
  复制链接到剪贴板

gRPC双向数据流的交互控制（go语言实现）

feichaiyu 发表于 2019/12/16 00:25:55 2019/12/16

【摘要】 ##gRPC简介 gRPC (https://grpc.io) 是一个由Google开发的高性能、开源、跨多种编程语言和通用的远程过程调用协议(RPC) 框架，用于客户端和服务器端之间的通信，使用HTTP/2协议并将 ProtoBuf (https://developers.google.com/protocol-buffers)作为序列化工具。##gRPC模式 gRPC主要有4种请求／响应...

##gRPC简介 gRPC (https://grpc.io) 是一个由Google开发的高性能、开源、跨多种编程语言和通用的远程过程调用协议(RPC) 框架，用于客户端和服务器端之间的通信，使用HTTP/2协议并将 ProtoBuf (https://developers.google.com/protocol-buffers)作为序列化工具。

##gRPC模式 gRPC主要有4种请求／响应模式，分别是：

(1) 简单模式（Simple RPC）

这种模式最为传统，即客户端发起一次请求，服务端响应一个数据，这和大家平时熟悉的RPC没有什么大的区别，所以不再详细介绍。

(2) 服务端数据流模式（Server-side streaming RPC）

这种模式是客户端发起一次请求，服务端返回一段连续的数据流。典型的例子是客户端向服务端发送一个股票代码，服务端就把该股票的实时数据源源不断的返回给客户端。

(3) 客户端数据流模式（Client-side streaming RPC）

与服务端数据流模式相反，这次是客户端源源不断的向服务端发送数据流，而在发送结束后，由服务端返回一个响应。典型的例子是物联网终端向服务器报送数据。

(4) 双向数据流模式（Bidirectional streaming RPC）

顾名思义，这是客户端和服务端都可以向对方发送数据流，这个时候双方的数据可以同时互相发送，也就是可以实现实时交互。典型的例子是聊天机器人。

##双向数据流实战在gRPC中文文档(http://doc.oschina.net/grpc?t=60133)中有上述4种模式的实例，但是其中双向数据流的例子过于简单，没有体现出双向控制的特点，所以本文创建一个新的例子（用golang实现），用以展示gRPC双向数据流的交互（关于proto如何定义、相关包如何安装，在文档中都有介绍，所以本文略去此部分）。

###1、proto定义 syntax = "proto3"; // 语法使用 protocol buffer proto3

###服务端程序 server.go package main

import (	"io"
	"log"
	"net"
	"strconv"
	"google.golang.org/grpc"
	proto "chat" // 自动生成的 proto代码)// Streamer 服务端type Streamer struct{}// BidStream 实现了 ChatServer 接口中定义的 BidStream 方法func (s *Streamer) BidStream(stream proto.Chat_BidStreamServer) error {
	ctx := stream.Context()	for {		select {		case <-ctx.Done():
			log.Println("收到客户端通过context发出的终止信号")			return ctx.Err()		default:			// 接收从客户端发来的消息
			输入, err := stream.Recv()			if err == io.EOF {
				log.Println("客户端发送的数据流结束")				return nil
			}			if err != nil {
				log.Println("接收数据出错:", err)				return err
			}			// 如果接收正常，则根据接收到的 字符串 执行相应的指令
			switch 输入.Input {			case "结束对话\n":
				log.Println("收到'结束对话'指令")				if err := stream.Send(&proto.Response{Output: "收到结束指令"}); err != nil {					return err
				}				// 收到结束指令时，通过 return nil 终止双向数据流
				return nil

			case "返回数据流\n":
				log.Println("收到'返回数据流'指令")				// 收到 收到'返回数据流'指令， 连续返回 10 条数据
				for i := 0; i < 10; i++ {					if err := stream.Send(&proto.Response{Output: "数据流 #" + strconv.Itoa(i)}); err != nil {						return err
					}
				}			default:				// 缺省情况下， 返回 '服务端返回: ' + 输入信息
				log.Printf("[收到消息]: %s", 输入.Input)				if err := stream.Send(&proto.Response{Output: "服务端返回: " + 输入.Input}); err != nil {					return err
				}
			}
		}
	}
}func main() {
	log.Println("启动服务端...")
	server := grpc.NewServer()	// 注册 ChatServer
	proto.RegisterChatServer(server, &Streamer{})

	address, err := net.Listen("tcp", ":3000")	if err != nil {		panic(err)
	}	if err := server.Serve(address); err != nil {		panic(err)
	}
}

###客户端程序 client.go package main