大部分读者都会想到用策略模式去代替if-else。策略模式主要是定义统一行为（接口或抽象类），并实现不同策略下的处理逻辑（对应实现类）。
注解实现
首先定义一个客户实体类：
public class Customer {
/**
* 客户姓名
*/
private String name;
/**
* 客户类型
*/
private String type;
/**
* 客户区域
*/
private String area;
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
public String getType() {
return type;
}
public void setType(String type) {
this.type = type;
}
public String getArea() {
return area;
}
public void setArea(String area) {
this.area = area;
}
}
假如对于不同类型（新客户、老客户、VIP客户）的客户有不同的折扣。通常会有这样一个sevice，里面有一大堆的if -else的代码，目的是根据客户的类型做不同的折扣。
@Service
public class CustomerService {
public BigDecimal quote(BigDecimal originalPrice, Customer customer) {
if (customer.getType().equals("新客户")) {
// 新客户的逻辑
return this.quoteNewCustomer(originalPrice);
} else if (customer.getType().equals("老客户")) {
// 老客户的逻辑
return this.quoteOldCustomer(originalPrice);
} else if (customer.getType().equals("VIP客户")) {
// VIP客户的逻辑
return this.quoteVIPCustomer(originalPrice);
}
// 其他人员的逻辑
return originalPrice;
}
private BigDecimal quoteNewCustomer(BigDecimal originalPrice) {
System.out.println("抱歉！新客户没有折扣！");
return originalPrice;
}
private BigDecimal quoteOldCustomer(BigDecimal originalPrice) {
System.out.println("恭喜！老客户打9折");
originalPrice = originalPrice.multiply(new BigDecimal(0.9)).setScale(2, BigDecimal.ROUND_HALF_UP);
return originalPrice;
}
private BigDecimal quoteVIPCustomer(BigDecimal originalPrice) {
System.out.println("恭喜！VIP客户打8折");
originalPrice = originalPrice.multiply(new BigDecimal(0.8)).setScale(2, BigDecimal.ROUND_HALF_UP);
return originalPrice;
}
}
当我们新增一个客户类型的时候，首先要添加一个该种客户类型的报价算法方法，然后再quote方法中再加一个else if的分支，是不是感觉很是麻烦呢？而且这也违反了设计原则之一的开闭原则（open-closed-principle）.
那有没有什么办法使得我们的报价管理即可扩展、可维护，又可以方便的响应变化呢？当然有解决方案啦，就是我们下面要讲的策略模式。
1、首先定义一个CustomerHandler接口，此接口规定了处理报价的方法。
public interface CustomerHandler {
void quote(BigDecimal originalPrice, Customer customer);
}
2、定义一个CustomerHandlerType注解，来表示某个类是用来处理何种类型的客户。
@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Service
public @interface CustomerHandlerType {
String type();
}
3、接下来就是实现不同客户处理各自的handler，并加上我们所定义的CustomerHandlerType注解。
@CustomerHandlerType(type = "新客户")
public class NewCustomerHandler implements CustomerHandler {
@Override
public BigDecimal quote(BigDecimal originalPrice, Customer customer) {
System.out.println("抱歉！新客户没有折扣！");
return originalPrice;
}
}
@CustomerHandlerType(type = "老客户")
public class OldCustomerHandler implements CustomerHandler {
@Override
public BigDecimal quote(BigDecimal originalPrice, Customer customer) {
System.out.println("恭喜！老客户打9折");
originalPrice = originalPrice.multiply(new BigDecimal(0.9)).setScale(2, BigDecimal.ROUND_HALF_UP);
return originalPrice;
}
}
@CustomerHandlerType(type = "VIP客户")
public class VIPCustomerHandler implements CustomerHandler {
@Override
public BigDecimal quote(BigDecimal originalPrice, Customer customer) {
System.out.println("恭喜！VIP客户打8折");
originalPrice = originalPrice.multiply(new BigDecimal(0.8)).setScale(2, BigDecimal.ROUND_HALF_UP);
return originalPrice;
}
}
4、以上准备就绪后，就是向spring容器中注入各种客户处理的handler，并在CustomerService.quote方法中，通过策略（客户类型）去决定选择哪一个CustomerHandler去处理折扣。我们可以这样做：
@Service
public class CustomerService {
private Map<String, CustomerHandler> customerHandlerMap;
@Autowired
public void setCustomerHandlerMap(List<CustomerHandler> customerHandlers) {
this.customerHandlerMap = customerHandlers.stream().collect(
Collectors.toMap(customerHandler -> AnnotationUtils.findAnnotation(customerHandler.getClass(), CustomerHandlerType.class).type(),
v -> v, (v1, v2) -> v1)
);
}
public void quote(Customer customer) {
// 获取不同客户的报价
CustomerHandler handler = customerHandlerMap.get(customer.getType());
BigDecimal price = handler.quote(new BigDecimal(10000), customer);
}
}
在CustomerService中，维护了一个customerHandlerMap，它的key为客户类型，value为对应的客户处理器Handler。通过@Autowired去初始化customerHandlerMap。
不论以后业务如何发展致使客户类型种类增加，CustomerService 的核心逻辑不会改变，我们只需要实现新增客户类型的CustomerHandler即可。
到此，似乎已经讲完了通过注解实现策略模式，干掉if-else的方法，就这样结束了吗？不，真正的重点从现在才开始。
现在回过头看customerHandlerMap这个Map，它的key是客户类型，假如，我们想通过客户类型+客户区域这两个属性来决定到底使用哪一种CustomerHandler怎么办？
此时可以把key的类型设置为CustomerHandlerType？就像这样。
private Map<CustomerHandlerType, CustomerHandler> customerHandlerMap;
如此一来，不管决定订单处理器orderhandler的因素怎么变，我们便可以以不变应万变（这不就是我们所追求的代码高扩展性和灵活性么）。那当我们的map的key变成了OrderHandlerType之后，注入和获取的逻辑就要相应改变，注入的地方很好改变，如下：
@Autowired
public void setCustomerHandlerMap(List<CustomerHandler> customerHandlers) {
this.customerHandlerMap = customerHandlers.stream().collect(
Collectors.toMap(customerHandler -> AnnotationUtils.findAnnotation(customerHandler.getClass(), CustomerHandlerType.class),
v -> v, (v1, v2) -> v1)
);
}
问题变成了如何关联customer的类型和区域与CustomerHandlerType注解。
public class CustomerHandlerTypeImpl implements CustomerHandlerType {
private String type;
private String area;
public CustomerHandlerTypeImpl(String type, String area) {
this.type = type;
this.area = area;
}
@Override
public String type() {
return type;
}
@Override
public String area() {
return area;
}
@Override
public Class<? extends Annotation> annotationType() {
return CustomerHandlerType.class;
}
}
我们这样获取对应的CustomerHandler，
public void quote(Customer customer) {
// 获取不同客户的报价
CustomerHandlerType customerHandlerType = new CustomerHandlerTypeImpl(customer.getType(), customer.getArea());
CustomerHandler handler = customerHandlerMap.get(customerHandlerType);
BigDecimal price = handler.quote(new BigDecimal(10000), customer);
}
运行起来报错，获取不到对应的CustomerHandler。
梳理下，在Autowied时，我们放进customerHandleMap的key是动态代理类对象，而获取时，自定义了CustomerHandlerTypeI实现类CustomerHandlerTypeImpl，而又没有重写hashCode和equals方法，才导致从map中没有获取到我们所想要的CustomerHandler，那么，我们把实现类CustomerHandlerTypeImpl的hashCode和equals这两个方法重写，保持跟动态代理类生成的一样不就行了吗？
那我们就按照AnnotationInvocationHandler中的实现，在我们的CustomerHandlerTypeImpl中按照相同的逻辑重写下这两个方法，如下
@Override
public int hashCode() {
int hashCode = 0;
hashCode += (127 * "type".hashCode()) ^ type.hashCode();
hashCode += (127 * "area".hashCode()) ^ area.hashCode();
return hashCode;
}
@Override
public boolean equals(Object obj) {
if (!(obj instanceof CustomerHandlerType)) {
return false;
}
CustomerHandlerType other = (CustomerHandlerType) obj;
return type.equals(other.type()) && area.equals(other.area());
}
再次运行看看。
