我們知道因為編程語言的限制，歷史遺留下來的系統總是有很多的毛病，不夠面向對象，尤其是很多系統濫用if else。我曾經見過一個項目，大家基本上就是寫一個方法，然后在里面if else套if esle得嵌套了好幾層，難看就不必說了，這種代碼根本就沒法維護。
今天我就使用從實際項目中提煉出來的例子來講解一下如何將這類代碼變得更加面向對象 - 重構成模式并且添加測試代碼，
先來看一個丑陋的類：
package de.jingge.refactoring;
public class SystemManager {
public static final int LOGGEDIN = 0;
public static final int LOGGEDOUT = 1;
public static final int IDLE = 2;
int state;
public void login() {
// call service#login()
updateState(LOGGEDIN);
}
public void logout() {
// call service#logout()
updateState(LOGGEDOUT);
}
public void idle() {
// call some other services
updateState(IDLE);
}
public void updateState(int state) {
if (state == LOGGEDIN) {
// do something after logging in is successful,
// for example: show welcome dialog, open the last edit document, etc.
} else if (state == LOGGEDOUT) {
// do something after logging out is successful,
// for example: free used resource, dispose GUI components, etc.
} else if (state == IDLE) {
// do something after the user is idle,
// for example: save the application state temporarily, lock the application, etc.
} else {
throw new IllegalArgumentException("unknown state");
}
this.state = state;
}
}
這里我們展示了一個 SystemManager，它負責處理用戶在系統中的狀態：登入（logged in），登出（logged
out）,以及空閑（idle）。從代碼中可以看到，這個類用了int來定義狀態并且因此導致了updatteState（）方法里面出現大量if
else。從目前看來這些if else是無法避免的，應為這個類需要針對不同的狀態作出反應。隨著狀態的增加，if
else的數量也會繼續增加。這個解決方案顯然很差。
那么怎么樣才能讓這個類更加地面向對象呢？
在處理面向對象之前，我們首先要編寫一個測試類，這也是處理這類歷史遺留下來代碼所必需做的第一步，只有在測試代碼的保護下，我們才能放心大膽地進行重構。
初步的測試代碼如下：
package de.jingge.refactoring;
import org.junit.AfterClass;
import org.junit.BeforeClass;
import org.junit.Test;
import static org.junit.Assert.*;
public class SystemManagerTest {
private static SystemManager manager;
@BeforeClass
public static void setUpClass() throws Exception {
manager = new SystemManager();
// add some service mock objects
}
@AfterClass
public static void tearDownClass() throws Exception {
}
@Test
public void login() {
manager.login();
assertEquals(manager.state, SystemManager.LOGGEDIN);
}
@Test
public void logout() {
manager.logout();
assertEquals(manager.state, SystemManager.LOGGEDOUT);
}
@Test
public void idle() {
manager.idle();
assertEquals(manager.state, SystemManager.IDLE);
}
}
運行測試代碼->通過。
在下一篇文章我們將正式開始重構。地址：
使用重構移除丑陋的if else代碼（2）
聲明：本文版權歸作者所有，如需轉載請注明出處。