使用 JDBC 插入数据并获取自动生成的主键（如 MySQL 的 AUTO_INCREMENT 或 Oracle 的序列）的完整示例代码，包含详细注释

以下是使用 JDBC 插入数据并获取自动生成的主键（如 MySQL 的 AUTO_INCREMENT 或 Oracle 的序列） 的完整示例代码，包含详细注释：

import java.sql.*;public class GeneratedKeysExample {// 数据库连接参数private static final String URL = "jdbc:mysql://localhost:3306/mydatabase";private static final String USER = "root";private static final String PASSWORD = "your_password";public static void main(String[] args) {Connection conn = null;try {// 1. 加载数据库驱动Class.forName("com.mysql.cj.jdbc.Driver");// 2. 获取数据库连接conn = DriverManager.getConnection(URL, USER, PASSWORD);// 3. 定义插入语句（包含自增主键）String sql = "INSERT INTO users (name, email) VALUES (?, ?)";// 4. 创建 PreparedStatement 并启用返回生成的键try (PreparedStatement pstmt = conn.prepareStatement(sql, Statement.RETURN_GENERATED_KEYS // 关键：启用生成键的返回)) {// 5. 设置参数pstmt.setString(1, "Alice");pstmt.setString(2, "alice@example.com");// 6. 执行插入操作int affectedRows = pstmt.executeUpdate();System.out.println("插入成功，影响行数：" + affectedRows);// 7. 获取生成的主键try (ResultSet generatedKeys = pstmt.getGeneratedKeys()) {if (generatedKeys.next()) {long userId = generatedKeys.getLong(1); // 获取第一列（主键）System.out.println("生成的用户ID：" + userId);} else {throw new SQLException("创建用户时未获取到主键！");}}}} catch (ClassNotFoundException e) {System.out.println("JDBC驱动未找到！");e.printStackTrace();} catch (SQLException e) {System.out.println("数据库操作失败！");e.printStackTrace();} finally {// 8. 关闭连接try {if (conn != null) conn.close();} catch (SQLException e) {e.printStackTrace();}}}
}

代码说明

1. 核心步骤

启用生成键返回：
在 prepareStatement() 中传入 Statement.RETURN_GENERATED_KEYS，告知 JDBC 需要返回生成的键。
执行插入并获取键：
- executeUpdate() 执行插入操作。
- getGeneratedKeys() 返回包含生成键的 ResultSet。
处理结果集：
- generatedKeys.next() 移动到第一条记录（假设每次插入一条）。
- getLong(1) 获取第一列（主键列）的值。

2. 数据库表结构（MySQL 示例）

CREATE TABLE users (id BIGINT AUTO_INCREMENT PRIMARY KEY, -- 自增主键name VARCHAR(50),email VARCHAR(100)
);

关键注意事项

数据库支持：
- MySQL：使用 AUTO_INCREMENT 主键。
- Oracle：需指定 RETURNING 子句或使用序列（代码需调整）。
- PostgreSQL：支持 RETURNING 子句，但需调整 SQL 语法。
多行插入：
如果插入多行，getGeneratedKeys() 返回所有生成的键，需遍历 ResultSet。
列索引：
generatedKeys.getLong(1) 中的 1 表示主键是结果集的第一列。若表有多个生成列，需根据列顺序调整。

扩展场景：Oracle 示例

Oracle 需要使用 RETURNING 子句和序列：

// Oracle 示例代码片段
String sql = "INSERT INTO users (id, name, email) " +"VALUES (user_seq.NEXTVAL, ?, ?) " +"RETURNING id INTO ?";
try (PreparedStatement pstmt = conn.prepareStatement(sql)) {pstmt.setString(1, "Alice");pstmt.setString(2, "alice@example.com");pstmt.registerOutParameter(3, OracleTypes.NUMBER); // 注册输出参数int rows = pstmt.executeUpdate();long userId = pstmt.getLong(3); // 通过输出参数获取主键
}

常见问题

Q1：为什么获取不到生成的键？

原因：未在 prepareStatement() 中启用 RETURN_GENERATED_KEYS。
解决：确保代码中包含 Statement.RETURN_GENERATED_KEYS。

Q2：如何处理批量插入的生成键？

// 批量插入示例
pstmt.addBatch(); // 添加多条数据
pstmt.executeBatch();// 获取所有生成的键
try (ResultSet keys = pstmt.getGeneratedKeys()) {while (keys.next()) {long id = keys.getLong(1);// 处理每个生成的键}
}