Question

我目前正在构建一个处理IEnumerable<TSource>的扩展方法。对于源集合中的每个项目，我需要生成一个或两个结果类型的项目。

这是我的第一种方法，它失败了，因为该方法留在第一个return语句中，并在第二个return语句被命中时忘记它的状态。

public static IEnumerable<TResult> DoSomething<TSource>(this IEnumerable<TSource> source)
    where TResult : new()
{
    foreach(item in source)
    {
        if (item.SomeSpecialCondition)
        {
            yield return new TResult();
        }
        yield return new TResult();
    }
}

我如何正确实施此方案？

Answer 1

您的解决方案应该有效。这是一个完整的示例程序，演示了该方法的工作原理：

using System;
using System.Collections.Generic;
using System.Linq;

namespace Demo
{
    class Program
    {
        void run()
        {
            var test = DuplicateOddNumbers(Enumerable.Range(1, 10));

            foreach (var n in test)
                Console.WriteLine(n);
        }

        public IEnumerable<int> DuplicateOddNumbers(IEnumerable<int> sequence)
        {
            foreach (int n in sequence)
            {
                if ((n & 1) == 1)
                    yield return n;

                yield return n;
            }
        }

        static void Main(string[] args)
        {
            new Program().run();
        }
    }
}

另外，请注意Dmitry Dovgopoly关于使用正确的TSource和TResult的评论。

如何实现扩展方法，为每个输入值产生两个结果？

1 个答案: