CTO写的代码，真是绝了！-51CTO.COM

最近我看到某厂的 CTO 写的代码，被全网吐槽，我们一起来欣赏一下!

[[338627]]
图片来自 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 使得代码更加优雅。
如果需要新增渠道，我们只需要在编写具体规则实现类并继承 GeneralChannelRule 抽象类，并在枚举类中新增的枚举，而不需要改动到原先的任何代码。这符合了开发封闭原则。

最后

以上是通过枚举来巧妙干掉 if-else 的方案，对于减少 if-else 还有很多有趣的解决方案(如：状态设计模式等)，感兴趣的朋友去查阅相关的资料。

作者：聚 IT

编辑：陶家龙

出处：toutiao.com/i6847406631983153672