JDBC 返回主键

转载至:https://www.liyongzhen.com/

上一节课里我们学习通过PreparedStatement对象执行带参数的查询SQL和修改SQL。

这节课我们学习使用 PreparedStatement对象执行插入SQL，并且返回主键。

要返回主键，有一个前提，就是数据库表必须有主键自增。

下面的代码演示了执行插入SQL并返回主键的用法。

String sql = "INSERT INTO products(`ProductName`,`SupplierID`,`CategoryID`,`Unit`,`Price`)VALUES(?,?,?,?,?);";
PreparedStatement statement = 
   connection.prepareStatement(sql,Statement.RETURN_GENERATED_KEYS); //加入参数Statement.RETURN_GENERATED_KEYS
statement.setString(1, "milk"); //设置参数……
statement.setLong(2, 16);
statement.setLong(3, 1);
statement.setString(4, "500ml");
statement.setBigDecimal(5, new BigDecimal(5.6));
int value = statement.executeUpdate();
 
System.out.println("受影响的记录数 "+value);
 
int id = 0;
ResultSet rs = statement.getGeneratedKeys();  //得到生成的主键，主键放在ResultSet对象里
if(rs.next()){ //提取主键值
    id = rs.getInt(1);
}
 
System.out.println("主键 "+id);

关键技术：

在创建PreparedStatement对象时，指定一个常量Statement.RETURN_GENERATED_KEYS，表示有返回主键的能力。
执行SQL语句时，仍然调用executeUpdate()方法。另外要调用getGeneratedKeys()方法，返回ResultSet对象。 ResultSet对象是结果集，存储查询结果，主键就保存在这里。
第三步，从ResultSet对象里提取主键。

完整代码：

import java.math.BigDecimal;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
 
public class Insert {
 
    public static void main(String[] args) {
        String url = "jdbc:mysql://127.0.0.1:3306/northwind";
        String user = "northwind";
        String password = "northwind";
 
        try (Connection connection = DriverManager.getConnection(url, user, password)) {
 
            String sql = "INSERT INTO products(`ProductName`,`SupplierID`,`CategoryID`,`Unit`,`Price`)VALUES(?,?,?,?,?);";
            PreparedStatement statement = connection.prepareStatement(sql,Statement.RETURN_GENERATED_KEYS);
            statement.setString(1, "milk");
            statement.setLong(2, 16);
            statement.setLong(3, 1);
            statement.setString(4, "500ml");
            statement.setBigDecimal(5, new BigDecimal(5.6));
            int value = statement.executeUpdate();
             
            System.out.println("受影响的记录数 "+value);
             
            int id = 0;
            ResultSet rs = statement.getGeneratedKeys();
            if(rs.next()){
                id = rs.getInt(1);
            }
             
            System.out.println("主键 "+id);
 
            rs.close();
            statement.close(); // 关闭statement，释资源
 
        } catch (SQLException e) {
            e.printStackTrace();
        }
 
    }
 
}