JDBC中获取自增主键：使用getGeneratedKeys()方法

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

public class GetGeneratedKeysExample {

    private static final String DB_URL = "jdbc:postgresql://localhost:5432/your_database";
    private static final String USER = "your_username";
    private static final String PASS = "your_password";

    public static void main(String[] args) {
        Connection connection = null;
        PreparedStatement pstmt = null;
        ResultSet generatedKeys = null;

        try {
            // 1. 建立数据库连接
            connection = DriverManager.getConnection(DB_URL, USER, PASS);
            connection.setAutoCommit(false); // 开启事务

            // 2. 准备SQL插入语句。假设表名为 'the_table'，有一个自增主键 'id' 和一个普通列 'some_column'
            String sql = "INSERT INTO the_table(some_column) VALUES (?)";

            // 3. 创建PreparedStatement，并指示返回生成的键
            // 方式一：指定列名
            // pstmt = connection.prepareStatement(sql, new String[]{"id"});
            // 方式二：使用常量
            pstmt = connection.prepareStatement(sql, PreparedStatement.RETURN_GENERATED_KEYS);

            // 4. 设置参数
            pstmt.setInt(1, 100); // 插入一个整数值

            // 5. 执行插入操作
            int affectedRows = pstmt.executeUpdate();

            if (affectedRows > 0) {
                // 6. 获取生成的键
                generatedKeys = pstmt.getGeneratedKeys();
                if (generatedKeys.next()) {
                    // 7. 从ResultSet中获取ID。通常，第一个列就是生成的ID。
                    int newId = generatedKeys.getInt(1);
                    System.out.println("新插入记录的ID: " + newId);
                } else {
                    System.out.println("未获取到生成的ID。");
                }
            } else {
                System.out.println("插入操作失败，没有行受到影响。");
            }

            connection.commit(); // 提交事务

        } catch (SQLException e) {
            e.printStackTrace();
            if (connection != null) {
                try {
                    connection.rollback(); // 发生异常时回滚事务
                } catch (SQLException ex) {
                    ex.printStackTrace();
                }
            }
        } finally {
            // 8. 关闭资源
            try {
                if (generatedKeys != null) generatedKeys.close();
                if (pstmt != null) pstmt.close();
                if (connection != null) connection.close();
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }
    }
}

CREATE TABLE the_table (
    id SERIAL PRIMARY KEY,
    some_column INT NOT NULL
);

// ... (之前的连接和PreparedStatement准备代码类似) ...

// 假设是批量插入
pstmt.setInt(1, 101);
pstmt.addBatch();
pstmt.setInt(1, 102);
pstmt.addBatch();

int[] affectedRows = pstmt.executeBatch(); // 执行批量操作

// 获取生成的键
ResultSet generatedKeys = pstmt.getGeneratedKeys();
System.out.println("批量插入生成的所有ID:");
while (generatedKeys.next()) {
    int newId = generatedKeys.getInt(1);
    System.out.println("  - " + newId);
}

// ... (关闭资源) ...