设计模式之状态模式（State Pattern）

no_mkd

一.什么是状态模式？

把所有动作都封装在状态对象中，状态持有者将行为委托给当前状态对象

也就是说，状态持有者（比如汽车，电视，ATM机都有多个状态）并不知道动作细节，状态持有者只关心自己当前所处的状态（持有的状态对象是哪个），再把一切事情都交给当前状态对象去打理就好了，甚至都不用去控制状态切换（当然，状态持有者有权利控制状态切换，也可以选择做甩手掌柜。。）

二.举个例子

假设我们要模拟一个ATM机，有以下需求：

• 取款，验证卡密，吐出现钞，结束服务
• 若卡密验证失败或者余额不足，则直接弹出卡片，结束本次服务
• 机器内无存钞，显示No Cash，并向银行发送无钞信息
• 机器故障，显示No Service，并向维修人员发送维修请求

那么用户取款的过程应该是这样的：

这是用户操作的流程，对于ATM机而言还需要注意下面几点：

• 获得用户输入金额后，不仅要验证用户卡内余额，还要验证ATM机内余额
• 每一次成功的取款服务结束后，都要检查ATM机内余额（若无钞则进行相应处理）
• 任何环节出现无法处理的错误，都按照故障来处理

想想上面的取款过程，我们还必须考虑各种非法操作，比如：

• 不插卡直接输入密码
• 机器故障时仍然进行操作
• 机内无钞时要求取款

细分之后，我们会发现细节问题变得很麻烦，机器就摆在那里，所有的用户接口都开放着，用户完全有可能在任何时候使用任何接口。为了防止这些非法操作，我们不得不添加一系列的判断，比如：

• 验证密码之前应该检查是否已经插卡，以及机器是否发生了故障
• 插卡之前应该检查机器是否发生了故障
• 取款操作之前应该检查机器是否故障，是否无钞，是否。。。

我们的代码中需要设置大量的成员变量，用来标识机器的状态，更可怕的是我们的逻辑块里面存在大量的if-else，而且每一个操作里面的if-else块都只有细微的差别，看起来像冗余，但又不好处理（即使我们把每一个判断都提出来作为独立的方法，可以缩减代码规模，但在处理过程上仍然是冗余。。）

更好的处理方法就是使用状态模式，能够完全消除状态判断部分的冗余，并提供清晰整洁的代码结构。下面用状态模式来实现例子中的需求

（1）首先找出ATM提供的所有接口

1. 插卡
2. 提交密码
3. 取款（假设取款按钮是物理键）
4. 查询（假设同上）
5. 取卡

（2）再找出ATM的所有状态以及各个状态对应的动作

• 准备就绪（Ready），可用接口：全部
• 无钞（NoCash），可用接口：1，2，4，5
• 故障（NoService），可用接口：无

（3）编码实现

先定义State基类，类中封装了（1）列出的所有接口：

package StatePattern;
/**


定义ATM机状态


@author ayqy
/
public interface ATMState {
/*

插卡
*/
public abstract void insertCard();

/**

提交密码
*/
public abstract void submitPwd();

/**

取款
*/
public abstract void getCash();

/**

查询余额
*/
public abstract void queryBalance();

/**

取卡
*/
public abstract void ejectCard();
}

再逐一实现三个状态

ReadyState：

package StatePattern;
/**


实现ATM就绪状态


@author ayqy
*/
public class ReadyState implements ATMState{
private ATM atm;//保留状态持有者的引用，以便对其进行操作
public ReadyState(ATM atm){
this.atm = atm;
}
@Override
public void insertCard() {
System.out.println("插卡完成");
}
@Override
public void submitPwd() {
System.out.println("密码提交完成");
//验证密码并做相应处理
if("123".equals(atm.getPwd())){
System.out.println("密码验证通过");
}
else{
System.out.println("密码验证失败");
//弹出卡片
ejectCard();
}
}
@Override
public void getCash() {
if(atm.getTotalAmount() >= atm.getAmount() && atm.getBalance() >= atm.getAmount()){
//更新账户余额
atm.setBalance(atm.getBalance() - atm.getAmount());
//更新机内现钞总数
atm.setTotalAmount(atm.getTotalAmount() - atm.getAmount());
System.out.println("吐出￥" + atm.getAmount());
System.out.println("取款完成");
//弹出卡片
ejectCard();
//检查机内余钞
if(atm.getTotalAmount() == 0){//若无钞，切换到NoService状态
atm.setCurrState(atm.getNoCashState());
System.out.println("无钞信息已经发送至银行");
}
}
else{
System.out.println("取款失败，余额不足");
//弹出卡片
ejectCard();
}
}
@Override
public void queryBalance() {
System.out.println("余额￥" + atm.getBalance());
System.out.println("余额查询完成");
}
@Override
public void ejectCard() {
System.out.println("取卡完成");
}
}

注意我们在状态类中进行状态切换的部分：

if(atm.getTotalAmount() == 0){//若无钞，切换到NoService状态
	atm.setCurrState(atm.getNoCashState());
}

我们并不是直接new具体状态对象，而是使用了ATM提供的set接口，这样做是为了尽量的解耦（兄弟对象彼此之间并不认识），获取更多的弹性

实现NoCashState：

package StatePattern;
/**


实现ATM无钞状态


@author ayqy
*/
public class NoCashState implements ATMState{
private ATM atm;//保留状态持有者的引用，以便对其进行操作
public NoCashState(ATM atm){
this.atm = atm;
}
@Override
public void insertCard() {
System.out.println("插卡完成");
}
@Override
public void submitPwd() {
System.out.println("密码提交完成");
//验证密码并做相应处理
if("123".equals(atm.getPwd())){
System.out.println("密码验证通过");
}
else{
System.out.println("密码验证失败");
//弹出卡片
ejectCard();
}
}
@Override
public void getCash() {
System.out.println("取款失败，机内无钞");
}
@Override
public void queryBalance() {
System.out.println("余额￥" + atm.getBalance());
System.out.println("余额查询完成");
}
@Override
public void ejectCard() {
System.out.println("取卡完成");
}
}

实现NoServiceState：

package StatePattern;
/**


实现ATM故障状态


@author ayqy
*/
public class NoServiceState implements ATMState{
private ATM atm;//保留状态持有者的引用，以便对其进行操作
public NoServiceState(ATM atm){
this.atm = atm;
}
@Override
public void insertCard() {
System.out.println("插卡失败，机器发生了故障");
}
@Override
public void submitPwd() {
System.out.println("密码提交失败，机器发生了故障");
}
@Override
public void getCash() {
System.out.println("取款失败，机器发生了故障");
}
@Override
public void queryBalance() {
System.out.println("余额查询失败，机器发生了故障");
}
@Override
public void ejectCard() {
System.out.println("取卡失败，机器发生了故障");
}
}

实现了具体的状态，就可以构造ATM类了，就像这样：

package StatePattern;
/**


实现ATM机


@author ayqy
*/
public class ATM {
/所有状态/
private ATMState readyState;
private ATMState noCashState;
private ATMState noServiceState;
private ATMState currState;//当前状态
private int totalAmount;//机内现钞总数
/测试用的临时变量/
private String pwd;//密码
private int balance;//余额
private int amount;//取款金额
public ATM(int totalAmount, int balance, int amount, String pwd) throws Exception{
//初始化所有状态
readyState = new ReadyState(this);
noCashState = new NoCashState(this);
noServiceState = new NoServiceState(this);
 if(totalAmount > 0){
 	currState = readyState;
 }
 else if(totalAmount == 0){
 	currState = noCashState;
 }
 else{
 	throw new Exception();
 }
 
 //初始化测试数据
 this.totalAmount = totalAmount;
 this.balance = balance;
 this.amount = amount;
 this.pwd = pwd;

}
/把具体行为委托给状态对象/
/**

插卡
*/
public void insertCard(){
currState.insertCard();
}

/**

提交密码
*/
public void submitPwd(){
currState.submitPwd();
}

/**

取款
*/
public void getCash(){
currState.getCash();
}

/**

查询余额
*/
public void queryBalance(){
currState.queryBalance();
}

/**

取卡
*/
public void ejectCard(){
currState.ejectCard();
}

public String toString(){
return "现钞总数￥" + totalAmount;
}
/此处略去大量getter and setter/
}

一切都做好了，迫不及待的测试一下吧

三.运行示例

首先实现一个Test类：

package StatePattern;
import java.util.Scanner;
/**


实现测试类


@author ayqy
/
public class Test {
public static void main(String[] args) {
/测试数据/
/ 机内总数	 账户余额	 取款金额	密码
* 1000	500     200		123
* 1000	300     500		123
* 0	500		200		123
* */
try {
test(1000, 500, 200, "123");
System.out.println("-------");
test(1000, 300, 500, "123");
System.out.println("-------");
test(0, 500, 200, "123");
} catch (Exception e) {
System.out.println("机器故障，维修请求已经发送至维修方");
}
}
private static void test(int totalAmount, int balance, int amount, String pwd)throws Exception{
//创建ATM
ATM atm;
atm = new ATM(totalAmount, balance, amount, pwd);
//输出初始状态
System.out.println(atm.toString());
atm.insertCard();
atm.submitPwd();
atm.getCash();
//输出结束状态
System.out.println(atm.toString());
}
}

我们设计的三个用例（正常取款，取大于余额的款，机内无现钞）运行结果如下：

四.状态模式与策略模式

还记得策略模式吗？难道不觉得这两者很相像吗？

没错，这两种模式的类图是完全相同的，解释一下：

• 状态主体（拥有者）持有状态对象，运行时可以通过动态指定状态对象来改变类的行为
• 策略主体持有算法族对象，运行时可以通过动态选择算法族中的算法（策略）来改变类的行为

也就是说，状态模式与策略模式都支持运行时的多态，并且其实现方式都是组合 + 委托。但是这并不代表这两种模式是相同的，因为它们的目标不同：

• 状态模式实现了算法流程可变（即状态切换，不同的状态有不同的流程）
• 策略模式实现了算法细节可选（即选择算法族内的算法，一个算法族包含多个可选算法）