.net lambda表示式合併

2022-10-21 15:01:07

事情的起因是公司一個小夥子問了我個問題 「海哥,來幫我看下這段程式碼怎麼不行」

Func<Report,bool> nameFilter = x=>x.Name == "test";
DbContext.Report.Where(x=>x.State==1 && nameFilter(x));

我一看,好傢伙,這麼騷的程式碼都能讓你想出來,正常情況下用Linq To Object是可以這麼操作的,但是EF的IQueryable查詢是不能這麼操作的。
Linq To Object是直接執行表示式,他就是個委託方法,裡面巢狀多少層委託和方法都是能直接執行的
IQueryable並不會執行表示式和方法,是把表示式轉換為對應的Sql語句來執行,解析到nameFilter的時候他就懵逼了,這是啥玩意兒啊,sql裡面沒有這種東西啊,他就轉換不了了。

小夥子知道後明細很失望,那不能啊,也不是我想顯擺我的技術,就是想讓小夥子能繼續他的騷操作,給他來點海克斯科技與狠活。

解決方案:

//表示式
Func<Report,bool> nameFilter = x=>x.Name == "test";
Func<Report,bool> stateFilter = x=>x.State==1;
//合併為
Func<Report,bool> whereFilter = x=>x.Name == "test" && x.State==1;

//呼叫
DbContext.Report.Where(whereFilter);

完美解決

那怎麼合併,當然得自己構造一個新的表示式,構造表示式需要用到Expression類,如果沒有用過這個類,可以按照下面的方式來偵錯看看一個表示式轉換為表示式樹是怎麼樣的。


TestExpression(x=>x.Name == "test",x=>x.State==1);

public static void TestExpression(Expression<Func<Report, bool>> left,Expression<Func<Report, bool>> right)
{ 
    //偵錯檢視expression物件
    var bodyLeft = left.Body;//這個就是x.Name == "test"
    var bodyRight = right.Body;//這個就是x.State==1
}

好,這裡我們能獲取到表示式的Body,然後使用Expression類能很好的合併兩個表示式的body

var andAlso = Expression.AndAlso(bodyLeft ,bodyRight);//x.Name == "test" && x.State==1

這樣還不行,這兩個表示式是兩個不同的委託物件,他們的引數x也是兩個不同的物件,合併了又沒完全合併

這就需要用到ExpressionVisitor類來遞迴表示式樹,把兩個表示式的引數替換為同一個引數。

    /// <summary>
    /// 替換表示式引數
    /// </summary>
    public class ReplaceExpressionVisitor : ExpressionVisitor
    {
        private Expression _leftParameter;

        public ReplaceExpressionVisitor(Expression leftParameter)
        {
            _leftParameter= leftParameter;
        }
        
        protected override Expression VisitParameter(ParameterExpression node)
        {
            return _leftParameter;
        }
    }

最終


TestExpression(x=>x.Name == "test",x=>x.State==1);

public static void TestExpression(Expression<Func<Report, bool>> left,Expression<Func<Report, bool>> right)
{ 
    //偵錯檢視expression物件
    var bodyLeft = left.Body;//這個就是x.Name == "test"
    var bodyRight = right.Body;//這個就是x.State==1
    var leftParameter = left.Parameters[0];
    //表示式遞迴存取
    var visitor =new ReplaceExpressionVisitor(leftParameter);
    //替換引數
    bodyRight = visitor.Visit(bodyRight);
    //合併表示式
    var expression = Expression.AndAlso(bodyLeft , bodyRight);
    //構建表示式
    var whereExpression= Expression.Lambda<Func<Report, bool>>(expression , left.Parameters);
    //編譯表示式
    var whereFilter = whereExpression.Compile();
    //使用
    DbContext.Report.Where(whereFilter);
}

正想給小老弟顯擺一下的時候,他又去寫其他騷程式碼了

騷不過騷不過,完善一下列子,下面是完整的程式碼

小嫩手不想動的小夥伴可以直接nuget上查詢DynamicExpression.Core,直接使用

更多原始碼看本人github

    /// <summary>
    /// 替換表示式引數
    /// </summary>
    public class ReplaceExpressionVisitor : ExpressionVisitor
    {
        private Dictionary<Expression, Expression> _parameters;

        public ReplaceExpressionVisitor(Dictionary<Expression,Expression> parameters)
        {
            _parameters = parameters;
        }

        protected override Expression VisitParameter(ParameterExpression node)
        {
            if (_parameters.TryGetValue(node, out Expression _newValue))
            {
                return _newValue;
            }
            return base.Visit(node);
        }
    }
    /// <summary>
    /// 表示式擴充套件
    /// </summary>
    public static class ExpressionExtension
    {

        /// <summary>
        /// 使用AndAlso合併表示式
        /// </summary>
        /// <param name="exprs"></param>
        /// <returns></returns>
        public static Expression<T> AndAlso<T>(this IList<Expression<T>> exprs)
        {
            if (exprs.Count == 0) return null;
            if (exprs.Count == 1) return exprs[0];

            var leftExpr = exprs[0];
            var left = leftExpr.Body;
            for (int i = 1; i < exprs.Count; i++)
            {
                var expr = exprs[i];
                var visitor = GetReplaceExpressionVisitor(expr.Parameters, leftExpr.Parameters);
                var right = visitor.Visit(expr.Body);
                left = Expression.AndAlso(left, right);
            }
            return Expression.Lambda<T>(left, leftExpr.Parameters);
        }

        /// <summary>
        /// 使用AndAlso合併表示式
        /// </summary>
        /// <typeparam name="T"></typeparam>
        /// <param name="left"></param>
        /// <param name="right"></param>
        /// <returns>left AndAlso right</returns>
        public static Expression<T> AndAlso<T>(this Expression<T> left, Expression<T> right)
        {
            return AndAlso(new List<Expression<T>>() { left, right });
        }

        /// <summary>
        /// 使用OrElse合併表示式
        /// </summary>
        /// <typeparam name="T"></typeparam>
        /// <param name="exprs"></param>
        /// <returns></returns>
        public static Expression<T> OrElse<T>(this IList<Expression<T>> exprs)
        {
            if (exprs.Count == 0) return null;
            if (exprs.Count == 1) return exprs[0];

            var leftExpr = exprs[0];
            var left = leftExpr.Body;
            for (int i = 1; i < exprs.Count; i++)
            {
                var expr = exprs[i];
                var visitor = GetReplaceExpressionVisitor(expr.Parameters, leftExpr.Parameters);
                var right = visitor.Visit(expr.Body);
                left = Expression.OrElse(left, right);
            }
            return Expression.Lambda<T>(left, leftExpr.Parameters);
        }

        /// <summary>
        /// 使用OrElse合併表示式
        /// </summary>
        /// <typeparam name="T"></typeparam>
        /// <param name="left"></param>
        /// <param name="right"></param>
        /// <returns>left OrElse right</returns>
        public static Expression<T> OrElse<T>(this Expression<T> left, Expression<T> right)
        {
            return OrElse(new List<Expression<T>>() { left, right });
        }


        /// <summary>
        /// 構建visitor
        /// </summary>
        /// <param name="oldParameters"></param>
        /// <param name="newParameters"></param>
        /// <returns></returns>
        private static ReplaceExpressionVisitor GetReplaceExpressionVisitor(ReadOnlyCollection<ParameterExpression> oldParameters, ReadOnlyCollection<ParameterExpression> newParameters)
        {
            Dictionary<Expression, Expression> dic = new Dictionary<Expression, Expression>();
            for (int i = 0; i < oldParameters.Count; i++)
            {
                dic.Add(oldParameters[i],newParameters[i]);
            }
            return new ReplaceExpressionVisitor(dic);
        }
    }

使用

string connectString = "Data Source=.;Initial Catalog=RportTest;Integrated Security=True";
var optionsBuilder = new DbContextOptionsBuilder<TestContext>();
optionsBuilder.UseSqlServer(connectString);
using (TestContext ctx = new TestContext(optionsBuilder.Options))
{

    Expression<Func<ReportData, bool>> epxr1 = report => report.ID == 2023;
    Expression<Func<ReportData, bool>> epxr2 = report => report.Name == "test1";

    var epxr3 = new List<Expression<Func<ReportData, bool>>>() { epxr1, epxr2 };

    var andPredicate = epxr3.AndAlso();
    var andQuery = ctx.ReportData.Where(andPredicate);
    string andSql = andQuery.ToQueryString();
    var andResult = andQuery.ToList();

    var orPredicate = epxr3.OrElse();
    var orQuery = ctx.ReportData.Where(orPredicate);
    string orSql = orQuery.ToQueryString();
    var orResult = orQuery.ToList();
}