这可以通过单个 LINQ 查询来实现吗？

Question

假设我有一个 IEnumerable 类型的给定对象，它是方法 SomeMethod() 的返回值，并且不包含重复元素。我希望能够在单个 LINQ 查询中“压缩”以下几行：

IEnumerable<string> someList = SomeMethod();

if (someList.Contains(givenString))
{
    return (someList.Where(givenString));
}
else
{
    return (someList);
}

编辑： 我错误地使用了 Single 而不是 First。现已更正。

我知道我可以使用三元运算符“压缩”它，但这不是重点。我只想列出能够用一行来实现这一点。这可能吗？

Answer 1

这将返回具有给定字符串的项目或所有项目（如果给定的字符串不存在于列表中）：

someList.Where(i => i == givenString || !someList.Contains(givenString))

Answer 2

您所需输出的性质要求您要么像现在一样发出两个数据请求，要么缓冲不匹配项，以便在未找到匹配项时返回。后者在实际获取数据是一个相对昂贵的调用的情况下特别有用（例如：数据库查询或 WCF 服务）。缓冲方法如下所示：

static IEnumerable<T> AllIfNone<T>(this IEnumerable<T> source, 
                                   Func<T, bool> predicate)
{
    //argument checking ignored for sample purposes
    var buffer = new List<T>();
    bool foundFirst = false;
    foreach (var item in source)
    {
        if (predicate(item))
        {
            foundFirst = true;
            yield return item;
        }
        else if (!foundFirst)
        {
            buffer.Add(item);
        }
    }

    if (!foundFirst)
    {
        foreach (var item in buffer)
        {
            yield return item;
        }
    }
}

此方法的惰性是 Where 或 ToList 的惰性，具体取决于集合是否包含匹配项。如果是这样，您应该获得类似于 Where 的执行。如果没有，您将大致了解调用 ToList 的执行情况（包含所有失败的过滤器检查的开销）并迭代结果。

Answer 3

三元运算符有什么问题？

someList.Any(s => s == givenString) ? someList.Where(s => s == givenString) : someList;

最好先执行“Where”，然后执行“Any”，但我不知道如何单行执行此操作。

var reducedEnumerable = someList.Where(s => s == givenString);

return reducedEnumerable.Any() ? reducedEnumerable : someList;

Answer 4

无法更改方法的返回类型，这就是您所要求的。第一个条件返回字符串，第二个条件返回字符串集合。

只需返回 IEnumerable 集合，然后对返回值调用 Single 即可，如下所示：

string test = ReturnCollectionOfStrings().Single(x => x == "test");