新Java運動：測試驅動開發3---使用者註冊4

完成資料庫操作的基礎架構之後，就是我們真正進行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頁面進行註冊測試。

 以上就是新Java運動：測試驅動開發3---用戶註冊4的內容，更多相關內容請關注PHP中文網（www.php.cn）！