来改善你的switch语句

善用多态来改善你的switch语句

        下面举的这一个例子,其实和<<重构>>第一章里面的那一个经典实例很像的,<<重构>>一本书绝对是一本好书,非常不错值得推荐的一本书。

举一个例子,是我现在的一个项目里面一位同事写的一小段代码,代码如下,它可以根据不同的操作符进行加减乘除方面的操作,而且这一些操作很有可能发生一些变化。当重构成我的现在每天的面包和水的时候，我看到switch语句，就想能不能把改良一下。虽然改善后，代码量是增加，但从可维护的角度来讲，我个人认为还是值得的,特别在你需要的操作的Switch语句里面很有可能发生需求变化的时候。

private void cacuData(DataRange dataRange, char oper) {
		double firstData = Double.parseDouble(dataRange.getFirstData());
		double secondData = Double.parseDouble(dataRange.getSecondData());
		switch (oper) {
		case '+':
			dataStack.push("" + (firstData + secondData));
			break;
		case '-':
			dataStack.push("" + (firstData - secondData));
			break;
		case '*':
			dataStack.push("" + (firstData * secondData));
			break;
		case '/':
			divisionProcess(firstData, secondData);
			break;
		}
	}

	/** 处理除数方法； */
	private void divisionProcess(double firstData, double secondData) {
		if (Math.abs(secondData) < Math.pow(10, -5)) {
			dataStack.push("-1");
		} else {
			dataStack.push("" + (firstData / secondData));
		}
	}

   我看到firstData,secondData就想能不能把它们放到一个Range范围的类里面是不是更好呢,于是新建了一个
 

 public class DoubleDataRange{
	private final double firstData;

	private final double secondData;

	public DoubleDataRange(double firstData, double secondData) {
		this.firstData = firstData;
		this.secondData = secondData;
	}
	
	public double getFirstData() {
		return firstData;
	}

	public double getSecondData() {
		return secondData;
	}
  }

然后我们再新建一系列操作方面的类，为每一个操作类新增一个类,也是这一次重构最核心的东西，其代码如下：

abstract class Operation{
	abstract void  cacuData(DoubleDataRange dataRange,SSheetStack dataStack);
}

class PlusOperation extends Operation {
	void cacuData(DoubleDataRange range, SSheetStack dataStack) {
		dataStack.push("" + (range.getFirstData() + range.getSecondData()));
	}
}

class SubtractOperation extends Operation {
	void cacuData(DoubleDataRange range, SSheetStack dataStack) {
		dataStack.push("" + (range.getFirstData() - range.getSecondData()));
	}
}

class MultiplyOperation extends Operation {
	void cacuData(DoubleDataRange range, SSheetStack dataStack) {
		dataStack.push("" + (range.getFirstData() * range.getSecondData()));
	}
}

class DivisionOperation extends Operation {
	void cacuData(DoubleDataRange range, SSheetStack dataStack) {
		if (Math.abs(range.getSecondData()) < Math.pow(10, -5)) {
			dataStack.push("-1");
		} else {
			dataStack.push("" + (range.getFirstData() / range.getSecondData()));
		}
	}
}

接下来的工作就很简单啦，把以前两段代码全部删除掉，换上如下的代码,虽然这一个时候也有Switch语句，但它只是纯粹地变成了一个取operation的类而已，并没有包含任何的操作。

private void cacuData(DataRange dataRange, char oper) {
		setOperation(oper);
		double firstData = Double.parseDouble(dataRange.getFirstData());
		double secondData = Double.parseDouble(dataRange.getSecondData());
		operation.cacuData(new DoubleDataRange(firstData,secondData), dataStack);
	}
	
	private Operation operation;
	
	private void  setOperation(char oper){
		switch (oper) {
		case '+':
			operation = new PlusOperation();
			break;
		case '-':
			operation = new SubtractOperation();
			break;
		case '*':
			operation = new MultiplyOperation();
			break;
		case '/':
			operation = new DivisionOperation();
			break;
		}
	}
}

评论

2 楼 wenjixiao 2009-08-04  

我只想知道怎样优雅的去掉switch，现在，switch还在。
现在这种微小的改善没啥意思。

1 楼 CherryRemind 2008-01-08  

修改好像就是封装了下operation， 隐藏了operation的细节