java 代码重构-第一章(分解并重组statement())

 

上一篇文章:java 代码重构-第一章(起点)

下一篇文章:java 代码重构-第一章(类自己该做自己的事)

 

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<Rental> 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<Rental> 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<Rental> enu_rentals = rentals.elements();
        String result = "Rental Record for "+ this.getName() + " 
";
        while(enu_rentals.hasMoreElements()){
            double thisAmount = 0;
            Rental each = enu_rentals.nextElement();

            thisAmount = 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