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