PL/pgSQL函数中捕获特定命名唯一约束异常的方法

这个需求其实很常见，我之前处理过类似的场景。虽然EXCEPTION WHEN unique_violation会把所有唯一约束冲突都拦下来，但只要在异常块里加个判断，就能精准只处理你指定的那个约束，其他情况直接把原始错误抛给客户端。

两种实现方式，推荐第二种更可靠： #

方式一：通过错误消息字符串匹配（简单但依赖消息格式）

当PostgreSQL触发唯一约束冲突时，错误消息（SQLERRM变量）里会明确写出违反的约束名，格式大概是duplicate key value violates unique constraint "your_constraint_name"。我们可以直接截取这个名称做判断：

CREATE OR REPLACE FUNCTION create_user(p_email text, p_name text)
RETURNS void AS $$
BEGIN
    INSERT INTO users (email, name) VALUES (p_email, p_name);
EXCEPTION
    WHEN unique_violation THEN
        -- 从错误消息里提取约束名
        IF substring(SQLERRM FROM 'unique constraint "([^"]+)"') = 'users_email_unique' THEN
            -- 这里写你针对该约束的自定义逻辑，比如提示用户邮箱已存在
            RAISE NOTICE '该邮箱已被注册，请更换其他邮箱';
            -- 如果不需要提示，直接静默处理也可以
        ELSE
            -- 不是目标约束，原样抛出原始错误
            RAISE;
        END IF;
END;
$$ LANGUAGE plpgsql;

方式二：通过诊断字段获取约束名（更可靠，推荐）

上面的字符串匹配可能会因为错误消息的语言或格式变化失效，PostgreSQL 9.3及以上版本支持用GET STACKED DIAGNOSTICS直接获取约束名称，这是系统提供的标准字段，完全不受消息格式影响：

CREATE OR REPLACE FUNCTION create_user(p_email text, p_name text)
RETURNS void AS $$
DECLARE
    target_constraint text := 'users_email_unique'; -- 这里替换成你的目标约束名
    violated_constraint text;
BEGIN
    INSERT INTO users (email, name) VALUES (p_email, p_name);
EXCEPTION
    WHEN unique_violation THEN
        -- 获取实际违反的约束名称
        GET STACKED DIAGNOSTICS violated_constraint = CONSTRAINT_NAME;
        
        IF violated_constraint = target_constraint THEN
            -- 执行你的自定义处理逻辑
            RAISE NOTICE '该邮箱已被注册，请更换其他邮箱';
        ELSE
            -- 非目标约束，重新抛出原始异常
            RAISE;
        END IF;
END;
$$ LANGUAGE plpgsql;

几个注意点： #

• 一定要确认目标约束的准确名称：如果创建约束时用了双引号，名称会保留大小写；没加的话会自动转为小写，别写错了
• RAISE;不带任何参数时，会完整抛出原始异常，客户端收到的错误和没捕获时完全一样
• 如果处理目标约束时不想返回任何错误，直接把RAISE NOTICE换成你的业务逻辑就行，比如记录日志啥的

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