c# 扩展方法奇思妙用基础篇 六：WhereIf 扩展

一、Where 扩展的不足

如下界面，可通过姓名、编号和地址对人员进行模糊查询：

我们通常会写出如下代码：

 

 

public IQueryable<Person> Query(IQueryable<Person> source, string name, string code, string address)
{
    var result = source;
    if(string.IsNullOrEmpty(name) == false)
        result = source.Where(p => p.Name.Contains(name));
    if (string.IsNullOrEmpty(code) == false)
        result = source.Where(p => p.Code.Contains(code));
    if (string.IsNullOrEmpty(address) == false)
        result = source.Where(p => p.Code.Contains(address));
    return result;
}

以上代码有大量的 if 显得很繁琐，代码可读性不好。

二、创建并使用 WhereIf 扩展

WhereIf 扩展比较简单，代码如下：

public static IQueryable<T> WhereIf<T>(this IQueryable<T> source, Expression<Func<T, bool>> predicate, bool condition)
{
    return condition ? source.Where(predicate) : source;
}

上面的代码可简化成：

public IQueryable<Person> Query(IQueryable<Person> source, string name, string code, string address)
{
    return source
        .WhereIf(p => p.Name.Contains(name), string.IsNullOrEmpty(name) == false)
        .WhereIf(p => p.Code.Contains(code), string.IsNullOrEmpty(code) == false)
        .WhereIf(p => p.Code.Contains(address), string.IsNullOrEmpty(address) == false);
}

是不是更易读一些！

当然，我们还需要一个 IEnumerable<T> 版本的 WhereIf 扩展，方便对集合进行查询：

public static IEnumerable<T> WhereIf<T>(this IEnumerable<T> source, Func<T, bool> predicate, bool condition)
{
    return condition ? source.Where(predicate) : source;
}

三、WhereIf 完整代码

namespace System.Linq
{
    public static class WhereIfExtension
    {
        public static IQueryable<T> WhereIf<T>(this IQueryable<T> source, Expression<Func<T, bool>> predicate, bool condition)
        {
            return condition ? source.Where(predicate) : source;
        }
        public static IQueryable<T> WhereIf<T>(this IQueryable<T> source, Expression<Func<T, int, bool>> predicate, bool condition)
        {
            return condition ? source.Where(predicate) : source;
        }
        public static IEnumerable<T> WhereIf<T>(this IEnumerable<T> source, Func<T, bool> predicate, bool condition)
        {
            return condition ? source.Where(predicate) : source;
        }
        public static IEnumerable<T> WhereIf<T>(this IEnumerable<T> source, Func<T, int, bool> predicate, bool condition)
        {
            return condition ? source.Where(predicate) : source;
        }
    }
}

将类放入 System.Linq 命名空间中，随时使用，更方便。