PostgREST实现条件性更新插入的技术方案

在PostgREST应用中，开发者经常会遇到需要根据特定条件执行更新或插入操作的需求。本文将以订单表为例，探讨如何实现只有当新记录的序列号大于现有记录时才执行更新操作的技术方案。

问题背景

假设我们有一个订单表(orders)，其中包含两个关键字段：

• order_number(订单号)：作为主键
• seq_num(序列号)：用于记录版本或顺序的数值字段

业务需求是：只有当新记录的seq_num值大于表中已存在记录的seq_num值时，才执行更新操作；否则保持原记录不变。

技术限制分析

PostgREST原生提供的PUT方法无法直接实现这种条件性更新插入操作。这是因为：

1. PUT请求本身是无条件的，它会直接覆盖现有记录
2. REST协议本身不提供这种条件判断机制

解决方案

方案一：使用触发器实现

可以在数据库层面创建一个BEFORE INSERT OR UPDATE触发器，在数据变更前进行条件判断：

CREATE OR REPLACE FUNCTION check_seq_num()
RETURNS TRIGGER AS $$
BEGIN
    IF TG_OP = 'UPDATE' OR TG_OP = 'INSERT' THEN
        IF EXISTS (SELECT 1 FROM orders WHERE order_number = NEW.order_number AND seq_num >= NEW.seq_num) THEN
            RETURN NULL; -- 取消操作
        END IF;
    END IF;
    RETURN NEW;
END;
$$ LANGUAGE plpgsql;

CREATE TRIGGER orders_seq_num_trigger
BEFORE INSERT OR UPDATE ON orders
FOR EACH ROW EXECUTE FUNCTION check_seq_num();

方案二：使用存储过程封装

创建一个专用的存储过程来处理这种条件性更新插入：

CREATE OR REPLACE FUNCTION upsert_order(
    p_order_number text,
    p_seq_num integer,
    -- 其他字段参数
) RETURNS json AS $$
DECLARE
    result json;
BEGIN
    -- 检查是否存在记录且seq_num小于传入值
    IF EXISTS (
        SELECT 1 FROM orders 
        WHERE order_number = p_order_number 
        AND seq_num < p_seq_num
    ) THEN
        -- 执行更新
        UPDATE orders SET
            seq_num = p_seq_num,
            -- 其他字段更新
        WHERE order_number = p_order_number
        RETURNING row_to_json(orders.*) INTO result;
        
        RETURN result;
    ELSIF NOT EXISTS (SELECT 1 FROM orders WHERE order_number = p_order_number) THEN
        -- 执行插入
        INSERT INTO orders(order_number, seq_num, ...)
        VALUES (p_order_number, p_seq_num, ...)
        RETURNING row_to_json(orders.*) INTO result;
        
        RETURN result;
    ELSE
        -- 返回原记录
        SELECT row_to_json(orders.*) INTO result 
        FROM orders WHERE order_number = p_order_number;
        
        RETURN result;
    END IF;
END;
$$ LANGUAGE plpgsql;

方案三：使用PostgREST的RPC功能

通过PostgREST的函数调用功能来执行条件性操作：

1. 首先创建上述存储过程
2. 然后通过POST请求调用该函数

请求示例：

POST /rpc/upsert_order
Content-Type: application/json

{
    "p_order_number": "ORD123",
    "p_seq_num": 5,
    ...
}

性能考虑

1. 触发器方案对应用透明，但可能影响批量操作的性能
2. 存储过程方案更灵活，可以包含更复杂的业务逻辑
3. 两种方案都需要确保order_number和seq_num字段有适当的索引

事务完整性

无论采用哪种方案，都需要注意：

1. 确保操作在事务中执行
2. 考虑并发情况下的锁竞争问题
3. 在高并发场景下可能需要使用SELECT FOR UPDATE来避免竞态条件

总结

在PostgREST中实现条件性更新插入操作，虽然不能直接通过标准的REST方法实现，但可以通过数据库触发器或自定义存储过程的方式优雅地解决。选择哪种方案取决于具体的业务需求、性能要求和开发团队的偏好。