Snowflake中如何实现类似PostgreSQL的RETURNING子句效果？

Snowflake目前已经原生支持RETURNING子句，和PostgreSQL的用法基本一致，实现插入后返回自增ID的需求非常简便。

原生RETURNING语法（推荐） #

你不需要改写原有PostgreSQL风格的SQL，直接在Snowflake中执行即可：

INSERT INTO EXAMPLE_TABLE(COL_2)
VALUES('ABCD')
RETURNING COL_1;

执行后语句会直接返回新插入行生成的自增COL_1值3，表内也会正确新增COL_1=3、COL_2='ABCD'的记录，和你预期的效果完全一致。
这个语法同样支持UPDATE、DELETE操作后返回变更行的指定字段，用法和PostgreSQL没有明显差异。

旧版本兼容方案 #

如果你使用的Snowflake账号版本较老，暂不支持原生RETURNING子句，可以通过反查的方式实现同等效果：

• 第一步：先执行插入语句，执行完成后不要运行其他SQL，避免会话级的最后查询ID被覆盖

INSERT INTO EXAMPLE_TABLE(COL_2)
VALUES('ABCD');

• 第二步：用插入时的唯一业务字段反查刚写入的记录，拿到自增ID：

SELECT COL_1
FROM EXAMPLE_TABLE
WHERE COL_2 = 'ABCD'
-- 为了避免同COL_2值重复导致查错，按自增ID倒序取最新的一条即可
ORDER BY COL_1 DESC
LIMIT 1;

注意：不要直接用SELECT MAX(COL_1) FROM EXAMPLE_TABLE拿最新ID，并发插入场景下这个写法会拿到其他会话写入的ID，导致数据错误。
如果是批量插入场景，你可以提前为每一行插入数据生成唯一的批次标识，插入后按批次标识反查就能拿到本次插入所有行生成的自增ID；如果主键是自定义序列而非默认IDENTITY字段，也可以提前调用序列的NEXTVAL生成所有主键值，插入时直接写入主键字段，提前拿到所有ID不需要反查。

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