动手实现一个简单的 rpc 框架到入门 grpc (下)

之前手动实现了一次简陋的 rpc 调用,为了简单使用了 json 编码信息,其实这是非常不可靠的,go 中 json 解析会有一些问题,比如整数会变成浮点数,而且 json 字符串比较占空间。

gRPC 由 google 开发,是一款语言中立、平台中立、开源的 RPC 框架,默认使用 protocol buffers 来序列化和传输消息,基于 http2。

protobuf

Protocol Buffers 是一种轻便高效的结构化数据存储格式,可以用于结构化数据串行化,或者说序列化。它很适合做数据存储或 RPC 数据交换格式。可用于通讯协议、数据存储等领域的语言无关、平台无关、可扩展的序列化结构数据格式。

先编写 proto 文件,再编译成 go 文件,新建 proto/hello.proto 文件

// proto/hello.proto syntax = "proto3"; package proto; message String { string value = 1; // 类型 字段 字段标识符 } // 定义服务 service HelloService { rpc Hello (String) returns (String); }

安装 protobuf 编译器用于编译 proto 文件,比如使用 scoop 安装 scoop install protobuf,或者手动下载安装

protobuf 编译工具可以把 proto 模板编译成多种语言,默认不支持 go,安装一下 go 插件

$ go get -u github.com/golang/protobuf/proto $ go get -u github.com/golang/protobuf/protoc-gen-go

进入 hello.proto 文件目录,编译生成 hello.pb.go,后续为了方便可以把编译指令写成 shell 脚本

$ protoc -I . --go_out=plugins=grpc:. ./hello.proto

生成的 hello.pb.go 有一些接口和方法(省略了其他的),使用时候只需要实现接口调用对应方法

type HelloServiceServer interface { Hello(context.Context, *String) (*String, error) } func RegisterHelloServiceServer(s *grpc.Server, srv HelloServiceServer) { s.RegisterService(&_HelloService_serviceDesc, srv) } type helloServiceClient struct { cc grpc.ClientConnInterface } func NewHelloServiceClient(cc grpc.ClientConnInterface) HelloServiceClient { return &helloServiceClient{cc} } func (c *helloServiceClient) Hello(ctx context.Context, in *String, opts ...grpc.CallOption) (*String, error) { out := new(String) err := c.cc.Invoke(ctx, "/proto.HelloService/Hello", in, out, opts...) if err != nil { return nil, err } return out, nil } grpc

安装 grpc

$ go get -u google.golang.org/grpc 服务端

在 proto 文件夹同级新建 sever/server.go 文件,实现 server 接口,注册服务

type HelloServiceImpl struct{} func (p *HelloServiceImpl) Hello(ctx context.Context, args *proto.String) (*proto.String, error) { reply := &proto.String{Value: "hello " + args.GetValue()} return reply, nil } func main() { grpcServer := grpc.NewServer() proto.RegisterHelloServiceServer(grpcServer, new(HelloServiceImpl)) lis, err := net.Listen("tcp", ":1234") if err != nil { log.Fatal(err) } grpcServer.Serve(lis) } 客户端

在 proto 文件夹同级新建 sever/server.go 文件,调用服务

func main() { conn, err := grpc.Dial("localhost:1234", grpc.WithInsecure()) if err != nil { log.Fatal(err) } defer conn.Close() client := proto.NewHelloServiceClient(conn) reply, err := client.Hello(context.Background(), &proto.String{Value: "world"}) if err != nil { log.Fatal(err) } fmt.Println(reply.GetValue()) } 验证

首先运行服务端 go run server/server.go,再运行客户端

$ go run client/client.go hello world

这里只简单介绍了用法,关于 protobuf 的更多用法可以参考 https://developers.google.com/protocol-buffers/,关于 protobuf 的原理可以参考 https://blog.csdn.net/carson_ho/article/details/70568606,另外 grpc 还支持流式调用。

流式调用

数据传输是流式的,服务端和服务端不用输完全部数据后才响应,流式传输可以是客户端流式、或服务端流式,也可以同时流式传输,只需要一个 stream 标识

service HelloService { rpc Hello (String) returns (String); rpc HelloStream (String) returns (stream String); //服务端流式响应 //rpc HelloStream (stream String) returns (String); //客户端端流式发送 //rpc HelloStream (stream String) returns (stream String); //双向流 }

重新编译,然后在 server/server.go 新增方法

func (HelloServiceImpl) HelloStream(args *proto.String, stream proto.HelloService_HelloStreamServer) error { for i := 0; i < 10; i++ { _ = stream.Send(&proto.String{Value: "hello " + args.GetValue() + strconv.Itoa(i)}) } return nil }

在 client/client.go 中调用 HelloStream 方法

stream, _ := client.HelloStream(context.Background(), &proto.String{Value: "World"}) for { res, err := stream.Recv() if err == io.EOF { break } if err != nil { log.Printf("stream.Recv: %v", err) } log.Printf("%s", res.String()) }

验证,首先运行 server,然后运行 client

$ go run server/server.go $ go run client/client.go 2020/07/21 07:41:10 value:"hello World0" 2020/07/21 07:41:10 value:"hello World1" 2020/07/21 07:41:10 value:"hello World2" 2020/07/21 07:41:10 value:"hello World3" 2020/07/21 07:41:10 value:"hello World4" 2020/07/21 07:41:10 value:"hello World5" 2020/07/21 07:41:10 value:"hello World6" 2020/07/21 07:41:10 value:"hello World7" 2020/07/21 07:41:10 value:"hello World8" 2020/07/21 07:41:10 value:"hello World9"

内容版权声明:除非注明,否则皆为本站原创文章。

转载注明出处:https://www.heiqu.com/wpzxds.html