使用JOOQ通用插入操作时如何返回记录ID？

当然可以实现！针对你的MSSQL场景，结合JOOQ 3.9.5的特性，我们可以通过修改通用插入API来返回自动生成的主键ID——既然你所有表的主键都统一命名为id，实现起来非常直接。

修改后的通用插入方法 #

我们需要调整方法的返回类型，并利用JOOQ的returning()方法来指定返回主键字段，最后提取生成的ID值：

Long insert(@NonNull final Table<?> table, @NonNull final Collection<Field<?>> columns, @NonNull final Collection<Object> values) {
    // 获取表中的id字段，指定类型为Long（如果你的主键是Integer，换成Integer.class即可）
    Field<Long> idField = table.field("id", Long.class);
    
    return dslContext.insertInto(table)
              .columns(columns)
              .values(values)
              .returning(idField) // 指定返回生成的id字段
              .fetchOne() // 获取插入后的结果记录
              .getValue(idField); // 提取id值
}

关键细节说明 #

• returning()方法：针对MSSQL，JOOQ会自动生成包含OUTPUT子句的SQL，这是MSSQL获取自增主键的标准方式，JOOQ 3.9.5完全支持这个特性。
• 通用表的字段获取：因为你的方法接收的是Table<?>类型的通用表，我们通过table.field("id", Long.class)来动态获取主键字段，同时指定类型以避免类型转换问题。
• 结果提取：fetchOne()会返回插入操作生成的单条结果记录（因为你是单条插入），再通过getValue()直接提取主键值。

注意事项 #

• 如果你的主键类型是Integer而非Long，只需要把代码中的Long替换成Integer即可。
• 若插入操作失败（比如约束冲突），fetchOne()会返回null，此时调用getValue()会抛出NullPointerException，你可以根据业务需求添加异常处理逻辑，比如：

  Record record = dslContext.insertInto(table)
              .columns(columns)
              .values(values)
              .returning(idField)
              .fetchOne();
              
  if (record == null) {
      // 处理插入失败的情况，比如抛出自定义异常或返回null
      throw new RuntimeException("插入记录失败，未生成主键ID");
  }
  
  return record.getValue(idField);
  

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