DB2中避免MERGE INTO并发：原子完成增改查并获取QTY值

针对你遇到的DB2不支持SELECT FROM MERGE，需要原子完成检查、更新/插入、获取最新QTY且避免并发问题的需求，我整理了几个实用的方案，咱们逐个来看：

方案1：封装为带输出参数的存储过程（推荐） #

把锁表、MERGE、查询这一系列操作封装到存储过程中，利用存储过程的事务原子性来保证所有步骤在同一个事务内执行，同时通过输出参数直接返回最新的QTY值，避免外部查询的并发风险。

存储过程代码示例 #

CREATE OR REPLACE PROCEDURE UPDATE_DOC_COUNT(
    IN p_SEND_VALUE VARCHAR(100),  -- 根据实际字段类型调整
    IN p_RECEIVE_VALUE VARCHAR(100),
    IN p_DOC_ID_VALUE INT,
    OUT p_QTY INT
)
LANGUAGE SQL
BEGIN
    -- 排他锁定整个表，确保其他事务无法修改或读取表数据直到当前事务结束
    LOCK TABLE DOC_COUNT IN EXCLUSIVE MODE;

    -- 执行你的MERGE逻辑
    MERGE INTO DOC_COUNT AS mt 
    USING ( SELECT * FROM TABLE ( VALUES (p_SEND_VALUE, p_RECEIVE_VALUE, 1) ) ) AS vt (SEND, RECEIVE, QTY) 
    ON (mt.SEND = vt.SEND AND mt.RECEIVE = vt.RECEIVE) 
    WHEN MATCHED THEN UPDATE SET QTY = mt.QTY + 1 
    WHEN NOT MATCHED THEN INSERT (SEND, RECEIVE, QTY) 
        VALUES (vt.SEND, vt.RECEIVE, (SELECT COUNT(DOCUMENT_ID) AS DOC_CC_COUNT FROM V_DOC_COUNT WHERE DOCUMENT_ID <= p_DOC_ID_VALUE AND RECEIVE = p_RECEIVE_VALUE));

    -- 查询最新QTY并赋值给输出参数
    SELECT QTY INTO p_QTY FROM DOC_COUNT WHERE SEND = p_SEND_VALUE AND RECEIVE = p_RECEIVE_VALUE;
END@

方案优势 #

• 所有操作在同一个事务内完成，原子性有保障；
• 表锁生效后，其他事务无法执行任何修改或读取操作，彻底避免MERGE与查询之间的并发问题；
• 直接通过输出参数返回结果，无需在外部额外执行查询语句。

方案2：行级锁+事务包裹（性能更优） #

如果觉得表锁粒度太粗影响性能，可以换成行级锁，仅锁定目标行，同时把所有操作放在一个事务中，确保原子性和并发安全。

执行代码示例 #

-- 手动开启事务（如果你的客户端没有自动开启事务的话）
START TRANSACTION;

-- 尝试锁定目标行：如果行存在则立即锁定，不存在也不影响后续INSERT
SELECT 1 FROM DOC_COUNT WHERE SEND = ?SEND_VALUE? AND RECEIVE = ?RECEIVE_VALUE? FOR UPDATE;

-- 执行MERGE操作
MERGE INTO DOC_COUNT AS mt 
USING ( SELECT * FROM TABLE ( VALUES (?SEND_VALUE?, ?RECEIVE_VALUE?, 1) ) ) AS vt (SEND, RECEIVE, QTY) 
ON (mt.SEND = vt.SEND AND mt.RECEIVE = vt.RECEIVE) 
WHEN MATCHED THEN UPDATE SET QTY = mt.QTY + 1 
WHEN NOT MATCHED THEN INSERT (SEND, RECEIVE, QTY) 
    VALUES (vt.SEND, vt.RECEIVE, (SELECT COUNT(DOCUMENT_ID) AS DOC_CC_COUNT FROM V_DOC_COUNT WHERE DOCUMENT_ID <= ?DOC_ID_VALUE? AND RECEIVE = ?RECEIVE_VALUE));

-- 查询最新的QTY值（此时行已被锁定，其他事务无法修改）
SELECT QTY FROM DOC_COUNT WHERE SEND = ?SEND_VALUE? AND RECEIVE = ?RECEIVE_VALUE?;

-- 提交事务，释放锁
COMMIT;

方案优势 #

• 行级锁粒度更细，不会影响表中其他行的正常操作，性能比表锁更好；
• 事务内的所有操作原子执行，锁定行后其他事务无法修改或读取该行（取决于隔离级别，读已提交及以上级别下，其他事务看不到未提交的修改），确保查询到的是最新值。

方案3：利用触发器+会话变量获取结果 #

通过创建AFTER INSERT/UPDATE触发器，在数据修改后自动将最新QTY写入会话变量，无需再查询表即可获取结果。

步骤1：创建会话变量和触发器 #

-- 创建会话级变量，用于存储最新QTY
CREATE VARIABLE SESSION.QTY_RESULT INT;

-- 创建触发器，当目标行被修改/插入时更新会话变量
CREATE OR REPLACE TRIGGER TRG_DOC_COUNT_AFTER_CHANGE
AFTER INSERT OR UPDATE ON DOC_COUNT
REFERENCING NEW AS NEW_ROW
FOR EACH ROW
WHEN (NEW_ROW.SEND = SESSION.SEND_PARAM AND NEW_ROW.RECEIVE = SESSION.RECEIVE_PARAM)
SET SESSION.QTY_RESULT = NEW_ROW.QTY;

步骤2：执行操作并获取结果 #

START TRANSACTION;

-- 设置会话参数，让触发器匹配目标行
SET SESSION.SEND_PARAM = ?SEND_VALUE?;
SET SESSION.RECEIVE_PARAM = ?RECEIVE_VALUE?;

-- 锁定行或表（这里用行级锁示例）
SELECT 1 FROM DOC_COUNT WHERE SEND = ?SEND_VALUE? AND RECEIVE = ?RECEIVE_VALUE? FOR UPDATE;

-- 执行MERGE
MERGE INTO DOC_COUNT AS mt 
USING ( SELECT * FROM TABLE ( VALUES (?SEND_VALUE?, ?RECEIVE_VALUE?, 1) ) ) AS vt (SEND, RECEIVE, QTY) 
ON (mt.SEND = vt.SEND AND mt.RECEIVE = vt.RECEIVE) 
WHEN MATCHED THEN UPDATE SET QTY = mt.QTY + 1 
WHEN NOT MATCHED THEN INSERT (SEND, RECEIVE, QTY) 
    VALUES (vt.SEND, vt.RECEIVE, (SELECT COUNT(DOCUMENT_ID) AS DOC_CC_COUNT FROM V_DOC_COUNT WHERE DOCUMENT_ID <= ?DOC_ID_VALUE? AND RECEIVE = ?RECEIVE_VALUE));

-- 直接从会话变量获取最新QTY
SELECT SESSION.QTY_RESULT AS QTY;

COMMIT;

方案优势 #

• 无需额外查询表，直接从会话变量获取结果，减少一次表访问；
• 会话变量仅当前会话可见，不会与其他会话产生冲突。

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