聊聊dubbo-go的registryAwareCluster

时间:2022-07-22
本文章向大家介绍聊聊dubbo-go的registryAwareCluster,主要内容包括其使用实例、应用技巧、基本知识点总结和需要注意事项,具有一定的参考价值,需要的朋友可以参考一下。

本文主要研究一下dubbo-go的registryAwareCluster

registryAwareCluster

dubbo-go-v1.4.2/cluster/cluster_impl/registry_aware_cluster.go

type registryAwareCluster struct{}
​
func init() {
    extension.SetCluster("registryAware", NewRegistryAwareCluster)
}
​
// NewRegistryAwareCluster ...
func NewRegistryAwareCluster() cluster.Cluster {
    return &registryAwareCluster{}
}
​
func (cluster *registryAwareCluster) Join(directory cluster.Directory) protocol.Invoker {
    return newRegistryAwareClusterInvoker(directory)
}
  • registryAwareCluster的Join方法执行newRegistryAwareClusterInvoker(directory)

newRegistryAwareClusterInvoker

dubbo-go-v1.4.2/cluster/cluster_impl/registry_aware_cluster_invoker.go

type registryAwareClusterInvoker struct {
    baseClusterInvoker
}
​
func newRegistryAwareClusterInvoker(directory cluster.Directory) protocol.Invoker {
    return &registryAwareClusterInvoker{
        baseClusterInvoker: newBaseClusterInvoker(directory),
    }
}
  • newRegistryAwareClusterInvoker方法创建registryAwareClusterInvoker

Invoke

dubbo-go-v1.4.2/cluster/cluster_impl/registry_aware_cluster_invoker.go

func (invoker *registryAwareClusterInvoker) Invoke(ctx context.Context, invocation protocol.Invocation) protocol.Result {
    invokers := invoker.directory.List(invocation)
    //First, pick the invoker (XXXClusterInvoker) that comes from the local registry, distinguish by a 'default' key.
    for _, invoker := range invokers {
        if invoker.IsAvailable() && invoker.GetUrl().GetParam(constant.REGISTRY_DEFAULT_KEY, "false") == "true" {
            return invoker.Invoke(ctx, invocation)
        }
    }
​
    //If none of the invokers has a local signal, pick the first one available.
    for _, invoker := range invokers {
        if invoker.IsAvailable() {
            return invoker.Invoke(ctx, invocation)
        }
    }
    return nil
}
  • Invoke方法先通过invoker.directory.List(invocation)获取invokers,然后遍历invokers判断是否有来自local registry的invoker, 如果有则执行invoker.Invoke(ctx, invocation);否则再次遍历invokers,找到第一个available的执行invoker.Invoke(ctx, invocation)

小结

registryAwareCluster的Join方法执行newRegistryAwareClusterInvoker(directory);registryAwareClusterInvoker的Invoke方法先通过invoker.directory.List(invocation)获取invokers,然后遍历invokers判断是否有来自local registry的invoker, 如果有则执行invoker.Invoke(ctx, invocation),没有local的则遍历invokers,找到第一个available的执行invoker.Invoke(ctx, invocation)

doc