Grpc 自定义解析器 resolver

原创于 2021-11-16 20:24:39 发布 · 786 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#go #grpc

Go 同时被 2 个专栏收录

10 篇文章

订阅专栏

grpc

1 篇文章

订阅专栏

本文介绍如何在gRPC中使用自定义解析器resolver，通过导入自定义包'grpc_builder'，实现gRPC连接的个性化配置。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

package grpc_builder

import (
	"fmt"
	"google.golang.org/grpc/resolver"
)

const customResolverScheme = "custom"

type customResolverBuilder struct {
}

type customResolver struct {
	target resolver.Target
	cc     resolver.ClientConn
}

func (*customResolverBuilder) Build(target resolver.Target, cc resolver.ClientConn, opts resolver.BuildOptions) (resolver.Resolver, error) {
	r := &customResolver{
		target: target,
		cc: cc,
	}

	r.start()
	return r, nil
}

func (*customResolverBuilder) Scheme() string  {
	return customResolverScheme
}

func (r customResolver) start() {
	addrs := make([]resolver.Address, 0)
	for i:=0; i<9; i++ {
		addrs = append(addrs, resolver.Address{Addr: "192.168.237.12:9080"})
	}
	fmt.Println("addrs", addrs)
	r.cc.UpdateState(resolver.State{Addresses: addrs})
}

func (*customResolver) ResolveNow(o resolver.ResolveNowOptions) {}
func (*customResolver) Close()                                  {}

func init()  {
	resolver.Register(&customResolverBuilder{})
}