聊聊dubbo-go的registryAwareCluster

324 阅读1分钟

本文主要研究一下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