原理:
resolver解析地址,然后获取出 解析后的 resolveConnect;
调用 updateResolveStatus
然后 根据负载均衡 设置,生成负载均衡器;
负载均衡器内负责创建子链接;
子链接创建完后,将子链接传给picker管理,然后更新主链接的状态;
使用步骤:
- 定义自己的resolver
- resolve出相应的地址,传给负载均衡器builder
- 负载均衡器builder创建连接,传给picker
使用方法:
const (exampleScheme = "example"exampleServiceName = "lb.example.grpc.io")var addrs = []string{"localhost:50051", "localhost:50052"}func callUnaryEcho(c ecpb.EchoClient, message string) {ctx, cancel := context.WithTimeout(context.Background(), time.Second)defer cancel()r, err := c.UnaryEcho(ctx, &ecpb.EchoRequest{Message: message})if err != nil {log.Fatalf("could not greet: %v", err)}fmt.Println(r.Message)}func makeRPCs(cc *grpc.ClientConn, n int) {hwc := ecpb.NewEchoClient(cc)for i := 0; i < n; i++ {callUnaryEcho(hwc, "this is examples/load_balancing")}}func main() {pickfirstConn, err := grpc.Dial(fmt.Sprintf("%s:///%s", exampleScheme, exampleServiceName),// grpc.WithBalancerName("pick_first"), // "pick_first" is the default, so this DialOption is not necessary.grpc.WithInsecure(),grpc.WithBlock(),)if err != nil {log.Fatalf("did not connect: %v", err)}defer pickfirstConn.Close()fmt.Println("--- calling helloworld.Greeter/SayHello with pick_first ---")makeRPCs(pickfirstConn, 10)fmt.Println()// Make another ClientConn with round_robin policy.roundrobinConn, err := grpc.Dial(fmt.Sprintf("%s:///%s", exampleScheme, exampleServiceName),grpc.WithBalancerName("round_robin"), // This sets the initial balancing policy.grpc.WithInsecure(),grpc.WithBlock(),)if err != nil {log.Fatalf("did not connect: %v", err)}defer roundrobinConn.Close()fmt.Println("--- calling helloworld.Greeter/SayHello with round_robin ---")makeRPCs(roundrobinConn, 10)}// Following is an example name resolver implementation. Read the name// resolution example to learn more about it.type exampleResolverBuilder struct{}func (*exampleResolverBuilder) Build(target resolver.Target, cc resolver.ClientConn, opts resolver.BuildOptions) (resolver.Resolver, error) {r := &exampleResolver{target: target,cc: cc,addrsStore: map[string][]string{exampleServiceName: addrs,},}r.start()return r, nil}func (*exampleResolverBuilder) Scheme() string { return exampleScheme }type exampleResolver struct {target resolver.Targetcc resolver.ClientConnaddrsStore map[string][]string}func (r *exampleResolver) start() {addrStrs := r.addrsStore[r.target.Endpoint]addrs := make([]resolver.Address, len(addrStrs))for i, s := range addrStrs {addrs[i] = resolver.Address{Addr: s}}r.cc.UpdateState(resolver.State{Addresses: addrs})}func (*exampleResolver) ResolveNow(o resolver.ResolveNowOptions) {}func (*exampleResolver) Close() {}func init() {resolver.Register(&exampleResolverBuilder{})}
