Я думаю, вам нужно использовать рекурсивное общее табличное выражение .
РЕДАКТИРОВАТЬ: я также добавил ниже процедурную реализацию вместо использования рекурсивного общего табличного выражения. Я рекомендую использовать этот процедурный подход, поскольку я думаю, что может быть несколько сценариев данных, которые рекурсивный запрос CTE, который я включил, вероятно, не обрабатывает.
Запрос ниже дает правильные ответы для предоставленных вами сценариев, но вы, вероятно, захотите придумать некоторые дополнительные сложные сценарии и посмотреть, есть ли какие-либо ошибки.
Например, у меня есть ощущение, что этот запрос может выйти из строя, если у вас есть несколько предыдущих заказов, перекрывающихся с более поздним заказом.
with CurrentOrders (UserId, SubscriptionDays, StartDate, EndDate) as
(
select
userid,
sum(subscriptiondays),
min(orderdate),
dateadd(day, sum(subscriptiondays), min(orderdate))
from #orders
where
#orders.orderdate <= @currentdate
-- start with the latest order(s)
and not exists (
select 1
from #orders o2
where
o2.userid = #orders.userid
and o2.orderdate <= @currentdate
and o2.orderdate > #orders.orderdate
)
group by
userid
union all
select
#orders.userid,
#orders.subscriptiondays,
#orders.orderdate,
dateadd(day, #orders.subscriptiondays, #orders.orderdate)
from #orders
-- join any overlapping orders
inner join CurrentOrders on
#orders.userid = CurrentOrders.UserId
and #orders.orderdate < CurrentOrders.StartDate
and dateadd(day, #orders.subscriptiondays, #orders.orderdate) > CurrentOrders.StartDate
)
select
UserId,
sum(SubscriptionDays) as TotalSubscriptionDays,
min(StartDate),
sum(SubscriptionDays) - datediff(day, min(StartDate), @currentdate) as RemainingSubscriptionDays
from CurrentOrders
group by
UserId
;
Филипп упомянул обеспокоенность по поводу предела рекурсии для общих табличных выражений. Ниже приведена процедурная альтернатива с использованием табличной переменной и цикла while, который, как мне кажется, выполняет то же самое.
Хотя я убедился, что этот альтернативный код работает, по крайней мере для предоставленных образцов данных, я был бы рад услышать чьи-либо комментарии об этом подходе. Отличная идея? Плохая идея? Есть какие-либо проблемы, о которых нужно знать?
declare @ModifiedRows int
declare @CurrentOrders table
(
UserId int not null,
SubscriptionDays int not null,
StartDate date not null,
EndDate date not null
)
insert into @CurrentOrders
select
userid,
sum(subscriptiondays),
min(orderdate),
min(dateadd(day, subscriptiondays, orderdate))
from #orders
where
#orders.orderdate <= @currentdate
-- start with the latest order(s)
and not exists (
select 1
from #orders o2
where
o2.userid = #orders.userid
and o2.orderdate <= @currentdate
-- there does not exist any other order that surpasses it
and dateadd(day, o2.subscriptiondays, o2.orderdate) > dateadd(day, #orders.subscriptiondays, #orders.orderdate)
)
group by
userid
set @ModifiedRows = @@ROWCOUNT
-- perform an extra update here in case there are any additional orders that were made after the start date but before the specified @currentdate
update co set
co.SubscriptionDays = co.SubscriptionDays + #orders.subscriptiondays
from @CurrentOrders co
inner join #orders on
#orders.userid = co.UserId
and #orders.orderdate <= @currentdate
and #orders.orderdate >= co.StartDate
and dateadd(day, #orders.subscriptiondays, #orders.orderdate) < co.EndDate
-- Keep attempting to update rows as long as rows were updated on the previous attempt
while(@ModifiedRows > 0)
begin
update co set
SubscriptionDays = co.SubscriptionDays + overlap.subscriptiondays,
StartDate = overlap.orderdate
from @CurrentOrders co
-- join any overlapping orders
inner join (
select
#orders.userid,
sum(#orders.subscriptiondays) as subscriptiondays,
min(orderdate) as orderdate
from #orders
inner join @CurrentOrders co2 on
#orders.userid = co2.UserId
and #orders.orderdate < co2.StartDate
and dateadd(day, #orders.subscriptiondays, #orders.orderdate) > co2.StartDate
group by
#orders.userid
) overlap on
overlap.userid = co.UserId
set @ModifiedRows = @@ROWCOUNT
end
select
UserId,
sum(SubscriptionDays) as TotalSubscriptionDays,
min(StartDate),
sum(SubscriptionDays) - datediff(day, min(StartDate), @currentdate) as RemainingSubscriptionDays
from @CurrentOrders
group by
UserId
РЕДАКТИРОВАТЬ2: Я внес некоторые изменения в приведенный выше код, чтобы учесть различные особые случаи, например, если у пользователя просто два заказа, оба заканчиваются в одну и ту же дату.
Например, изменение данных настройки на следующие вызвало проблемы с исходным кодом, которые я сейчас исправил:
insert into #orders
select 1, 2, 10, '2011-01-01'
union
select 2, 1, 10, '2011-01-10'
union
select 3, 1, 10, '2011-01-15'
union
select 4, 2, 6, '2011-01-15'
union
select 5, 2, 4, '2011-01-17'
РЕДАКТИРОВАТЬ3: я сделал некоторые дополнительные корректировки для решения других особых случаев. В частности, предыдущий код столкнулся с проблемами со следующими данными настройки, которые я сейчас исправил:
insert into #orders
select 1, 2, 10, '2011-01-01'
union
select 2, 1, 6, '2011-01-10'
union
select 3, 1, 10, '2011-01-15'
union
select 4, 2, 10, '2011-01-15'
union
select 5, 1, 4, '2011-01-12'