PostgreSQL触发器函数中如何将查询结果存入变量并在IF中复用

没问题，我来帮你调整这个触发器函数，实现把查询结果存入变量再判断的需求。核心就是利用PL/pgSQL的变量声明与赋值能力，步骤很清晰：

1. 声明布尔类型变量：因为EXISTS查询返回的是布尔值（存在为true，不存在为false），用布尔变量存储这个结果最贴合需求。
2. 将查询结果赋值给变量：通过SELECT ... INTO语法，把EXISTS查询的结果存入提前声明的变量中。
3. 复用变量做逻辑判断：在IF语句里直接使用这个变量，判断是否执行后续的插入操作。

以下是修改后的完整触发器函数代码：

CREATE OR REPLACE FUNCTION add_row() 
RETURNS TRIGGER 
AS $$
DECLARE
    row_exists BOOLEAN; -- 声明布尔变量，用于存储查询结果
BEGIN
    -- 执行查询并将结果存入变量：检查今日是否已存在对应user_id的行
    SELECT EXISTS(
        SELECT 1 
        FROM some_table 
        WHERE "user_id" = new."user_id" 
        AND created >= now()::date
    ) INTO row_exists;

    -- 复用变量判断：如果不存在目标行则执行插入
    IF NOT row_exists THEN
        INSERT INTO another_table VALUES(1, 2, 3);
    END IF;

    RETURN NULL; -- 触发器需返回值，若你的触发器是BEFORE类型也可返回NEW，按需调整
END;
$$ LANGUAGE plpgsql;

补充说明： #

• 触发器函数必须返回值，我在代码里添加了RETURN NULL;，如果你的触发器是BEFORE INSERT这类场景，也可以返回NEW，具体根据触发器的触发时机和业务需求调整。
• 用EXISTS直接返回布尔值比存储其他类型更高效，因为数据库不需要返回实际行数据，仅需判断是否存在即可。

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