新Java運動：測試驅動開發3---用戶註冊3

直到目前為止，我們還沒有接觸到用戶註冊的實質問題，也就是在資料庫中新增使用者。我們現在來處理這個需求。

首先需要確定資料庫存取所使用的技術，這裡可以選擇Hibernate、JPA或JDBC。相信絕大多數應用程式都是採用Hibernate來作為資料庫存取技術，另外一部分人可能會選擇JPA，但是我們在這裡選擇JDBC。原因比較簡單，底層的東西看起來好像比較複雜，但一旦掌握，由於它的內容少，相對來講更容易精通。而這個O-R映射模型，增加了許多抽象概念和細節，我們通常只看這些架構冰山的一角，而如果想掌握冰山下面的東西，難度比直接掌握JDBC難上幾個數量級。

另外，由於我們測試驅動開發的架構，我們想要轉換為其他架構，可以對現有程式碼進行重構，在測試案例的保證下，我們可以放心地對程式碼進行修改。

好了，我們先用DAO模式，定義資料庫存取介面類別UserDao，程式碼如下：

public interface UserDao {  
    public long registerUser(Map<String, Object> userInfo);  
}

一般來講，我們使用Mysql資料庫，所以定義Dao的實作類別UserMysqlDao，程式碼如下圖：

public class UserMysqlDao implements UserDao {  
    @Override  
    public long registerUser(Map<String, Object> userInfo) {  
        // TODO Auto-generated method stub  
        return 0;  
    }

我們當然不希望使用者來確定所使用的資料庫，然後實例化對應的實作類，為此我們需要採用工廠模式，引入DaoFactory類，程式碼如下：

public class DaoFactory {  
    public static UserDao getUserDao() {  
        UserDao dao = null;  
        switch (dbms) {  
        case "mysql":  
            dao = new UserMysqlDao();  
            break;  
        }  
        return dao;  
    }  
      
    public static String getDbms() {  
        return dbms;  
    }  
  
    public static void setDbms(String dbms) {  
        DaoFactory.dbms = dbms;  
    }  
  
    private static String dbms = "mysql";  
}

由上面的程式碼可以看出，呼叫者只需呼叫DaoFactory.getUserDao()即可取得實作類別了。

下面就是資料庫連線的問題，我們希望在Jboss等應用程式伺服器上執行時採用資料庫連線池，而在單元測試中使用DriverManager，所以需要定義一個DataSource的封裝類別JdbcDs來處理上述情況，程式碼如下所示：

public class JdbcDs {	
	public static Connection getConnection() throws SQLException {
		Connection conn = null;
		if (iDebug <= 0) {
			conn = appDs.getConnection();
		} else {
			try {
				Class.forName("com.mysql.jdbc.Driver").newInstance();
			} catch (InstantiationException e) {
				// TODO Auto-generated catch block
				e.printStackTrace();
			} catch (IllegalAccessException e) {
				// TODO Auto-generated catch block
				e.printStackTrace();
			} catch (ClassNotFoundException e) {
				// TODO Auto-generated catch block
				e.printStackTrace();
			}
			Properties connectionProps = new Properties();
		    connectionProps.put("user", "root");
		    connectionProps.put("password", "yantao");
		    conn = DriverManager.getConnection(
	                   "jdbc:" + "mysql" + "://" +
	                   "localhost" +
	                   ":" + 3306 + "/XrcjDb",
	                   connectionProps);
		}
		return conn;
	}

	public static int iDebug = 1;
	public final static String APP_DS = "java:jboss/datasources/XcgDS";
	private static DataSource appDs = null; 
	static{
		if (null == appDs) {
			Properties env = new Properties();
			try {
				InitialContext ictx = new InitialContext(env);
				appDs = (DataSource)ictx.lookup(APP_DS);
			} catch (NamingException e1) {
				// TODO Auto-generated catch block
				e1.printStackTrace();
			}
		}
	}
	
}

當進行單元測試時，只需將iDebug置為1即可。否則會使用資料庫連線池。

做完所有的準備工作，下面就可以正式進行資料庫功能開發了。

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