我在之前的文章中使用枚举消除了if-else语句
这次我采用其他方式消除if-else。
你在平时开发中肯定有使用if-else语句的时候,然而大量的if-else语句不利于代码阅读,影响代码复杂度。反正我在消除Sonar异味的时候头疼过。
之前公司系统的代码中也存在if-else过多问题,导致代码不优雅,这里为了讲解,我将业务逻辑简化。
假如我有一个计算实际价格的方法,要求输入用户的级别和商品的实际价格,返回商品的真实价格,代码如下:
这里level
就是用户的级别,originalPrice
就是商品的原始价格。如果用户级别为"normal",返回的真实价格为原始价格乘以1.0。如果用户级别为"vip",返回的真实价格为原始价格乘以0.9。否则返回原始价格。
测试代码如下:
结果"vip"的价格为:
getActualPrice()测试的真实价格为:90.0
上面的代码你可能看着没毛病,但是用户的等级可能还有超级vip(svip),还可能有ssvip。还可能更多,那么这个方法if-else将会非常多。并且会经常修改这个方法,违背了设计原则中的开闭原则。为此,我思考能不能不修改这个方法也能实现level
级别的扩展呢。
想来想去,我决定使用设计模式来消除if-else。
public interface CalculateStrategy {
String userLevel();
BigDecimal discount(BigDecimal originalPrice);
}
这里先写一个计算的接口,定义了两个方法:获取用户等级,和相应价格计算的方法。
@Component
public class NormalStrategy implements CalculateStrategy {
@Override
public String userLevel() {
return "normal";
}
@Override
public BigDecimal discount(BigDecimal originalPrice) {
return originalPrice.multiply(new BigDecimal("1.0"));
}
}
normal级别的实现类。
@Component
public class VipStrategy implements CalculateStrategy {
@Override
public String userLevel() {
return "vip";
}
@Override
public BigDecimal discount(BigDecimal originalPrice) {
return originalPrice.multiply(new BigDecimal("0.9"));
}
}
vip级别的实现类。
我们计算真实价格的代码如下:
这里我创建了一个map,map的key存放用户级别level
,value存放CalculateStrategy
的对象。当系统启动时,IoC容器会给SaleService
创建对象,这时map会存入所有实现了CalculateStrategy
的对象。
然后编写一个getActualPriceWithStrategy()
方法,用于根据用户级别计算商品实际价格。如果增加用户级别,只需要增加相应的类,如:SvipStrategy
、SSvipStrategy
等。这样 就不用修改getActualPriceWithStrategy()
中的内容了,保证了这个方法的安全性。
测试代码如下:
测试结果:
getActualPriceWithStrategy()测试的真实价格为:90.0
其实上面我有使用到了设计模式中的策略模式,我将if-else中算法抽取出来,单独封装到了类中,这样使算法独立于调用者,如果改变算法也只需要改变算法类,不需要改变调用者。
当然只是用策略模式还不能消除if-else,大家可以思考下:
“上述代码还有用到其他设计模式么?”
扫码关注腾讯云开发者
领取腾讯云代金券
Copyright © 2013 - 2025 Tencent Cloud. All Rights Reserved. 腾讯云 版权所有
深圳市腾讯计算机系统有限公司 ICP备案/许可证号:粤B2-20090059 深公网安备号 44030502008569
腾讯云计算(北京)有限责任公司 京ICP证150476号 | 京ICP备11018762号 | 京公网安备号11010802020287
Copyright © 2013 - 2025 Tencent Cloud.
All Rights Reserved. 腾讯云 版权所有