Linq Distinct扩展

前言

问题描述：Person类有两个属性ID(int)、Name(string)属性。筛选序列中不重复的Person。

public class Person
{
    public int ID { get; set; }
    public string Name { get; set; }
}

好吧，看样子得用Distinct方法。不过Distinct方法有两个重载。

public static IEnumerable<TSource> Distinct<TSource>(
    this IEnumerable<TSource> source
)

public static IEnumerable<TSource> Distinct<TSource>(
    this IEnumerable<TSource> source,
    IEqualityComparer<TSource> comparer
)

看样子已经够用了。不过如果我想只按ID唯一，或者按Name唯一呢？派生IEqualityComparer倒是可以解决。不过那也忒坑爹了。还是自己试着写一个扩展吧。

Distinct扩展

版本1

参考封装的Distinct的方法先写一个。声明一个List<TSource>，循环的时候比较，如果不同就加进去。

public static IEnumerable<TSource> Distinct<TSource, TProperty>(this IEnumerable<TSource> source, Func<TSource, TProperty> selector)
{
    if (source == null) throw new ArgumentNullException("source");
    if (selector == null) throw new ArgumentNullException("selector");
    List<TSource> temp = new List<TSource>();
    foreach (var item in source)
    {
        if (temp.Any(t => selector(t).Equals(selector(item))))
        {
            continue;
        }
        temp.Add(item);
    }
    return temp;
}

测试一下，没什么问题。不过在函数里声明List<TSource>感觉怪怪的，想办法换种方式。

List<Person> list = new List<Person>()
{
    new Person() { ID = 1, Name = "Alen1" },
    new Person() { ID = 1, Name = "Alen2" },
    new Person() { ID = 2, Name = "Alen2" },
    new Person() { ID = 2, Name = "Alen1" }
};

var temp = list.Distinct(t => t.ID);

版本2

取指定属性的唯一序列，在循环的时候使用yield返回。

public static IEnumerable<TSource> Distinct<TSource, TProperty>(this IEnumerable<TSource> source, Func<TSource, TProperty> selector)
{
    if (source == null) throw new ArgumentNullException("source");
    if (selector == null) throw new ArgumentNullException("selector");

    var pdis = source.Select(t => selector(t)).Distinct();
    foreach (var item in pdis)
    {
        yield return source.First(t => selector(t).Equals(item));
    }
}

测试通过。

用Stopwatch分别测试两个函数循环不同次数的时间。对比一下，循环次数在10,000次以上时，版本2耗时大概是版本1的1/40，性能差距算是比较大。

后记

目前来说已经够用了，如果还有其他需求，再根据情况扩展吧。

list.Distinct(t => t.ID);
list.Distinct(t => new { t.ID, t.Name }); //根据ID及Name的值分别比较来取唯一，并非比较引用