java 嵌套if优化_利用策略模式改善代码中的if嵌套

什么是策略模式？其思想是针对一组算法，将每一种算法都封装到具有共同接口的独立的类中，从而是它们可以相互替换。策略模式的最大特点是使得算法可以在不影响客户端的情况下发生变化，从而改变不同的功能。

假如我们有一个根据不同用户类型返回不同折扣的方法，我们的实现可能是这样：

importorg.springframework.stereotype.Service;

@Servicepublic classCashService {public double cash(String type, doublemoney) {if ("svip".equals(type)) {return money * 0.75;

}else if ("vip".equals(type)) {return money * 0.9;

}else{returnmoney;

}

}

}

现在我们各个类型的用户折扣耦合在一起，修改一个用户的折扣力度有可能会对其他类型用户造成影响。根据策略模式的思想，我们需要把折扣力度封装成具体的方法并面向接口编程。我们首先定义公共的接口DiscountService,编写其实现类，则我们改造后的代码可能如下所示:

importcom.study.designer.strategy.NormalDiscountStrategy;importcom.study.designer.strategy.SvipDiscountStrategy;importcom.study.designer.strategy.VipDiscountStrategy;importorg.springframework.beans.factory.annotation.Autowired;importorg.springframework.stereotype.Service;

@Servicepublic classCashService {

@AutowiredprivateSvipDiscountStrategy svipDiscountStrategy;

@AutowiredprivateVipDiscountStrategy vipDiscountStrategy;

@AutowiredprivateNormalDiscountStrategy normalDiscountStrategy;public double cash(String type, doublemoney) {if ("svip".equals(type)) {returnsvipDiscountStrategy.getMoney(money);

}else if ("vip".equals(type)) {returnvipDiscountStrategy.getMoney(money);

}else{returnnormalDiscountStrategy.getMoney(money);

}

}

}

可以看到，改造后的CashService中还存在许多if判断，我们需要消除这些if判断。我们可以在CashService初始化时就获取到所有的折扣策略，然后根据具体类型计算具体折扣。获取所有策略可以交由Spring来完成,改造后的代码如下所示：

importcom.study.designer.strategy.inf.DiscountStrategy;importorg.springframework.stereotype.Service;importjava.util.HashMap;importjava.util.List;importjava.util.Map;

@Servicepublic classCashService {private Map strategyMap = new HashMap<>();public CashService(Liststrategies) {for(DiscountStrategy strategy : strategies) {

strategyMap.put(strategy.getType(), strategy);

}

}public double cash(String type, doublemoney) {returnstrategyMap.get(type).getMoney(money);

}

}