获取在Snowflake Datawarehouse中插入的行的标识

Question

如果我有一个带有自动递增ID列的表，我希望能够在该表中插入一行，并获取刚刚创建的行的ID。我知道，通常，StackOverflow问题需要某种经过尝试或研究的代码，但是我不确定从Snowflake入手。我已经翻阅了他们的文档，却对此一无所获。

到目前为止，我能做的最好的事情是tryresult_scan()和last_query_id()，但是这些操作并没有为我提供有关所插入行的任何相关信息，只是向我确认已插入行。

我相信我要的是与MS SQL Server的SCOPE_IDENTITY()功能相似的功能。

是否有适用于MS SQL Server的Snowflake等效功能SCOPE_IDENTITY()？

编辑：为了在这里有代码的缘故：

CREATE TABLE my_db..my_table
(
    ROWID INT IDENTITY(1,1),
    some_number INT,
    a_time TIMESTAMP_LTZ(9),
    b_time TIMESTAMP_LTZ(9),
    more_data VARCHAR(10)
);
INSERT INTO my_db..my_table
(
    some_number,
    a_time,
    more_data
)
VALUES
(1, my_time_value, some_data);

我想为ROWID我刚刚插入的这一行添加自动增量。

Answer 1

注意：在某些非常罕见的情况下，下面的答案可能不是100％正确，请参见下面的“更新”部分

雪花无法提供SCOPE_IDENTITY今天的效果。

但是，您可以在执行给定语句后立即利用Snowflake的时间旅行来检索列的最大值。

这是一个例子：

create or replace table x(rid int identity, num int);
insert into x(num) values(7);
insert into x(num) values(9);
-- you can insert rows in a separate transaction now to test it
select max(rid) from x AT(statement=>last_query_id());
----------+
 MAX(RID) |
----------+
 2        |
----------+

last_query_id()如果您以后想访问它，也可以将其保存到变量中，例如

insert into x(num) values(5);
set qid = last_query_id();
...
select max(rid) from x AT(statement=>$qid);

注意-通常这是正确的，但是如果用户（例如）rid手动将较大的值插入其中，则可能会影响此查询的结果。

更新

注意，我意识到上面的代码很少会产生错误的答案。

由于类似的分布式系统中查询的各个阶段的执行顺序Snowflake可能是不确定的，并且Snowflake允许html" target="_blank">并发INSERT语句，因此可能会发生以下情况

• 两个查询，Q1和Q2，做一个简单的单行INSERT，大约在同一时间开始
• Q1 开始，有点领先
• Q2 开始
• Q1创建一个行同值1从IDENTITY柱
• Q2创建一个行同值2从IDENTITY柱
• Q2领先Q1- 这是关键部分
• Q2 提交，被标记为已完成 T2
• Q1 提交，被标记为已完成 T1

请注意，此T1时间晚于T2。现在，当我们尝试执行操作时SELECT ... AT(statement=>Q1)，我们将看到状态为as
T1，包括之前语句的所有更改，因此包括2from的值Q2。这不是我们想要的。

解决这个问题的方法可能是向unique identifier每个INSERT对象添加一个（例如，来自单独的SEQUENCE对象），然后使用MAX。

对不起。分布式交易很困难:)