TerribleDev/LinqRb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

foreach should return the same enumerable

Browse Source
This commit is contained in:
Tommy Parnell
2016-05-10 17:48:06 -04:00
parent bb47821041
commit 13723cc553
1 changed files with 21 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/LinqRb/Extensions/IEnumerable.cs
View File

@@ -111,18 +111,19 @@ namespace System.Linq
}
}
public static void ForEach<T>(this IEnumerable<T> source, Action<T> action)
public static IEnumerable<T> ForEach<T>(this IEnumerable<T> source, Action<T> action)
{
Enumerate(source, action);
return Enumerate(source, action);
}
public static void ForEach<T>(this IEnumerable<T> source, Action<T, int> action)
public static IEnumerable<T> ForEach<T>(this IEnumerable<T> source, Action<T, int> action)
{
var index = 0;
var enumerator = source.GetEnumerator();
while(enumerator.MoveNext())
{
action?.Invoke(enumerator.Current, index);
yield return enumerator.Current;
index++;
}
}
@@ -141,12 +142,13 @@ namespace System.Linq
}
}
private static void Enumerate<T>(IEnumerable<T> source, Action<T> action)
private static IEnumerable<T> Enumerate<T>(IEnumerable<T> source, Action<T> action)
{
var enumerator = source.GetEnumerator();
while(enumerator.MoveNext())
{
action?.Invoke(enumerator.Current);
yield return enumerator.Current;
}
}
}