Mercurial > pub > bltoolkit
diff Source/Data/Linq/Builder/FirstSingleBuilder.cs @ 0:f990fcb411a9
Копия текущей версии из github
author | cin |
---|---|
date | Thu, 27 Mar 2014 21:46:09 +0400 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/Source/Data/Linq/Builder/FirstSingleBuilder.cs Thu Mar 27 21:46:09 2014 +0400 @@ -0,0 +1,172 @@ +using System; +using System.Linq; +using System.Linq.Expressions; + +namespace BLToolkit.Data.Linq.Builder +{ + using BLToolkit.Linq; + using Data.Sql; + using Reflection; + + class FirstSingleBuilder : MethodCallBuilder + { + public static string[] MethodNames = new[] { "First", "FirstOrDefault", "Single", "SingleOrDefault" }; + + protected override bool CanBuildMethodCall(ExpressionBuilder builder, MethodCallExpression methodCall, BuildInfo buildInfo) + { + return + methodCall.IsQueryable(MethodNames) && + methodCall.Arguments.Count == 1; + } + + protected override IBuildContext BuildMethodCall(ExpressionBuilder builder, MethodCallExpression methodCall, BuildInfo buildInfo) + { + var sequence = builder.BuildSequence(new BuildInfo(buildInfo, methodCall.Arguments[0])); + var take = 0; + + if (!buildInfo.IsSubQuery || builder.SqlProvider.IsSubQueryTakeSupported) + switch (methodCall.Method.Name) + { + case "First" : + case "FirstOrDefault" : + take = 1; + break; + + case "Single" : + case "SingleOrDefault" : + if (!buildInfo.IsSubQuery) + take = 2; + break; + } + + if (take != 0) + builder.BuildTake(sequence, new SqlValue(take)); + + return new FirstSingleContext(buildInfo.Parent, sequence, methodCall); + } + + protected override SequenceConvertInfo Convert( + ExpressionBuilder builder, MethodCallExpression methodCall, BuildInfo buildInfo, ParameterExpression param) + { + if (methodCall.Arguments.Count == 2) + { + var predicate = (LambdaExpression)methodCall.Arguments[1].Unwrap(); + var info = builder.ConvertSequence(new BuildInfo(buildInfo, methodCall.Arguments[0]), predicate.Parameters[0]); + + if (info != null) + { + info.Expression = methodCall.Convert(ex => ConvertMethod(methodCall, 0, info, predicate.Parameters[0], ex)); + info.Parameter = param; + + return info; + } + } + else + { + var info = builder.ConvertSequence(new BuildInfo(buildInfo, methodCall.Arguments[0]), null); + + if (info != null) + { + info.Expression = methodCall.Convert(ex => ConvertMethod(methodCall, 0, info, null, ex)); + info.Parameter = param; + + return info; + } + } + + return null; + } + + public class FirstSingleContext : SequenceContextBase + { + public FirstSingleContext(IBuildContext parent, IBuildContext sequence, MethodCallExpression methodCall) + : base(parent, sequence, null) + { + _methodCall = methodCall; + } + + readonly MethodCallExpression _methodCall; + + public override void BuildQuery<T>(Query<T> query, ParameterExpression queryParameter) + { + Sequence.BuildQuery(query, queryParameter); + + switch (_methodCall.Method.Name) + { + case "First" : query.GetElement = (ctx, db, expr, ps) => query.GetIEnumerable(ctx, db, expr, ps).First(); break; + case "FirstOrDefault" : query.GetElement = (ctx, db, expr, ps) => query.GetIEnumerable(ctx, db, expr, ps).FirstOrDefault(); break; + case "Single" : query.GetElement = (ctx, db, expr, ps) => query.GetIEnumerable(ctx, db, expr, ps).Single(); break; + case "SingleOrDefault" : query.GetElement = (ctx, db, expr, ps) => query.GetIEnumerable(ctx, db, expr, ps).SingleOrDefault(); break; + } + } + + static object SequenceException() + { + return new object[0].First(); + } + + public override Expression BuildExpression(Expression expression, int level) + { + if (expression == null) + { + if (Builder.SqlProvider.IsApplyJoinSupported && Parent.SqlQuery.GroupBy.IsEmpty) + { + var join = SqlQuery.OuterApply(SqlQuery); + + Parent.SqlQuery.From.Tables[0].Joins.Add(join.JoinedTable); + + var expr = Sequence.BuildExpression(expression, level); + var idx = SqlQuery.Select.Add(new SqlValue(1)); + + idx = ConvertToParentIndex(idx, this); + + var defaultValue = _methodCall.Method.Name.EndsWith("OrDefault") ? + Expression.Constant(TypeHelper.GetDefaultValue(expr.Type), expr.Type) as Expression : + Expression.Convert( + Expression.Call( + null, + ReflectionHelper.Expressor<object>.MethodExpressor(_ => SequenceException())), + expr.Type); + + expr = Expression.Condition( + Expression.Call( + ExpressionBuilder.DataReaderParam, + ReflectionHelper.DataReader.IsDBNull, + Expression.Constant(idx)), + defaultValue, + expr); + + return expr; + } + + if (Sequence.IsExpression(null, level, RequestFor.Object).Result) + return Builder.BuildMultipleQuery(Parent, _methodCall); + + return Builder.BuildSql(_methodCall.Type, Parent.SqlQuery.Select.Add(SqlQuery)); + } + + throw new InvalidOperationException(); + } + + public override SqlInfo[] ConvertToSql(Expression expression, int level, ConvertFlags flags) + { + return Sequence.ConvertToSql(expression, level + 1, flags); + } + + public override SqlInfo[] ConvertToIndex(Expression expression, int level, ConvertFlags flags) + { + return Sequence.ConvertToIndex(expression, level, flags); + } + + public override IsExpressionResult IsExpression(Expression expression, int level, RequestFor requestFlag) + { + return Sequence.IsExpression(expression, level, requestFlag); + } + + public override IBuildContext GetContext(Expression expression, int level, BuildInfo buildInfo) + { + throw new InvalidOperationException(); + } + } + } +}