如何编写Snowflake存储过程：左联查询、插入数据并返回结果集

核心结论 #

可以在返回数据前用查询结果完成插入操作——通过先暂存查询结果，执行插入后再返回该结果集，既保证数据同步，又能返回所需数据。

具体代码实现 #

假设Table_A的键字段为a_id，Table_B通过a_id关联去重，存储过程接受参数N指定返回行数：

CREATE OR REPLACE PROCEDURE get_and_sync_data(N INT)
RETURNS TABLE(/* 替换为实际需要返回的字段，例如：a_id INT, a_name VARCHAR, b_desc VARCHAR... */)
LANGUAGE SQL
AS
$$
DECLARE
    temp_result RESULTSET;
BEGIN
    -- 1. 查询未同步到B的A数据（左连+过滤），暂存结果
    temp_result := (
        SELECT *
        FROM Table_A a
        LEFT JOIN Table_B b ON a.a_id = b.a_id
        WHERE b.a_id IS NULL
        LIMIT :N
    );

    -- 2. 将暂存结果中的a_id插入Table B，完成同步
    INSERT INTO Table_B (a_id)
    SELECT a_id FROM TABLE(:temp_result);

    -- 3. 返回暂存的结果集
    RETURN TABLE(:temp_result);
END;
$$;

关键细节说明 #

• 数据一致性：用RESULTSET变量一次性捕获查询结果，避免两次查询导致数据不一致（比如两次查询期间有其他进程修改数据）。
• 去重保障：查询时通过WHERE b.a_id IS NULL过滤已同步数据，配合插入操作，确保下次调用不会返回相同行。建议给Table_B.a_id加唯一约束，防止重复插入报错。
• 并发场景优化：如果存在多进程调用的情况，用MERGE替代INSERT更安全：

  MERGE INTO Table_B b
  USING TABLE(:temp_result) t
  ON b.a_id = t.a_id
  WHEN NOT MATCHED THEN INSERT (a_id) VALUES (t.a_id);
  

• C#调用适配：通过Snowflake .NET驱动调用CALL get_and_sync_data(10)后，可直接获取返回的结果集，处理逻辑与SQL Server存储过程类似，确保驱动版本支持存储过程返回表类型结果。

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