1

我必须将 SQL Server 过程转换为 PostgreSQL，它将天数添加到给定日期。我做了一些研究并找到了一个解决方案，但它只适用于正数天数，当我将它传递一个负数（从给定日期移回）它永远不会起作用。以下是函数：

CREATE OR REPLACE FUNCTION public.add_business_day(
    from_date date,
    num_days integer)
    RETURNS date
    LANGUAGE 'sql'

    COST 100
    VOLATILE 
    
AS $BODY$
    select d
    from (
        select d::date, row_number() over (order by d)
        from generate_series(from_date+ 1, from_date+ num_days * 2 + 5, '1d') d
        where 
            extract('dow' from d) not in (0, 6) 
        ) s
    where row_number = num_days
$BODY$;

是否有任何修复/修改或替代此逻辑以使其适用于正数和负数天数？

4

1 回答 1

1

演示：db<>小提琴

我相信您需要一个条件查询来生成日期系列。像这样：

select 
    d::date, 
    CASE 
        WHEN num_days >= 0 THEN row_number() over (order by d) - 1
        ELSE row_number() over (order by d DESC) * -1 + 1
    END as row_number,
    extract('dow' from d)
from
    generate_series(
        CASE WHEN num_days >= 0 THEN current_date ELSE current_date + num_days * 2 - 5 END, 
        CASE WHEN num_days >= 0 THEN current_date + num_days * 2 + 5 ELSE current_date END, 
        '1d'
    ) d

这将切换函数的开始值和结束值generate_series()并更改row_number()窗口函数的顺序，以确保它以正确的顺序计算负值。

可以使用函数从函数中删除CASE子句。有了这个，您可以控制要添加的间隔，当然可以是负数：generate_series()sign()

演示：db<>小提琴

select 
    d::date, 
    CASE 
        WHEN num_days >= 0 THEN row_number() over (order by d) - 1
        ELSE row_number() over (order by d DESC) * -1 + 1
    END as row_number,
    extract('dow' from d)
from
    generate_series(current_date, current_date + num_days * 2 + sign(num_days)::int * 5, sign(num_days)::int * interval '1d') d
于 2020-11-18T22:06:56.133 回答