GRPC源码解读

最新推荐文章于 2024-05-14 09:30:00 发布

weixin_34167819

最新推荐文章于 2024-05-14 09:30:00 发布

阅读量810

点赞数

文章标签： python golang

原文链接：https://my.oschina.net/u/2004526/blog/840295

版权

2019独角兽企业重金招聘Python工程师标准>>>

GRPC源码解读

使用的protobuf文件

syntax = "proto3";

package servers;

// The greeter service definition.
service Greeter {
    // Sends a greeting
    rpc SayHello (HelloRequest) returns (HelloReply) {}
}

// The request message containing the user's name.
message HelloRequest {
    string name = 1;
}

// The response message containing the greetings
message HelloReply {
    string message = 1;
}

Client

连接

conn, err := grpc.Dial("localhost:8080", grpc.WithInsecure())

创建一个 ClientConn 并将传入的 DialOption 应用到 ClientConn 的 dopts (dialOptions) 字段上。设置默认的编解码类 protoCodec, backoff配置，负载均衡策略（roundRobin）。

从负载均衡器中拿到地址列表，启动goroutine连接各个地址（看下面的<连接address>），主协程设置超时时间（根据 dopts.timeout)），在超时时间内等待连接完成。

如果有地址列表监控Watcher(当负载均衡设置了naming.Resolver时会有)，异步监控地址列表的变动（变动时更新 ClientConn的连接列表 conns）。

连接address

如果 dopts.insecure 为false，检查证书相关信息。

将创建的 addrConn 放入到 ClientConn 的 conns map中，并销毁被替换的 addrConn。

重置 Transport:

连接地址，一旦出错，重试，重试间隔时间由 dopts.bs 指定。

如果没出错则调用负载均衡的Up方法（标记一下这个地址已经连接了而已）。

监控 Transport: 如果有错误发生就重置 Transport(关闭之前的，重新连接一个)。

调用方法

c := servers.NewGreeterClient(conn)
reply, err := c.SayHello(context.Background(), &servers.HelloRequest{"person"})

创建Client没有什么东西。

调用 SayHello 时主要代码就下面这句：

grpc.Invoke(ctx, "/servers.Greeter/SayHello", in, out, c.cc, opts...)

从 ClientConn 中获取 Transport。

利用获取到的transport 创建一个 Stream, Stream的id每次递增2，将 Stream 保存在transport的activeStreams中。设置各种 header 字段。将 header 按照 frame 发送给服务端以开启一个stream，frame最大16KB。

将 HelloRequest 编码写到 stream 中。

等待读取服务器的相应到 HelloReply。

最后关闭 stream。