实体框架日期组合查询



我有一个Apartments列表,每个Apartment都有它的ApartmentRooms,每个ApartmentRoom都有它的DateBatches。每个DateBatch都有代表占用日期的Dates列表(每个记录一个记录)。

Apartment 1
 Room 1
    DateBatch
      1.5.2015
      2.5.2015
    DateBatch 2
      8.5.2015
      9.5.2015
 Room 2
    DateBatch
      5.5.2015
      6.5.2015

由此,您可以看到这间公寓有2个房间,其中房间1在以下日期1.5,2.5,8.5和9.5入住,房间2在5.5和6.5入住。

用户可以输入所需的 N 天和他想要停留的连续天数的 X 天。

因此,例如,用户输入从 1.5 到 15.5 的时间段,他想睡 10 晚,我需要列出所有公寓,其中至少有一间公寓房间可用于任何可能的日期组合,在这种情况下将如下所示:

1.5-10.5
2.5-11.5
3.5-12.5
4.5-13.5
5.5-14.5

到目前为止,我已经尝试过这个,它仅适用于第一次 foreach 迭代,因为 foreach 将查询与 AND 条件连接起来,而不是 OR 标准,我认为这是一种非常糟糕的方法。

 public static IQueryable<Apartment> QueryByPeriod(this IQueryable<Apartment> apartments, DateTime PeriodStart, DateTime PeriodEnd, int StayDuration)
 {
        var possibleDateRanges = new List<List<DateTime>>();
        //set all possible start dates for the desired period
        for (var i = PeriodStart; i <= PeriodEnd.AddDays(-StayDuration); i = i.AddDays(1))
        {
            List<DateTime> dates = new List<DateTime>();
            foreach(var date in i.DateRange(i.AddDays(StayDuration-1)))
            {
                dates.Add(date);
            }
            possibleDateRanges.Add(dates);
        }
        //filter by date range
        //select apartment rooms where one of possible combinations is suitable for selected period
        foreach (var possibleDates in possibleDateRanges)
        {
            apartments = apartments.Where(m => m.ApartmentRooms
            .Any(g => g.OccupiedDatesBatches.Select(ob => ob.OccupiedDates).Any(od => od.Any(f => possibleDates.Contains(f.Date)))
            ) == false);
        }
        return apartments;
    }

有什么建议吗?

若要将多个条件与 OR 组合,可以使用(例如)LinqKit 库。通过 nuget 安装它,添加using LinqKit;然后:

    public static IQueryable<Apartment> QueryByPeriod(this IQueryable<Apartment> apartments, DateTime PeriodStart, DateTime PeriodEnd, int StayDuration) {
        var possibleDateRanges = new List<Tuple<DateTime, DateTime>>();
        // list all ranges, so for your example that would be:
        //1.5-10.5
        //2.5-11.5
        //3.5-12.5
        //4.5-13.5
        //5.5-14.5            
        var startDate = PeriodStart;
        while (startDate.AddDays(StayDuration - 1) < PeriodEnd) {
            possibleDateRanges.Add(new Tuple<DateTime, DateTime>(startDate, startDate.AddDays(StayDuration - 1)));
            startDate = startDate.AddDays(1);
        }
        Expression<Func<Apartment, bool>> condition = null;
        foreach (var range in possibleDateRanges) {                
            Expression<Func<Apartment, bool>> rangeCondition = m => m.ApartmentRooms       
                // find rooms where ALL occupied dates are outside target interval             
                .Any(g => g.OccupiedDatesBatches.SelectMany(ob => ob.OccupiedDates).All(f => f.Date < range.Item1 || f.Date > range.Item2)
                );
            // concatenate with OR if necessary
            if (condition == null)
                condition = rangeCondition;
            else
                condition = condition.Or(rangeCondition);
        }
        if (condition == null)
            return apartments;
        // note AsExpandable here
        return apartments.AsExpandable().Where(condition);
    }

请注意,我还修改了您的逻辑。当然,此逻辑非常适合单元测试,如果您正在处理一个严肃的项目 - 您应该针对不同的条件使用内存中 EF 提供程序(或模拟)对其进行明确测试。

这是一个纯(不需要外部包)LINQ to Entities 解决方案。

首先确定可能的开始日期列表:

var startDates = Enumerable.Range(0, PeriodEnd.Subtract(PeriodStart).Days - StayDuration + 1)
    .Select(offset => PeriodStart.AddDays(offset))
    .ToList();

然后使用以下查询:

var availableApartments = apartments.Where(a => a.ApartmentRooms.Any(ar =>
    startDates.Any(startDate => !ar.OccupiedDatesBatches.Any(odb => 
        odb.OccupiedDates.Any(od => 
            od.Date >= startDate && od.Date < DbFunctions.AddDays(startDate, StayDuration))))));

此解决方案的好处是可以轻松扩展。上面的查询返回可用的公寓,但不提供哪个房间可用以及何时可用的信息 - 您可能需要向用户提供这些信息。使用上述方法,您可以获得如下信息:

public class AvailableApartmentInfo
{
    public Apartment Apartment { get; set; }
    public Room Room { get; set; }
    public DateTime StartDate { get; set; }
}
var availableApartmentInfo =
    from a in apartments
    from ar in a.ApartmentRooms
    from startDate in startDates
    where !ar.OccupiedDatesBatches.Any(odb => 
        odb.OccupiedDates.Any(od => 
            od.Date >= startDate && od.Date < DbFunctions.AddDays(startDate, StayDuration)))
    select new AvailableApartmentInfo { Apartment = a, Room = ar, StartDate = startDate };

最新更新