C#Count()扩展方法性能

时间:2009-10-30 17:41:39

标签: c# linq count extension-methods

如果在具有Count()属性的IEnumerable<T>上调用LINQ Count扩展方法(例如List<T>),Count()方法是否会查找该属性并返回它(而不是通过枚举它们来计算项目)?以下测试代码似乎表明它确实:

using System;
using System.Collections;
using System.Collections.Generic;
using System.Diagnostics;
using System.Linq;

namespace CountSpeedTest
{
    // Output:
    // List      - CLR : 0 ms
    // Enumerate - CLR : 10 ms
    // List      - Mine: 12 ms
    // Enumerate - Mine: 12 ms
    class Program
    {
        private const int Runs = 10;
        private const int Items = 1000000;

        static void Main(string[] args)
        {
            var total = new long[] {0, 0, 0, 0};
            for (int i = 0; i < Runs; ++i)
            {
                var items = Enumerable.Range(0, Items).Select(o => o.ToString()).ToList();
                var list = new List<string>(items);
                var enumerate = new Enumerate<string>(items);
                total[0] += TimeCount(list, c => c.Count());
                total[1] += TimeCount(enumerate, c => c.Count());
                total[2] += TimeCount(list, c => c.SlowCount());
                total[3] += TimeCount(enumerate, c => c.SlowCount());
            }
            Console.WriteLine(String.Format("List      - CLR : {0} ms", total[0] / Runs));
            Console.WriteLine(String.Format("Enumerate - CLR : {0} ms", total[1] / Runs));
            Console.WriteLine(String.Format("List      - Mine: {0} ms", total[2] / Runs));
            Console.WriteLine(String.Format("Enumerate - Mine: {0} ms", total[3] / Runs));
            Console.ReadKey(true);
        }

        private static long TimeCount<T>(IEnumerable<T> collection, Func<IEnumerable<T>, int> counter)
        {
            var stopwatch = Stopwatch.StartNew();
            var count = counter(collection);
            stopwatch.Stop();
            if (count != Items) throw new Exception("Incorrect Count");
            return stopwatch.ElapsedMilliseconds;
        }
    }

    public static class CountExtensions
    {
        // Performs a simple enumeration based count.
        public static int SlowCount<T>(this IEnumerable<T> items)
        {
            var i = 0;
            var enumerator = items.GetEnumerator();
            while (enumerator.MoveNext()) i++;
            return i;
        }
    }

    // Wraps an IEnumerable<T> to hide its Count property.
    public class Enumerate<T> : IEnumerable<T>
    {
        private readonly IEnumerable<T> collection;
        public Enumerate(IEnumerable<T> collection) { this.collection = collection; }

        public IEnumerator<T> GetEnumerator() { return collection.GetEnumerator(); }
        IEnumerator IEnumerable.GetEnumerator() { return GetEnumerator(); }
    }
}

在相关说明中:实现IEnumerable<T>的自定义集合如何以CLR Count扩展方法的方式公开其自己的Count()属性可以利用它吗?

2 个答案:

答案 0 :(得分:12)

它不会按名称查找Count属性,但会检查它是否实现ICollection<T>,然后使用该类型的Count属性。来自documentation

  

如果源的类型实现   ICollection<T>,那   实现用于获取   元素数量。否则,这个   方法确定计数。

(显然这仅适用于不带谓词的重载。)

因此,如果您想有效地获得计数,请确保实施ICollection<T>

答案 1 :(得分:5)

是的,Enumerable.Count方法确实会查找ICollection<T>并使用它的Count属性(如果找到)。您可以通过查看反射器中的Enumerable.Count来验证这一点。

仅当您使用不带其他参数的Count扩展方法时,才会出现这种情况。如果您使用带谓词的版本,它将遍历可枚举元素。