如何使用 pg-promise 重复 SQL 插入直到成功？

Question

在我的程序中，我将一些数据插入到一个表中并取回它的 id，我需要确保我将该 id 输入到另一个具有唯一随机生成字符串的表中。但是，如果尝试插入已经存在的随机字符串而插入失败，我该如何重复插入直到成功？

我正在使用pg-promise 与 postgreSQL 对话。我可以运行这样的程序，将数据插入到两个表中，因为随机字符串不存在：

   db.none(
            `
            WITH insert_post AS
            (
                INSERT INTO table_one(text) VALUES('abcd123')
                RETURNING id
            )
            INSERT INTO table_two(id, randstr)
                    VALUES((SELECT id FROM insert_post), '${randStrFn()}')
            `
        )
    .then(() => console.log("Success"))
    .catch(err => console.log(err));

我不确定是否有任何基于 SQL/JS/pg-promise 的简单解决方案可供我使用。

Answer 1

我会鼓励问题的作者为他的问题寻求一个纯 SQL 解决方案，因为在性能方面它会比其他任何东西都更有效。

但由于问题是关于如何使用pg-promise 重新运行查询，除了已经发布的示例之外，我将提供一个示例，除非每次尝试都获取和释放连接，以及适当的数据完整性。

db.tx(t => {
    // BEGIN;
    return t.one('INSERT INTO table_one(text) VALUES($1) RETURNING id', 'abcd123', a => +a.id)
        .then(id => {
            var f = attempts => t.none('INSERT INTO table_two(id, randstr) VALUES($1, randStrFn())', id)
                .catch(error => {
                    if (--attempts) {
                        return f(attempts); // try again
                    }
                    throw error; // give up
                });
            return f(3); // try up to 3 times
        });
})
    .then(data => {
        // COMMIT;
        // success, data = null
    })
    .catch(error => {
        // ROLLBACK;
    });

由于您尝试重新运行依赖查询，因此不应让第一个查询保持成功，如果您对第二个查询的所有尝试都失败了，您应该回滚所有更改，即使用事务 - 方法 @ 987654322@，如代码所示。

这就是为什么我们将您的 WITH 查询拆分到事务中，以确保这种完整性。

更新

下面是一个更好的版本。因为事务内部的错误需要隔离，为了避免破坏事务堆栈，每次尝试都应该在自己的SAVEPOINT内部，这意味着使用另一个事务级别：

db.tx(t => {
    // BEGIN;
    return t.one('INSERT INTO table_one(name) VALUES($1) RETURNING id', 'abcd123', a => +a.id)
        .then(id => {
            var f = attempts => t.tx(sp => {
                // SAVEPOINT level_1;
                return sp.none('INSERT INTO table_two(id, randstr) VALUES($1, randStrFn())', id);
            })
                .catch(error => {
                    // ROLLBACK TO SAVEPOINT level_1;
                    if (--attempts) {
                        return f(attempts); // try again
                    }
                    throw error; // give up
                });
            return f(3); // try up to 3 times
        });
})
    .then(data => {
        // 1) RELEASE SAVEPOINT level_1;
        // 2) COMMIT;
    })
    .catch(error => {
        // ROLLBACK;
    });

我还建议使用pg-monitor，这样您就可以看到并理解下面发生的事情，以及实际上正在执行的查询。

---

附：我是pg-promise的作者。

Answer 2

最简单的方法是将它放入一个方法中，然后在 catch 中重新调用它：

const insertPost = (post, numRetries) => {
    return
       db.none(
                `
                WITH insert_post AS
                (
                    INSERT INTO table_one(text) VALUES('abcd123')
                    RETURNING id
                )
                INSERT INTO table_two(id, randstr)
                        VALUES((SELECT id FROM insert_post), '${randStrFn()}')
                `
            )
        .then(() => console.log("Success"))
        .catch(err => {
            console.log(err)
            if (numRetries < 3) {
              return self.insertPost(post, numRetries + 1);
            }
            throw err;
        });

}