代码改变世界

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

2011-02-17 19:24 by 鹤冲天, ... 阅读, ... 评论, 收藏, 编辑

一、Where 扩展的不足

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

PersonQuery

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

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 命名空间中,随时使用,更方便。