聊聊dubbo-go的availableCluster

序

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

• available_cluster