返回数据库的表自增主键值

有些开发者喜欢使用数据库自增字段作为表主键，即：主键值不在应用层产生，而是在新增记录时由数据库产生。这样，应用层在保存对象前并不知道对象主键值，而必须在保存数据后才能从数据库中返回主键值。在很多情况下，我们需要获取新对象持久化后的主键值。在Hibernate、JPA、等ORM框架中，新对象在持久化之后，主键值会自动绑定到对象上，给程序带来方便。

在JDBC3.0规范中，当新增记录时，允许将数据库自动产生的主键值绑定到Statement或PreparedStatement中。在使用Statement时，可以通过以下方法绑定到主键值。

int executeUpdate(String sql , int autoGeneratedKeys)

也可以通过Connection创建绑定自增主键值的Prepared Statement，如下：

PreparedStatement prepareStatement(String sql , int autoGenaratedKeys)

当autoGeneratedKeys参数设置为Statement.RETURN_GENERATED_KEYS时，即可绑定数据库产生的主键值；当设置为Statement.RETURN_GENERATED_KEYS时，不绑定主键值。下面的代码演示了Statement绑定并获取数据库产生的主键值得过程：

Statement stmt = conn.createStatement();
String sql = "insert into t_topic(topic_title,user_id) values('测试主题','123')";
//指定绑定表自增主键值
stmt.executeUpdate(sql , Statement.RETURN_GENERATED_KEYS);
ResultSet rs = stmt.getGeneratedKeys();
if(rs.next()){
   int key = rs.getInt();
}

Spring 利用者一技术，提供了一个可以返回新增记录对应主键值得方法。如下：

int update(PreparedStatementCreator psc , KeyHolder generatedKeyHolder);

Spring为KeyHolder接口指代了一个通用得实现类GeneratedKeyHolder，该类返回新增记录时得自增长主键值。假设希望在新增论坛版块对象后，将主键值加载到对象中，则可以按如下所示：

package com.smart.pojo;
//论坛实体类
public class Forum {

    private int forumId;
    private String forumName;
    private String forumDesc;

    public int getForumId() {
        return forumId;
    }

    public void setForumId(int forumId) {
        this.forumId = forumId;
    }

    public String getForumName() {
        return forumName;
    }

    public void setForumName(String forumName) {
        this.forumName = forumName;
    }

    public String getForumDesc() {
        return forumDesc;
    }

    public void setForumDesc(String forumDesc) {
        this.forumDesc = forumDesc;
    }
}

@Repository
public class ForumDao {

    private JdbcTemplate jdbcTemplate;

    @Autowired
    public void setJdbcTemplate(JdbcTemplate jdbcTemplate) {
        this.jdbcTemplate = jdbcTemplate;
    }

    public void addForumDao(final Forum forum){
       //新增后主键值绑定到对象的update方法
        final String sql = "INSERT INTO t_forum(forum_name,forum_desc) VALUEA(?,?)";
        KeyHolder keyHolder = new GeneratedKeyHolder();//创建一个主键持有者
        jdbcTemplate.update(new PreparedStatementCreator() {
            public PreparedStatement createPreparedStatement(Connection con) throws SQLException {
                PreparedStatement ps = con.prepareStatement(sql);
                ps.setString(1 , forum.getForumName());
                ps.setString(2 , forum.getForumDesc());
                return ps;
            }
        },keyHolder);
        forum.setForumId(keyHolder.getKey().intValue());//从主键持有者中获取主键
    }

这样，在调用addForum(final Forum forum)新增forum领域对象后，forum将拥有对应的主键，方便后继的使用。

返回数据库的表自增主键值

猜你喜欢