PL/pgSQL中如何断言约束异常并封装自定义断言函数？

当然可以实现这类针对约束违例的精准断言，甚至还能指定具体的约束类型！让我一步步给你拆解可行的方案：

PL/pgSQL原生的ASSERT语句只能检查布尔条件，没法直接断言「某个操作会触发特定约束违例」，但我们可以通过嵌套BEGIN...EXCEPTION块来实现这个逻辑，甚至能精确匹配约束类型或约束名称。

基础实现：捕获特定约束违例 #

先举个实际例子，假设我们有一张带唯一约束的表：

CREATE TABLE test_user (
    id INT PRIMARY KEY,
    username VARCHAR(50) UNIQUE NOT NULL
);

下面的PL/pgSQL块会断言「插入重复用户名会触发唯一约束违例」：

DO $$
DECLARE
    v_username VARCHAR(50) := 'johndoe';
BEGIN
    -- 先插入一条初始记录，确保后续操作会触发约束
    INSERT INTO test_user (id, username) VALUES (1, v_username);
    
    -- 嵌套块包裹要测试的操作
    BEGIN
        -- 尝试触发约束违例的操作
        INSERT INTO test_user (id, username) VALUES (2, v_username);
        
        -- 如果执行到这里，说明没触发预期的约束，主动抛出断言失败
        RAISE ASSERTION_FAILED USING MESSAGE = 'Expected unique constraint violation on test_user.username';
    EXCEPTION
        WHEN unique_violation THEN
            -- 捕获到预期的约束违例，断言通过，什么都不用做
            NULL;
        WHEN OTHERS THEN
            -- 捕获到其他异常，说明不是预期的，抛出断言失败
            RAISE ASSERTION_FAILED USING MESSAGE = 'Unexpected exception: ' || SQLERRM;
    END;
END $$;

这里的核心逻辑是：如果测试操作没有抛出预期的异常，就主动触发ASSERTION_FAILED；如果抛出了目标异常，就静默通过；其他异常则直接判定断言失败。

封装成自定义断言函数 #

为了避免重复写嵌套块的逻辑，我们可以把这个断言逻辑封装成通用函数。PostgreSQL没有宏，但自定义函数完全能满足需求：

基于SQLSTATE的通用断言函数 #

PostgreSQL的每个异常都对应一个标准SQLSTATE码（比如唯一约束违例是23505，外键约束违例是23503），用这个码来匹配异常会比异常名称更可靠：

CREATE OR REPLACE FUNCTION assert_constraint_violation(p_sql TEXT, p_expected_sqlstate TEXT)
RETURNS VOID AS $$
BEGIN
    BEGIN
        -- 执行要测试的SQL
        EXECUTE p_sql;
        
        -- 没抛出异常，断言失败
        RAISE ASSERTION_FAILED USING MESSAGE = 'Expected SQLSTATE ' || p_expected_sqlstate || ' but none was raised';
    EXCEPTION
        WHEN OTHERS THEN
            -- 检查异常的SQLSTATE是否匹配预期
            IF SQLSTATE != p_expected_sqlstate THEN
                RAISE ASSERTION_FAILED USING MESSAGE = 'Expected SQLSTATE ' || p_expected_sqlstate || ', got ' || SQLSTATE || ': ' || SQLERRM;
            END IF;
            -- 匹配成功，直接返回
            RETURN;
    END;
END $$ LANGUAGE plpgsql;

使用这个函数的示例：

DO $$
BEGIN
    INSERT INTO test_user (id, username) VALUES (3, 'janedoe');
    
    -- 断言插入重复用户名会触发唯一约束违例（SQLSTATE 23505）
    PERFORM assert_constraint_violation(
        'INSERT INTO test_user (id, username) VALUES (4, ''janedoe'')',
        '23505'
    );
END $$;

进阶：针对具体约束名称的断言 #

如果需要更精准地断言「触发的是某个特定约束」，可以检查异常消息中是否包含目标约束名：

CREATE OR REPLACE FUNCTION assert_specific_constraint_violation(p_sql TEXT, p_constraint_name TEXT)
RETURNS VOID AS $$
DECLARE
    v_error_msg TEXT;
BEGIN
    BEGIN
        EXECUTE p_sql;
        RAISE ASSERTION_FAILED USING MESSAGE = 'Expected violation of constraint ' || p_constraint_name || ' but none was raised';
    EXCEPTION
        WHEN OTHERS THEN
            v_error_msg := SQLERRM;
            -- PostgreSQL的约束违例消息会包含约束名称，比如"duplicate key value violates unique constraint "test_user_username_key""
            IF position(p_constraint_name IN v_error_msg) = 0 THEN
                RAISE ASSERTION_FAILED USING MESSAGE = 'Expected constraint ' || p_constraint_name || ', got error: ' || v_error_msg;
            END IF;
            RETURN;
    END;
END $$ LANGUAGE plpgsql;

使用示例：

PERFORM assert_specific_constraint_violation(
    'INSERT INTO test_user (id, username) VALUES (5, ''janedoe'')',
    'test_user_username_key'
);

• 原生ASSERT无法直接断言约束违例，但通过嵌套EXCEPTION块可以实现精准的断言逻辑
• 封装成自定义函数能大幅简化重复代码，推荐使用SQLSTATE来匹配异常类型（比异常名称更稳定）
• 如果需要更细粒度的控制，可以针对具体约束名称做断言

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