关于dubbo:聊聊dubbogo的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

• registry_aware_cluster