c# - 将此 foreach 产量重写为 linq 产量？

Question

假设我有以下代码(缩小上下文以限制问题范围)

public static IEnumerable<Color> GetThemColors(){
    var ids = GetThePrimaryIds();
    foreach (int id in ids){
        yield return GetColorById(id);
    }
    ids = GetTheOtherIds();
    foreach (int id in ids){
        yield return GetOtherColorsById(id);
    }
}

我想将它们重写为这样的东西(当然不能编译

public static IEnumerable<Color> GetThemColors(){
    GetThePrimaryIds().Select(id=>yield return GetColorById(id));
    GetTheOtherIds().Select(id=>yield return GetOtherColorsById(id));       
}

关键在于，在我的第一个代码片段中，我有两个 foreach 枚举器产生，我不知道如何在 linq 中做到这一点而不会失去我的延迟加载功能。

Answer 1

你想要Concat:

return GetThePrimaryIds().Select(id => GetColorById(id)).Concat(
    GetTheOtherIds().Select(id => GetOtherColorsById(id)));

另请注意，您不需要在 lambda 中使用 yield return。