写在最前: 这篇文章其实算是马后炮了，因为一直拖延症的问题，顺带过了一个五一假期，结果发现已经有社区贡献者提供了Consul的服务发现实现，于是本来写了一半的文章只能进行调整了。拖延症害人啊！几个草稿的文章看来要尽快赶出来了🤦♂️
在上一篇文章中，我其实遗留了一个问题：如何定义dapr的服务发现呢？其实在后面阅读dapr的源码之后也前一篇文章的评论中提到了答案：目前dapr提供了内置两种服务发现模式：K8s模式和用于独立部署的mDNS模式。mDNS模式在某些网络环境下可能存在问题（比如跨机房），不过没有关系，dapr同时提供了可扩展能力，可以通过定义自主的服务发现能力扩展dapr的边界。
从 NameResolution 到 Resolver 接口
在 pkg/components/nameresolution/registry.go 文件中，dapr定义了一个 NameResolution 结构体用于服务注册和发现：
type (
// NameResolution is a name resolution component definition.
NameResolution struct {
Name string
FactoryMethod func() nr.Resolver
}
// Registry handles registering and creating name resolution components.
Registry interface {
Register(components ...NameResolution)
Create(name, version string) (nr.Resolver, error)
}
nameResolutionRegistry struct {
resolvers map[string]func() nr.Resolver
}
)
其中真正的服务解析则是依靠 components-contrib 中实现了 Resolver 接口的具体实现执行。
// Resolver is the interface of name resolver.
type Resolver interface {
// Init initializes name resolver.
Init(metadata Metadata) error
// ResolveID resolves name to address.
ResolveID(req ResolveRequest) (string, error)
}
其中 Init 会在 Runtime 初始化时被调用，而 ResolveID 则会在服务查询时调用。比如在 pkg/messaging/direct_messaging.go 的方法 getRemoteApp 中进行服务的解析：
func (d *directMessaging) getRemoteApp(appID string) (remoteApp, error) {
id, namespace, err := d.requestAppIDAndNamespace(appID)
if err != nil {
return remoteApp{}, err
}
request := nr.ResolveRequest{ID: id, Namespace: namespace, Port: d.grpcPort}
address, err := d.resolver.ResolveID(request)
if err != nil {
return remoteApp{}, err
}
return remoteApp{
namespace: namespace,
id: id,
address: address,
}, nil
}
当然，事实上这样并不完全足够，还需要把这个服务注册放入dapr支持的服务中去：
runtime.WithNameResolutions(
nr_loader.New("mdns", func() nr.Resolver {
return nr_mdns.NewResolver(logContrib)
}),
nr_loader.New("kubernetes", func() nr.Resolver {
return nr_kubernetes.NewResolver(logContrib)
}),
nr_loader.New("consul", func() nr.Resolver {
return nr_consul.NewResolver(logContrib)
}),
),
上面的这些是设定的dpar目前支持的一些服务发现功能，而我们之前服务发现也一直使用的 Consul 实现，已经满足我们的需求了…😓拖延症害人啊！
从原理到实现
上面提到了我们需要实现一个 Resolver 接口的实现，我们可以预见到我们大概会需要这么一个东西:
type resolver struct {}
// NewResolver creates Consul name resolver.
func NewResolver() nr.Resolver
// Init will configure component. It will also register service or validate client connection based on config
func (r *resolver) Init(metadata nr.Metadata) error
// ResolveID resolves name to address via consul
func (r *resolver) ResolveID(req nr.ResolveRequest) (string, error)
接下来就需要一个 client *consul.Client 去实现服务的注册：
type resolver struct {
client *consul.Client
}
func (r *resolver) Init(metadata nr.Metadata) error {
// ...
if err := r.client.Agent().ServiceRegister(regData); err != nil {
return fmt.Errorf("failed to register consul service: %w", err)
}
// ...
}
注册服务完成后，在调用具体的服务时，我们需要获取具体的服务地址：
func (r *resolver) ResolveID(req nr.ResolveRequest) (string, error) {
// ...
services, _, err := r.client.Health().Service(req.ID, "", true, cfg.QueryOptions)
// ...
}
当然上面的演示代码只是部分核心功能代码，如果需要拓展更多的实现细节内容，需要查看具体的官方接收社区贡献的实现：components-contrib/nameresolution/consul