概要
本文在前两篇Take源码分析的基础上，着重分析Range参数中有倒数的情况，即分析TakeRangeFromEndIterator的源码实现。
源码及分析
TakeRangeFromEndIterator方法用于处理Range中的开始和结束索引存在倒数的情况。该方法位于Take.cs文件中。通过yield return/break的方式管理迭代过程。
TakeRangeFromEndIterator方法从整体上分为两部分，一部分是TryGetNonEnumeratedCount为真的情况，即souce实现了ICollection接口的情况；另一部分是souce没有实现ICollection接口，TryGetNonEnumeratedCount返回为False的情况。
private static IEnumerable<TSource> TakeRangeFromEndIterator<TSource>(IEnumerable<TSource> source, bool isStartIndexFromEnd, int startIndex, bool isEndIndexFromEnd, int endIndex)
{
if (source.TryGetNonEnumeratedCount(out int count))
{
startIndex = CalculateStartIndex(isStartIndexFromEnd, startIndex, count);
endIndex = CalculateEndIndex(isEndIndexFromEnd, endIndex, count);
if (startIndex < endIndex)
{
foreach (TSource element in
TakeRangeIterator(source, startIndex, endIndex))
{
yield return element;
}
}
yield break;
}
static int CalculateStartIndex(bool isStartIndexFromEnd, int startIndex, int count) =>
Math.Max(0, isStartIndexFromEnd ? count - startIndex : startIndex);
static int CalculateEndIndex(bool isEndIndexFromEnd, int endIndex, int count) =>
Math.Min(count, isEndIndexFromEnd ? count - endIndex : endIndex);
}
- 该函数有4个参数，开始索引是否倒数，开始索引值，结束索引是否为倒数，结束索引值；
- 如果source实现了ICollection接口，可以在不遍历souce序列的情况下，直接获取序列长度，则TryGetNonEnumeratedCount返回为True；
- 调用CalculateStartIndex和CalculateEndIndex内联方法，获取正数的索引值，假设Range是 ^3… ^1，元素共10个，则转换完成后是7…9，即从第7个取到第9个；
- 在开始索引值小于结束索引值的前提下，调TakeRangeIterator方法，按照普通正数Range的方法进行处理，并将结果以状态机的形式按照yield return/break方式返回。具体详见 C# Linq源码分析之Take （二）
注意在TryGetNonEnumeratedCount返回为True的情况下，因为可以直接取到序列的元素个数，不需要进行逐个迭代和累加。因此才可以将倒数的Range转换成正数的Range。对于无法获取序列元素的情况，我们看下面的代码分析。
在开始索引是倒数的情况下，进行如下处理，此时假设我们有如下序列，我们的Range是 ^3 … ^1，但是不知道序列内元素个数。
Queue<TSource> queue;
if (isStartIndexFromEnd)
{
// TakeLast compat: enumerator should be disposed before yielding the first element.
using (IEnumerator<TSource> e = source.GetEnumerator())
{
if (!e.MoveNext())
{
yield break;
}
queue = new Queue<TSource>();
queue.Enqueue(e.Current);
count = 1;
while (e.MoveNext())
{
if (count < startIndex)
{
queue.Enqueue(e.Current);
++count;
}
else
{
do
{
queue.Dequeue();
queue.Enqueue(e.Current);
checked { ++count; }
} while (e.MoveNext());
break;
}
}
Debug.Assert(queue.Count == Math.Min(count, startIndex));
}
startIndex = CalculateStartIndex(isStartIndexFromEnd: true, startIndex, count);
endIndex = CalculateEndIndex(isEndIndexFromEnd, endIndex, count);
Debug.Assert(endIndex - startIndex <= queue.Count);
for (int rangeIndex = startIndex; rangeIndex < endIndex; rangeIndex++)
{
yield return queue.Dequeue();
}
}
- 获取souce的迭代器e；
- 如果取一个元素失败，证明Range中指定的范围在实际序列中根本取不到，则通过yield break关闭状态机；
- 定义缓冲队列queue，将“A”放入队列，元素个数初始值设置为1；
- 迭代开始，count < startIndex在元素个数累加器小于启始索引的情况下，每次队列增加一个元素，直到count等于3，此时队列元素是"A", “B”, “C”;
- 然后每次删除队首元素，再添加新的元素到队尾，当遍历完整个source序列后，队列元素是“G”， “H”， “I”，此种方法遍历，算法只需要启始索引值，结束索引值完全忽略；
- 根据迭代中获取的元素个数count，计算出正数的开始和结束索引，本例应该是6…8;
- 将缓冲队列queue按照状态机的形式，通过yield return方式返回；因为rangeIndex < endIndex;，所以最后的返回值是“G”， “H”没有“I”，只会取到结束索引对应元素的前一个元素。
在开始索引不是倒数的情况下， 进行如下处理，此时假设我们有如下序列，我们的Range是 3 … ^2，此时我们并不清楚集合内元素的个数。
请注意在现有的情况下，如果开始索引是正数，结尾索引一定是倒数的。如果结尾索引是正数，更加之前的代码分析，只会进入C# Linq源码分析之Take （二）所介绍的TakeRangeIterator方法。
假设我们使用的数据如下：
else
{
Debug.Assert(!isStartIndexFromEnd && isEndIndexFromEnd);
// SkipLast compat: the enumerator should be disposed at the end of the enumeration.
using IEnumerator<TSource> e = source.GetEnumerator();
count = 0;
while (count < startIndex && e.MoveNext())
{
++count;
}
if (count == startIndex)
{
queue = new Queue<TSource>();
while (e.MoveNext())
{
if (queue.Count == endIndex)
{
do
{
queue.Enqueue(e.Current);
yield return queue.Dequeue();
} while (e.MoveNext());
break;
}
else
{
queue.Enqueue(e.Current);
}
}
}
}
- 定义迭代器e 和 集合内初始元素个数计数器设置为0；
- 在初始化操作后，count是3，迭代器指向元素C；
- 如果count等于 startIndex，证明Range中的取值在下面的序列中可以取到。如果不等，则返回空；
- 因为是倒数第2个，所以endIndex的数值是2，
- D和E进入缓冲队列；
- 因为缓冲队列中的长度必须和endIndex相等，队列每进入一个元素到队尾，然后删除队首元素并按照yield return的状态机方式返回。
最后将所有需要的元素返回。上面算法并不需要直到序列内元素的个数。
