使用枚举来巧妙干掉 if-else，使代码更加优雅

不念不忘少年蓝@ 2021-09-01 04:48 271阅读 0赞

点击上方 [Java后端][Java]，选择 设为星标

优质文章，及时送达

--------------------

作者：聚IT

来源：toutiao.com/i6847406631983153672

本文通过一个简单的例子来展示如何通过枚举巧妙地干掉if-else，使代码看起来更佳优雅。

场景：当我们接收到一些数据需要对其进行处理时，由于它们来自于不同的渠道（如：腾讯，头条），不同渠道所需的处理方式不同，下面我们写一个简单Demo来实现该的场景。

我把 Java基础 相关的文章整理成了 PDF，关注微信公众号 Java后端，回复 666 下载。

## **解决思路** ##

1、首先构建一个 GeneralChannelRule 基础规则抽象类，定义一个抽象方法process()，不同的渠道都需要实现该抽象方法。

public abstract class GeneralChannelRule {
     public abstract void process();
    }

2、编写一个腾讯的规则类，定义具体对于腾讯渠道数据的处理逻辑

public class TencentChannelRule extends GeneralChannelRule
       @Override
        public void process() {
            // Tencent处理逻辑
        }
    }

3、编写一个头条的规则类，定义具体对于头条数据的处理逻辑

public class TouTiaoChannelRule extends GeneralChannelRule
       @Override
        public void process() {
            // TouTiao处理逻辑
        }
    }

4、建立一个简单的枚举类

public enum ChannelRuleEnum {
        /**
         * 头条
         */
        TOUTIAO("TOUTIAO"),
        /**
         * 腾讯
         */
        TENCENT("TENCENT"),
        ;
      ....
    }

5、使用规则对数据进行处理。

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();
        }

解析：如果通过上面的方式，则存在则两个缺点。

当我们需要新增新的渠道的时候，需要对main方法中的逻辑进行修改调整。这违背了设计模式中的开放封闭规则。开放封闭原bai则的核心的思想是软件实体是可扩du展，而不可zhi修改的。

也就是说，对扩展是开dao放的，而对修改是封闭的

新增渠道后，修改代码会产生大量的if else，不太优雅。为了解决以上的两个问题，我们可以借助枚举类来巧妙优化。

## **新的思路** ##

1、下面我们调整一下枚举类，增加一个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;
        }
    }

2、改写程序

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 还有很多有趣的解决方案（如：状态设计模式等），感兴趣的朋友去查阅相关的资料。

**- END -**

最近整理一份面试资料**《Java技术栈学习手册》**，覆盖了Java技术、面试题精选、Spring全家桶、Nginx、SSM、微服务、数据库、数据结构、架构等等。

获取方式：点“ **在看**，关注公众号 **Java后端** 并回复 **777** 领取，更多内容陆续奉上。

**推****荐****阅****读 **

***1.*** [如下的准备可以大大提升面试成功率?][Link 1]

***2. ***[Docker 入门终极指南：边学边用][Docker]

***3.*** [卧槽！GitHub 又诞生一个沙雕项目？！][GitHub]

***4.*** [15 种微服务架构框架][15]

***5.*** [Linux 用着太爽啦！！！][Linux]

![format_png][]

**喜****欢****文****章****，****点****个****在看 **![format_png 1][]

[Java]: http://mp.weixin.qq.com/s?__biz=Mzg2MjEwMjI1Mg%3D%3D&chksm=ce0da17bf97a286df119047a25b7de51621fe542816673bb4aa3ebd8ab97b09a86ed7db2e2ee&idx=3&mid=2247489272&scene=21&sn=ce74938072fc73a808d83ccb474db46d#wechat_redirect
[Link 1]: http://mp.weixin.qq.com/s?__biz=Mzg2MjEwMjI1Mg%3D%3D&chksm=ce0e49d8f979c0cef3d9dffe050258555c4d82a7e9d1b1137f3c2b1aa6b6477581e42c907ef0&idx=2&mid=2247499355&scene=21&sn=7dad79948b393a7e330e5342f0bba6e3#wechat_redirect
[Docker]: http://mp.weixin.qq.com/s?__biz=Mzg2MjEwMjI1Mg%3D%3D&chksm=ce0e49bef979c0a8412ca40629a6a9a7d94a0f94930256f2d9a4f097975b0e7890c9308663f3&idx=2&mid=2247499325&scene=21&sn=19d106931001aa1012e2921af5e551f6#wechat_redirect
[GitHub]: http://mp.weixin.qq.com/s?__biz=Mzg2MjEwMjI1Mg%3D%3D&chksm=ce0e4a68f979c37ea521ea31d6fea56f6fa34ff81950e032e8c3d1b744605790a4a7cc13f51f&idx=1&mid=2247499243&scene=21&sn=3294e9ec235c16fac227b11942abecf7#wechat_redirect
[15]: http://mp.weixin.qq.com/s?__biz=Mzg2MjEwMjI1Mg%3D%3D&chksm=ce0e49bef979c0a8b7a248b77d1394e0a56931917569e0a6676545d5df3a19dd7a251ed94b53&idx=4&mid=2247499325&scene=21&sn=305b6a6d9cd5de79ccef162af6b90ead#wechat_redirect
[Linux]: http://mp.weixin.qq.com/s?__biz=Mzg2MjEwMjI1Mg%3D%3D&chksm=ce0e4ac1f979c3d7009a5a4a682e28959bd67c64ec87cffbecea8f6fc7322dd974a237da8128&idx=1&mid=2247499074&scene=21&sn=4454cf7653521fb834dabd16a064d269#wechat_redirect
[format_png]: /images/20210730/3b4a79ba69674d9e8479a7941f1d8424.png
[format_png 1]: /images/20210730/c7b9e0cdf2b24b1a8f39197227d02d8d.png