如何在PL/pgSQL函数中使用条件语句筛选指定记录？

首先先明确你的表结构（方便后续参考）：

CREATE TABLE addresses ( 
    gid serial NOT NULL, 
    housenum character varying(30), 
    prefix character varying(10), 
    name character varying(100), 
    type character varying(16) 
);

我来帮你补全那个未完成的getAllFoo()函数，同时演示如何用IF语句筛选符合条件的记录：

方法1：循环+IF逐行判断（适合复杂逻辑） #

CREATE OR REPLACE FUNCTION "geomCheck".getAllFoo() 
RETURNS SETOF addresses 
AS $BODY$
DECLARE
    r addresses%rowtype; -- 定义变量存储每行数据
BEGIN
    -- 遍历addresses表的所有记录
    FOR r IN SELECT * FROM addresses LOOP
        -- 这里写你的筛选条件，示例：筛选type为'Foo'且housenum不为空的记录
        IF r.type = 'Foo' AND r.housenum IS NOT NULL THEN
            RETURN NEXT r; -- 返回符合条件的行
        END IF;

        -- 也可以添加ELSEIF分支处理其他情况
        -- ELSIF r.name LIKE '%Street%' THEN
        --     RETURN NEXT r;
        -- END IF;
    END LOOP;

    RETURN; -- 结束函数
END;
$BODY$ LANGUAGE plpgsql STABLE;

关键说明： #

• FOR r IN SELECT ...会遍历查询结果的每一行，把数据存入变量r
• IF语句里可以写任意你需要的条件组合，支持AND/OR、字段判断、函数调用等复杂逻辑
• 符合条件的行通过RETURN NEXT r返回，最终函数会返回所有符合条件的记录集合

方法2：RETURN QUERY直接返回（简单筛选更高效） #

如果你的筛选逻辑只是基础的条件判断，强烈推荐用这种方式，因为PostgreSQL会直接优化SQL查询，比逐行循环效率高很多：

CREATE OR REPLACE FUNCTION "geomCheck".getAllFoo() 
RETURNS SETOF addresses 
AS $BODY$
BEGIN
    -- 直接用SQL WHERE子句筛选，然后返回结果集
    RETURN QUERY
        SELECT * FROM addresses
        WHERE type = 'Foo' AND housenum IS NOT NULL; -- 替换成你的筛选条件

    RETURN;
END;
$BODY$ LANGUAGE plpgsql STABLE;

什么时候用哪种方法？ #

• 当筛选逻辑需要复杂的分支判断、调用其他函数、或者逐行处理数据时，用方法1的循环+IF
• 当只是简单的字段匹配、范围筛选等，用方法2的RETURN QUERY配合SQL语句，性能更好

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