C# 如何自定义LINQ扩展方法 - 为IEnumerable添加新功能

来源:这里教程网 时间:2026-02-21 17:35:27 作者:

在 C# 中,为

IEnumerable<t></t>
添加自定义 LINQ 扩展方法,本质是写一个 静态类中的静态方法,且第一个参数用
this
修饰符绑定到
IEnumerable<t></t>
类型。这样就能像原生
Where
Select
那样链式调用。

扩展方法必须满足的三个条件

缺一不可,否则编译器不认作扩展方法:

方法必须定义在非泛型静态类中(比如叫
EnumerableExtensions
方法本身必须是静态方法 第一个参数必须是
this IEnumerable<t> source</t>
(或具体类型如
this IEnumerable<string> source</string>
),且
T
要在方法签名中声明为泛型参数

写一个实用的扩展:GetOrDefault

类似字典的

TryGetValue
,但用于集合——安全取第 N 个元素,越界时返回默认值而非异常:

public static class EnumerableExtensions
{
    public static T GetOrDefault<T>(this IEnumerable<T> source, int index, T defaultValue = default)
    {
        if (source == null) throw new ArgumentNullException(nameof(source));
        if (index     // 尽量避免 ToList(),用迭代器高效处理
    var enumerator = source.GetEnumerator();
    for (int i = 0; i <= index; i++)
    {
        if (!enumerator.MoveNext())
            return defaultValue;
        if (i == index)
            return enumerator.Current;
    }
    return defaultValue;
}

}

使用示例:

var list = new[] { "a", "b", "c" };
Console.WriteLine(list.GetOrDefault(1));     // "b"
Console.WriteLine(list.GetOrDefault(5));     // null(string 默认值)
Console.WriteLine(list.GetOrDefault(5, "N/A")); // "N/A"

注意性能与空值安全

扩展方法不是魔法,它只是语法糖。实际调用仍走迭代逻辑,所以要注意:

避免在方法内部无脑调用
ToList()
ToArray()
,尤其对大数据流或 IO 枚举器(如文件行枚举)会造成额外内存和延迟
始终校验
source
是否为
null
,并给出清晰异常信息
如果扩展行为依赖索引(如分页、取第 N 项),考虑是否支持
IList<t></t>
优化路径(用
source is IList<t> list ? list[index] : ...</t>

让扩展方法支持链式调用和延迟执行

保持与标准 LINQ 一致的行为:返回

IEnumerable<t></t>
、不立即执行、支持
yield return

public static IEnumerable<T> WhereNotNull<T>(this IEnumerable<T> source) where T : class
{
    if (source == null) throw new ArgumentNullException(nameof(source));
    foreach (T item in source)
    {
        if (item != null) yield return item;
    }
}

这样就能无缝接入现有链式调用:

var result = items.Where(x => x.Length > 3)
                  .WhereNotNull()
                  .Select(x => x.ToUpper());

基本上就这些。核心就是“静态类 + 静态方法 + this 参数”,再加一点对延迟执行和空值的敬畏。写多了你会发现,自己写的扩展和 LINQ 原生方法用起来几乎没区别。

相关推荐