Как бы вы реализовали интерфейс IEnumerator?
-
09-06-2019 - |
Вопрос
У меня есть класс, который сопоставляет объекты с объектами, но в отличие от словаря он отображает их в обоих направлениях.Сейчас я пытаюсь реализовать собственный интерфейс IEnumerator, который перебирает значения.
public class Mapper<K,T> : IEnumerable<T>, IEnumerator<T>
{
C5.TreeDictionary<K,T> KToTMap = new TreeDictionary<K,T>();
C5.HashDictionary<T,K> TToKMap = new HashDictionary<T,K>();
public void Add(K key, T value)
{
KToTMap.Add(key, value);
TToKMap.Add(value, key);
}
public int Count
{
get { return KToTMap.Count; }
}
public K this[T obj]
{
get
{
return TToKMap[obj];
}
}
public T this[K obj]
{
get
{
return KToTMap[obj];
}
}
public IEnumerator<T> GetEnumerator()
{
return KToTMap.Values.GetEnumerator();
}
public T Current
{
get { throw new NotImplementedException(); }
}
public void Dispose()
{
throw new NotImplementedException();
}
object System.Collections.IEnumerator.Current
{
get { throw new NotImplementedException(); }
}
public bool MoveNext()
{
;
}
public void Reset()
{
throw new NotImplementedException();
}
}
Решение
Во-первых, не заставляйте свой объект коллекции реализовывать IEnumerator<>.Это приводит к ошибкам.(Рассмотрим ситуацию, когда два потока выполняют итерацию по одной и той же коллекции).
Корректная реализация перечислителя оказывается нетривиальной задачей, поэтому в C# 2.0 для этого добавлена специальная языковая поддержка, основанная на операторе return return.
Недавняя серия публикаций в блоге Рэймонда Чена («Реализация итераторов в C# и ее последствия») — хороший повод для начала.
Другие советы
Просто реализуйте интерфейс IEnumerable, нет необходимости реализовывать IEnumerator, если вы не хотите делать какие-то специальные действия в перечислителе, которые в вашем случае, похоже, не нужны.
public class Mapper<K,T> : IEnumerable<T> {
public IEnumerator<T> GetEnumerator()
{
return KToTMap.Values.GetEnumerator();
}
}
вот и все.
CreateEnumerable()
возвращает IEnumerable
который реализует GetEnumerator()
public class EasyEnumerable : IEnumerable<int> {
IEnumerable<int> CreateEnumerable() {
yield return 123;
yield return 456;
for (int i = 0; i < 6; i++) {
yield return i;
}//for
}//method
public IEnumerator<int> GetEnumerator() {
return CreateEnumerable().GetEnumerator();
}//method
IEnumerator IEnumerable.GetEnumerator() {
return CreateEnumerable().GetEnumerator();
}//method
}//class
Используйте доходность.
Вот пример из книги Роберта Седжвика «Алгоритмы (4-е издание)».
Он был написан на Java, и я практически переписал его на C#.
public class Stack<T> : IEnumerable<T>
{
private T[] array;
public Stack(int n)
{
array = new T[n];
}
public Stack()
{
array = new T[16];
}
public void Push(T item)
{
if (Count == array.Length)
{
Grow(array.Length * 2);
}
array[Count++] = item;
}
public T Pop()
{
if (Count == array.Length/4)
{
Shrink(array.Length/2);
}
return array[--Count];
}
private void Grow(int size)
{
var temp = array;
array = new T[size];
Array.Copy(temp, array, temp.Length);
}
private void Shrink(int size)
{
Array temp = array;
array = new T[size];
Array.Copy(temp,0,array,0,size);
}
public int Count { get; private set; }
public IEnumerator<T> GetEnumerator()
{
return new ReverseArrayIterator(Count,array);
}
IEnumerator IEnumerable.GetEnumerator()
{
return GetEnumerator();
}
// IEnumerator implementation
private class ReverseArrayIterator : IEnumerator<T>
{
private int i;
private readonly T[] array;
public ReverseArrayIterator(int count,T[] array)
{
i = count;
this.array = array;
}
public void Dispose()
{
}
public bool MoveNext()
{
return i > 0;
}
public void Reset()
{
}
public T Current { get { return array[--i]; } }
object IEnumerator.Current
{
get { return Current; }
}
}
}