C#具有空检查的重复代码

时间:2016-12-03 13:23:11

标签: c# rpc null-check

我正在使用RPC(protobuf-remote),我需要做一些检查以防另一端(服务器)关闭。让我们说很多RPC方法,比如:

public FirstObj First(string one, string two)
{
    if (rpc == null)
        return (FirstObj)Activator.CreateInstance(typeof(FirstObj));

    return rpc.First(one, two);
}

public SecondObj Second(string one)
{
    if (rpc == null)
        return (SecondObj)Activator.CreateInstance(typeof(SecondObj));

    return rpc.Second(one);
}

public ThirdObj Third()
{
    if (rpc == null)
        return (ThirdObj)Activator.CreateInstance(typeof(ThirdObj));

    return rpc.Third();
}

有没有改变这个重复的空检查代码?所以我可以这样写:

public FirstObj First(string one, string two)
{
    return rpc.First(one, two);
}

如果RPC服务器关闭,那将进行空检查并按类型创建对象,因此我将获得所需对象的默认值。

2 个答案:

答案 0 :(得分:4)

您可以创建此类扩展方法:

public static class RpcExtension
{
    public static T GetObject<T>(this RPC rpc, Func<RPC, T> func)
        where T: class , new ()
    {
        if (rpc == null)
        {
            return Activator.CreateInstance<T>();
        }
        return func(rpc);
    }
}

用于此用途:

var first = rpc.GetObject(r => r.First(a, b));

答案 1 :(得分:2)

您可以使用通用方法简化代码:

private static T Make<T>() {
    return (T)Activator.CreateInstance(typeof(T));
}
public FirstObj First(string one, string two) {
    return rpc == null ? Make<FirstObj>() : rpc.First(one, two);
}
public SecondObj Second(string one) {
    return rpc == null ? Make<SecondObj>() : rpc.Second(one);
}
public ThirdObj Third() {
    return rpc == null ? Make<ThirdObj>() : rpc.Third();
}

如果FirstObjSecondObjThirdObj类型都是类,而不是struct或基元,而rpc永远不会返回null对他们来说,你可以这样做:

public static T RpcOrDefault<T>(T obj) where T : class {
    return obj ?? (T)Activator.CreateInstance(typeof(T));
}

并将其命名为

FirstObj first = RpcOrDefault(rpc?.First(one, two));
//                               ^

请注意?中的?.,以免您null引用例外。