CockroachDB不支持UDF，如何模拟Postgres的digest校验报错逻辑？

由于CockroachDB暂不支持用户定义函数（UDF），可以通过以下几种方式实现当digest()返回值与预期不匹配时抛出错误的逻辑：

方法1：使用CHECK约束 #

直接在表结构中添加CHECK约束，将digest校验逻辑内嵌其中。当插入或更新数据时，若校验不通过，数据库会自动抛出约束违反错误。

示例代码：

CREATE TABLE example_table (
    id INT PRIMARY KEY,
    data TEXT NOT NULL,
    expected_digest BYTEA NOT NULL,
    CHECK (digest(data, 'sha256') = expected_digest)
);

尝试插入不符合校验条件的数据时：

INSERT INTO example_table (id, data, expected_digest)
VALUES (1, 'test', '\xdeadbeef');

CockroachDB会抛出check constraint failed: digest(data, 'sha256') = expected_digest的错误，直接阻断操作。

方法2：使用PL/pgSQL触发器 #

如果校验逻辑涉及更复杂的场景（如动态获取预期值、关联多字段校验），可以借助触发器实现。CockroachDB支持PL/pgSQL触发器函数，可在数据写入前执行自定义校验逻辑，不匹配则主动抛出错误。

示例代码：

-- 创建触发器函数
CREATE OR REPLACE FUNCTION validate_digest()
RETURNS TRIGGER AS $$
BEGIN
    IF digest(NEW.data, 'sha256') != NEW.expected_digest THEN
        RAISE EXCEPTION 'Digest mismatch: computed value does not match expected';
    END IF;
    RETURN NEW;
END;
$$ LANGUAGE plpgsql;

-- 绑定触发器到目标表
CREATE TRIGGER check_digest_trigger
BEFORE INSERT OR UPDATE ON example_table
FOR EACH ROW EXECUTE FUNCTION validate_digest();

触发校验时，会抛出自定义的错误信息，明确告知校验失败原因。

方法3：应用层前置校验 #

若无需依赖数据库层面的强制校验，可在应用代码中提前完成digest计算与对比，仅当匹配时才执行数据库操作，不匹配则直接在应用层抛出错误。这种方式灵活性强，无需修改数据库结构。

以Python为例：

import hashlib

def insert_valid_data(db_conn, record_id, data, expected_digest):
    # 计算数据的SHA256 digest
    computed_digest = hashlib.sha256(data.encode()).digest()
    if computed_digest != expected_digest:
        raise ValueError("Digest mismatch: computed value does not match expected")
    # 执行数据库插入
    with db_conn.cursor() as cursor:
        cursor.execute(
            "INSERT INTO example_table (id, data, expected_digest) VALUES (%s, %s, %s)",
            (record_id, data, expected_digest)
        )
    db_conn.commit()

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