快捷導(dǎo)航

golang grpc 負(fù)載均衡的方法

更新時間：2018年07月11日 09:47:29 作者：xjtuhit

這篇文章主要介紹了golang grpc 負(fù)載均衡的方法，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧

微服務(wù)架構(gòu)里面，每個服務(wù)都會有很多節(jié)點，如果流量分配不均勻，會造成資源的浪費，甚至將一些機器壓垮，這個時候就需要負(fù)載均衡，最簡單的一種策略就是輪詢，順序依次選擇不同的節(jié)點訪問。

grpc 在客戶端提供了負(fù)載均衡的實現(xiàn)，并提供了服務(wù)地址解析和更新的接口(默認(rèn)提供了 DNS 域名解析的支持)，方便不同服務(wù)的集成

使用示例

conn, err := grpc.Dial(
  "",
  grpc.WithInsecure(),
  // 負(fù)載均衡，使用 consul 作服務(wù)發(fā)現(xiàn)
  grpc.WithBalancer(grpc.RoundRobin(grpclb.NewConsulResolver(
    "127.0.0.1:8500", "grpc.health.v1.add",
  ))),
)

創(chuàng)建連接的時候可以使用 WithBalancer 選項來指定負(fù)載均衡策略，這里使用 RoundRobin 算法，其實就是輪詢策略

與 consul 的集成

有了負(fù)載均衡策略，還需要一個地址解析和更新策略，可以使用 DNS 服務(wù)來實現(xiàn)，但如果我們使用 consul 來做服務(wù)的注冊和發(fā)現(xiàn)，可以通過實現(xiàn) ‘naming.Resolver' 和 ‘naming.Watcher' 接口來支持

naming.Resolver: 實現(xiàn)地址解析
naming.Watcher: 實現(xiàn)節(jié)點的變更，添加或者刪除

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, // 同步點，這個調(diào)用將一直阻塞，直到有新的更新
    })
    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/blob/master/doc/naming.md

Load Balancing in gRPC: https://github.com/grpc/grpc/blob/master/doc/load-balancing.md

dns_resolver: https://github.com/grpc/grpc-go/blob/30fb59a4304034ce78ff68e21bd25776b1d79488/naming/dns_resolver.go

代碼地址: https://github.com/hatlonely/hellogolang/blob/master/sample/addservice/cmd/client/main.go

以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章: