日期:2014-05-16  浏览次数:20394 次

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

我们在做项目开发时很多时候都需要数据库的增删改查,而添加数据后又想得到该项数据在数据库中的主键值,这里有一个非常方便的方法可以在插入数据的同时获得主键值。

	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);
	}

注意上面代码:要添加Statement.RETURN_CENERATED_KEYS

 conn.prepareStatement(sql,Statement.RETURN_GENERATED_KEYS);
就可以用 pst.getGenerateKeys()方法得到主键。