组成
从名字常来看,IEnumerator是枚举器的意思,IEnumerable是可枚举的意思。这两个都是个接口
接下来我们看一下IEnumerator和IEnumerable的源码
public interface IEnumerable
{
IEnumerator GetEnumerator();
}
IEnumerable非常简单,就一个GetEnumerator()方法,但是这个方法的返回值却是一个IEnumerator对象,从注释我们可以得知IEnumerable代表继承此接口的类可以获取一个IEnumerator来实现枚举这个类中包含的集合中的元素的功能(比如List<T>,ArrayList,Dictionary等继承了IEnumeratble接口的类,接下来我们看下IEnumerator这个接口中有什么。
public interface IEnumerator
{
object Current { get; }
bool MoveNext();
void Reset();
}
IEnumerator有一个object类型的属性,还有一个返回值为bool值的MoveNext方法和一个无返回值的Reset方法
实现接口
我们模仿ArrayList来实现一个简单的MyList,然后用foreach遍历。
public class MyList : IEnumerable
{
public int[] _data = new int[10] { 1, 5, 7, 9, 7, 8, 7, 8, 7, 4 };
public int this[int index]
{
get
{
return _data[index];
}
}
IEnumerator IEnumerable.GetEnumerator()
{
Debug.Log("foreach调用 GetEnumerator");
return new MIEnumtor(this);
}
}
我们的MyList提供了一个数组用来遍历,然后提供一个索引器取得里面的元素,然后我们实现IEnumerable这个接口的GetEnumerator方法。
public class MIEnumtor : IEnumerator
{
private MyList myList;
private int index;
public MIEnumtor(MyList my)
{
index = -1;
myList = my;
}
public object Current
{
get
{
Debug.Log("foreach调用 Current");
return myList[index];
}
}
public bool MoveNext()
{
Debug.Log("foreach调用 MoveNext");
if (index < myList._data.Length - 1)
{
index++;
return true;
}
index = -1;
return false;
}
public void Reset()
{
}
}
然后我们提供一个实现了IEnumerator接口的MIEnumtor 类,我们定义了一个索引index用来指向当前集合的元素位置,在属性Current中我们返回index位置的元素,每调用一次MoveNext我们就将索引向下移动,知道集合的最后一个元素时,重置索引并返回false,然后我们用Foreach来遍历这个集合。
public class Test2 : MonoBehaviour
{
private void Awake()
{
MyList my = new MyList();
foreach (var item in my)
{
Debug.Log(item);
}
}
}
最后我们的日志输出了如下结果
分析
通过输出结果,我们可以发现,foreach在运行时会先调用MyList的GetIEnumerator函数获取一个MIEnumtor,之后通过循环调用MIEnumtor的MoveNext函数,index后移,更新Current属性,然后返回Current属性,直到MoveNext返回false。
总结
GetIEnumerator()负责获取枚举器。
MoveNext()负责让Current获取下一个值,并判断遍历是否结束。
Current负责返回当前指向的值。
Rest()负责重置枚举器的状态(在foreach中没有用到)
这些就是IEnumerable,IEnumerator的基本工作原理了。
其次我们发现
MyList my = new MyList();
foreach (var item in my)
{
Debug.Log(item);
}
其实就等价于
MyList my = new MyList();
MIEnumtor mIEnumtor = my.GetEnumerator();
while (mIEnumtor.MoveNext())
{
Debug.Log(mIEnumtor.Current);
}
也就是说foreach其实是一种语法糖,用来简化对可枚举元素的遍历代码。而被遍历的类通过实现IEnumerable接口和一个相关的IEnumerator枚举器来实现遍历功能。