场景:当我们的项目中需要接收到一些数据并且需要对其进行处理时,由于它们来自于不同的渠道(如:阿里、腾讯),不同渠道所需要的处理方式不同,下面我们写一个常见的 Demo 来实现该场景。
1、首先,构建一个基础规则的抽象类 GeneralChannelRule,定义一个抽象方法 process(),不同的渠道都需要实现该抽象方法。
public abstract class GeneralChannelRule {
public abstract void process();
}
2、建立一个简单的枚举类。
public enum ChannelRuleEnum {
ALIBABA("Alibaba"),
TENCENT("Tencent");
private String code;
ChannelRuleEnum(String channel) {
this.code = channel;
}
public String getCode() {
return code;
}
public void setCode(String code) {
this.code = code;
}
}
3、定义一个阿里的规则类,用来处理对于阿里渠道数据的具体处理逻辑。
public class AlibabaChannelRule extends GeneralChannelRule {
@Override
public void process() {
// Alibaba 数据处理逻辑
System.out.println("This is Alibaba");
}
}
4、定义一个腾讯的规则类,用来处理对于腾讯渠道数据的具体处理逻辑。
public class TencentChannelRule extends GeneralChannelRule {
@Override
public void process() {
// Tencent 数据处理逻辑
System.out.println("This is Tencent");
}
}
5、使用规则对数据进行处理。
public class Demo {
public static void main(String[] args) {
// 模拟接收到的数据是来自 Alibaba
String channel = "Alibaba";
GeneralChannelRule rule = null;
if (ChannelRuleEnum.ALIBABA.getCode().equals(channel)) {
rule = new AlibabaChannelRule();
} else if (ChannelRuleEnum.TENCENT.getCode().equals(channel)) {
rule = new TencentChannelRule();
} else {
// 无匹配项
}
// 执行逻辑
rule.process();
}
}
6、如图为实现结果。
通过如上的方式,能够实现需求。
但是存在两个缺点:
其一、当我们需要新增新的渠道的时候,需要对 main 方法中的逻辑进行修改调整。这一点违背了设计模式中的开放封闭规则。开放封闭规则的核心思想是软件实体是可扩展而不可修改的。也就是说,对扩展是开放的,而对修改则是封闭的。
其二、新增渠道后,修改代码会产生大量的 if - else,不太优雅。
为了解决以上的两个问题,我们可以借助枚举类来进行优化。
1、调整一下枚举类,增加一个属性 GeneralChannelRule,且给对应的渠道构建对应的实现类 GeneralChannelRule,同时新增一个匹配方法 match()。
public enum ChannelRuleEnum {
ALIBABA("Alibaba", new AlibabaChannelRule()),
TENCENT("Tencent", new TencentChannelRule());
private String code;
private GeneralChannelRule channelRule;
/**
* 匹配
* @param code 渠道值
* @return
*/
public static ChannelRuleEnum match(String code){
ChannelRuleEnum[] values = ChannelRuleEnum.values();
for (ChannelRuleEnum value : values) {
if(value.code.equals(code)){
return value;
}
}
return null;
}
ChannelRuleEnum(String channel, GeneralChannelRule channelRule) {
this.code = channel;
this.channelRule = channelRule;
}
public GeneralChannelRule getChannelRule() {
return channelRule;
}
public void setChannel(GeneralChannelRule channelRule) {
this.channelRule = channelRule;
}
public String getCode() {
return code;
}
public void setCode(String code) {
this.code = code;
}
}
2、改写程序。
public class Demo {
public static void main(String[] args) {
// 模拟接收到的数据是来自 Tencent
String channel = "Tencent";
ChannelRuleEnum channelRule = ChannelRuleEnum.match(channel);
GeneralChannelRule rule = channelRule.getChannelRule();
rule.process();
}
}
3、如下为实现结果。
通过使用枚举类,在枚举中将 key 与渠道规则具体实现进行绑定。通过优化,可以减少 if - else 使得代码更加优雅。同时,如果需要新增渠道,还需要做两步操作即可完成。这样就不需要改动原有的代码逻辑,符合了开放封闭原则。 操作步骤如下:
- 定义新增渠道实现类,继承 GeneralChannelRule 抽象类,且实现其抽象方法 process()。
- 在枚举类 ChannelRuleEnum 新增对应的渠道实现类。
以上就是通过枚举类的方式来巧妙替代 if - else 的方案,对于替代 if - else 的优秀解决方案还有很多(如:策略模式、工厂模式等),感兴趣的朋友可以查阅相关的资料。