本文紧接
使用重构移除丑陋的if else代码（3）
OK, 到目前为止，所有的逻辑代码已经从SystemManager重构到了SystemStatePerformer。下一步应该继续重构SystemManager， 将SystemState替换为performer：
1， 使用IDE的重构功能，将变量SystemState改为SystemStatePerformer
2. 在updateState()方法中调用SystemStatePerformerFactory
3. 在测试代码里面，调用manager.statePerformer.getState()
重构后的代码如下：
package de.jingge.refactoring;
import static de.jingge.refactoring.SystemState.*;
public class SystemManager {
SystemStatePerformer statePerformer;
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(SystemState state) {
this.statePerformer = SystemStatePerformerFactory.getInstance()
getSystemStatePerformer(state);
statePerformer.perform();
}
}
可以看到if else已经消失了。
测试代码也要做相应修改：
package de.jingge.refactoring;
import org.junit.AfterClass;
import org.junit.BeforeClass;
import org.junit.Test;
import static org.junit.Assert.*;
import static de.jingge.refactoring.SystemState.*;
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.statePerformer.getState(), LOGGEDIN);
}
@Test
public void logout() {
manager.logout();
assertEquals(manager.statePerformer.getState(), LOGGEDOUT);
}
@Test
public void idle() {
manager.idle();
assertEquals(manager.statePerformer.getState(), IDLE);
}
}
到这里重构已经差不多完成了，代码已经更加面向对象了。这里还有一个小问题，在factory里面还有一个switch，这个和if else其实是没有本质区别的，也就是说if else并没有被完全移除掉。
那么如何能够彻底把这个switch也移除掉呢？很简单，我们只需要在getSystemStatePerformer()方法被调用之前先创建所有
performer匿名类的实例，然后在该方法被调用时直接返回对应的实力。 那么具体如何实现呢，请看下一篇文章使用重构移除丑陋的if else代码（5）。
声明：本文版权归作者所有，如需转载请注明出处。