最近我看到某廠的 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