Entity Framework中复用Railcar数量计算表达式遇DataReader错误的解决方法

问题根源 #

你当前的实现使用Func<Railcar, double>，这会触发客户端求值：EF会先将Railcar数据加载到内存，再对每个内存中的实例调用委托。而委托内的dbContext.Transfers查询会为每个Railcar发起新的数据库请求，导致同一数据库连接上同时打开多个DataReader，触发异常。同时这种方式会产生N+1查询，性能极差。

解决方案：使用表达式树（Expression<Func<...>>） #

将计算逻辑定义为表达式树，而非普通委托，这样EF能将整个逻辑翻译成SQL，在数据库端一次性完成计算，避免客户端多次查询和DataReader冲突。

方案1：定义可复用的表达式树 #

public static class QueryHelper
{
    // 返回表达式树，EF可将其翻译成SQL
    public static Expression<Func<Railcar, double>> GetRailcarQuantity()
    {
        return r => r.InboundQuantity
            // 转成可空类型处理无匹配记录的情况，避免Sum返回null
            - DbContext.Transfers
                .Where(t => t.FromId == r.Id && t.FromType == TransferType.Railcar)
                .Sum(t => (double?)t.Quantity) ?? 0
            + DbContext.Transfers
                .Where(t => t.ToId == r.Id && t.ToType == TransferType.Railcar)
                .Sum(t => (double?)t.Quantity) ?? 0;
    }
}

使用方式：

var quantityExpr = QueryHelper.GetRailcarQuantity();

var railcars = dbContext.Railcars
    .Where(r => r.FacilityId == UserContext.FacilityId)
    .Select(r => new
    {
        r.RailcarNumber,
        Quantity = quantityExpr.Invoke(r)
    })
    .ToList();

EF Core会自动将Invoke的表达式展开，翻译成完整的SQL语句，在数据库端完成所有计算。

方案2：编写IQueryable扩展方法（更简洁） #

如果需要频繁复用包含该计算的投影，可以直接写扩展方法：

public static class RailcarQueryExtensions
{
    public static IQueryable<(string RailcarNumber, double Quantity)> WithQuantity(this IQueryable<Railcar> railcars)
    {
        return railcars.Select(r => new
        {
            r.RailcarNumber,
            Quantity = r.InboundQuantity
                - DbContext.Transfers
                    .Where(t => t.FromId == r.Id && t.FromType == TransferType.Railcar)
                    .Sum(t => (double?)t.Quantity) ?? 0
                + DbContext.Transfers
                    .Where(t => t.ToId == r.Id && t.ToType == TransferType.Railcar)
                    .Sum(t => (double?)t.Quantity) ?? 0
        })
        .Select(x => (x.RailcarNumber, x.Quantity));
    }
}

使用方式：

var railcars = dbContext.Railcars
    .Where(r => r.FacilityId == UserContext.FacilityId)
    .WithQuantity()
    .ToList();

额外优化建议 #

• 为Transfers表的FromId+FromType、ToId+ToType字段创建联合索引，提升Sum查询的性能。
• 用(double?)t.Quantity ?? 0处理无匹配转移记录的情况，避免Sum返回null导致的计算错误。

内容的提问来源于stack exchange，提问作者Jonathan Wood