MYSQL获取自增主键【4种方法】

时间:2022-03-08 20:57:55

通常我们在应用中对mysql执行了insert操作后,需要获取插入记录的自增主键。本文将介绍java环境下的4种方法获取insert后的记录主键auto_increment的值:

  1. 通过JDBC2.0提供的insertRow()方式
  2. 通过JDBC3.0提供的getGeneratedKeys()方式
  3. 通过SQL select LAST_INSERT_ID()函数
  4. 通过SQL @@IDENTITY 变量

1. 通过JDBC2.0提供的insertRow()方式

自jdbc2.0以来,可以通过下面的方式执行。

[java] view plain copy
  1. Statement stmt = null;  
  2. ResultSet rs = null;  
  3. try {  
  4.     stmt = conn.createStatement(java.sql.ResultSet.TYPE_FORWARD_ONLY,  // 创建Statement  
  5.                                 java.sql.ResultSet.CONCUR_UPDATABLE);  
  6.     stmt.executeUpdate("DROP TABLE IF EXISTS autoIncTutorial");  
  7.     stmt.executeUpdate(                                                // 创建demo表  
  8.             "CREATE TABLE autoIncTutorial ("  
  9.             + "priKey INT NOT NULL AUTO_INCREMENT, "  
  10.             + "dataField VARCHAR(64), PRIMARY KEY (priKey))");  
  11.     rs = stmt.executeQuery("SELECT priKey, dataField "                 // 检索数据  
  12.        + "FROM autoIncTutorial");  
  13.     rs.moveToInsertRow();                                              // 移动游标到待插入行(未创建的伪记录)  
  14.     rs.updateString("dataField""AUTO INCREMENT here?");              // 修改内容  
  15.     rs.insertRow();                                                    // 插入记录  
  16.     rs.last();                                                         // 移动游标到最后一行  
  17.     int autoIncKeyFromRS = rs.getInt("priKey");                        // 获取刚插入记录的主键preKey  
  18.     rs.close();  
  19.     rs = null;  
  20.     System.out.println("Key returned for inserted row: "  
  21.         + autoIncKeyFromRS);  
  22. }  finally {  
  23.     // rs,stmt的close()清理  
  24. }  
优点:早期较为通用的做法

缺点:需要操作ResultSet的游标,代码冗长。

2. 通过JDBC3.0提供的getGeneratedKeys()方式

[java] view plain copy
  1. Statement stmt = null;  
  2. ResultSet rs = null;  
  3. try {  
  4.     stmt = conn.createStatement(java.sql.ResultSet.TYPE_FORWARD_ONLY,  
  5.                                 java.sql.ResultSet.CONCUR_UPDATABLE);    
  6.     // ...  
  7.     // 省略若干行(如上例般创建demo表)  
  8.     // ...  
  9.     stmt.executeUpdate(  
  10.             "INSERT INTO autoIncTutorial (dataField) "  
  11.             + "values ('Can I Get the Auto Increment Field?')",  
  12.             Statement.RETURN_GENERATED_KEYS);                      // 向驱动指明需要自动获取generatedKeys!  
  13.     int autoIncKeyFromApi = -1;  
  14.     rs = stmt.getGeneratedKeys();                                  // 获取自增主键!  
  15.     if (rs.next()) {  
  16.         autoIncKeyFromApi = rs.getInt(1);  
  17.     }  else {  
  18.         // throw an exception from here  
  19.     }   
  20.     rs.close();  
  21.     rs = null;  
  22.     System.out.println("Key returned from getGeneratedKeys():"  
  23.         + autoIncKeyFromApi);  
  24. }  finally { ... }  
这种方式只需要2个步骤:1. 在executeUpdate时激活自动获取key; 2.调用Statement的getGeneratedKeys()接口
优点:
1. 操作方便,代码简洁 2. jdbc3.0的标准 3. 效率高,因为没有额外访问数据库
这里补充下, a.在jdbc3.0之前,每个jdbc driver的实现都有自己获取自增主键的接口。在mysql jdbc2.0的driver org.gjt.mm.mysql中,getGeneratedKeys()函数就实现在org.gjt.mm.mysql.jdbc2.Staement.getGeneratedKeys()中。这样直接引用的话,移植性会有很大影响。JDBC3.0通过标准的getGeneratedKeys很好的弥补了这点。 b.关于getGeneratedKeys(),官网还有更详细解释:OracleJdbcGuide

3. 通过SQL select LAST_INSERT_ID()

[java] view plain copy
  1. Statement stmt = null;  
  2. ResultSet rs = null;  
  3. try {  
  4.     stmt = conn.createStatement();  
  5.     // ...  
  6.     // 省略建表  
  7.     // ...  
  8.     stmt.executeUpdate(  
  9.             "INSERT INTO autoIncTutorial (dataField) "  
  10.             + "values ('Can I Get the Auto Increment Field?')");  
  11.     int autoIncKeyFromFunc = -1;  
  12.     rs = stmt.executeQuery("SELECT LAST_INSERT_ID()");             // 通过额外查询获取generatedKey  
  13.     if (rs.next()) {  
  14.         autoIncKeyFromFunc = rs.getInt(1);  
  15.     }  else {  
  16.         // throw an exception from here  
  17.     }   
  18.     rs.close();  
  19.     System.out.println("Key returned from " +  
  20.                        "'SELECT LAST_INSERT_ID()': " +  
  21.                        autoIncKeyFromFunc);  
  22. }  finally {...}  
这种方式没什么好说的,就是额外查询一次函数LAST_INSERT_ID().
优点:简单方便 缺点:相对JDBC3.0的getGeneratedKeys(),需要额外多一次数据库查询。
补充: 1. 这个函数,在mysql5.5手册的定义是:“returns a BIGINT (64-bit) value representing the first automatically generated value successfully inserted for an AUTO_INCREMENT column as a result of the most recently executed INSERT statement.”。文档点此 2. 这个函数,在connection维度上是“线程安全的”。就是说,每个mysql连接会有个独立保存LAST_INSERT_ID()的结果,并且只会被当前连接最近一次insert操作所更新。也就是2个连接同时执行insert语句时候,分别调用的LAST_INSERT_ID()不会相互覆盖。举个栗子:连接A插入表后LAST_INSERT_ID()返回100,连接B插入表后LAST_INSERT_ID()返回101,但是连接A重复执行LAST_INSERT_ID()的时候,始终返回100,而不是101。这个可以通过监控mysql连接数和执行结果来验证,这里不详述实验过程。 3. 在上面那点的基础上,如果在同一个连接的前提下同时执行insert,那可能2次操作的返回值会相互覆盖。因为LAST_INSERT_ID()的隔离程度是连接级别的。这点,getGeneratedKeys()是可以做的更好,因为getGeneratedKeys()是statement级别的。同个connection的多次statement,getGeneratedKeys()是不会被相互覆盖。

4. 通过SQL SELECT @@IDENTITY

这个方式和LAST_INSERT_ID()效果是一样的。官网文档如此表述:“This variable is a synonym for the last_insert_id variable. It exists for compatibility with other database systems. You can read its value with SELECT @@identity, and set it using SET identity.” 文档点此

重要补充: 无论是SELECT LAST_INSERT_ID()还是SELECT @@IDENTITY,对于一条insert语句插入多条记录,永远只会返回第一条插入记录的generatedKey.如: [java] view plain copy
  1. INSERT INTO t VALUES  
  2.     -> (NULL, 'Mary'), (NULL, 'Jane'), (NULL, 'Lisa');  
LAST_INSERT_ID(), @@IDENTITY都只会返回'Mary'所在的那条记录的generatedKey

小结

所以,最好还是通过JDBC3 提供的getGeneratedKeys()函数来获取insert记录的主键。不但简单,而且效率高。
在mybatis中,就有相关设置: [java] view plain copy
  1. <insert id="save" parameterType="MappedObject" useGeneratedKeys="true" keyProperty="id">  
  2. </insert>  
可参考mybatis文档:sqlmap-xml

参考资料

http://dev.mysql.com/doc/connector-j/en/connector-j-usagenotes-last-insert-id.html
http://www.onjava.com/pub/a/onjava/excerpt/javaentnut_2/index2.html?page=3
http://docs.oracle.com/javase/1.4.2/docs/guide/jdbc/getstart/statement.html#1000569
http://dev.mysql.com/doc/refman/5.5/en/information-functions.html#function_last-insert-id
http://nic.jlu.edu.cn/newcourse/dbxz/jdbc20.pdf
http://www.enidc.com/help/newscontent/110507