java 代码重构-第一章(分解并重组statement())
2.1 把switch代码在方法中抽出
第一个明显引起我注意的就是长得离谱的statement() 。每当看到这样长长的函数,我就想把它大卸八块。要知道,代码区块愈小,代码的功能就愈容易管理,代码的处理和搬移也都愈轻松。
重构过程的第一阶段中,我将说明如何把长长的函数切开,并把较小块的代码移至更合适的class 内。我希望降低代码重复量,从而使新的(打印HTML 报表用的)函数更容易撰写。
第一个步骤是找出代码的逻辑泥团(logical clump)并运用 Extract Method。本例一个明显的逻辑泥团就是switch 语句,把它提炼(extract)到独立函数中似乎比较好。
和任何重构准则一样,当我提炼一个函数时,我必须知道可能出什么错。如果我提炼得不好,就可能给程序引入臭虫。所以重构之前我需要先想出安全作法。
首先我得在这段代码里头找出函数内的局部变量(local variables)和参数(parameters)。我找到了两个:each 和thisAmount,前者并未被修改,后者会被修改。任何不会被修改的变量都可以被我当成参数传入新的函数,至于会被修改的变量就需格外小心。如果只有一个变量会被修改,我可以把它当作返回值。thisAmount 是个临时变量,其值在每次循环起始处被设为0,并且在switch 语句之前不会改变,所以我可以直接把新函数的返回值赋予它。
下面展示重构前后的代码。重构前的代码在下面类对比.凡是从函数提炼出来的代码,以及新代码所做的任何修改,只要我觉得不是明显到可以一眼看出,就以粗体字标示出来特别提醒你。
上一篇文章,最开始的代码
Customer
/**
* 通计清单
* @return
*/
public String statement(){
double totalAmount = 0;//合计
int frequentRentePoints = 0;
Enumeration enu_rentals = rentals.elements();
String result = "Rental Record for "+ this.getName() + "
";
while(enu_rentals.hasMoreElements()){
double thisAmount = 0;
Rental each = enu_rentals.nextElement();
switch (each.getMovie().getPriceCode()) {
case Movie.REGULAR:
thisAmount += 2;
if(each.getDeysRented() > 2){
thisAmount += (each.getDeysRented() - 2) * 1.5;
}
break;
case Movie.NEW_RELEASE:
thisAmount += each.getDeysRented() * 3;
break;
case Movie.CHILDRENS:
thisAmount += 1.5;
if(each.getDeysRented() > 3){
thisAmount += (each.getDeysRented() -3 ) * 1.5;
}
break;
}
frequentRentePoints ++;
if((each.getMovie().getPriceCode() == Movie.NEW_RELEASE) && each.getDeysRented() > 1){
frequentRentePoints ++;
}
result += ""+each.getMovie().getTitle() + "" +String.valueOf(thisAmount) + "
";
totalAmount += thisAmount;
}
result += "Amount owed is " + String.valueOf(totalAmount) + "
";
result += "You earned " + String.valueOf(frequentRentePoints) + " frequent renter points";
return result;
}
跟下面重构的代码对比(Customer 完整代码)
package com.mkfree.refactoring.shap1;
import java.util.Enumeration;
import java.util.Vector;
/**
* 顾客
* @author hk
*
* 2012-12-25 下午10:59:03
*/
public class Customer {
private String name;
private Vector rentals = new Vector<>();
public Customer(String name) {
this.name = name;
}
/**
* 添加
* @param rental
*/
public void addRentals(Rental rental){
rentals.add(rental);
}
public String getName() {
return name;
}
/**
* 通计清单
* @return
*/
public String statement(){
double totalAmount = 0;//合计
int frequentRentePoints = 0;
Enumeration enu_rentals = rentals.elements();
String result = "Rental Record for "+ this.getName() + "
";
while(enu_rentals.hasMoreElements()){
double thisAmount = 0;
Rental each = enu_rentals.nextElement();
amountFor(each);//计算一笔租片费
frequentRentePoints ++;
if((each.getMovie().getPriceCode() == Movie.NEW_RELEASE) && each.getDeysRented() > 1){
frequentRentePoints ++;
}
result += ""+each.getMovie().getTitle() + "" +String.valueOf(thisAmount) + "
";
totalAmount += thisAmount;
}
result += "Amount owed is " + String.valueOf(totalAmount) + "
";
result += "You earned " + String.valueOf(frequentRentePoints) + " frequent renter points";
return result;
}
/**
* 把计算一笔租片
* @param each
* @return
*/
private int amountFor(Rental each) { // 计算一笔租片费。
int thisAmount = 0;
switch (each.getMovie().getPriceCode()) {
case Movie.REGULAR: // 普通片
thisAmount += 2;
if (each.getDeysRented() > 2)
thisAmount += (each.getDeysRented() - 2) * 1.5;
break;
case Movie.NEW_RELEASE: // 新片
thisAmount += each.getDeysRented() * 3;
break;
case Movie.CHILDRENS: // 儿童。
thisAmount += 1.5;
if (each.getDeysRented() > 3)
thisAmount += (each.getDeysRented() - 3) * 1.5;
break;
}
return thisAmount;
}
}
每次做完这样的修改之后,我都要编译并测试。这一次起头不算太好——测试失败了,有两笔测试数据告诉我发生错误。一阵迷惑之后我明白了自己犯的错误。我愚蠢地将amountFor() 的返回值型别声明为int,而不是double 。
以上代码就已经消除了switch里的代码...是不是,整个类好看些呢呢?
本文章来自:http://blog.mkfree.com/posts/18