如何在 Java 中安全获取 SQL 插入语句生成的主键 ID

在 Java 应用开发中，向数据库插入数据时，如果表的主键是数据库自动生成的，比如 MySQL 的自增主键、PostgreSQL 的序列主键，我们往往需要在插入完成后拿到这个生成的主键 ID，用于后续关联数据的插入或者业务状态更新。如果采用查询表中最大 ID 的方式获取，在多线程并发插入的场景下，很容易拿到其他线程插入的数据 ID，导致业务数据错乱，因此必须采用安全的获取方式。

标准安全获取方式：使用 getGeneratedKeys

JDBC 规范中提供了标准的获取自动生成主键的方式，核心是通过 PreparedStatement 的 getGeneratedKeys 方法实现，这种方式由数据库驱动层保证线程安全，不会出现并发问题。

单条插入获取主键示例

以 MySQL 数据库为例，插入用户数据后获取自增主键的完整代码如下：

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;

public class GetPrimaryKeyDemo {
    public static void main(String[] args) {
        // 数据库连接信息，实际开发中建议放在配置文件中
        String url = "jdbc:mysql://127.0.0.1:3306/test_db?useSSL=false&serverTimezone=UTC";
        String username = "root";
        String password = "123456";
        Connection conn = null;
        PreparedStatement pstmt = null;
        ResultSet rs = null;
        try {
            // 1. 获取数据库连接
            conn = DriverManager.getConnection(url, username, password);
            // 2. 编写插入SQL，不需要指定自增主键字段
            String sql = "INSERT INTO user (name, age) VALUES (?, ?)";
            // 3. 创建PreparedStatement时，指定需要返回自动生成的主键，参数为Statement.RETURN_GENERATED_KEYS
            pstmt = conn.prepareStatement(sql, PreparedStatement.RETURN_GENERATED_KEYS);
            // 4. 设置参数
            pstmt.setString(1, "张三");
            pstmt.setInt(2, 25);
            // 5. 执行插入操作
            int affectedRows = pstmt.executeUpdate();
            if (affectedRows > 0) {
                // 6. 获取自动生成的主键结果集
                rs = pstmt.getGeneratedKeys();
                if (rs.next()) {
                    // 获取主键ID，列索引从1开始
                    long userId = rs.getLong(1);
                    System.out.println("插入成功，生成的主键ID为：" + userId);
                }
            }
        } catch (SQLException e) {
            e.printStackTrace();
        } finally {
            // 7. 关闭资源，注意关闭顺序
            try {
                if (rs != null) rs.close();
                if (pstmt != null) pstmt.close();
                if (conn != null) conn.close();
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }
    }
}

关键步骤说明

• 创建 PreparedStatement 时，第二个参数必须传入 PreparedStatement.RETURN_GENERATED_KEYS，告诉数据库驱动需要返回自动生成的主键，否则 getGeneratedKeys 会返回空结果集。
• 执行插入后，通过 pstmt.getGeneratedKeys() 获取主键结果集，该方法返回的是一个 ResultSet 对象，和查询返回的结果集处理方式一致。
• 如果插入操作影响了多行（比如批量插入），结果集中会包含多个主键，需要遍历获取。

批量插入场景下的主键获取

如果需要批量插入多条数据，同样可以使用 getGeneratedKeys 方法一次性获取所有生成的主键，示例代码如下：

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;

public class BatchInsertDemo {
    public static void main(String[] args) {
        String url = "jdbc:mysql://127.0.0.1:3306/test_db?useSSL=false&serverTimezone=UTC";
        String username = "root";
        String password = "123456";
        Connection conn = null;
        PreparedStatement pstmt = null;
        ResultSet rs = null;
        try {
            conn = DriverManager.getConnection(url, username, password);
            String sql = "INSERT INTO user (name, age) VALUES (?, ?)";
            pstmt = conn.prepareStatement(sql, PreparedStatement.RETURN_GENERATED_KEYS);
            // 添加批量参数
            pstmt.setString(1, "李四");
            pstmt.setInt(2, 22);
            pstmt.addBatch();
            pstmt.setString(1, "王五");
            pstmt.setInt(2, 28);
            pstmt.addBatch();
            pstmt.setString(1, "赵六");
            pstmt.setInt(2, 30);
            pstmt.addBatch();
            // 执行批量插入
            pstmt.executeBatch();
            // 获取所有生成的主键
            rs = pstmt.getGeneratedKeys();
            List<Long> primaryKeys = new ArrayList<>();
            while (rs.next()) {
                primaryKeys.add(rs.getLong(1));
            }
            System.out.println("批量插入生成的主键列表：" + primaryKeys);
        } catch (SQLException e) {
            e.printStackTrace();
        } finally {
            try {
                if (rs != null) rs.close();
                if (pstmt != null) pstmt.close();
                if (conn != null) conn.close();
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }
    }
}

不同数据库的适配注意事项

不同数据库的自动主键生成机制不同，使用 getGeneratedKeys 时需要注意以下适配问题：

常见错误及避坑指南

• 不要使用 SELECT MAX(id) FROM 表名 的方式获取主键，并发场景下会出现主键错乱，这是最典型的错误用法。
• 创建 PreparedStatement 时不要忘记传入 PreparedStatement.RETURN_GENERATED_KEYS 参数，否则无法获取主键。
• 如果表有多个自动生成的列，getGeneratedKeys 会返回所有自动生成的列值，需要根据列顺序或者列名准确获取目标主键。
• 获取主键后要及时处理，避免结果集未关闭导致的资源泄漏问题。

注意：如果使用的 ORM 框架比如 MyBatis、Hibernate，框架已经封装了主键获取的逻辑，不需要手动调用 getGeneratedKeys 方法，只需要按照框架的配置方式指定主键生成策略即可，比如 MyBatis 中可以在 insert 标签上配置 useGeneratedKeys 和 keyProperty 属性自动回填主键。

JavaSQL主键_IDPreparedStatement数据库操作修改时间：2026-07-05 23:18:35