举一个自动取款机的场景来说明状态模式的使用。去自动提款机取钱首先得有插卡的动作,最后还会有取卡的动作。对应两个状态,一个是无卡状态,另外一个是有卡状态。
一、代码实现
public class ATM {
final static int NO_CARD = 0;
final static int HAVE_CARD = 1;
//提款机初始状态为无卡
int state = NO_CARD;
//插卡
public void insertCard(){
if(state==NO_CARD){
System.out.println("插入卡片");
//设置状态为有卡状态
setState(HAVE_CARD);
}else if(state==HAVE_CARD){
System.out.println("已插入银行卡");
}
}
//取卡
public void quitCard(){
if(state==NO_CARD){
System.out.println("你没有插入银行卡");
}else if(state==HAVE_CARD){
System.out.println("退出卡片");
//设置状态为无卡状态
setState(NO_CARD);
}
}
public void setState(int state){
this.state = state;
}
}
从上面的代码可以预见到,如果再增加状态,那么对应的操作里面就需要增加一组if-else,越到后面代码将越来越难以维护。
二、使用状态模式改造
//状态接口
public interface State {
//插入卡片
public void insertCard();
//取出卡片
public void quitCard();
}
//无卡状态
public class NoCard implements State{
NewATM atm;
public NoCard(NewATM newATM){
this.atm = newATM;
}
@Override
public void insertCard() {
System.out.println("插入卡片");
atm.setState(NewATM.HAVE_CARD);
}
@Override
public void quitCard() {
System.out.println("你没有插入银行卡");
}
}
//有卡状态
public class HaveCard implements State{
NewATM atm;
public HaveCard(NewATM newATM){
this.atm = newATM;
}
@Override
public void insertCard() {
System.out.println("已插入银行卡");
}
@Override
public void quitCard() {
System.out.println("退出卡片");
atm.setState(NewATM.NO_CARD);
}
}
//使用状态模式改造后的业务代码
public class NewATM {
static State NO_CARD;
static State HAVE_CARD;
private State state = NO_CARD;
public NewATM(){
NO_CARD = new NoCard(this);
HAVE_CARD = new HaveCard(this);
}
//插卡
public void insertCard(){
state.insertCard();
}
//取卡
public void quitCard(){
state.quitCard();
}
public void setState(State state){
this.state = state;
}
}
经过改造,彻底干掉了繁琐的if-else状态判断。当有业务变更,或是新状态加入时可灵活的调整代码。
系列文章:
设计模式系列之一单例模式
设计模式系列之二策略模式
设计模式系列之三观察者模式
设计模式系列之四装饰者模式
设计模式系列之五工厂模式
设计模式系列之六命令模式
设计模式系列之八外观模式
设计模式系列之七适配器模式
设计模式系列之九模板方法模式
设计模式系列之十迭代器模式
设计模式系列之十一组合模式