63 lines
1.8 KiB
C#
63 lines
1.8 KiB
C#
namespace Azusa.Shared.Extensions;
|
||
|
||
/// <summary>
|
||
/// 使用Range作为参数的迭代器方法
|
||
/// <br/>
|
||
/// 扩展foreach关键字来实现类似<c>foreach (var i in 1..5)</c>的效果
|
||
/// </summary>
|
||
public static class ForeachExtensions
|
||
{
|
||
/// <summary>
|
||
/// 拓展Range结构实现GetEnumerator方法供foreach读取，实现foreach(var i in x..y)
|
||
/// </summary>
|
||
/// <returns></returns>
|
||
public static CustomIntEnumerator GetEnumerator(this Range range)
|
||
{
|
||
return new CustomIntEnumerator(range);
|
||
}
|
||
|
||
/// <summary>
|
||
/// 拓展int类实现GetEnumerator方法供foreach读取，实现foreach(var i in x)
|
||
/// </summary>
|
||
/// <param name="end"></param>
|
||
/// <returns></returns>
|
||
public static CustomIntEnumerator GetEnumerator(this int end)
|
||
{
|
||
return new CustomIntEnumerator(end);
|
||
return new CustomIntEnumerator(new Range(0, end));//在执行空函数时性能比上一句低10倍，Why
|
||
}
|
||
}
|
||
|
||
//使用引用结构体增强性能
|
||
public ref struct CustomIntEnumerator
|
||
{
|
||
private int _current;
|
||
private readonly int _end;
|
||
|
||
public CustomIntEnumerator(Range range)
|
||
{
|
||
//避免某些时候从结尾开始编制
|
||
// x.. 时会产生Range(x,^0)
|
||
if (range.End.IsFromEnd)
|
||
{
|
||
throw new NotSupportedException("不支持从结尾编制索引");
|
||
}
|
||
_current = range.Start.Value - 1;
|
||
_end = range.End.Value - 1;//迭代器不包含范围的尾部
|
||
}
|
||
|
||
public CustomIntEnumerator(int end)
|
||
{
|
||
_current = -1;
|
||
_end = end;
|
||
}
|
||
|
||
/* 注意，供foeach使用的迭代器不需要实现IEnumerator接口，只需要提供Current属性以及MoveNext方法即可，*/
|
||
public int Current => _current;
|
||
|
||
public bool MoveNext()
|
||
{
|
||
_current++;
|
||
return _current <= _end;
|
||
}
|
||
} |