聊聊dubbo-go的failsafeCluster

type failsafeCluster struct{}

const failsafe = "failsafe"

func init() {
    extension.SetCluster(failsafe, NewFailsafeCluster)
}

// NewFailsafeCluster ...
func NewFailsafeCluster() cluster.Cluster {
    return &failsafeCluster{}
}

func (cluster *failsafeCluster) Join(directory cluster.Directory) protocol.Invoker {
    return newFailsafeClusterInvoker(directory)
}

type failsafeClusterInvoker struct {
    baseClusterInvoker
}

func newFailsafeClusterInvoker(directory cluster.Directory) protocol.Invoker {
    return &failsafeClusterInvoker{
        baseClusterInvoker: newBaseClusterInvoker(directory),
    }
}

func (invoker *failsafeClusterInvoker) 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{}
    }

    url := invokers[0].GetUrl()
    methodName := invocation.MethodName()
    //Get the service loadbalance config
    lb := url.GetParam(constant.LOADBALANCE_KEY, constant.DEFAULT_LOADBALANCE)
    //Get the service method loadbalance config if have
    if v := url.GetMethodParam(methodName, constant.LOADBALANCE_KEY, ""); v != "" {
        lb = v
    }
    loadbalance := extension.GetLoadbalance(lb)

    invoked := make([]protocol.Invoker, 0)
    var result protocol.Result

    ivk := invoker.doSelect(loadbalance, invocation, invokers, invoked)
    //DO INVOKE
    result = ivk.Invoke(ctx, invocation)
    if result.Error() != nil {
        // ignore
        logger.Errorf("Failsafe ignore exception: %v.\n", result.Error().Error())
        return &protocol.RPCResult{}
    }
    return result
}

// GetLoadbalance ...
func GetLoadbalance(name string) cluster.LoadBalance {
    if loadbalances[name] == nil {
        panic("loadbalance for " + name + " is not existing, make sure you have import the package.")
    }

    return loadbalances[name]()
}