在 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 原生方法用起来几乎没区别。
