聊聊dubbo-go的availableCluster

巧了我就是萌 提交于 2020-08-12 02:40:46

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

NewAvailableCluster

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

type availableCluster struct{}

const available = "available"

func init() {
	extension.SetCluster(available, NewAvailableCluster)
}

// NewAvailableCluster ...
func NewAvailableCluster() cluster.Cluster {
	return &availableCluster{}
}
  • NewAvailableCluster方法实例化availableCluster

Join

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

func (cluser *availableCluster) Join(directory cluster.Directory) protocol.Invoker {
	return NewAvailableClusterInvoker(directory)
}
  • Join方法执行NewAvailableClusterInvoker

NewAvailableClusterInvoker

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

type availableClusterInvoker struct {
	baseClusterInvoker
}

// NewAvailableClusterInvoker ...
func NewAvailableClusterInvoker(directory cluster.Directory) protocol.Invoker {
	return &availableClusterInvoker{
		baseClusterInvoker: newBaseClusterInvoker(directory),
	}
}
  • NewAvailableClusterInvoker方法实例化availableClusterInvoker

Invoke

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

func (invoker *availableClusterInvoker) Invoke(ctx context.Context, invocation protocol.Invocation) protocol.Result {
	invokers := invoker.directory.List(invocation)
	err := invoker.checkInvokers(invokers, invocation)
	if err != nil {
		return &protocol.RPCResult{Err: err}
	}

	err = invoker.checkWhetherDestroyed()
	if err != nil {
		return &protocol.RPCResult{Err: err}
	}

	for _, ivk := range invokers {
		if ivk.IsAvailable() {
			return ivk.Invoke(ctx, invocation)
		}
	}
	return &protocol.RPCResult{Err: errors.New(fmt.Sprintf("no provider available in %v", invokers))}
}
  • Invoke方法先通过invoker.directory.List(invocation)获取invokers,之后遍历invokers,对于ivk.IsAvailable()为true的执行ivk.Invoke(ctx, invocation)

小结

availableCluster的Join方法执行NewAvailableClusterInvoker,而availableClusterInvoker的Invoke方法先通过invoker.directory.List(invocation)获取invokers,之后遍历invokers,对于ivk.IsAvailable()为true的执行ivk.Invoke(ctx, invocation)

doc

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!