在实际操作中,我们常常有一些各种古怪的需求,比如现在这个。数据库的table中有一个自增的id字段,插入数据库后,要求返回刚刚插入那条数据的id。 在单线程情况下,这当然是一个很简单的问题。首先获得数据库的最大id,给将要插入的语句id赋值为MAX+1,然后插入即可。 Oracle也很容易实现,给这个table建立一个sequence,每次都向这个sequence获取下一个id,效率也是价格公道童叟无欺。 但是,给多线程下的mysql返回刚刚插入的id,这似乎有点小为难。一个立刻可以想到的方法是,给插入操作加个锁,插入后查询MAX id,然后解开锁。但是这个显然是不怎么讲究了。
网上翻了下,有很多种方法解决这个问题,不过最后选择了getGeneratedKeys这个方法。 getGeneratedKeys是Statement下的一个方法,说明如下:
Retrieves any auto-generated keys created as a result of executing this Statement object. If this Statement object did not generate any keys, an empty ResultSet object is returned.
Note:If the columns which represent the auto-generated keys were not specified,the JDBC driver implementation will determine the columns which best represent the auto-generated keys. Return: a ResultSet object containing the auto-generated key(s) generated by the execution of this Statement object Exception:
- SQLException if a database access error occurs or this method is called on a closed Statement
- SQLFeatureNotSupportedException if the JDBC driver does not support this method since 1.4
public int insert(String name) { Connection conn = null; PreparedStatement ps = null; ResultSet rs = null; int retNum = -1; try { conn = iDBPool.tryGetConnection();// get Connection if (conn == null) return -1; String sql = "insert into example_table(name) values(?)"; ps = conn.prepareStatement(sql,Statement.RETURN_GENERATED_KEYS); // ps.setString(1,name); ps.executeUpdate(); rs = ps.getGeneratedKeys(); if(rs.next()){ retNum = rs.getInt(1); }else { retNum = -1; } } catch (Exception e) { Log.getLog().info(e.getMessage()); e.printStackTrace(); } finally { iDBPool.release(conn); try { if (null != ps) ps.close(); if (null != rs) rs.close(); } catch (SQLException e) { e.printStackTrace(); } } return retNum; }如上,完毕。
6 Comments
mooc · June 17, 2016 at 09:15
C++就好办多了,atomic 使 lock-free 很方便
yu · June 17, 2016 at 18:25
@mooc 多机并行的时候光 atomic 并不能 assign 一个唯一的 id,而且大多数语言都可以做 CAS
恋羽 · November 27, 2013 at 11:23
记得在高并发的时候mysql取这个值还是会出问题
yu · November 27, 2013 at 22:34
soga
但“为什么线程安全”在文中说过了,既然这个内容已经存储在独立的对象,为什么还是会出现线程不安全问题呢?求详细内容。
Leniy · November 4, 2013 at 15:37
java不懂
yu · November 4, 2013 at 15:59
java带有自解释属性,哪怕你没学过不会写,基本看代码也知道这是干毛用的。。。