插入一条记录返回记录的主键，在普通的jdbc操作中实现比较简单
/**
* 插入一条记录，并返回主键
**/
public Integer addCase(CaseName caseN) {
Connection conn = null;
PreparedStatement pst = null;
ResultSet rest = null;
String sql = "insert into CaseName(Name,DepID) values(?,?) select SCOPE_IDENTITY()";
Integer row = null;
try {
conn = JdbcPool.dataSource.getConnection();
pst = conn.prepareStatement(sql);
pst.setString(1, caseN.getName());
pst.setInt(2, caseN.getDepID());
row = pst.executeUpdate();
rest = pst.getGeneratedKeys();
if (rest.next()) {
row = rest.getInt(1);
}
} catch (SQLException e) {
e.printStackTrace();
} finally {
JdbcUtils.closeConn(conn, pst, null);
}
return row;
}
主要是通过sql 语句的select SCOPE_IDENTITY()";和rest = pst.getGeneratedKeys()实现。但是在SpringJdbcTemplate 中执行update语句返回的是表中生效的行数，并不是我们需要的主键
@Override
public Integer updateCaseByNode(Integer id, String node) {
String sql = "update CaseName set Node = ? where ID = ?";
// 返回 生效 的数据行数
return jdbcTemplate.update(sql,node, id);
}
这时问题就变得非常棘手，在网上查阅资料后我总结了下面的两种实现方式
1.sql语句中包含select SCOPE_IDENTITY()
/**
* dao层插入数据的接口实现方法
* @param caseName
* @return
*/
@Override
public Integer saveCase(CaseName caseName) {
String sql = "insert into CaseName(Name,DepID) values(?,?) select SCOPE_IDENTITY() ";
// 调用insertCase
return insertCase(sql, caseName.getName(), caseName.getDepId()+"");
}
/**
* 具体的插入方法
* @param sql
* @param caseName
* @param depId
* @return
*/
private Integer insertCase(String sql, String caseName, String depId){
KeyHolder keyHolder = new GeneratedKeyHolder();
jdbcTemplate.update(new PreparedStatementCreator() {
@Override
public PreparedStatement createPreparedStatement(Connection conn)
throws SQLException {
// 预处理
PreparedStatement ps = conn.prepareStatement(sql);
ps.setString(1, caseName);
ps.setString(2, depId);
return ps;
}
}, keyHolder);
// 返回主键
System.out.println("插入后数据主键：" + keyHolder.getKey().intValue());
return keyHolder.getKey().intValue();
}
2.sql语句中不含select SCOPE_IDENTITY()，在PrepareStatement参数中加入PreparedStatement.RETURN_GENERATED_KEYS
/**
* dao层插入数据的接口实现方法
* @param caseName
* @return
*/
@Override
public Integer saveCase(CaseName caseName) {
// 注意sql 和方式1 的区别
String sql = "insert into CaseName(Name,DepID) values(?,?) ";
// 调用insertCase
return insertCase(sql, caseName.getName(), caseName.getDepId()+"");
}
/**
* 具体的插入方法
* @param sql
* @param caseName
* @param depId
* @return
*/
private Integer insertCase(String sql, String caseName, String depId){
KeyHolder keyHolder = new GeneratedKeyHolder();
jdbcTemplate.update(new PreparedStatementCreator() {
@Override
public PreparedStatement createPreparedStatement(Connection conn)
throws SQLException {
// 预处理 注意参数 PreparedStatement.RETURN_GENERATED_KEYS
PreparedStatement ps = conn.prepareStatement(sql,PreparedStatement.RETURN_GENERATED_KEYS);
ps.setString(1, caseName);
ps.setString(2, depId);
return ps;
}
}, keyHolder);
// 返回主键
System.out.println("插入后数据主键：" + keyHolder.getKey().intValue());
return keyHolder.getKey().intValue();
}
效果如下
图1.png
以上两种方式，推荐第二种，因为select SCOPE_IDENTITY()在不同的数据库中可能不存在这种语法。
