Question

如何向数据库添加时同时返回ID

Answer 1

public void save(Person p) throws SQLException {
String sql = "insert into person (name,pwd,sex,age,email,loves) values(?,?,?,?,?,?)";
Connection conn = DbUtil.getConnection();
PreparedStatement pst = conn.prepareStatement(sql,Statement.RETURN_GENERATED_KEYS);
pst.setString(1, p.getName());
pst.setString(2, p.getPwd());
pst.setString(3, p.getSex());
pst.setInt(4, p.getAge());
pst.setString(5, p.getEmail());
StringBuffer loves = new StringBuffer("");
//将爱好信息拼成字符串
for(String s:p.getLoves()){
loves.append(s);
loves.append(",");
}
//如果用户没选择爱好，则写入""
if(loves.length()>0){
loves.substring(0, loves.length()-1);
pst.setString(6, loves.toString());
}else{
pst.setString(6,"");
}
pst.executeUpdate();
ResultSet rs = pst.getGeneratedKeys();
rs.next();
int id = rs.getInt(1);
p.setId(id);
DbUtil.closeConnection(conn);
}