酷应用

CTO写的代码，真是绝了！

百家作者：51CTO技术栈 2020-08-19 21:51:56

“

最近我看到某厂的 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 与规则具体实现进行绑定。

通过改变：