首先是插入一条数据到数据库,并返回插入时的id号,数据的设计 id号的是自动递增的
public static int saveHotel(Hotel hotel) {
PreparedStatement pstmt = null;
Connection conn = DB.getConnection();
try {
String sql = "insert into t_hotel values(null , ?,? ,? ,?,? ,? ,?,?,?)";
//插入数据时返回的id号
pstmt = conn.prepareStatement(sql, Statement.RETURN_GENERATED_KEYS);
pstmt.setString(1, hotel.getName());
pstmt.setString(2, hotel.getPassword());
pstmt.setString(3, hotel.getAddress());
pstmt.setString(4, hotel.getPhone());
pstmt.setInt(5, hotel.getPrice());
pstmt.setString(6, hotel.getImg());
pstmt.setString(7, hotel.getCuisine());
pstmt.setString(8, hotel.getArea());
pstmt.setString(9, hotel.getStreet());
pstmt.executeUpdate();
//获得返回的id号
ResultSet rs = pstmt.getGeneratedKeys();
rs.next();
int returnKey = rs.getInt(1);
return returnKey ;
} catch (SQLException e) {
e.printStackTrace();
return 0 ;
} finally {
DB.closepStmt(pstmt);
DB.closeConn(conn);
}
}
批量插入数据并返回id号
public static ResultSet saveHotel(List<hotel> hotels) {
PreparedStatement pstmt = null;
Connection conn = DB.getConnection();
try {
String sql = "insert into t_hotel values(null , ?,? ,? ,?,? ,? ,?,?,?)";
//插入数据时返回的id号
pstmt = conn.prepareStatement(sql, Statement.RETURN_GENERATED_KEYS);
for(Hotel hotel :hotels){
pstmt.setString(1, hotel.getName());
pstmt.setString(2, hotel.getPassword());
pstmt.setString(3, hotel.getAddress());
pstmt.setString(4, hotel.getPhone());
pstmt.setInt(5, hotel.getPrice());
pstmt.setString(6, hotel.getImg());
pstmt.setString(7, hotel.getCuisine());
pstmt.setString(8, hotel.getArea());
pstmt.setString(9, hotel.getStreet());
pstmt.addBatch();
}
pstmt.executeBatch();
//获得返回的id号,是一个set
ResultSet keyRs =pstmt.getGeneratedKeys();
} catch (SQLException e) {
e.printStackTrace();
return 0 ;
} finally {
DB.closepStmt(pstmt);
DB.closeConn(conn);
}
return keyRs ;
}