PostgreSQL函数：获取UPDATE影响行数并实现结果通知与行数提示

刚好我之前处理过类似的需求，给你一个完整的实现方案，完全满足你的要求：

基础版函数实现 #

首先是针对你给出的distributors表更新场景的具体实现：

CREATE OR REPLACE FUNCTION update_distributors()
RETURNS TEXT AS $$
DECLARE
    affected_rows INTEGER;
BEGIN
    -- 用WITH子句捕获更新的行，统计受影响行数
    WITH rows AS (
        UPDATE distributors 
        SET dname = 'JKL Widgets' 
        WHERE did <= 10 
        RETURNING 1
    )
    SELECT count(*) INTO affected_rows FROM rows;

    -- 输出受影响行数到通知日志
    RAISE NOTICE 'Updated % rows', affected_rows;

    -- 根据行数返回对应的结果提示
    IF affected_rows > 0 THEN
        RETURN 'update success';
    ELSE
        RETURN 'update failed';
    END IF;
END;
$$ LANGUAGE plpgsql;

关键部分解释 #

• 捕获受影响行数：用你提供的WITH子句方式，通过RETURNING 1捕获所有被更新的行，再用count(*)统计数量，存入affected_rows变量。
• 输出通知：通过RAISE NOTICE输出具体的行数，调用函数时在控制台或日志里能看到这条信息。
• 返回结果判断：简单的条件判断，只要有受影响的行就返回成功提示，否则返回失败。

调用方式 #

执行下面的语句就能调用这个函数：

SELECT update_distributors();

调用后会先输出类似NOTICE: Updated 3 rows的通知，然后返回'update success'或'update failed'字符串。

通用版函数（推荐） #

如果想让函数更灵活，支持动态传入更新值和条件参数，可以改成这样：

CREATE OR REPLACE FUNCTION update_distributors(p_new_dname TEXT, p_max_did INTEGER)
RETURNS TEXT AS $$
DECLARE
    affected_rows INTEGER;
BEGIN
    WITH rows AS (
        UPDATE distributors 
        SET dname = p_new_dname 
        WHERE did <= p_max_did 
        RETURNING 1
    )
    SELECT count(*) INTO affected_rows FROM rows;

    RAISE NOTICE 'Updated % rows', affected_rows;

    -- 用CASE表达式简化判断逻辑
    RETURN CASE WHEN affected_rows > 0 THEN 'update success' ELSE 'update failed' END;
END;
$$ LANGUAGE plpgsql;

调用时传入参数即可：

SELECT update_distributors('JKL Widgets', 10);

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