概要

Take方法作为IEnumerable的扩展方法，具体对应两个重载方法。本文主要分析第一个接收整数参数的重载方法。

源码解析

Take方法的基本定义

public static System.Collections.Generic.IEnumerable Take (this System.Collections.Generic.IEnumerable source, int count);

基本功能是从序列source中，返回指定个数count的相邻元素。

源码分析

Take.cs

public static IEnumerable<TSource> Take<TSource>(this IEnumerable<TSource> source, int count)
{if (source == null){ThrowHelper.ThrowArgumentNullException(ExceptionArgument.source);}return count <= 0 ?Empty<TSource>() :TakeIterator<TSource>(source, count);
}

Take方法本身代码很简单，首先作了一个空序列的检查，如果序列为空，则抛出异常。然后如果count是0，即取前0项相邻元素，等价于什么也不作，直接返回，否则调用TakeIterator方法。

Take.SizeOpt.cs

private static IEnumerable<TSource> TakeIterator<TSource>(IEnumerable<TSource> source, int count)
{Debug.Assert(count > 0);foreach (TSource element in source){yield return element;if (--count == 0) break;}
}

TakeIterator方法并没有像我们之前分析的Where，Select等方法那样，根据功能，定于很多Iterator的派生类来实现具体的功能，而是使用了yield return的方式。

按照count的个数取出对应的元素，以yield return的方式返回。

下面我们使用相同的代码，定义我们自己的扩展方法take 和takeIterator，通过log来搞清楚yield return方式的实现细节。

 public static IEnumerable<TSource> take<TSource>(this IEnumerable<TSource> source, int count)
{Console.WriteLine("take is called !");if (source == null){             ThrowHelper.ThrowArgumentNullException(ExceptionArgument.source);}return count <= 0 ?Empty<TSource>() :takeIterator<TSource>(source, count);
}private static IEnumerable<TSource> takeIterator<TSource>(IEnumerable<TSource> source, int count)
{Console.WriteLine("TakeIterator is called !");Debug.Assert(count > 0);foreach (TSource element in source){Console.WriteLine("Enter takeIterator Foreach");yield return element;Console.WriteLine("Return " + element + " from takeIterator Foreach");if (--count == 0) break;}
}

Case 1 不通过toList或foreach循环来调用take的返回值。

  static void Main(string[] args){var list = Enumerable.Range(1,10).take(2);}

执行结果如下：

在这里插入图片描述
我们可以看到takeIterator并未被调用。

Case 2: 通过foreach循环来调用take的返回值

static void Main(string[] args){var list = Enumerable.Range(1,10).take(2);foreach (var item in list){Console.WriteLine("Enter foreach Main function's  foreach");Console.WriteLine("Print " + item + " in Main function");}}