完成数据库操作的基础架构之后,就是我们真正进行JDBC数据操作的时候了。所涉及的数据库表ER图如下所示:
如上图所示,我们第一步是向t_user表中添加记录。由于用户注册需要操作多张表,因此需要用到事务,先写出一个简单的基于JDBC的事务框架,代码如下所示:
@Override
public long registerUser(Map<String, Object> userInfo) {
Connection conn = null;
long userId = 0;
try {
conn = JdbcDs.getConnection();
conn.setAutoCommit(false);
userId = addUser(conn, userInfo);
if (userId <= 0) {
throw new SQLException("Fail to add user in t_user");
}
conn.commit();
} catch (SQLException e) {
// TODO Auto-generated catch block
e.printStackTrace();
try {
conn.rollback();
} catch (SQLException e1) {
// TODO Auto-generated catch block
e1.printStackTrace();
}
userId = -1;
} finally {
try {
conn.close();
} catch (SQLException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
return userId;
}
其次是具体的用户添加操作,具体代码如下所示:
private long addUser(Connection conn, Map<String, Object> userInfo) {
long userId = -1;
PreparedStatement stmt = null;
ResultSet rst = null;
String sql = "insert into t_user(user_name, user_group_id, user_level_id) values(?, 2, 1)";
try {
stmt = conn.prepareStatement(sql, Statement.RETURN_GENERATED_KEYS);
stmt.setString(1, (String)userInfo.get("userName"));
int affectedNum = stmt.executeUpdate();
if (1 == affectedNum) {
rst = stmt.getGeneratedKeys();
if (rst.next()) {
userId = rst.getLong(1);
}
} else {
userId = -1;
}
} catch (SQLException e) {
// TODO Auto-generated catch block
e.printStackTrace();
userId = -1;
} finally {
try {
if (rst != null) {
rst.close();
}
if (stmt != null) {
stmt.close();
}
} catch (Exception ex) {
}
}
return userId;
}
最后是改变测试用例中的判断成功条件为所返回的userId大于0。
运行测试用例,应该可以成功通过测试用例。
经过以上几篇文章,我们终于可以进行有意义的开发工作了。下一步就是实现所有用户注册业务逻辑,还有一块是异常情况的处理,例如userName重复的情况。当完成所有这些功能后,我们还需要进行端到端测试,这就涉及通过JSP页面进行注册测试。