如何根据 Snowflake 中先前查询的结果提交/回滚事务？

Question

我有这样的交易

BEGIN;

...

SELECT * FROM TABLE(RESULT_SCAN(LAST_QUERY_ID())); -- Returns TRUE/FALSE

+----------------+
| Result         |
|----------------|
| FALSE          |
+----------------+

-- If TRUE
COMMIT;

-- If FALSE
-- ROLLBACK;

在这种情况下，我如何根据上一个查询的结果“FALSE”进行回滚？我想自动化这个脚本，而不是自己在运行时做出决定。

Answer 1

也许您可以创建一个存储过程并根据参数执行提交或回滚：

CREATE OR REPLACE PROCEDURE commit_or_rollback( result BOOLEAN ) 
returns BOOLEAN
language javascript
AS
$$
    if ( RESULT ) { snowflake.execute( {sqlText: "COMMIT" }) } 
    else {
        snowflake.execute( {sqlText: "ROLLBACK" }) };
    return RESULT;        
$$
;

所以你的脚本会是这样的：

BEGIN

...

CALL commit_or_rollback(  SELECT * FROM TABLE(RESULT_SCAN(LAST_QUERY_ID())) ); 

一个测试用例：

BEGIN;

SELECT CURRENT_TRANSACTION();

+-----------------------+
| CURRENT_TRANSACTION() |
+-----------------------+
| 1591113827577         |
+-----------------------+

CALL commit_or_rollback( ( SELECT TRUE ) ); 

+--------------------+
| COMMIT_OR_ROLLBACK |
+--------------------+
| TRUE               |
+--------------------+

SELECT CURRENT_TRANSACTION();

最后一个命令返回 NULL。我还可以看到在查询历史记录中执行了 COMMIT。