2 // QueryableTransformer.cs
5 // Roei Erez (roeie@mainsoft.com)
6 // Jb Evain (jbevain@novell.com)
8 // Copyright (C) 2008 Novell, Inc (http://www.novell.com)
10 // Permission is hereby granted, free of charge, to any person obtaining
11 // a copy of this software and associated documentation files (the
12 // "Software"), to deal in the Software without restriction, including
13 // without limitation the rights to use, copy, modify, merge, publish,
14 // distribute, sublicense, and/or sell copies of the Software, and to
15 // permit persons to whom the Software is furnished to do so, subject to
16 // the following conditions:
18 // The above copyright notice and this permission notice shall be
19 // included in all copies or substantial portions of the Software.
21 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
22 // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
23 // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
24 // NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
25 // LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
26 // OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
27 // WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
31 using System.Collections;
32 using System.Collections.Generic;
33 using System.Collections.ObjectModel;
35 using System.Linq.Expressions;
36 using System.Reflection;
37 using System.Runtime.CompilerServices;
39 namespace System.Linq {
41 class QueryableTransformer : ExpressionTransformer {
43 protected override Expression VisitMethodCall (MethodCallExpression methodCall)
45 if (IsQueryableExtension (methodCall.Method))
46 return ReplaceQueryableMethod (methodCall);
48 return base.VisitMethodCall (methodCall);
51 protected override Expression VisitLambda (LambdaExpression lambda)
56 protected override Expression VisitConstant (ConstantExpression constant)
58 var qe = constant.Value as IQueryableEnumerable;
62 var enumerable = qe.GetEnumerable ();
63 if (enumerable != null)
64 return Expression.Constant (enumerable);
66 return Visit (qe.Expression);
69 static bool IsQueryableExtension (MethodInfo method)
71 return HasExtensionAttribute (method) &&
72 method.GetParameters () [0].ParameterType.IsAssignableTo (typeof (IQueryable));
75 static bool HasExtensionAttribute (MethodInfo method)
77 return method.GetCustomAttributes (typeof (ExtensionAttribute), false).Length > 0;
80 MethodCallExpression ReplaceQueryableMethod (MethodCallExpression old)
82 Expression target = null;
83 if (old.Object != null)
84 target = Visit (old.Object);
86 var method = ReplaceQueryableMethod (old.Method);
87 var parameters = method.GetParameters ();
88 var arguments = new Expression [old.Arguments.Count];
90 for (int i = 0; i < arguments.Length; i++) {
91 arguments [i] = UnquoteIfNeeded (
92 Visit (old.Arguments [i]),
93 parameters [i].ParameterType);
96 return Expression.Call (target, method, arguments);
99 static Expression UnquoteIfNeeded (Expression expression, Type delegateType)
101 if (expression.NodeType != ExpressionType.Quote)
104 var lambda = (LambdaExpression) ((UnaryExpression) expression).Operand;
105 if (lambda.Type == delegateType)
111 static Type GetTargetDeclaringType (MethodInfo method)
113 return method.DeclaringType == typeof (Queryable) ? typeof (Enumerable) : method.DeclaringType;
116 static MethodInfo ReplaceQueryableMethod (MethodInfo method)
118 var target_type = GetTargetDeclaringType (method);
119 var result = GetMatchingMethod (method, target_type);
124 throw new InvalidOperationException (
126 "There is no method {0} on type {1} that matches the specified arguments",
128 target_type.FullName));
131 static MethodInfo GetMatchingMethod (MethodInfo method, Type declaring)
133 foreach (var candidate in declaring.GetMethods ()) {
134 if (!MethodMatch (candidate, method))
137 if (method.IsGenericMethod)
138 return candidate.MakeGenericMethodFrom (method);
146 static bool MethodMatch (MethodInfo candidate, MethodInfo method)
148 if (candidate.Name != method.Name)
151 if (!HasExtensionAttribute (candidate))
154 var parameters = method.GetParameterTypes ();
156 if (parameters.Length != candidate.GetParameters ().Length)
159 if (method.IsGenericMethod) {
160 if (!candidate.IsGenericMethod)
163 if (candidate.GetGenericArguments ().Length != method.GetGenericArguments ().Length)
166 candidate = candidate.MakeGenericMethodFrom (method);
169 if (!TypeMatch (candidate.ReturnType, method.ReturnType))
172 var candidate_parameters = candidate.GetParameterTypes ();
174 if (candidate_parameters [0] != GetComparableType (parameters [0]))
177 for (int i = 1; i < candidate_parameters.Length; ++i)
178 if (!TypeMatch (candidate_parameters [i], parameters [i]))
184 static bool TypeMatch (Type candidate, Type type)
186 if (candidate == type)
189 return candidate == GetComparableType (type);
192 static Type GetComparableType (Type type)
194 if (type.IsGenericInstanceOf (typeof (IQueryable<>)))
195 type = typeof (IEnumerable<>).MakeGenericTypeFrom (type);
196 else if (type.IsGenericInstanceOf (typeof (IOrderedQueryable<>)))
197 type = typeof (IOrderedEnumerable<>).MakeGenericTypeFrom (type);
198 else if (type.IsGenericInstanceOf (typeof (Expression<>)))
199 type = type.GetFirstGenericArgument ();
200 else if (type == typeof (IQueryable))
201 type = typeof (IEnumerable);