将此 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));       
}

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

Answer 1

你想要Concat:

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

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