golang grpc 負載均衡

微服務架構裏面，每一個服務都會有不少節點，若是流量分配不均勻，會形成資源的浪費，甚至將一些機器壓垮，這個時候就須要負載均衡，最簡單的一種策略就是輪詢，順序依次選擇不一樣的節點訪問

grpc 在客戶端提供了負載均衡的實現，並提供了服務地址解析和更新的接口(默認提供了 DNS 域名解析的支持)，方便不一樣服務的集成

使用示例

conn, err := grpc.Dial(
    "",
    grpc.WithInsecure(),
    // 負載均衡，使用 consul 做服務發現
    grpc.WithBalancer(grpc.RoundRobin(grpclb.NewConsulResolver(
        "127.0.0.1:8500", "grpc.health.v1.add",
    ))),
)

建立鏈接的時候能夠使用 WithBalancer 選項來指定負載均衡策略，這裏使用 RoundRobin 算法，其實就是輪詢策略

與 consul 的集成

有了負載均衡策略，還須要一個地址解析和更新策略，能夠使用 DNS 服務來實現，但若是咱們使用 consul 來作服務的註冊和發現，能夠經過實現 naming.Resolver 和 naming.Watcher 接口來支持

• naming.Resolver: 實現地址解析
• naming.Watcher: 實現節點的變動，添加或者刪除

func NewConsulResolver(address string, service string) naming.Resolver {
    return &consulResolver{
        address: address,
        service: service,
    }
}

type consulResolver struct {
    address string
    service string
}

func (r *consulResolver) Resolve(target string) (naming.Watcher, error) {
    config := api.DefaultConfig()
    config.Address = r.address
    client, err := api.NewClient(config)
    if err != nil {
        return nil, err
    }

    return &consulWatcher{
        client:  client,
        service: r.service,
        addrs:   map[string]struct{}{},
    }, nil
}

type consulWatcher struct {
    client    *api.Client
    service   string
    addrs     map[string]struct{}
    lastIndex uint64
}

func (w *consulWatcher) Next() ([]*naming.Update, error) {
    for {
        services, metainfo, err := w.client.Health().Service(w.service, "", true, &api.QueryOptions{
            WaitIndex: w.lastIndex, // 同步點，這個調用將一直阻塞，直到有新的更新
        })
        if err != nil {
            logrus.Warn("error retrieving instances from Consul: %v", err)
        }
        w.lastIndex = metainfo.LastIndex

        addrs := map[string]struct{}{}
        for _, service := range services {
            addrs[net.JoinHostPort(service.Service.Address, strconv.Itoa(service.Service.Port))] = struct{}{}
        }

        var updates []*naming.Update
        for addr := range w.addrs {
            if _, ok := addrs[addr]; !ok {
                updates = append(updates, &naming.Update{Op: naming.Delete, Addr: addr})
            }
        }

        for addr := range addrs {
            if _, ok := w.addrs[addr]; !ok {
                updates = append(updates, &naming.Update{Op: naming.Add, Addr: addr})
            }
        }

        if len(updates) != 0 {
            w.addrs = addrs
            return updates, nil
        }
    }
}

func (w *consulWatcher) Close() {
    // nothing to do
}

參考連接

• gRPC Name Resolution: https://github.com/grpc/grpc/...
• Load Balancing in gRPC: https://github.com/grpc/grpc/...
• dns_resolver: https://github.com/grpc/grpc-...
• 代碼地址: https://github.com/hatlonely/...

轉載請註明出處
本文連接： http://www.hatlonely.com/2018...