GRPC golang版源码分析之客户端(二)

Table of Contents

• 1. 前言
• 2. 负载均衡
• 3. 相关链接

1 前言

前面一篇文章分析了一个grpc call的大致调用流程，顺着源码走了一遍，但是grpc中有一些特性并没有进行分析，这篇blog主要就是分析负载均衡这个特性。负载均衡可以让你在使用grpc调用方法是连接不同的服务器。在启动的时候，grpc client会建立对所有服务器的连接，然后通过轮询算法拿不同的服务器。

2 负载均衡

负载均衡是在客户端实现的，通过在初始化一个grpc客户端传入一个负载均衡器，默认是通过轮询算法每次调用时切换调用的服务器，由于是在客户端做的，所以只能让服务器平摊压力，并不能实时的根据服务器的状态来进行负载均衡。比如某一刻所有的grpc客户端都轮到了同一个服务器（这里只是举例）。

//在clientconn.go的DialContext函数中 go func() {var addrs []Addressif cc.dopts.balancer == nil {// Connect to target directly if balancer is nil.//如果没有设置负载均衡器，则直接连接addrs = append(addrs, Address{Addr: target})} else {var credsClone credentials.TransportCredentialsif creds != nil {credsClone = creds.Clone()}config := BalancerConfig{DialCreds: credsClone,}//启动一个负载均衡器,start函数会启动一个watch监听地址的变化，而Notify返回一个通道，在每次服务器地址变化后的最新地址信息.if err := cc.dopts.balancer.Start(target, config); err != nil {waitC <- errreturn}ch := cc.dopts.balancer.Notify()if ch == nil {// There is no name resolver installed.addrs = append(addrs, Address{Addr: target})} else {addrs, ok = <-chif !ok || len(addrs) == 0 {waitC <- errNoAddrreturn}}}//对每一个地址进行连接, 因为这是客户端启动时，所以需要对所有地址操作.for _, a := range addrs {if err := cc.resetAddrConn(a, false, nil); err != nil {waitC <- errreturn}}close(waitC) }() .............. //省略一些代码 if ok {//这里开启一个监听goroutine，主要是监听服务器地址变化并对新的地址建立连接，对老的地址关闭连接go cc.lbWatcher() }

上面的代码大致是说明了一个负载均衡器是怎么使用的，下面看看负载均衡具体的结构:

type Balancer interface {//启动一个负载均衡，内部会启动一个名称服务器的watcher，不断监听地址的变化Start(target string, config BalancerConfig) errorUp(addr Address) (down func(error))//得到下一次连接的地址Get(ctx context.Context, opts BalancerGetOptions) (addr Address, put func(), err error)//从服务器得到更新的地址,这个地址是更新后的所有地址Notify() <-chan []Address//关闭负载均衡器Close() error }

负载均衡器有个默认实现在balancer.go中，使用的轮询算法。这个实现中包含了一个名字服务器的服务，我们可以通过实现一个名字服务器的接口，然后封装到这个负载均衡器中，这样就不需要自己实现整个负载均衡器。名字服务器的接口如下:

type Update struct {// Op indicates the operation of the update.Op Operation// Addr is the updated address. It is empty string if there is no address update.Addr string// Metadata is the updated metadata. It is nil if there is no metadata update.// Metadata is not required for a custom naming implementation.Metadata interface{} } // Resolver creates a Watcher for a target to track its resolution changes. type Resolver interface {// Resolve creates a Watcher for target.//通过一个名字得到一个watcher，监听服务器地址变化。Resolve(target string) (Watcher, error) } // Watcher watches for the updates on the specified target. type Watcher interface {// Next blocks until an update or error happens. It may return one or more// updates. The first call should get the full set of the results. It should// return an error if and only if Watcher cannot recover.// 得到下次更新的地址Next() ([]*Update, error)// Close closes the Watcher.Close() }

负载均衡器的全貌大概就这些了，其中还有些细节我没有看到，目前来说我也不希望过早深入这些细节。所以负载均衡就到这里啦。

3 相关链接

http://www.grpc.io/docs/guides/auth.html#overview 验证文档

https://guidao.github.io/grpc_balancer.html

总结

以上是生活随笔为你收集整理的GRPC golang版源码分析之客户端(二)的全部内容，希望文章能够帮你解决所遇到的问题。

如果觉得生活随笔网站内容还不错，欢迎将生活随笔推荐给好友。