AsyncQueryableEx.Generated.tt 8.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287
  1. // Licensed to the .NET Foundation under one or more agreements.
  2. // The .NET Foundation licenses this file to you under the Apache 2.0 License.
  3. // See the LICENSE file in the project root for more information.
  4. <#@ template debug="false" hostspecific="false" language="C#" #>
  5. <#@ assembly name="System.Core" #>
  6. <#@ assembly name="System.Runtime" #>
  7. <#@ assembly name="$(ProjectDir)\..\System.Linq.Async\bin\$(Configuration)\net46\System.Threading.Tasks.Extensions.dll" #>
  8. <#@ assembly name="$(ProjectDir)\..\System.Linq.Async\bin\$(Configuration)\net46\System.Linq.Async.dll" #>
  9. <#@ assembly name="$(ProjectDir)\..\System.Interactive.Async\bin\$(Configuration)\net46\System.Interactive.Async.dll" #>
  10. <#@ import namespace="System.Linq" #>
  11. <#@ import namespace="System.Text" #>
  12. <#@ import namespace="System.Threading" #>
  13. <#@ import namespace="System.Threading.Tasks" #>
  14. <#@ import namespace="System.Collections.Generic" #>
  15. <#@ output extension=".cs" #>
  16. <#
  17. var nullableParameterNames = new[] { "comparer" };
  18. var exclude = new[] { "ForEach", "ForEachAsync", "ToEnumerable", "ToAsyncEnumerable", "ToObservable", "AsAsyncEnumerable" };
  19. var toQuotedImpl = default(Func<Type, int, bool, string>);
  20. toQuotedImpl = (t, i, b) =>
  21. {
  22. var name = t.Name;
  23. if (t.IsGenericType)
  24. {
  25. var genDef = t.GetGenericTypeDefinition();
  26. name = genDef.Name.Substring(0, genDef.Name.LastIndexOf('`'));
  27. var genArgs = "<" + string.Join(", ", t.GetGenericArguments().Select(a => toQuotedImpl(a, i, false))) + ">";
  28. if (b)
  29. {
  30. if (name == "Func" || name == "Action")
  31. {
  32. name = "Expression<" + name + genArgs + ">";
  33. }
  34. else if (name == "IAsyncEnumerable" && i == 0)
  35. {
  36. name = "IAsyncQueryable" + genArgs;
  37. }
  38. else if (name == "IOrderedAsyncEnumerable" && i == 0)
  39. {
  40. name = "IOrderedAsyncQueryable" + genArgs;
  41. }
  42. else
  43. {
  44. name += genArgs;
  45. }
  46. }
  47. else
  48. {
  49. if (name == "Nullable")
  50. {
  51. name = genArgs.Substring(1, genArgs.Length - 2) + "?";
  52. }
  53. else
  54. {
  55. name += genArgs;
  56. }
  57. }
  58. }
  59. else if (t.IsArray)
  60. {
  61. var elem = toQuotedImpl(t.GetElementType(), i, b);
  62. name = elem + "[]";
  63. }
  64. else
  65. {
  66. if (t == typeof(int))
  67. {
  68. name = "int";
  69. }
  70. else if (t == typeof(long))
  71. {
  72. name = "long";
  73. }
  74. else if (t == typeof(float))
  75. {
  76. name = "float";
  77. }
  78. else if (t == typeof(double))
  79. {
  80. name = "double";
  81. }
  82. else if (t == typeof(decimal))
  83. {
  84. name = "decimal";
  85. }
  86. else if (t == typeof(bool))
  87. {
  88. name = "bool";
  89. }
  90. else if (t == typeof(object))
  91. {
  92. name = "object";
  93. }
  94. }
  95. return name;
  96. };
  97. var toQuoted = new Func<Type, int, string>((t, i) => toQuotedImpl(t, i, true));
  98. #>
  99. using System.Collections.Generic;
  100. using System.Linq.Expressions;
  101. using System.Reflection;
  102. using System.Threading;
  103. using System.Threading.Tasks;
  104. namespace System.Linq
  105. {
  106. public static partial class AsyncQueryableEx
  107. {
  108. <#
  109. // NOTE: Just including extension methods
  110. foreach (var m in typeof(AsyncEnumerableEx).GetMethods()
  111. .Where(m => m.IsStatic)
  112. .Where(m => !exclude.Contains(m.Name))
  113. .Where(m => m.IsDefined(typeof(System.Runtime.CompilerServices.ExtensionAttribute), true))
  114. .Where(m =>
  115. {
  116. var p0 = m.GetParameters()[0].ParameterType;
  117. if (p0.IsGenericType)
  118. {
  119. var p0d = p0.GetGenericTypeDefinition();
  120. return p0d == typeof(IAsyncEnumerable<>) || p0d == typeof(IOrderedAsyncEnumerable<>);
  121. }
  122. return false;
  123. })
  124. .OrderBy(m => m.Name)
  125. .ThenBy(m => m.IsGenericMethod ? m.GetGenericArguments().Length : 0)
  126. .ThenBy(m => m.GetParameters().Length)
  127. .ThenBy(m => string.Join(", ", m.GetParameters().Select((p, i) => toQuoted(p.ParameterType, i) + " " + p.Name))))
  128. {
  129. var genArgs = m.GetGenericArguments();
  130. var ret = toQuoted(m.ReturnType, 0);
  131. var name = m.Name;
  132. if (genArgs.Length > 0)
  133. {
  134. name += "<" + string.Join(", ", genArgs.Select(a => a.Name)) + ">";
  135. }
  136. var isParams = false;
  137. var parCount = m.GetParameters().Length;
  138. if (parCount != 0)
  139. {
  140. if (m.GetParameters().Last().IsDefined(typeof(ParamArrayAttribute), true))
  141. {
  142. isParams = true;
  143. }
  144. }
  145. var pars = string.Join(", ", m.GetParameters().Select((p, i) => (i == parCount - 1 && isParams ? "params " : "") + toQuoted(p.ParameterType, i) + " " + p.Name));
  146. var quotedPars = string.Join(", ", m.GetParameters().Select((p, i) => "default(" + toQuoted(p.ParameterType, i) + ")"));
  147. if (m.IsDefined(typeof(System.Runtime.CompilerServices.ExtensionAttribute), true))
  148. {
  149. pars = "this " + pars;
  150. }
  151. var mtd = "(MethodInfo)MethodBase.GetCurrentMethod()";
  152. if (m.IsGenericMethod)
  153. {
  154. mtd = "(" + mtd + ").MakeGenericMethod(" + string.Join(", ", genArgs.Select(a => "typeof(" + a.Name + ")")) + ")";
  155. }
  156. var provider = m.GetParameters()[0].Name + ".Provider";
  157. var factory = "";
  158. var rem = "";
  159. var cast = "";
  160. var quotedArgs = new List<string>();
  161. if (m.ReturnType.IsGenericType)
  162. {
  163. var td = m.ReturnType.GetGenericTypeDefinition();
  164. if (td == typeof(Task<>))
  165. {
  166. factory = "ExecuteAsync<" + toQuotedImpl(m.ReturnType.GetGenericArguments()[0], -1, false) + ">";
  167. var last = m.GetParameters().Last();
  168. if (last.ParameterType == typeof(CancellationToken))
  169. {
  170. rem = ", " + last.Name;
  171. }
  172. else
  173. {
  174. rem = ", CancellationToken.None";
  175. }
  176. }
  177. else if (td == typeof(IAsyncEnumerable<>) || td == typeof(IOrderedAsyncEnumerable<>))
  178. {
  179. factory = "CreateQuery<" + toQuotedImpl(m.ReturnType.GetGenericArguments()[0], -1, false) + ">";
  180. if (td == typeof(IOrderedAsyncEnumerable<>))
  181. {
  182. cast = "(" + toQuoted(m.ReturnType, 0) + ")";
  183. }
  184. }
  185. }
  186. var n = 0;
  187. foreach (var p in m.GetParameters())
  188. {
  189. var pt = p.ParameterType;
  190. var add = false;
  191. if (pt.IsGenericType)
  192. {
  193. var ptd = pt.GetGenericTypeDefinition();
  194. if (ptd == typeof(IAsyncEnumerable<>) || ptd == typeof(IOrderedAsyncEnumerable<>))
  195. {
  196. if (n == 0)
  197. {
  198. quotedArgs.Add(p.Name + ".Expression");
  199. }
  200. else
  201. {
  202. quotedArgs.Add("GetSourceExpression(" + p.Name + ")");
  203. }
  204. add = true;
  205. }
  206. else if (ptd.Name.StartsWith("Func") || ptd.Name.StartsWith("Action"))
  207. {
  208. quotedArgs.Add(p.Name);
  209. add = true;
  210. }
  211. }
  212. if (!add)
  213. {
  214. quotedArgs.Add("Expression.Constant(" + p.Name + ", typeof(" + toQuoted(pt, -1) + "))");
  215. }
  216. n++;
  217. }
  218. var expr = "Expression.Call(" + mtd + ", " + string.Join(", ", quotedArgs) + ")";
  219. var infoofmtd = "InfoOf(() => AsyncQueryableEx." + name + "(" + quotedPars + "))";
  220. var infoofexpr = "Expression.Call(" + infoofmtd + ", " + string.Join(", ", quotedArgs) + ")";
  221. #>
  222. public static <#=ret#> <#=name#>(<#=pars#>)
  223. {
  224. <#
  225. var any = false;
  226. foreach (var p in m.GetParameters())
  227. {
  228. if (!p.ParameterType.IsValueType && !p.ParameterType.IsGenericParameter && !nullableParameterNames.Contains(p.Name))
  229. {
  230. any = true;
  231. #>
  232. if (<#=p.Name#> == null)
  233. throw new ArgumentNullException(nameof(<#=p.Name#>));
  234. <#
  235. }
  236. }
  237. #>
  238. <#
  239. if (any)
  240. {
  241. #>
  242. <#
  243. }
  244. #>
  245. #if CRIPPLED_REFLECTION
  246. return <#=cast#><#=provider#>.<#=factory#>(<#=infoofexpr#><#=rem#>);
  247. #else
  248. return <#=cast#><#=provider#>.<#=factory#>(<#=expr#><#=rem#>);
  249. #endif
  250. }
  251. <#
  252. }
  253. #>
  254. }
  255. }