PostgreSql | 数据库 |自定义函数的设计和实现

-- 创建一个自定义函数，用于计算两个日期之间的工作日天数
CREATE OR REPLACE FUNCTION count_workdays(_start DATE, _end DATE)
RETURNS INTEGER AS $$
DECLARE
    workdays INTEGER := 0;
    current_date DATE := _start;
BEGIN
    WHILE current_date <= _end LOOP
        -- 检查当前日期是否是工作日（这里假设工作日为周一至周五）
        IF EXTRACT(DOW FROM current_date) BETWEEN 1 AND 5 THEN
            workdays := workdays + 1;
        END IF;
        -- 移动到下一天
        current_date := current_date + INTERVAL '1 day';
    END LOOP;
    RETURN workdays;
END;
$$ LANGUAGE plpgsql IMMUTABLE;
 
-- 使用自定义函数
SELECT count_workdays('2023-01-01'::date, '2023-01-07'::date) AS workdays;

这个自定义函数count_workdays接收两个日期参数，并计算这两个日期之间的工作日数。函数内部使用了一个WHILE循环来遍历每一天，并通过EXTRACT函数检查每一天的是否是工作日（假设工作日为周一至周五）。最后返回工作日的总数。这个函数可以直接在PostgreSQL数据库中使用，并可通过SELECT语句进行调用。