Kratos 路由与负载均衡

接口实现

路由与负载均衡主要的接口是 Selector，在同目录下也提供了一个默认的 Selector 实现，该实现可以通过替换 NodeBuilder、Filter、Balancer 来分别实现节点权重计算算法、服务路由过滤策略、负载均衡算法的可插拔

type Selector interface {
  // Selector 内部维护的服务节点列表通过 Rebalancer 接口来更新
  Rebalancer

  // Select nodes
  // if err == nil, selected and done must not be empty.
  Select(ctx context.Context, opts ...SelectOption) (selected Node, done DoneFunc, err error)
}

// 通过Rebalancer实现服务节点变更感知
type Rebalancer interface {
  Apply(nodes []Node)
}

已支持的实现：

• wrr : Weighted round robin (Kratos Client内置默认算法)
• p2c : Power of two choices
• random : Random

使用方式

HTTP Client

import  "github.com/go-kratos/kratos/v2/selector/p2c"
import  "github.com/go-kratos/kratos/v2/selector/filter"

// 创建路由 Filter：筛选版本号为"2.0.0"的实例
filter :=  filter.Version("2.0.0")
// 创建 P2C 负载均衡算法 Selector，并将路由 Filter 注入
selector := p2c.New(p2c.WithFilter(filter))

hConn, err := http.NewClient(
  context.Background(),
  http.WithEndpoint("discovery:///helloworld"),
  http.WithDiscovery(r),
  // 通过 http.WithSelector 将 Selector 注入 HTTP Client 中
  http.WithSelector(
    p2c.New(p2c.WithFilter(filter.Version("2.0.0"))),
  )
)

gRPC Client

import  "github.com/go-kratos/kratos/v2/selector/p2c"
import  "github.com/go-kratos/kratos/v2/selector/filter"

// 创建路由 Filter：筛选版本号为"2.0.0"的实例
filter :=  filter.Version("2.0.0")

conn, err := grpc.DialInsecure(
  context.Background(),
  grpc.WithEndpoint("discovery:///helloworld"),
  grpc.WithDiscovery(r),
  // 由于 gRPC 框架的限制，只能使用全局 balancer name 的方式来注入 selector
  grpc.WithBalancerName(wrr.Name),
  // 通过 grpc.WithFilter 注入路由 Filter
  grpc.WithFilter(filter),
)