如何创建 IEnumerable哪些多重枚举是不可能的？

Question

当我枚举 IEnumerable 两次时，Resharper 抱怨 可能多次枚举 IEnumerable。我知道，在某些数据库查询的情况下，当您枚举两次时，您会得到异常。

我想在测试中重现这种行为。所以，我基本上希望抛出以下函数（因为多个枚举）：

    private void MultipleEnumerations(IEnumerable<string> enumerable)
    {
        MessageBox.Show(enumerable.Count().ToString());
        MessageBox.Show(enumerable.Count().ToString());
    }

我应该传递什么给它？所有列表、集合等都可以使用多个枚举。 即使这种 IEnumerable 也不例外：

    private IEnumerable<string> GetIEnumerable()
    {
        yield return "a";
        yield return "b";
    }

谢谢。

Answer 1

您可能只想要一个自定义类：

public class OneShotEnumerable<T> : IEnumerable<T>
{
    private readonly IEnumerable<T> _source;
    private bool _shouldThrow = false;

    public OneShotEnumerable(IEnumerable<T> source)
    {
        this._source = source;
    }

    public IEnumerator<T> GetEnumerator()
    {
        if (_shouldThrow) throw new InvalidOperationException();
        _shouldThrow = true;

        return _source.GetEnumerator();
    }

    IEnumerator IEnumerable.GetEnumerator()
    {
        return GetEnumerator();
    }
}

Answer 2

创建您自己的类来实现 IEnumerable，并在调用 GetEnumerator() 两次时引发异常（使用布尔实例字段）。

或者，创建一个使用标志字段的迭代器，以确保它不能被调用两次（枚举迭代器两次将执行整个方法两次）。

Answer 3

我从 John Gietzen 的答案中复制的自定义类（经过一些更正）可以有效地与扩展方法结合起来，以创建一种非常简单的方法来执行此操作。

public class OneShotEnumerable<T> : IEnumerable<T>
{
    private readonly IEnumerable<T> source;
    private bool shouldThrow = false;

    public OneShotEnumerable(IEnumerable<T> source)
    {
        this.source = source;
    }

    public IEnumerator<T> GetEnumerator()
    {
        if (shouldThrow) 
            throw new InvalidOperationException("This enumerable has already been enumerated.");

        shouldThrow = true;

        return this.source.GetEnumerator();
    }
}

public static clas OneShotEnumerableExtension
{
    public static IEnumerable<T> SingleUse<T>(this IEnumerable<T> source)
    {
#if (DEBUG)
        return new OneShotEnumerableExtension(source);
#else
        return source;
#endif
    }
}

然后你可以通过简单地把一些东西传递给你以前的方法

MultipleEnumerations(MyEnumerable.SingleUse());