Azure SQL数据仓库中无法通过JDBC获取生成键的解决方案咨询

确实，Azure SQL Data Warehouse（现在也叫Synapse SQL池）对自增列的支持有不少限制——SCOPE_IDENTITY()、IDENT_CURRENT()这类函数以及OUTPUT子句都不被支持，这直接导致JDBC标准的getGeneratedKeys()方法彻底失效。我这里整理了几个可行的替代方案，你可以根据自己的场景选择：

方案1：用SEQUENCE替代IDENTITY列（推荐，可控性强） #

如果你的表结构还能调整，把IDENTITY列换成SEQUENCE对象是最稳妥的办法。这样你可以在插入前就拿到要使用的ID，完全避开后续获取的麻烦：

1. 创建SEQUENCE对象 #

CREATE SEQUENCE seq_t_identity
    START WITH 1
    INCREMENT BY 1;

2. 修改表结构替换IDENTITY列 #

-- 先删除原IDENTITY列
ALTER TABLE t_identity
DROP COLUMN id;

-- 添加新列，默认值用SEQUENCE的下一个值
ALTER TABLE t_identity
ADD id INTEGER NOT NULL DEFAULT NEXT VALUE FOR seq_t_identity;

3. 调整JDBC代码逻辑 #

先获取序列值，再显式插入到表中，这样你直接就拿到了生成的ID：

try (Connection c = DriverManager.getConnection(url, properties);
     Statement s = c.createStatement()) {
    // 获取下一个序列值
    ResultSet rs = s.executeQuery("SELECT NEXT VALUE FOR seq_t_identity");
    rs.next();
    Long generatedId = rs.getLong(1);
    
    // 显式插入ID和业务字段
    s.executeUpdate(String.format("INSERT INTO t_identity (id, val) VALUES (%d, 1)", generatedId));
    
    System.out.println("生成的ID：" + generatedId);
}

这个方案的优势是逻辑清晰、可控，完全避开SQL DW的限制；缺点是需要修改表结构，如果表已有大量数据，需要先迁移数据。

方案2：插入后通过唯一业务字段查询ID（无需改表） #

如果无法修改表结构，那只能依赖插入时的唯一业务字段（比如假设val在当前场景下是唯一的，或者你可以组合多个字段形成唯一条件），插入后执行查询获取ID：

try (Connection c = DriverManager.getConnection(url, properties);
     Statement s = c.createStatement()) {
    int val = 1;
    // 执行插入
    s.executeUpdate(String.format("INSERT INTO t_identity (val) VALUES (%d)", val));
    
    // 通过唯一业务字段查询生成的ID
    ResultSet rs = s.executeQuery(String.format("SELECT id FROM t_identity WHERE val = %d", val));
    if (rs.next()) {
        System.out.println("生成的ID：" + rs.getObject(1));
    }
}

⚠️ 注意：这个方案有风险，如果查询条件不能唯一定位到刚插入的行，会返回错误结果。如果没有天然的唯一业务字段，可以临时插入一个UUID作为标识，查询到ID后再删除或更新这个标识字段。

方案3：批量插入场景下的范围查询 #

如果是批量插入数据，可以先获取当前表的最大ID，插入后查询所有大于该值的ID：

try (Connection c = DriverManager.getConnection(url, properties);
     Statement s = c.createStatement()) {
    // 获取当前表的最大ID（如果表为空，返回0）
    ResultSet maxRs = s.executeQuery("SELECT ISNULL(MAX(id), 0) FROM t_identity");
    maxRs.next();
    long currentMaxId = maxRs.getLong(1);
    
    // 执行批量插入
    s.executeUpdate("INSERT INTO t_identity (val) VALUES (1), (2), (3)");
    
    // 查询本次插入生成的所有ID
    ResultSet rs = s.executeQuery(String.format("SELECT id FROM t_identity WHERE id > %d", currentMaxId));
    while (rs.next()) {
        System.out.println("生成的ID：" + rs.getObject(1));
    }
}

这个方案适合批量场景，但要注意并发问题——如果有其他会话同时插入数据，可能会拿到不属于当前批次的ID。建议在事务中执行这个逻辑，利用事务的隔离性避免冲突。

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