Snowflake存储过程事务Begin/End块异常：提交后进入错误块

我在Snowflake中编写了名为fullvsdeltaload的SQL存储过程，用于实现全量/增量数据加载。当以reload_val=1（增量加载）通过Snowflake任务执行该存储过程时，Merge语句执行完成后已执行COMMIT，但仍进入异常块触发STATEMENT_ERROR。现附上代码，询问是否需要添加额外的BEGIN TRANSACTION语句，以确保事务正常提交并退出存储过程。

存储过程代码如下：

CREATE or REPLACE PROCEDURE fullvsdeltaload(reload number)
RETURNS string
LANGUAGE sql 
EXECUTE AS OWNER
AS
$$

    BEGIN
        
        LET reload_val INT := :reload;

        if (reload_val = 0)   -- do a full load
        THEN
        BEGIN
        create or replace table drop_and_recreate_main_table  
            (
            ..
            ..
            )
            AS
            WITH 
            ..   ;

        COMMIT; 

        RETURN 'Success fully reloading the main table' ;
        
        ELSEIF  (reload_val = 1) THEN  -- do a delta load
        
        BEGIN
        create or replace table delta_table_with_changed_records 
            (
                ..
            )
            AS
            WITH 
            .. ;
        
        MERGE INTO main_table 
        USING delta_table 
            ON ..
            AND ..
                WHEN MATCHED THEN 
                    UPDATE SET ..
                        WHEN NOT MATCHED THEN 
                            INSERT ..
                            VALUES ..;
    
        COMMIT;
            
        RETURN 'Success merging changed records into the main table' ;     
        END IF;

       EXCEPTION

        WHEN STATEMENT_ERROR THEN
        ROLLBACK; 
        
        let errMsg string:= OBJECT_CONSTRUCT('SUCCESS', FALSE,
                                'ERROR TYPE', 'STATEMENT_ERROR',
                                'SQLCODE', SQLCODE,
                                'SQLERRM', SQLERRM,
                                'SQLSTATE', SQLSTATE);  
        RETURN  errMsg;
                    
                                
        WHEN EXPRESSION_ERROR THEN
        ROLLBACK;
        
        let errMsg string:= OBJECT_CONSTRUCT('SUCCESS', FALSE,
                                'ERROR TYPE', 'EXPRESSION_ERROR',
                                'SQLCODE', SQLCODE,
                                'SQLERRM', SQLERRM,
                                'SQLSTATE', SQLSTATE);     
        RETURN  errMsg;    
                                
        WHEN OTHER THEN
        ROLLBACK;
        
        let errMsg string:= OBJECT_CONSTRUCT('SUCCESS', FALSE,
                                'ERROR TYPE', 'OTHER',
                                'SQLCODE', SQLCODE,
                                'SQLERRM', SQLERRM,
                                'SQLSTATE', SQLSTATE);  
             
        RETURN  errMsg;      
    END;
$$
; 

1. 不需要额外添加BEGIN TRANSACTION #

Snowflake的SQL存储过程默认采用自动事务模式：每个DML/DDL语句执行时会自动开启事务，除非你显式使用BEGIN TRANSACTION手动控制事务边界。你的代码中已经显式调用了COMMIT，再加BEGIN TRANSACTION反而会导致事务嵌套，引发不必要的异常。

2. 现有代码的问题分析 #

你遇到的异常触发问题，根源不在事务开启，而是代码结构的几个问题：

• 内部冗余的BEGIN块：在IF和ELSEIF分支里额外嵌套了BEGIN，但没有对应的END，这会导致语法解析错误，触发STATEMENT_ERROR。
• COMMIT后的执行路径：即使MERGE和COMMIT成功，后续的代码逻辑可能因为语法问题（比如未闭合的块）进入异常分支。
• 事务回滚的冗余性：COMMIT执行成功后，事务已经结束，此时再触发ROLLBACK会报错，反而可能成为异常来源。

3. 代码修正方案 #

针对上述问题，调整代码结构如下：

CREATE or REPLACE PROCEDURE fullvsdeltaload(reload number)
RETURNS string
LANGUAGE sql 
EXECUTE AS OWNER
AS
$$
BEGIN
    LET reload_val INT := :reload;

    IF (reload_val = 0) THEN  -- 全量加载
        CREATE OR REPLACE TABLE drop_and_recreate_main_table (
            -- 表结构定义
        ) AS
        WITH 
            -- CTE逻辑
        SELECT ...;

        RETURN 'Success fully reloading the main table';
    ELSEIF (reload_val = 1) THEN  -- 增量加载
        CREATE OR REPLACE TABLE delta_table_with_changed_records (
            -- 表结构定义
        ) AS
        WITH 
            -- CTE逻辑
        SELECT ...;
    
        MERGE INTO main_table 
        USING delta_table_with_changed_records 
            ON -- 匹配条件
            AND -- 附加条件
        WHEN MATCHED THEN 
            UPDATE SET -- 更新字段
        WHEN NOT MATCHED THEN 
            INSERT -- 插入字段
            VALUES -- 对应值;

        RETURN 'Success merging changed records into the main table';     
    END IF;

EXCEPTION
    WHEN STATEMENT_ERROR THEN
        -- 仅当事务未提交时回滚
        IF (CURRENT_TRANSACTION() IS NOT NULL) THEN
            ROLLBACK;
        END IF;
        RETURN OBJECT_CONSTRUCT(
            'SUCCESS', FALSE,
            'ERROR TYPE', 'STATEMENT_ERROR',
            'SQLCODE', SQLCODE,
            'SQLERRM', SQLERRM,
            'SQLSTATE', SQLSTATE
        );
                    
    WHEN EXPRESSION_ERROR THEN
        IF (CURRENT_TRANSACTION() IS NOT NULL) THEN
            ROLLBACK;
        END IF;
        RETURN OBJECT_CONSTRUCT(
            'SUCCESS', FALSE,
            'ERROR TYPE', 'EXPRESSION_ERROR',
            'SQLCODE', SQLCODE,
            'SQLERRM', SQLERRM,
            'SQLSTATE', SQLSTATE
        );    
                                
    WHEN OTHER THEN
        IF (CURRENT_TRANSACTION() IS NOT NULL) THEN
            ROLLBACK;
        END IF;
        RETURN OBJECT_CONSTRUCT(
            'SUCCESS', FALSE,
            'ERROR TYPE', 'OTHER',
            'SQLCODE', SQLCODE,
            'SQLERRM', SQLERRM,
            'SQLSTATE', SQLSTATE
        );      
END;
$$
;

关键调整点 #

• 移除分支内部的冗余BEGIN块，确保IF/ELSEIF/END IF结构闭合。
• 去掉显式的COMMIT：Snowflake中，CREATE OR REPLACE TABLE是DDL语句，会自动提交事务；MERGE作为DML语句，在存储过程中如果没有显式事务控制，执行完成后会自动提交（除非后续有其他语句）。
• 异常块中增加CURRENT_TRANSACTION() IS NOT NULL判断，避免对已提交的事务执行ROLLBACK。

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