如何将 PostgreSQL“merge_db”(又名 upsert)函数翻译成 MySQL
Posted
技术标签:
【中文标题】如何将 PostgreSQL“merge_db”(又名 upsert)函数翻译成 MySQL【英文标题】:How to translate PostgreSQL "merge_db" (aka upsert) function into MySQL 【发布时间】:2012-07-07 01:10:37 【问题描述】:直接来自手册,这里是canonical example of merge_db in PostgreSQL:
CREATE TABLE db (a INT PRIMARY KEY, b TEXT);
CREATE FUNCTION merge_db(key INT, data TEXT) RETURNS VOID AS
$$
BEGIN
LOOP
-- first try to update the key
UPDATE db SET b = data WHERE a = key;
IF found THEN
RETURN;
END IF;
-- not there, so try to insert the key
-- if someone else inserts the same key concurrently,
-- we could get a unique-key failure
BEGIN
INSERT INTO db(a,b) VALUES (key, data);
RETURN;
EXCEPTION WHEN unique_violation THEN
-- Do nothing, and loop to try the UPDATE again.
END;
END LOOP;
END;
$$
LANGUAGE plpgsql;
SELECT merge_db(1, 'david');
SELECT merge_db(1, 'dennis');
这是否可以在 mysql 中表示为用户定义的函数,如果可以,如何表示?与 MySQL 的标准 INSERT...ON DUPLICATE KEY UPDATE
相比有什么优势吗?
注意:我专门寻找用户定义的函数,而不是INSERT...ON DUPLICATE KEY UPDATE
。
【问题讨论】:
【参考方案1】:在 MySQL 5.5.14 上测试。
CREATE TABLE db (a INT PRIMARY KEY, b TEXT);
DELIMITER //
CREATE PROCEDURE merge_db(k INT, data TEXT)
BEGIN
DECLARE done BOOLEAN;
REPEAT
BEGIN
-- If there is a unique key constraint error then
-- someone made a concurrent insert. Reset the sentinel
-- and try again.
DECLARE ER_DUP_UNIQUE CONDITION FOR 23000;
DECLARE CONTINUE HANDLER FOR ER_DUP_UNIQUE BEGIN
SET done = FALSE;
END;
SET done = TRUE;
SELECT COUNT(*) INTO @count FROM db WHERE a = k;
-- Race condition here. If a concurrent INSERT is made after
-- the SELECT but before the INSERT below we'll get a duplicate
-- key error. But the handler above will take care of that.
IF @count > 0 THEN
UPDATE db SET b = data WHERE a = k;
ELSE
INSERT INTO db (a, b) VALUES (k, data);
END IF;
END;
UNTIL done END REPEAT;
END//
DELIMITER ;
CALL merge_db(1, 'david');
CALL merge_db(1, 'dennis');
一些想法:
您不能先进行更新,然后再检查@ROW_COUNT()
,因为它会返回实际更改的行数。如果该行已经具有您尝试更新的值,则该值可能为 0。
另外,@ROW_COUNT()
不是复制安全的。
您可以使用REPLACE...INTO
。
如果使用 InnoDB 或支持事务的表,您也许可以使用 SELECT...FOR UPDATE
(未经测试)。
与仅使用 INSERT...ON DUPLICATE KEY UPDATE
相比,我认为此解决方案没有任何优势。
【讨论】:
以上是关于如何将 PostgreSQL“merge_db”(又名 upsert)函数翻译成 MySQL的主要内容,如果未能解决你的问题,请参考以下文章
如何使用 PostgreSQL 将数据从 AngularJS 前端传递到 Nodejs 后端?
如何将 Postgresql 布尔值转换为 MySQL Tinyint?
如何使用 Pscycopg2 将字典插入 Postgresql 表