如何修改PlPGSQL函数以自定义数据类型返回两行数据？

你的核心问题是没有把两次查询到的结果赋值到自定义类型的字段里，同时缺少返回语句。下面是修改后的完整方案，我会一步步解释关键改动：

修改后的完整代码 #

CREATE TABLE foo( 
    dt DATE NOT NULL, 
    f1 REAL NOT NULL, 
    f2 REAL NOT NULL, 
    f3 REAL NOT NULL, 
    f4 REAL NOT NULL
); -- 修正了原表定义末尾多余的逗号，避免语法错误

CREATE TYPE start_finish_type AS (
    start_f1 REAL, 
    start_f2 REAL, 
    start_f3 REAL, 
    start_f4 REAL, 
    finish_f1 REAL, 
    finish_f2 REAL, 
    finish_f3 REAL, 
    finish_f4 REAL
);

CREATE FUNCTION fetch_begin_end_data(start_date DATE, finish_date DATE) 
RETURNS start_finish_type AS $$
DECLARE
    actual_start_date DATE;
    actual_finish_date DATE ;
    retval start_finish_type;
BEGIN
    -- 获取>=start_date的最早日期
    SELECT MIN(dt) INTO actual_start_date FROM foo WHERE dt >= start_date;
    
    -- 这里把MIN改成了MAX：你应该是要取<=finish_date的最晚行，而非最早行，若需求特殊可改回MIN
    SELECT MAX(dt) INTO actual_finish_date FROM foo WHERE dt <= finish_date;

    -- 将起始日期对应的f1-f4赋值到retval的start前缀字段
    IF actual_start_date IS NOT NULL THEN
        SELECT f1, f2, f3, f4 
        INTO retval.start_f1, retval.start_f2, retval.start_f3, retval.start_f4 
        FROM foo WHERE dt = actual_start_date;
    ELSE
        -- 可选：无匹配起始行时的处理，这里给默认值，也可以抛出异常
        retval.start_f1 := 0;
        retval.start_f2 := 0;
        retval.start_f3 := 0;
        retval.start_f4 := 0;
    END IF;

    -- 将结束日期对应的f1-f4赋值到retval的finish前缀字段
    IF actual_finish_date IS NOT NULL THEN
        SELECT f1, f2, f3, f4 
        INTO retval.finish_f1, retval.finish_f2, retval.finish_f3, retval.finish_f4 
        FROM foo WHERE dt = actual_finish_date;
    ELSE
        -- 可选：无匹配结束行时的处理
        retval.finish_f1 := 0;
        retval.finish_f2 := 0;
        retval.finish_f3 := 0;
        retval.finish_f4 := 0;
    END IF;

    -- 返回组装好的自定义类型变量
    RETURN retval;
END $$ LANGUAGE plpgsql;

关键改动说明 #

• 修正语法错误：原表定义最后一列后多了个逗号，直接执行会报错，已移除。
• 优化日期查询逻辑：原代码用MIN(dt)获取结束日期，会拿到<=finish_date的最早行，这大概率不符合需求，改成MAX(dt)获取最晚的结束行。
• 赋值自定义类型字段：使用SELECT ... INTO语法，把两次查询到的f1-f4值，分别存入retval变量对应的start_*和finish_*字段。
• 增加空值保护：如果没有找到符合条件的起始/结束日期，避免查询抛出错误，这里设置了默认值，你也可以换成抛出异常（比如RAISE EXCEPTION 'No start date matched';）。
• 添加返回语句：原函数缺少返回逻辑，必须加上RETURN retval;才能返回组装好的自定义类型数据。

测试示例 #

你可以用以下语句验证函数效果：

-- 插入测试数据
INSERT INTO foo VALUES ('2024-01-01', 1.1, 2.2, 3.3, 4.4);
INSERT INTO foo VALUES ('2024-01-10', 5.5, 6.6, 7.7, 8.8);

-- 调用函数
SELECT * FROM fetch_begin_end_data('2024-01-01', '2024-01-10');

执行后会返回一行包含所有start和finish字段的结果，完全匹配你的自定义类型结构。

内容的提问来源于stack exchange，提问作者Homunculus Reticulli