Ни для кого не секрет, что хотя LINQ позиционируется как универсальный язык запросов к коллекциям разного происхождения (как коллекциям в памяти, так и различным удаленным источникам данных, например, базам данных), однако на деле результаты одинаковых запросов получаются разными в зависимости от того, к какой коллекции был запрос. В частности, при наличии null в свойстве Property1 используемом в выражении в методе .Select(c=>c.Property1.Property2) можно получить как NullReferenceException, так и null в качестве результата.
Пример:
Предположим у вас есть DbContext, предоставляющий вам доступ к таблице книг. При этом у каждой книги может быть указан автор, представленный в таблице авторов.
Соответственно, представление автора в коде выглядит как-то так:
public class Book { public string Title{get;set;} public string ISBN{get;set;} public Author Author{get;set;} ... } public class Author { public string Name {get;set;} ... }
Предположим, также, что кроме локальной бд в качестве источника данных мы используем так же веб-сервис, который предоставляет нам информацию откуда-то извне. Мы обернули его в адаптер, так что он предоставляет нам коллекцию точно таких же книг с авторами, однако, внутри она представляет собой List.
Для унификации обернем оба источника данных в интерфейс IBookSource:
public interface IBookSource { T GetBookData<T>(string isbn, Expression<Func<Book, T>> selector) where T:class; }
Такая Generic реализация позволяет не забирать из бд всегда целиком сущность Book с ее навигационными свойствами, а получать запросом только нужную информацию. Реализация этого метода для DbContext тривиальна:
public T GetBookData<T>(string isbn, Expression<Func<Book, T>> selector) where T:class { return Set<Book>().Where(b=>b.ISBN == isbn).Select(selector).SingleOrDefault(); }
Что произойдет, если мы вызовем этот метод для книги у которой не указан автор следующим образом:
var authorName = GetBookData(isbn, b=>b.Author.Name);
В переменной authorName окажется null.
Теперь попробуем реализовать этот же метод в том случае, когда источником данных на самом деле является коллекция в памяти (List booklist, который мы получили запросом к сервису поставщику данных):
public T GetBookData<T>(string isbn, Expression<Func<Book, T>> selector) where T:class { List<Book> books = GetDataFromService(); return books.Where(b=>b.ISBN == isbn).Select(selector.Compile()).SingleOrDefault(); }
Обратите внимание, что, поскольку books это IEnumerable а не IQueriable, нам необходимо скомпилировать выражение прежде чем передавать его в метод Select.
Что произойдет, если мы опять вызовем метод для книги у которой не указан автор следующим образом:
var authorName = GetBookData(isbn, b=>b.Author.Name);
Мы получим NullReferenceException. Возникает вопрос, что же делать, мы ведь хотим унифицированно извлекать данные независимо от того что находится за интерфейсом IBookSource. Значит нам нужно свести поведение IQueriable и IEnumerable в вопросе доступа с свойствам класса к одинаковому поведению. Мне в данном случае больше нравится поведение IQueriable, поэтому я преобразую Expression, который выполняется над IEnumerable к такому же поведению.
Для этого нужно чтобы для каждого PropertyGetter в выражении будет проверяюлось не равно ли значение на котором вызывается PropertyGetter null, и если это так — вместо вызова PropertyGetter возвращалось null. Вот метод реализующий это поведение:
public static TResult With<TSource, TResult>(TSource source, Func<TSource, TResult> action) where TSource : class { if (source != default(TSource)) return action(source); return default(TResult); }
Однако, как добавить такое поведение в уже готовое выражение (Expression)?
На помощь приходит класс ExpressionVisitor. Это специальный класс, который позволяет пройти по всем узлам дерева выражений (ExpressionTree) и модифицировать их так как нам нужно.
public class AddMaybeVisitor : ExpressionVisitor { //Этот метод мы будем вызывать над выражением, которое нужно преобразовать. public Expression<Func<T1, T2>> Modify<T1, T2>(Expression<Func<T1, T2>> expression) { return (Expression<Func<T1, T2>>)Visit(expression); } // Этот метод вызывается в том случае, если узлом дерева выражений является обращение к свойству или полю, как раз то, что нам и требуется. protected override Expression VisitMember(MemberExpression node) { Visit(node.Expression); var expressionType = node.Expression.Type; var memberType = node.Type; var withMethodinfo = typeof(AddMaybeVisitor) .GetMethod("With") .MakeGenericMethod(expressionType, memberType); var p = Expression.Parameter(expressionType); var l = Expression.Lambda(Expression.MakeMemberAccess(p, node.Member), p); return Expression.Call(withMethodinfo, node.Expression, Expression.Constant(l.Compile(), typeof(Func<,>).MakeGenericType(expressionType, memberType)) ); } public static TResult With<TSource, TResult>(TSource source, Func<TSource, TResult> action) where TSource : class { if (source != default(TSource)) return action(source); return default(TResult); } }
Как использовать этот ExpressionVisitor? Очень просто:
public T GetBookData<T>(string isbn, Expression<Func<Book, T>> selector) where T:class { List<Book> books = GetDataFromService(); var modifiedSelector = new AddMaybeVisitor().Modify(selector); return books.Where(b=>b.ISBN == isbn).Select(modifiedSelector.Compile()).SingleOrDefault(); }
Теперь если запросить имя автора у книги, для которой не указан автор, мы получим null, а не NullReferenceException, абсолютно точно так же как и при запросе в бд через DbContext.
ссылка на оригинал статьи http://habrahabr.ru/post/256257/
Добавить комментарий