软件设计重构秘笈31式-30使用多态代替条件判断

概念

本文中的”使用多态代替条件判断”是指如果你需要检查对象的类型或者根据类型执行一些操作时，一种很好的办法就是将算法封装到类中，并利用多态性进行抽象调用。

意图

本文展示了面向对象编程的基础之一“多态性”， 有时你需要检查对象的类型或者根据类型执行一些操作时，一种很好的办法就是将算法封装到类中，并利用多态性进行抽象调用。

如下代码所示，OrderProcessor 类的ProcessOrder方法根据Customer 的类型分别执行一些操作，正如上面所讲的那样， 我们最好将OrderProcessor 类中这些算法（数据或操作）封装在特定的Customer 子类中。

案例

public class OrderProcessor {

    public double processOrder(Customer customer, List<Product> products)
    {
        // do some processing of order
        double orderTotal = products.stream().mapToDouble(Product::getPrice).sum();

        String customerType = customer.getType();
        if (customerType.equals("employee"))
        {
            orderTotal -= orderTotal * 0.15;
        }
        else if (customerType.equals("nonemployee"))
        {
            orderTotal -= orderTotal * 0.05;
        }
        return orderTotal;
    }

}

public abstract class Customer {

    protected abstract String getType();

}

public class Employee extends Customer {

    @Override
    protected String getType() {
        return "employee";
    }
}

public class NonEmployee extends Customer {

    @Override
    protected String getType() {
        return "nonemployee";
    }

}

public class Product {

    private double price;

    public double getPrice() {
        return price;
    }

    public void setPrice(double price) {
        this.price = price;
    }
}

重构

重构后的代码如下，每个Customer 子类都封装自己的算法，然后OrderProcessor 类的ProcessOrder方法的逻辑也变得简单并且清晰了。

public class OrderProcessor {

    public double processOrder(Customer customer, List<Product> products)
    {
        // do some processing of order
        double orderTotal = products.stream().mapToDouble(Product::getPrice).sum();

        orderTotal -= orderTotal * customer.discountPercentage();
        return orderTotal;
    }

}

public abstract class Customer {

    protected abstract String getType();

    public abstract double discountPercentage();

}

public class Employee extends Customer {

    @Override
    protected String getType() {
        return "employee";
    }

    @Override
    public double discountPercentage() {
        return 0.15;
    }
}

public class NonEmployee extends Customer {

    @Override
    protected String getType() {
        return "nonemployee";
    }

    @Override
    public double discountPercentage() {
        return 0.05;
    }
}

public class Product {

    private double price;

    public double getPrice() {
        return price;
    }

    public void setPrice(double price) {
        this.price = price;
    }
}

总结

”使用多态代替条件判断“这个重构在很多时候会出现设计模式中（常见的工厂家族、策略模式等都可以看到它的影子）， 因为运用它可以省去很多的条件判断，同时也能简化代码、规范类和对象之间的职责。