如何为多个foreach实现正确的IEnumerator接口？

Question

我有这样的代码：

        class T : IEnumerable, IEnumerator
        {
            private int position = -1;

            public T() { }

            public IEnumerator GetEnumerator() { return this; }

            public object Current { get { return position; } }

            public bool MoveNext()
            {
                position++;
                return (position < 5);
            }

            public void Reset() { position = -1; }
        }

//Using in code:
T t = new T();
foreach (int i in t)
 //to do something

在上面的代码中，所有内容都工作正常，但是当我使用 next 时：

foreach (int i in t)
   if (i == 2)
     foreach (int p in t)
       //print p
   else
       //print i

它打印（在括号中的第二个循环中）： 0 1 (3 4) 2 而不是 0 1 (0 1 2 3 4) 2 3 4 我在列表和集合上测试了它，他们做得正确。 我怎样才能达到我所需要的？

Answer 1

你不能，因为你已经让你的代码表面成为一个枚举器，在我看来，这本身就是一个错误。对我来说，更好的版本是：

class T : IEnumerable<int> {
    public IEnumerator<int> GetEnumerator() {
        int i = 0;
        while(i < 5) {
            yield return i;
            i++;
        }
    }
    IEnumerator IEnumerable.GetEnumerator() { return GetEnumerator(); }
}

编译器将创建正确的设备来使用单独的枚举器来实现此目的。

除非您正在为 .NET 1.1 编写代码，否则如果您发现自己手动编写了一个枚举器，那么很有可能您的做法很困难，而且还可能会出错。

如果您确实必须采取困难的方式：

class T : IEnumerable<int>
{
    public T() { }

    public IEnumerator<int> GetEnumerator() { return new TEnumerator(); }
    IEnumerator IEnumerable.GetEnumerator() { return GetEnumerator(); }
    private class TEnumerator : IEnumerator<int>
    {
        private int position = -1;
        public int Current { get { return position; } }
        object IEnumerator.Current { get { return Current; } }
        void IDisposable.Dispose() {}
        public bool MoveNext()
        {
            position++;
            return (position < 5);
        }
        public void Reset() { position = -1; }
    } 
}

这里的意义在于TEnumerator的不同实例允许相同要单独迭代的 T 实例。

Answer 2

foreach (int i in t)
   if (i == 2)
     foreach (int p in t)
       //print p
   else
       //print i

首先总是使用大括号，当你缩进匹配另一个if时会发生什么，这会让事情变得混乱。

foreach (int i in t) {
   if (i == 2) {
     foreach (int p in t) {
       //print p
      }
   } else {
       //print i
   }
 }

但您会遇到问题：每个 T 实例只有一个计数器，并且您使用的是同一个实例。因此你做一次。如果您希望允许并发枚举，则枚举器对象需要与每次返回一个新实例的 GetEnumerator 分开。