Python Glue 作业 - Snowflake 存储过程未返回确切的错误消息

Question

我正在尝试通过 python 粘合作业执行雪花存储过程，但是当我在存储过程中的 SQL 语句中遇到任何错误时，我希望该过程退出并且粘合作业应该失败。

我在我的雪花存储过程中使用此代码：

create or replace procedure schema.PROC_TEST 
    (input_sql varchar(16777216))
returns string not null
language javascript
as
$$
snowflake.execute({sqlText:`BEGIN TRANSACTION;`});
try
                {
                    var QueryExec = INPUT_SQL+ ";";
                    var QueryResult = snowflake.execute({sqlText: QueryExec});
                }
                catch (err)                                     // Catching any error if occurs in INPUT_SQL
                {
                    result = 'Error: ' + err;
                    snowflake.execute({sqlText:`ROLLBACK;`});
                    return result;
                    throw err.message;
                }
$$;

现在，如果我传递如下输入查询（这proc_test_final1是不正确的表名称）：

insert into schema.proc_test_final1 
    select id, name, sal 
    from schema.proc_test_src;

该过程不会抛出错误“表名不正确”，而是返回一个通用错误，例如

错误：090232 (25000)：存储过程执行错误：在存储过程中启动的作用域事务不完整，已回滚

我认为这可能是因为 ROLLBACK 命令，所以我尝试删除该命令，但仍然失败并出现与上述相同的错误。我还尝试删除 catch 块中的结果变量，但仍然捕获了上述错误。

我想在胶水作业中捕获确切的特定 SQL 错误（语法错误/代码错误）。

下面是我的胶水工作代码，我在其中发现了错误：

input_sql = 'insert into schema.proc_test_final1 select id, name, sal from schema.proc_test_src'
try:
    f_CallProc = "call TEST_PROC('"+input_sql+"');"
    cur.execute(f_CallProc)
    print(cur.fetchall())
    cur.close()
    ctx.close()

except snowflake.connector.errors.ProgrammingError as err:
    print('Error: {0} ({1})'.format(err.msg, err.sfqid))
    cur.close()
    ctx.close()
    sys.exit(1)

Answer 1

Scoped transaction started in stored procedure is incomplete and it was rolled back当您运行具有显式 START TRANSACTION 但没有显式“ROLLBACK”或“COMMIT”的 SP 时，将发生错误“ ”。

在我的测试中，如果我没有以下行：

snowflake.execute({sqlText:`ROLLBACK;`});

您的 SP 将因此类错误而失败。

但是，当我有这一行时，它将失败并出现以下错误：

Error: SQL compilation error:
Schema 'ERICLIN.SCHEMA' does not exist or not authorized.

我认为这就是你所追求的。

您能再仔细检查一下您的 SP 吗？您复制并粘贴的内容正确吗？

顺便说一句，你原来的SP不会运行，因为没有名为“input_sql”的变量，它应该是“INPUT_SQL”，因为Snowflake的SP中的每个变量都是大写的。

=========================

更新

请看下面的执行输出：

create or replace procedure PROC_TEST 
    (input_sql varchar(16777216))
returns string not null
language javascript
as
$$
    snowflake.execute({sqlText:`BEGIN TRANSACTION;`});
    try {
        var QueryExec = INPUT_SQL + ";";
        var QueryResult = snowflake.execute({sqlText: QueryExec});
    }
    catch (err)  {
        result = 'Error: ' + err;
        snowflake.execute({sqlText:`ROLLBACK;`});
        return result;
    }
$$;
call PROC_TEST('insert into schema.proc_test_final1 select id, name, sal from schema.proc_test_src');

+-----------------------------------------------------------+
| PROC_TEST                                                 |
|-----------------------------------------------------------|
| Error: SQL compilation error:                             |
| Schema 'ERICLIN.SCHEMA' does not exist or not authorized. |
+-----------------------------------------------------------+

如果没有 ROLLBACK 调用，它将产生与您相同的错误：

create or replace procedure PROC_TEST 
    (input_sql varchar(16777216))
returns string not null
language javascript
as
$$
    snowflake.execute({sqlText:`BEGIN TRANSACTION;`});
    try {
        var QueryExec = INPUT_SQL + ";";
        var QueryResult = snowflake.execute({sqlText: QueryExec});
    }
    catch (err)  {
        result = 'Error: ' + err;
        return result;
    }
$$;
call PROC_TEST('insert into schema.proc_test_final1 select id, name, sal from schema.proc_test_src');

090232 (25000): Stored procedure execution error: Scoped transaction started in stored procedure is incomplete and it was rolled back.