CachedEnumerator

CachedEnumerator.cs

// Source: https://wilsonhut.wordpress.com/2012/03/15/obvious-extension-methods-ienumerable-cache/

public static class Extensions
{
    public static CachedEnumerable<T> Cache<T>(this IEnumerable<T> enumerable)
    {
        return new CachedEnumerable<T>(enumerable);
    }
}

public class CachedEnumerable<T> : IEnumerable<T>
{
    IEnumerator<T> _originalEnumerator;
    readonly IEnumerable<T> _originalEnumerable;
    readonly List<T> _cache = new List<T>();
    public CachedEnumerable(IEnumerable<T> enumerable)
    {
        _originalEnumerable = enumerable;
    }
    public IEnumerator<T> GetEnumerator()
    {
        foreach (var t in _cache)
        {
            yield return t;
        }
        if (_originalEnumerator == null)
        {
            _originalEnumerator = _originalEnumerable.GetEnumerator();
        }
        while (_originalEnumerator.MoveNext())
        {
            _cache.Add(_originalEnumerator.Current);
            yield return _originalEnumerator.Current;
        }
    }
    IEnumerator IEnumerable.GetEnumerator()
    {
        return GetEnumerator();
    }
}