LambdaExtension.cs 4.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141
  1. using System;
  2. using System.Collections.Generic;
  3. using System.Linq;
  4. using System.Linq.Expressions;
  5. using System.Text;
  6. using System.Threading.Tasks;
  7. namespace Common
  8. {
  9. public static class LambdaExtension
  10. {
  11. public static Expression Property(this Expression expression, string propertyName)
  12. {
  13. return Expression.Property(expression, propertyName);
  14. }
  15. /// <summary>
  16. /// 组合AndAlso
  17. /// </summary>
  18. /// <param name="left"></param>
  19. /// <param name="right"></param>
  20. /// <returns></returns>
  21. public static Expression AndAlso(this Expression left, Expression right)
  22. {
  23. return Expression.AndAlso(left, right);
  24. }
  25. public static Expression Call(this Expression instance, string methodName, params Expression[] arguments)
  26. {
  27. return Expression.Call(instance, instance.Type.GetMethod(methodName) ?? throw new InvalidOperationException(), arguments);
  28. }
  29. public static Expression GreaterThan(this Expression left, Expression right)
  30. {
  31. return Expression.GreaterThan(left, right);
  32. }
  33. /// <summary>
  34. /// 转成Lambda表达式
  35. /// </summary>
  36. /// <typeparam name="T"></typeparam>
  37. /// <param name="body"></param>
  38. /// <param name="parameters"></param>
  39. /// <returns></returns>
  40. public static Expression<T> ToLambda<T>(this Expression body, params ParameterExpression[] parameters)
  41. {
  42. return Expression.Lambda<T>(body, parameters);
  43. }
  44. /// <summary>
  45. /// True
  46. /// </summary>
  47. /// <typeparam name="T"></typeparam>
  48. /// <returns></returns>
  49. public static Expression<Func<T, bool>> True<T>() { return param => true; }
  50. /// <summary>
  51. /// False
  52. /// </summary>
  53. /// <typeparam name="T"></typeparam>
  54. /// <returns></returns>
  55. public static Expression<Func<T, bool>> False<T>() { return param => false; }
  56. /// <summary>
  57. /// 组合And
  58. /// </summary>
  59. /// <returns></returns>
  60. public static Expression<Func<T, bool>> And<T>(this Expression<Func<T, bool>> first, Expression<Func<T, bool>> second)
  61. {
  62. return first.Compose(second, Expression.AndAlso);
  63. }
  64. /// <summary>
  65. /// 组合Or
  66. /// </summary>
  67. /// <returns></returns>
  68. public static Expression<Func<T, bool>> Or<T>(this Expression<Func<T, bool>> first, Expression<Func<T, bool>> second)
  69. {
  70. return first.Compose(second, Expression.OrElse);
  71. }
  72. /// <summary>
  73. /// Combines the first expression with the second using the specified merge function.
  74. /// </summary>
  75. private static Expression<T> Compose<T>(this Expression<T> first, Expression<T> second, Func<Expression, Expression, Expression> merge)
  76. {
  77. var map = first.Parameters
  78. .Select((f, i) => new { f, s = second.Parameters[i] })
  79. .ToDictionary(p => p.s, p => p.f);
  80. var secondBody = ParameterRebinder.ReplaceParameters(map, second.Body);
  81. return Expression.Lambda<T>(merge(first.Body, secondBody), first.Parameters);
  82. }
  83. /// <summary>
  84. /// ParameterRebinder
  85. /// </summary>
  86. private class ParameterRebinder : ExpressionVisitor
  87. {
  88. /// <summary>
  89. /// The ParameterExpression map
  90. /// </summary>
  91. private readonly Dictionary<ParameterExpression, ParameterExpression> _map;
  92. /// <summary>
  93. /// Initializes a new instance of the <see cref="ParameterRebinder"/> class.
  94. /// </summary>
  95. /// <param name="map">The map.</param>
  96. private ParameterRebinder(Dictionary<ParameterExpression, ParameterExpression> map)
  97. {
  98. this._map = map ?? new Dictionary<ParameterExpression, ParameterExpression>();
  99. }
  100. /// <summary>
  101. /// Replaces the parameters.
  102. /// </summary>
  103. /// <param name="map">The map.</param>
  104. /// <param name="exp">The exp.</param>
  105. /// <returns>Expression</returns>
  106. public static Expression ReplaceParameters(Dictionary<ParameterExpression, ParameterExpression> map, Expression exp)
  107. {
  108. return new ParameterRebinder(map).Visit(exp);
  109. }
  110. /// <summary>
  111. /// Visits the parameter.
  112. /// </summary>
  113. /// <param name="p">The p.</param>
  114. /// <returns>Expression</returns>
  115. protected override Expression VisitParameter(ParameterExpression p)
  116. {
  117. if (_map.TryGetValue(p, out ParameterExpression replacement))
  118. {
  119. p = replacement;
  120. }
  121. return base.VisitParameter(p);
  122. }
  123. }
  124. }
  125. }