Snowflake 中的异步查询执行：SQL 执行被取消

Question

我正在使用 Snowflake 数据库即服务来存储和处理我们的数据。由于处理大量数据，我想运行一个查询，获取查询ID并让它异步执行查询。系统的另一部分将通过使用该查询 ID 检查查询历史表来监视查询的状态。

我正在使用 Snowflake Python 连接器。

这是我迄今为止所拥有的示例：

from __future__ import print_function
import io, os, sys, time, datetime
modules_path = os.path.join(os.path.dirname(__file__), 'modules')
sys.path.append(modules_path)

import snowflake.connector

def async_query(data):
    connection = snowflake.connector.connect(
        user=data['user'],
        password=data['password'],
        account=data['account'],
        region=data['region'],
        database=data['database'],
        warehouse=data['warehouse'],
        schema=data['schema']
    )

    cursor = connection.cursor()
    cursor.execute(data['query'], _no_results=True)
    print(cursor.sfqid)

    return cursor.sfqid

这段代码似乎有效，即我正在获取查询 ID，但有一个问题 - SQL 查询失败，并出现错误“SQL 执行已取消”。在雪花中。如果我删除该_no_results=True参数，查询运行良好，但随后我必须等待它完成，这不是所需的行为。

有什么想法导致“SQL 执行取消”失败吗？

更多信息：我不想等待的原因是我正在 AWS Lambda 上运行代码，而 Lambda 的最长运行时间为 5 分钟。

Answer 1

如果未指定_no_results=True，则执行是同步的，因此应用程序必须等待查询完成。如果指定，查询将变为异步，因此应用程序将继续运行，但连接的析构函数最终将关闭会话，并且所有活动查询将被取消。看来这就是“SQL执行取消”的原因。

AWS lambda 将执行时间限制为 5 分钟，因此如果查询时间超过限制，它将无法工作。

Btw _no_results=True 是 SnowSQL 使用的内部参数，其行为将来可能会发生变化。