获取插入到雪花数据仓库中的行的标识

Question

如果我有一个具有自动递增ID列的表，我希望能够在该表中插入一行，并获得我刚刚创建的行的ID。我知道，一般来说，StackOverflow问题需要一些尝试过的代码或研究成果，但我不确定从雪花开始。我翻了他们的文件，什么也没找到。

编辑：为了这里有代码：

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%正确，请参见下面的更新部分

Snowflake现在没有提供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        |
----------+

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

注意，我意识到上面的代码很少会产生不正确的答案。

由于在像Snowflake这样的分布式系统中查询各个阶段的执行顺序可能是不确定的，而且Snowflake允许并发INSERT语句，因此可能会发生以下情况

• 两个查询，Q1和Q2，执行简单的单行插入，大致同时开始
• Q1开始，有点超前
• Q2开始
• Q1从Identity列
创建值 1的行
• Q2从标识列
创建值为 2的行
• Q2领先于Q1-这是关键部分
• Q2提交，在T2
时标记为完成
• Q1提交，在T1
时标记为完成

方法可以是向每个insert添加唯一标识符（例如，从一个单独的序列对象），然后使用max。

对不起。分布式事务很难：）