“
最近我看到某厂的 CTO 写的代码,被全网吐槽,我们一起来欣赏一下!
图片来自 Pexels
本文通过一个简单的例子来展示如何通过枚举巧妙地干掉 if-else,使代码看起来更佳优雅。
场景:当我们接收到一些数据需要对其进行处理时,由于它们来自于不同的渠道(如:腾讯,头条),不同渠道所需的处理方式不同,下面我们写一个简单 Demo 来实现该的场景。
解决思路
①首先构建一个 GeneralChannelRule 基础规则抽象类,定义一个抽象方法process(),不同的渠道都需要实现该抽象方法。
public abstract class GeneralChannelRule {
public abstract void process();
}
②编写一个腾讯的规则类,定义具体对于腾讯渠道数据的处理逻辑。
代码如下:
public class TencentChannelRule extends GeneralChannelRule
@Override
public void process() {
// Tencent处理逻辑
}
}
③编写一个头条的规则类,定义具体对于头条数据的处理逻辑。
代码如下:
public class TouTiaoChannelRule extends GeneralChannelRule
@Override
public void process() {
// TouTiao处理逻辑
}
}
④建立一个简单的枚举类。
代码如下:
public enum ChannelRuleEnum {
/**
* 头条
*/
TOUTIAO("TOUTIAO"),
/**
* 腾讯
*/
TENCENT("TENCENT"),
;
....
}
⑤使用规则对数据进行处理。
代码如下:
public static void main(String[] args) {
//这里我们模拟接收到的数据,其渠道为为TOUTIAO,来自头条的数据
String sign = "TOUTIAO";
GeneralChannelRule rule;
//根据对应渠道获取对应的具体规则实现类
if (ChannelRuleEnum.TENCENT.code.equals(sign)) {
rule = new TencentChannelRule();
} else if (ChannelRuleEnum.TOUTIAO.code.equals(sign)) {
rule = new TouTiaoChannelRule();
} else {
//匹配不到
}
//执行
rule.process();
}
解析:如果通过上面的方式,则存在则两个缺点。
a.当我们需要新增新的渠道的时候,需要对 main 方法中的逻辑进行修改调整。
这违背了设计模式中的开放封闭规则。开放封闭原则的核心的思想是软件实体是可扩展,而不可修改的。也就是说,对扩展是开放的,而对修改是封闭的。
b.新增渠道后,修改代码会产生大量的 if else,不太优雅。
为了解决以上的两个问题,我们可以借助枚举类来巧妙优化。
新的思路
①下面我们调整一下枚举类,增加一个 GeneralChannelRule 属性,并且给对应渠道构建对应的 GeneralChannelRule 实现类,新增一个 match() 匹配方法。
代码如下:
public enum ChannelRuleEnum {
/**
* 头条
*/
TOUTIAO("TOUTIAO",new TouTiaoChannelRule()),
/**
* 腾讯
*/
TENCENT("TENCENT",new TencentChannelRule()),
;
public String name;
public GeneralChannelRule channel;
ChannelRuleEnum(String name, GeneralChannelRule channel) {
this.name = name;
this.channel = channel;
}
//匹配
public static ChannelRuleEnum match(String name){
ChannelRuleEnum[] values = ChannelRuleEnum.values();
for (ChannelRuleEnum value : values) {
if(value.name.equals(name)){
return value;
}
}
return null;
}
public String getName() {
return name;
}
public GeneralChannelRule getChannel() {
return channel;
}
}
②改写程序,代码如下:
public static void main(String[] args) {
String sign = "TOUTIAO";
ChannelRuleEnum channelRule = ChannelRuleEnum.match(sign);
GeneralChannelRule rule = channelRule.channel;
rule.process(sign);
}
解析:通过使用枚举类,在枚举中将 key 与规则具体实现进行绑定。
通过改变:
最后
以上是通过枚举来巧妙干掉 if-else 的方案,对于减少 if-else 还有很多有趣的解决方案(如:状态设计模式等),感兴趣的朋友去查阅相关的资料。
作者:聚 IT
编辑:陶家龙
出处:toutiao.com/i6847406631983153672
在公众号菜单中可自行获取专属架构视频资料,包括不限于 java架构、python系列、人工智能系列、架构系列,以及最新面试、小程序、大前端均无私奉献,你会感谢我的哈
往期热门文章: