c# - 仅当第二个 TSource 不为 Null 时联合集合

Question

我有以下 linq 语句:

List<Type> allTypes = group.GetTypes().Union(group2.GetTypes()).ToList();

当 group2 为 null 时可能会抛出 NullReferenceException

解决这个问题的一种方法是在之前执行空检查，例如:

if (group2 != null)
{
    List<Type> allTypes = group.GetTypes().Union(group2.GetTypes()).ToList();
}
else 
{
   List<Type> allTypes = group.GetTypes();
}

但问题是我对不同类型有很多类似的赋值，不想以这种方式对它们中的每一个进行 if 语句，但我宁愿将空检查放在一行中，例如:

 List<Type> allTypes = group.GetTypes().Union((if group2 != null)group2.GetTypes()).ToList();

但不确定如何使用 linq 来完成。

Answer 1

这里的问题不是 TSource 为空；它是您要从中获取源的对象是 null (group2)。

您始终可以使用 Enumerable.Empty 来保存您的神奇 one liner。

List<Type> allTypes = group.GetTypes().Union(group2 != null ? group2.GetTypes() : Enumerable.Empty<Type>()).ToList();

或者您可以使用可重用的 Union 重载:

public static IEnumerable<T> Union<T>(this IEnumerable<IEnumerable<T>> source)
{
    var set = new HashSet<T>();
    foreach (var s in source)
    {
       foreach (var item in s)
       {
           if (set.Add(item))
               yield return item;
       }
    }
}

然后你的代码变成:

var allTypes = new [] { group, group2 }.Where(x => x != null).Select(x => x.GetTypes()).Union().ToList();

这种方法的优点是你可以有两个以上的序列形成一个联合。