导语?|?gRPC也是RPC技术家族的一种,它由Google主导开发,是一个跨平台的调用框架,其中和go语言结合的是最紧密的,在go语言的开发和调用中占据主导地位。gRPC采用protobuf作为配置载体来实现通讯和调用。本文主要实战演示一下gRPC的几种调用通讯模式(普通、客户端流、服务端流、双向流)以及和PHP客户端的联通调用。
在学习gRPC之前,我们需要了解一下ptorobuf语法和protoc的命令,能帮助我们更加深入的学习和理解gRPC。
一、需求分析
我们这次只搞个很简单的需求,搞个用户server系统,提供2个接口给外部,1个是保存用户信息,1个是根据用户UID查询用户信息,就这2个,不搞复杂了。
简单的用代码描述下就是:
function1 saveUser(name, age) return (UID)
function1 getUserInfo(UID) return (UID, name, age)
需求分析结束。
二、编写protobuf文件
需求分析结束之后,我们明确了我们需要干什么了,我们需要对外提供2个gRPC接口,那么我们就开始编写protobuf吧:
先定义2个rpc服务函数:
service UserServer {
rpc SaveUser (UserParams) returns (Id) {}
rpc GetUserInfo (Id) returns (UserInfo) {}
}
定义好了rpc之后,我们知道,函数里面的参数和返回值得都得是message信息,我们就开始创建参数和返回值对应的3个message吧。
//用户ID
message Id {
int32 id = 1;
}
//名字
message Name {
string name = 1;
}
//年龄
message Age {
int32 age = 1;
}
// 用户信息
message UserInfo {
int32 id = 1;
string name = 2;
int32 age = 3;
}
// 用户参数
message UserParams {
Name name = 1;
Age age = 2;
}
上面的写法是,将age和name单独出来搞成2个message,然后在UserParams里面搞成嵌套message,目的是想看下嵌套message在不同语言中的用法。但是总体也是比较简单的。
完整的ptoro文件为:
//userServer.proto
syntax = "proto3";
package proto;
option go_package = ".;proto";
message ...
service ...
三、编译成go版本的服务端
和php版本的客户端文件
编写完成protobuf文件了,接下来就是编译成不同的语言了,我们将使用protoc命令,来编译生成不同语言的版本库。我们用go语言作为服务端语言,用php和go分别作为客户端语言,完成本次的调用。
如果你机器上还未安装protobuf工具,安装很简单,如果是Mac的话,一条命令就搞定了:
brew install protobuf
如果是其他系统,请参考官方文档,也很简单。安装好的服务,命令执行的关键字是protoc--go_out=xx xx。
(一)编译成go版本
先编译生成go版本的服务端和客户端:
protoc --go_out=plugins=grpc:. userServer.proto
需要注意的是我们需要加上gRPC的支持,生成gRPC服务的代码。如果你执行报错,可能是protoc-gen-go扩展没安装,安装很简单:
go get -u github.com/golang/protobuf/protoc-gen-go
它会在$GOPATH/bin目录下生成1个可执行的protoc-gen-go文件。所以,这个文件不要删了。不然你使用--go_out时,会找不到protoc-gen-go,提示报错。
执行完毕之后,就会在proto目录下生成userServer.pg.go的文件,里面将proto里的message都转换成了go语言的struct,并且也把RPC也转换生成了2个可调用的客户端函数。
我们看下生成后目录结构,生成的ph.go文件在proto目录下。
├── client
│ └── simple_client
│ └── client.go
├── go.mod
├── go.sum
├── proto
│ ├── userServer.pb.go
│ └── userServer.proto
└── server
└── simple_server
└── server.go
(二)编译成php客户端
我们在php里面去调用go提供的gRPC服务,那么php就是一个客户端,同理,在php里面使用,其实也需要编译这个protobuf文件,需要用到--php_out这个参数。
利用prcl快速安装protobuf和gRPC这2个php扩展
sudo pecl install protobuf
sudo pecl install grpc
下载gRPC源码,这一步是为了生成php的proto生成器,可以给我们生成client服务代码,当然你也可以自己去写代码,不用这个生成器。但是对于新手,我建议你下载安装。
git clone -b v1.34.0 https://github.com/grpc/grpc #可能需要一段时间
git submodule update --init #可能需要一段时间
make grpc_php_plugin
php生成器位置生成在:
/Users/small/www/grpc/bins/opt/grpc_php_plugin
我们把同一份userServer.proto文件,拷贝到我们的php环境目录下。然后执行命令,生成php和gRPC服务类php文件:
protoc -I=. userServer.proto --php_out=. --grpc_out=. --plugin=protoc-gen-grpc=/Users/smallyang/www/grpc/bins/opt/grpc_php_plugin
在项目更目录下新建composer.json文件
{
"name": "grpc-go-php",
"require": {
"grpc/grpc": "^v1.34.0",
"google/protobuf": "^v3.14.0"
},
"autoload": {
"psr-4": {
"GPBMetadata\\": "GPBMetadata/", //自动生成的php类文件夹
"Proto\\": "Proto/" //proto的文件夹
}
}
}
执行下载2个依赖的库。这2个库类似于sdk,它们封装了很多方法方便我们应用层去使用,它们底层会去调用上面用prcl生成的2个php扩展的方法。
composer install
这一通,完成后,我们看下目录结构:
├── GPBMetadata
│ └── UserServer.php
├── Proto
│ ├── Age.php
│ ├── Id.php
│ ├── Name.php
│ ├── UserInfo.php
│ ├── UserParams.php
│ └── UserServerClient.php
├── composer.json
├── composer.lock
├── main.php
├── userServer.proto
└── vendor
├── autoload.php
├── composer
├── google
│ └── protobuf
│
└── grpc
└─
其中GPBMetadata和Proto文件夹是自动生成的。vendor里的2个扩展也是composer自动下载生成的。
一切准备好了,go版本的服务端和客户端准备就绪。php版本的客户端也准备就绪。
四、普通调用模式
普通模式,也叫一元模式,它是最常见,也是使用做多的方式,和普通的http请求模式是一样的。
客户端像服务端发送1次请求,然后服务端给出响应结果。很简单的模式。
client --1---> server
client <--2--- server
那我们来看下,我们如何实现这种普通模式的调用。
(一)go语言的调用实现
我们先用go来实现,上面go的服务端和客户端代码都生成好了。我们先在server和client目录下,分别新建自己的代码:
├── client
│ └── simple_client
│ └── client.go
└── server
└── simple_server
└── server.go
我们先来完成server端的代码逻辑,server端上面的逻辑,主要分3方面:
新建tcp连接。
注册gRPC服务,并把它挂到tcp上。
完成对外提供的几个rpc方法的逻辑。
我们就按照这个逻辑来开始写:
package main
import (
"context"
"fmt"
"go-grpc-example/proto"
"google.golang.org/grpc"
"log"
"math/rand"
"net"
)
//新建1个结构体,下面绑定了2个方法,实现了UserServerServer接口。
type UserServer struct{}
func main() {
//监听tcp
listen, err := net.Listen("tcp", "127.0.0.1:9527")
if err != nil {
log.Fatalf("tcp listen failed:%v", err)
}
//新建gRPC
server := grpc.NewServer()
fmt.Println("userServer grpc services start success")
//rcp方法注册到grpc
proto.RegisterUserServerServer(server, &UserServer{})
//监听tcp
_ = server.Serve(listen)
}
//保存用户
//第一个参数是固定的context
func (Service *UserServer) SaveUser(ctx context.Context, params *proto.UserParams) (*proto.Id, error) {
id := rand.Int31n(100) //随机生成id 模式保存成功
res := &proto.Id{Id: id}
fmt.Printf("%+v", params.GetAge())
fmt.Printf("%+v\n", params.GetName())
return res, nil
}
//获取用户信息
//第一个参数是固定的context
func (Service *UserServer) GetUserInfo(ctx context.Context, id *proto.Id) (*proto.UserInfo, error) {
res := &proto.UserInfo{Id: id.GetId(), Name: "test", Age: 31}
return res, nil
}
2个rpc方法很简单,只是mock数据,并没有真实实现。后期有时间,再来实现吧。值得注意是,rpc的函数,第一个参数是固定的ctx context.Context,这是用于控制信号和超时的,是固定写法。有空我专门来搞一起context包的学习。
我们运行一下, gRPC服务启动成功:
$ go run server/simple_server/server.go
userServer grpc services start success
在来看看client端,如何利用生成的pb.go文件,来实现逻辑呢?也是一样
监听server启动的tcp的ip:端口。
新建连接client服务,并绑定tcp。
去调用这2个rpc的函数。
开始写逻辑:
package main
import (
"context"
"fmt"
"go-grpc-example/proto"
"google.golang.org/grpc"
"log"
)
var client proto.UserServerClient
func main() {
//链接tcp端口
connect, err := grpc.Dial("127.0.0.1:9527", grpc.WithInsecure())
if err != nil {
log.Fatalln(err)
}
//新建client
client = proto.NewUserServerClient(connect)
//调用
SaveUser()
GetUserInfo()
}
func SaveUser() {
params := proto.UserParams{}
params.Age = &proto.Age{Age: 31}
params.Name = &proto.Name{Name: "test"}
res, err := client.SaveUser(context.Background(), ¶ms)
if err != nil {
log.Fatalf("client.SaveUser err: %v", err)
}
fmt.Printf("%+v\n", res.Id)
}
func GetUserInfo() {
res, err := client.GetUserInfo(context.Background(), &proto.Id{Id: 1})
if err != nil {
log.Fatalf("client.userInfo err: %v", err)
}
fmt.Printf("%+v\n", res)
}
调用的rpc的方法,是pb.go文件里已经帮我们生成了,我们直接调用即可。这2个函数的参数和返回值,和刚在server定义的得保持一致。第一个参数得是context.Context。
我们测试一下client的代码,是否可以调通:
$ go run client/simple_client/client.go
23
id:1 name:"test" age:31
我们可以看到有返回值了,在server那边也有了打印:
$ go run server/simple_server/server.go
userServer grpc services start success
age:31name:"test"
age:31name:"test"
age:31name:"test"
age:31name:"test"
到此为止,go版本的客户端和服务端的gRPC通讯成功了。
(二)php语言的客户端调用
我们再用php来调用go的gRPC服务,看下具体是怎么操作。首先,我们在上面已经自动帮我们生成了gRPC的client的server类,那我们就直接调用。
操作逻辑,和go client的步骤是一样的,分成2步:
监听server启动的tcp的ip:端口,并直接新建连接client服务
去调用这2个rpc的函数
我们具体看下,代码应该怎么写:
<?php
//引入 composer 的自动载加
require __DIR__ . '/vendor/autoload.php';
//连接gRPC服务端
//并连接客户端
$client = new \\Proto\\UserServerClient('127.0.0.1:9527', [
'credentials' => Grpc\ChannelCredentials::createInsecure()
]);
//实例化 $UserParams 请求类
$UserParams = new \\Proto\\UserParams();
$age = new \\Proto\\Age();
$age->setAge("18");
$UserParams->setAge($age);
$name = new \\Proto\\Name();
$name->setName("jack");
$UserParams->setName($name);
//调用远程服务
/**
* @var $Id \Proto\Id
*/
list($Id, $status) = $client->SaveUser($UserParams)->wait();
var_dump($status, $Id->getId());
//实例化Id类
$Id = new \\Proto\\Id();
//赋值
$Id->setId("1");
//调用
/**
* @var $User \Proto\UserInfo
*/
list($UserInfo, $status) = $client->GetUserInfo($Id)->wait();
var_dump($status, $UserInfo->getId(), $UserInfo->getName(), $UserInfo->getAge());
调用方式,看上去写法有点难受,没有go简洁,这是因为,php的调用方式都是用类的方式呈现的,所以调用传值以及返回的都是对应的类。所以只能用setXXX()这种模式来赋值的,以及用getXXX()方式来获取值。
我们执行一下:
php client.php
0, 58
1, test, 31
我们再去go server端看下输出:
age:18name:"jack"
age:18name:"jack"
age:18name:"jack"
age:18name:"jack"
php作为客户端调用go的gRPC server成功!
五、Client-side streaming?RPC
客户端流模式调用
什么是客户端流呢?也就是客户端在一次请求中,不断的将内容像流水一样,传给服务端。而服务端,则是需要不段的循环获取数据。
client -1-> -2-> -3-> -4-> server
client <--------1---------- server
为什么会有这种场景,因为存在一个痛点,就是客户端大包发送。
所以,流的方式传输还是有很大的使用场景的。那我们先来看看,流式调用有啥不同的地方。
最大的不同就是在protobuf文件中,定义一个rpc函数时候,得加一个stream关键字,就表示这是一个流媒体的传输调用。
service UserServer {
rpc SaveUser (stream UserParams) returns (Id) {} //客户端流
rpc GetUserInfo (Id) returns (stream UserInfo) {} //服务端流
rpc DeleteUser(stream Id) returns (stream Status){} //双向流
}
上面我们定义了3个rpc方法,如果是在函数的参数前,加stream,表示是客户端发送流式的请求,反之,如果是返回的参数前面,加stream,表示是客户端发送流式的请求。同理,2个都加,则表示是双向的,2边都在流式的发送,这种情况就复杂一些,我们分别来试一试。
(一)go语言的客户端流调用
我们先完善一下protobuf文件,我们在proto目录下新建1个stream流式的文件,名字叫:streamArticleServer.proto:
syntax = "proto3";
package proto;
option go_package = ".;proto";
//ID
message Aid {
int32 id = 1;
}
//作者
message Author {
string author = 1;
}
//标题
message Title {
string title = 1;
}
//内容
message Content {
string content = 1;
}
// 文章信息
message ArticleInfo {
int32 id = 1;
string author = 2;
string title = 3;
string content = 4;
}
// 保存文章信息
message ArticleParam {
Author author = 2;
Title title = 3;
Content content = 4;
}
//删除状态
message Status{
bool code = 1;
}
// 声明那些方法可以使用rpc
service ArticleServer {
rpc SaveArticle (stream ArticleParam) returns (Aid) {}
rpc GetArticleInfo (Aid) returns (stream ArticleInfo) {}
rpc DeleteArticle(stream Aid) returns (stream Status){}
}
//执行 :protoc --go_out=plugins=grpc:. streamArticleServer.proto
然后,我们执行一下,生成对应的go服务端和客户端文件:
protoc --go_out=plugins=grpc:. streamArticleServer.proto
这样,就在proto目录下,生成了一个新的streamArticleServer.pb.go文件,gRPC需要的client和server相关的调用都在这里面生成好了。
由于,我们本次只是看客户端流调用,那么我们只看SaveArticle这个方法。
接下来,我们开始写client和server的调用代码。我们先来完成server端的代码逻辑,server端上面的逻辑,主要分3方面:
新建tcp连接。
注册gRPC服务,并把它挂到tcp上。
完成对外提供的几个rpc方法的逻辑。
这和普通的server是一样的,只是在处理具体的请求的时候,会用for循环
package main
//流式服务
import (
"fmt"
"go-grpc-example/proto"
"google.golang.org/grpc"
"log"
"net"
)
type StreamArticleServer struct {
}
func (server *StreamArticleServer) SaveArticle(stream proto.ArticleServer_SaveArticleServer) error {
return nil
}
func (server *StreamArticleServer) GetArticleInfo(aid *proto.Aid, stream proto.ArticleServer_GetArticleInfoServer) error {
return nil
}
func (server *StreamArticleServer) DeleteArticle(stream proto.ArticleServer_DeleteArticleServer) error {
return nil
}
func main() {
listen, err := net.Listen("tcp", "127.0.0.1:9527")
if err != nil {
log.Fatalf("tcp listen failed:%v", err)
}
server := grpc.NewServer()
proto.RegisterArticleServerServer(server, &StreamArticleServer{})
fmt.Println("article stream Server grpc services start success")
_ = server.Serve(listen)
}
我们先把架子搭起来,值得注意的是:绑定3个方法的时候,他们的参数,和普通的模式不一样了。
我们做的时候可以看下RegisterArticleServerServer(s *grpc.Server,srv ArticleServerServer)的第二个参数ArticleServerServer,可以看下,这个接口是怎么写的,因为我们需要实现这个接口。
接口定义如下:
type ArticleServerServer interface {
SaveArticle(ArticleServer_SaveArticleServer) error
GetArticleInfo(*Aid, ArticleServer_GetArticleInfoServer) error
DeleteArticle(ArticleServer_DeleteArticleServer) error
}
这样,我们自己去实现这3个接口的时候,也就知道自己的参数和返回值怎么写了,这也是一种学习方法。
架子搭完,我们就来实现以下client流的情况下,sever的实现方式,也就是SaveArticle这个函数的内容实现:
func (server *StreamArticleServer) SaveArticle(stream proto.ArticleServer_SaveArticleServer) error {
for {
id := rand.Int31n(100)
r, err := stream.Recv()
if err == io.EOF {
fmt.Println("读取数据结束")
res := &proto.Aid{Id: id}
return stream.SendAndClose(res)
}
if err != nil {
return err
}
fmt.Printf("stream.rev author: %s, title: %s, context: %s", r.Author.GetAuthor(), r.Title.GetTitle(), r.Content.GetContent())
}
}
首先是有个for循环,源源不断的接受client的流数据,然后通过判断err==io.EOF来判断客户端额流水结束。然后整体返回1个随机的ID mock假数据。这样,server端就实现完毕了。
那接着我们来实现client流怎么写。老规矩,先定义出client的架子:
package main
import (
"context"
"fmt"
"go-grpc-example/proto"
"google.golang.org/grpc"
"log"
)
var client proto.ArticleServerClient
func main() {
connect, err := grpc.Dial("127.0.0.1:9527", grpc.WithInsecure())
if err != nil {
log.Fatal("connect grpc fail")
}
defer connect.Close()
client = proto.NewArticleServerClient(connect)
//SaveArticle()
//GetArticleInfo()
//DeleteArticle()
}
func SaveArticle() {
}
func GetArticleInfo() {
}
func DeleteArticle() {
}
前面的链接gRPC的部分和普通的模式是一样的。重点是本次的client流式该怎么写呢?也就是SaveArticle方法的写法:
func SaveArticle() {
//定义一组数据
SaveList := map[string]proto.ArticleParam{
"1": {Author: &proto.Author{Author: "tony"}, Title: &proto.Title{Title: "title1"}, Content: &proto.Content{Content: "content1"}},
"2": {Author: &proto.Author{Author: "jack"}, Title: &proto.Title{Title: "title2"}, Content: &proto.Content{Content: "content2"}},
"3": {Author: &proto.Author{Author: "tom"}, Title: &proto.Title{Title: "title3"}, Content: &proto.Content{Content: "content3"}},
"4": {Author: &proto.Author{Author: "boby"}, Title: &proto.Title{Title: "title4"}, Content: &proto.Content{Content: "content4"}},
}
//先调用函数
stream, err := client.SaveArticle(context.Background())
if err != nil {
log.Fatal("SaveArticle grpc fail", err.Error())
}
//再循环发送
for _, info := range SaveList {
err = stream.Send(&info)
if err != nil {
log.Fatal("SaveArticle Send info fail", err.Error())
}
}
//发送关闭新号,并且获取返回值
resp, err := stream.CloseAndRecv()
if err != nil {
log.Fatal("SaveArticle CloseAndRecv fail", err.Error())
}
fmt.Printf("resp: id = %d", resp.GetId())
}
这里需要注意的是,我们搞了个map,来循环给server发送数据,然后再就是关闭新号发送,再接受数据。可以看出,和server的方法是反着来的,比较好记。
我们测试一下。首先是启动server,启动成功:
$ go run server/stream_server/server.go
article stream Server grpc services start success
然后,我们编辑client.go里面main函数里的SaveArticle的注释,客户端执行调用一下:
$ go run client/stream_client/client.go
resp: id = 81
调用成功,返回ID=81。返回的是service端最后一次的生成的id。再看下server那边的输出:
$ go run server/stream_server/server.go
article stream Server grpc services start success
stream.rev author: jack, title: title2, context: content2
stream.rev author: tom, title: title3, context: content3
stream.rev author: boby, title: title4, context: content4
stream.rev author: tony, title: title1, context: content1
读取数据结束
server也输出正常。完全联通!当然,你说,客户端是流模式,就一定得搞个for循环去发送数据么?当然不是!
func SaveArticle2() {
//定义一组数据
SaveInfo := proto.ArticleParam {
Author: &proto.Author{Author: "mark"}, Title: &proto.Title{Title: "title5"}, Content: &proto.Content{Content: "content5"},
}
//先调用函数
stream, err := client.SaveArticle(context.Background())
if err != nil {
log.Fatal("SaveArticle grpc fail", err.Error())
}
//发送
err = stream.Send(&SaveInfo)
if err != nil {
log.Fatal("SaveArticle Send info fail", err.Error())
}
发送关闭新号,并且获取返回值
resp, err := stream.CloseAndRecv()
if err != nil {
log.Fatal("SaveArticle CloseAndRecv fail", err.Error())
}
fmt.Printf("resp: id = %d", resp.GetId())
}
这样也是可以的。相当于数组是1。循环了1次而已。
(二)php语言的客户端流调用
我们看下php版本的client流模式如何写呢?直接上代码吧:
<?php
//引入 composer 的自动载加
require __DIR__ . '/vendor/autoload.php';
SaveArticle();
function SaveArticle()
{
//连接 gRPC服务端
$client = new \\Proto\\ArticleServerClient('127.0.0.1:9527', [
'credentials' => Grpc\ChannelCredentials::createInsecure()
]);
//请求 SaveArticle 方法
$stream = $client->SaveArticle();
$ArticleParam = new \\Proto\\ArticleParam();
//循环流式写入数据
for ($i = 0; $i < 10; $i++) {
$ArticleParam->setAuthor((new \\Proto\\Author())->setAuthor("kevin1"));
$ArticleParam->setTitle((new \\Proto\\Title())->setTitle("title_php_" . $i));
$ArticleParam->setContent((new \\Proto\\Content())->setContent("content_php_" . $i));
$stream->write($ArticleParam);
}
//关闭并返回结果
/**
* @var $aid \proto\Aid
*/
list($aid, $status) = $stream->wait();
//打印AID
var_dump($aid->getId());
}
我们执行一下,打印:98,同时服务端server也输出了响应的流水数据:
stream.rev author: kevin1, title: title_php_0, context: content_php_0
stream.rev author: kevin1, title: title_php_1, context: content_php_1
stream.rev author: kevin1, title: title_php_2, context: content_php_2
stream.rev author: kevin1, title: title_php_3, context: content_php_3
stream.rev author: kevin1, title: title_php_4, context: content_php_4
stream.rev author: kevin1, title: title_php_5, context: content_php_5
stream.rev author: kevin1, title: title_php_6, context: content_php_6
stream.rev author: kevin1, title: title_php_7, context: content_php_7
stream.rev author: kevin1, title: title_php_8, context: content_php_8
stream.rev author: kevin1, title: title_php_9, context: content_php_9
读取数据结束
php的client的很多地方处理方式,和go client很类似,比如,先调用client->SaveArticle(),参数是空,啥也不传,然后再循环写入(write/Send)。然后,再发送关闭,等待结果返回。
六、服务端流模式调用
有了前面客户端流模式的铺垫,服务端流模式就简单了很多。无非是将之前的操作反过来操作下。server不断的循环发送给client,然后client循环接受。套路是一样的。整个通讯过程就变成了这样:
client ----1------------------> server
client <-5- <-4- <-3- <-2- <-1- server
(一)go语言的服务端流调用
我们先来看下go服务端代码的编写,也就是实现server.go中GetArticleInfo函数里面的内容。
func (server *StreamArticleServer) GetArticleInfo(aid *proto.Aid, stream proto.ArticleServer_GetArticleInfoServer) error {
for i := 0; i < 6; i++ {
id := strconv.Itoa(int(aid.GetId()))
err := stream.Send(&proto.ArticleInfo{
Id: aid.GetId(),
Author: "jack",
Title: "title_go_" + id,
Content: "content_go_" + id,
})
if err != nil {
return err
}
}
fmt.Println("发送完毕")
return nil
}
单纯的服务端流,就比较简单,1个for循环6次,每次send数据即可,也不需要关闭。
我们再看下go client怎么实现呢?也就是client.go中现实GetArticleInfo函数里面的内容。
func GetArticleInfo() {
Aid := proto.Aid{
Id: 2,
}
//请求
stream, err := client.GetArticleInfo(context.Background(), &Aid)
if err != nil {
log.Fatal("GetArticleInfo grpc fail", err.Error())
}
//循环接受server流发来数据
for {
r, err := stream.Recv()
if err == io.EOF {
fmt.Println("读取数据结束")
break
}
if err != nil {
log.Fatal("GetArticleInfo Recv fail", err.Error())
}
fmt.Printf("stream.rev aid: %d, author: %s, title: %s, context: %s\n", r.GetId(), r.GetAuthor(), r.GetTitle(), r.GetContent())
}
}
client代码也同样比较简单,搞个for死循环去Recv就可以了。判断是否是EOF了,则表示sever发送结束了,就可以跳出循环,结束。
我们运行下,看下client的输出:
$ go run client/stream_client/client.go
stream.rev aid: 2, author: jack, title: title_go_2, context: content_go_2
stream.rev aid: 2, author: jack, title: title_go_2, context: content_go_2
stream.rev aid: 2, author: jack, title: title_go_2, context: content_go_2
stream.rev aid: 2, author: jack, title: title_go_2, context: content_go_2
stream.rev aid: 2, author: jack, title: title_go_2, context: content_go_2
stream.rev aid: 2, author: jack, title: title_go_2, context: content_go_2
读取数据结束
server端的输出:
$ go run server/stream_server/server.go
article stream Server grpc services start success
发送完毕
go语言版本的client和server成功,结束!
(二)php语言的服务端流调用
由于,我们本次只要php作为client调用,所以,我们只看下php如何接受go的server流的数据,其实和前面的client server类似,反过来即可。我们直接看下GetArticleInfo函数代码怎么实现:
function GetArticleInfo()
{
//连接 gRPC服务端
$client = new \\Proto\\ArticleServerClient('127.0.0.1:9527', [
'credentials' => Grpc\ChannelCredentials::createInsecure()
]);
//请求 SaveArticle 方法
$stream = $client->GetArticleInfo((new \\Proto\\Aid())->setId("668"));
//获取服务流的数据
$features = $stream->responses();
//循环遍历打印出来
/**
* @var $feature \proto\ArticleInfo
*/
foreach ($features as $feature) {
echo $feature->getId() . "--" . $feature->getAuthor() . "--" . $feature->getTitle() . "--" . $feature->getContent() . PHP_EOL;
}
}
需要注意的地方就是语法的改变,普通模式是使用wait方法就可以直接获取结果了,在服务流模式下,client写法就不一样了,得先response,再foreach循环这个值。
我们运行下:
$ php stream.php
668--jack--title_go_668--content_go_668
668--jack--title_go_668--content_go_668
668--jack--title_go_668--content_go_668
668--jack--title_go_668--content_go_668
668--jack--title_go_668--content_go_668
668--jack--title_go_668--content_go_668
客户端获取数据成功,这些数据都是server通过流模式传输过来的。再看下server端的输出:
$ go run server/stream_server/server.go
article stream Server grpc services start success
668 发送完毕
好了,整个通讯就完成了。
七、双向流模式调用
双向模式顾名思义,就是client和server都是流水模式,2边一起流水。
client -1-> -2-> -3-> --4> -5-> server
client <-5- <-4- <-3- <-2- <-1- server
通过前面的单独的流水模式,我们应该可以猜到代码该怎么写了,无非就是把之前的send啊,recv啊一起上呗,for循环也一起都用。
(一)go语言的双向流流调用
首先是go语言的服务端的写法,我们直接写吧,也就是实现函数DeleteArticle内的方法实现:
//双端
func (server *StreamArticleServer) DeleteArticle(stream proto.ArticleServer_DeleteArticleServer) error {
for {
//循环接收client发送的流数据
r, err := stream.Recv()
if err == io.EOF {
fmt.Println("read done!")
return nil
}
if err != nil {
return err
}
fmt.Printf("stream.rev aid: %d\n", r.GetId())
//循环发流数据给client
err = stream.Send(&proto.Status{Code: true})
if err != nil {
return err
}
//fmt.Println("send done!")
}
}
代码也比较好理解,先1个for循环,里面先去Rev,再去Send。当然,反着来也是可以的。
我们再看下client怎么实现:
//双向流
func DeleteArticle() {
//链接rpc
stream, err := client.DeleteArticle(context.Background())
if err != nil {
log.Fatal("DeleteArticle grpc fail", err.Error())
}
for i := 0; i < 6; i++ {
//先发
err = stream.Send(&proto.Aid{Id: int32(i)})
if err != nil {
log.Fatal("DeleteArticle Send fail", err.Error())
}
//再收
r, err := stream.Recv()
if err == io.EOF {
break
}
if err != nil {
log.Fatal("GetArticleInfo Recv fail", err.Error())
}
fmt.Printf("stream.rev status: %v\n", r.GetCode())
}
//发送结束
_ = stream.CloseSend()
}
客户端也差不多,先来1个6for循环,然后先Send,再Recv。这次不能反着来,不能就阻塞了。for循环结束后,可以主动发送一个CloseSend,这样server就可以手动手动EOF的信息了。
我们先运行server,再运行client,看下打印输出:
#client
$ go run client/stream_client/client.go
stream.rev status: true
stream.rev status: true
stream.rev status: true
stream.rev status: true
stream.rev status: true
stream.rev status: true
再看下server端的输出:
$ go run server/stream_server/server.go
article stream Server grpc services start success
stream.rev aid: 0
stream.rev aid: 1
stream.rev aid: 2
stream.rev aid: 3
stream.rev aid: 4
stream.rev aid: 5
read done!
通讯成功!
(二)php语言的双向流流调用
go的client已经OK了,我们继续看下php作为client的情况。直接上代码:
function DeleteArticle()
{
//连接 gRPC服务端
$client = new \\Proto\\ArticleServerClient('127.0.0.1:9527', [
'credentials' => Grpc\ChannelCredentials::createInsecure()
]);
//请求 SaveArticle 方法
$stream = $client->DeleteArticle();
$AidParam = new \\Proto\\Aid();
//循环流式写入数据
for ($i = 0; $i < 6; $i++) {
$AidParam->setId($i);
$stream->write($AidParam);
}
//写入结束
$stream->writesDone();
/**
* @var $reply \proto\Status
*/
while ($reply = $stream->read()) {
var_dump($reply->getCode());
}
}
写法和go的client稍有不同,先自己for 6次写,再调用writesDone写入结束,再while循环read,打印出code信息。
我们运行下:
$ php stream.php
bool(true)
bool(true)
bool(true)
bool(true)
bool(true)
bool(true)
服务端:
stream.rev aid: 0
stream.rev aid: 1
stream.rev aid: 2
stream.rev aid: 3
stream.rev aid: 4
stream.rev aid: 5
read done!
整个gRPC的通讯和跨语言的调用就结束了,还是收获满满的。接下来,我们要看下gRPC tls加密通讯,以及设置超时的context,还有就是如何同时提供http的Restful的接口方式,以及如何部署,服务发现以及负载均衡的实现。
八、TLS加密通讯
上面的这些例子都是讲的明文通讯,在某些情况下很容易被截获的,还是有点危险的。因为gRPC是基于http2的,所以我们看下,如何配置tls,使其支持https的特性呢?
那么回顾下,https的核心逻辑:
server?采用非对称加密,生成一个公钥public1和私钥private1
server?把公钥public1传给client
client 采用对称加密生成1个秘钥A (或者2个秘钥A,内容都是一样)
client 用server给自己的公钥public1加密自己生成的对称秘钥A。生成了一个秘钥B.
client 把秘钥B传给server。
client 用秘钥A加密需要传输的数据Data,并传给server。
server 收到秘钥B后,用自己的私钥private1解开了,得到了秘钥A。
server 收到加密后的data后,用秘钥A解开了,获得了元素数据。
简而言之,就是采用非对称加密+对称加密的方式。其中,对称加密产生的秘钥,是既可以加密,又可以解密的,加密解密速度很快。而采用非对称加密,则不可以,必须公钥解密私钥,或者私钥加密公钥,加解密速度慢。这样一个组合,就可以保障数据得到加密,又不会影响速度。
知道了原理之后,我们看下具体在代码里如何实现。首先,我们要生成server的公钥public1和私钥private1。那就得用到openssl命令了。需要注意的是go在1.15版本,X509无法使用了,需要用Sans算法代替。
这样,我们就得到了2个key,1个是test.pem,它就是公钥。1个是test.key,它是私钥。其中,我们设置openssl.cnf中alt_names为:
[ alt_names ]
DNS.1 = www.zchd.ltd
DNS.2 = www.test.zchd.ltd
顾明思义,设置的通用名称是这个。这个在client调用中会用到,不清楚先别急。
(一)golang中使用tls加密
我们先在golang中的server和clent中使用tls,看看怎么做,首先是server
package main
//采用https的token加密
import (
"context"
"fmt"
"go-grpc-example/proto"
"google.golang.org/grpc"
"google.golang.org/grpc/credentials"
"log"
"math/rand"
"net"
)
type UserServer struct{}
func main() {
//读2个证书
c, err := credentials.NewServerTLSFromFile("/Users/small/www/gowww/go-grpc-example/conf/test.pem", "/Users/small/www/gowww/go-grpc-example/conf/test.key")
if err != nil {
log.Fatalf("new tls server err:", err.Error())
}
//监听端口
listen, err := net.Listen("tcp", "127.0.0.1:9528")
if err != nil {
log.Fatalf("tcp listen failed:%v", err)
}
//新建gRPC服务,并且传入证书handle
server := grpc.NewServer(grpc.Creds(c))
fmt.Println("userServer grpc services start success")
//注册本次的UserServer 服务
proto.RegisterUserServerServer(server, &UserServer{})
_ = server.Serve(listen)
}
//保存用户
func (Service *UserServer) SaveUser(ctx context.Context, params *proto.UserParams) (*proto.Id, error) {
id := rand.Int31n(100) //随机生成id 模式保存成功
res := &proto.Id{Id: id}
fmt.Printf("%+v ", params.GetAge())
fmt.Printf("%+v\n", params.GetName())
return res, nil
}
func (Service *UserServer) GetUserInfo(ctx context.Context, id *proto.Id) (*proto.UserInfo, error) {
res := &proto.UserInfo{Id: id.GetId(), Name: "test", Age: 31}
return res, nil
}
我们可以发现,除了注册gRPC使用证书不同之外,其他的rpc函数和非tls上是一致的。我们看下client怎么写:
package main
import (
"context"
"fmt"
"go-grpc-example/proto"
"google.golang.org/grpc"
"google.golang.org/grpc/credentials"
"log"
)
var client proto.UserServerClient
func main() {
//读取证书和服务名
crt, err := credentials.NewClientTLSFromFile("/Users/small/www/gowww/go-grpc-example/conf/test.pem", "www.zchd.ltd")
if err != nil {
panic(err.Error())
}
//监听端口,并传入证书handle
connect, err := grpc.Dial("127.0.0.1:9528", grpc.WithTransportCredentials(crt))
if err != nil {
log.Fatalln(err)
}
defer connect.Close()
//新建服务客户端
client = proto.NewUserServerClient(connect)
SaveUser()
//GetUserInfo()
}
func SaveUser() {
params := proto.UserParams{}
params.Age = &proto.Age{Age: 31}
params.Name = &proto.Name{Name: "test"}
res, err := client.SaveUser(context.Background(), ¶ms)
if err != nil {
log.Fatalf("client.SaveUser err: %v", err)
}
fmt.Println(res.Id)
}
func GetUserInfo() {
res, err := client.GetUserInfo(context.Background(), &proto.Id{Id: 1})
if err != nil {
log.Fatalf("client.userInfo err: %v", err)
}
fmt.Printf("%+v\n", res)
}
代码总体也很简单,需要注意的是NewClientTLSFromFile()这个函数,第一个参数需要传pem公钥文件,第一个参数传serverNameOverride,也就是我们在OpenSSL.cnf里面设置DNS的名字。
我们运行一下:
$ go run client/simple_token_client/client.go
47
服务端也有输出:
$ go run server/simple_tls_server/server.go
age:31 name:"test"
成功连接。需要注意的是2个证书的生成,涉及很多openssl命令,要注意别搞错了,这个搞错就很同意连接不成功,出现各种问题。
(二)php client 使用tls加密连接
老规矩,我们在php中的client,也可以用这种方式来加密连接一下服务端,直接上代码:
<?php
//引入 composer 的自动载加
require __DIR__ . '/vendor/autoload.php';
//公钥内容
$pem = file_get_contents("/Users/small/www/gowww/go-grpc-example/conf/test.pem");
//导入公钥证书和DNS name名
$client = new \\Proto\\UserServerClient('127.0.0.1:9528', [
'credentials' => \Grpc\ChannelCredentials::createSsl($pem),
'grpc.ssl_target_name_override' => 'www.zchd.ltd',
]);
//实例化 $UserParams 请求类
$UserParams = new \\Proto\\UserParams();
$age = new \\Proto\\Age();
$age->setAge(18);
$UserParams->setAge($age);
$name = new \\Proto\\Name();
$name->setName("jack");
$UserParams->setName($name);
//调用远程服务
/**
* @var $Id \Proto\Id
*/
list($Id, $status) = $client->SaveUser($UserParams)->wait();
var_dump($status->code, $Id->getId());
//实例化Id类
$Id = new \\Proto\\Id();
//赋值
$Id->setId("1");
//调用
/**
* @var $User \Proto\UserInfo
*/
list($UserInfo, $status) = $client->GetUserInfo($Id)->wait();
var_dump($status->code, $UserInfo->getId(), $UserInfo->getName(), $UserInfo->getAge());
需要注意的是证书的导入和写法,有点区别。运行一下:
$ php main_tls.php
int(0)
int(59)
int(0)
int(1)
string(4) "test"
int(31)
成功了。
九、超时控制
我们平时在代码中通过curl调用1个http请求的时候,都会设置timeout超时,这个是非常重要的,之前笔者就经历过1个接口没设置超时时间,由于1个接口读取时间很长,导致请求长时间等待,由于http请求堆积太多导致,服务线程池飙升,就导致节点死机。所以这是很严重的一个事情。
那么,我们再client中去调用gRPC的服务请求的时候,也应该要设置超时时间,这个超时可以通过context来实现。
所以,核心是用到context这个包,来设置,有2种方式,都可以:
//设置超时时间为1秒
ctx, cancel = context.WithDeadline(context.Background(), time.Now().Add(1*time.Second))
//更好的写法
ctx, cancel = context.WithTimeout(context.Background(), 1*time.Second)
直接上client代码吧:
package main
//超时控制
import (
"context"
"fmt"
"go-grpc-example/proto"
"google.golang.org/grpc"
"google.golang.org/grpc/codes"
"google.golang.org/grpc/status"
"log"
"time"
)
var client proto.UserServerClient
var ctx context.Context
var cancel context.CancelFunc
func main() {
connect, err := grpc.Dial("127.0.0.1:9527", grpc.WithInsecure())
if err != nil {
log.Fatalln(err)
}
defer connect.Close()
//ctx, cancel = context.WithDeadline(context.Background(), time.Now().Add(1*time.Second))
//另一种写法,1秒超时
ctx, cancel = context.WithTimeout(context.Background(), 1*time.Second)
defer cancel()
client = proto.NewUserServerClient(connect)
SaveUser()
}
func SaveUser() {
params := proto.UserParams{}
params.Age = &proto.Age{Age: 31}
params.Name = &proto.Name{Name: "test"}
//打印当前时间
fmt.Println(time.Now().Format("2006-01-02 15:04:05"))
//开始请求
res, err := client.SaveUser(ctx, ¶ms)
fmt.Println(err)
if err != nil {
got := status.Code(err)
//客户端自己超时控制
if got == codes.DeadlineExceeded {
log.Println("client.SaveUser err: deadline")
}
log.Printf("client.SaveUser err: %+v", err)
} else {
fmt.Println(res.Id)
}
}
client先是设置了context的WithTimeout时间为1秒,然后判断调用gRPC函数SaveUser的错误返回值,如果限制超时,就终止请求。
server端其实也需要对这个超时时间做及时的判断,因为server端可能请求了很多协程服务,client已经停止了,那么server端也应该要及时的停止了,而不是还在后端运行和计算,这样也可以节省服务器很多资源:
package main
import (
"context"
"fmt"
"go-grpc-example/proto"
"google.golang.org/grpc"
"google.golang.org/grpc/codes"
"google.golang.org/grpc/status"
"log"
"math/rand"
"net"
"time"
)
type UserServer struct{}
func main() {
listen, err := net.Listen("tcp", "127.0.0.1:9527")
if err != nil {
log.Fatalf("tcp listen failed:%v", err)
}
server := grpc.NewServer()
fmt.Println("userServer grpc services start success")
proto.RegisterUserServerServer(server, &UserServer{})
_ = server.Serve(listen)
}
//保存用户
func (Service *UserServer) SaveUser(ctx context.Context, params *proto.UserParams) (*proto.Id, error) {
time.Sleep(3*time.Second)
//检测是否超时
timeD, ok := ctx.Deadline()
if ok {
fmt.Println(timeD.Format("2006-01-02 15:04:05"), ctx.Err())
return nil, status.Errorf(codes.Canceled, "UserServer.SaveUser Deadline")
}
id := rand.Int31n(100) //随机生成id 模式保存成功
res := &proto.Id{Id: id}
fmt.Printf("%+v, ", params.GetAge())
fmt.Printf("%+v\n", params.GetName())
return res, nil
}
我们再server端,模拟了3秒超时。我们运行一下:
$ go run client/simple_timeout_client/client.go
2021/09/17 23:20:22
2021/09/17 23:20:23 client.SaveUser err: deadline
2021/09/17 23:20:23 client.SaveUser err: rpc error: code = DeadlineExceeded desc = context deadline exceeded
$ go run server/simple_timeout_server/server.go
userServer grpc services start success
2021/09/17 23:20:23 context deadline exceeded
可以看到,在1秒后,deadline了。成功!
当然,也可以用select+ctx.Done()的模式,来监听client的取消事件的:
关键代码如下:
select {
case <-ctx.Done():
fmt.Println("ctx.Done", ctx.Err())
return nil, status.Errorf(codes.Canceled, "UserServer.SaveUser Deadline")
}
参考资料:
1.gRPC及相关介绍
2.gRPC微服务框架
3.go-grpc-example
4.php gRPC官方demo?
5.Go1.15解决gRPC X509
?作者简介
杨义(smallyang)
腾讯高级工程师
腾讯高级工程师,主要负责IEG游戏活动运营及高可用平台的建设,对云服务、k8s以及高性能服务上也有很深的了解。
?推荐阅读
详细解答!从C++转向Rust需要注意哪些问题?
如何保证MySQL和Redis的数据一致性?10张图带你搞定!
前端推荐!10分钟带你了解Konva运行原理
Golang原生json可以一库走天下吗?