替代if else 的方法---巧用枚举类和抽象方法

替代if else 的方法—巧用枚举类和抽象方法

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

场景复现

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

package com.cyc.basic.test.rule;

public abstract class GeneralChannelRule {
    /**
     * 处理逻辑
     */
    public abstract void process();
}

2. 编写一个腾讯的规则类，设计对来及腾讯渠道的数据进行处理逻辑

package com.cyc.basic.test.rule;

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

3. 编写一个巨量的规则类，定义具体对于巨量数据的处理逻辑

package com.cyc.basic.test.rule;


public class OceanEngineChannelRule extends GeneralChannelRule{
   @Override
    public void process() {
        // 巨量处理逻辑
    }
}

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

package com.cyc.basic.test.rule;

import lombok.Getter;

@Getter
public enum ChannelRuleEnum {
    /**
     * 巨量
     */
    OCEANENGINE("OCEANENGINE"),
    /**
     * 腾讯
     */
    TENCENT("TENCENT");

    public String code;

    ChannelRuleEnum(String code) {
        this.code = code;
    }
}

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

package com.cyc.basic.test.rule;

import com.cyc.common.exception.ServiceException;

public class ProcessTest {
    public static void main(String[] args) {
        //模拟接收到的数据，其渠道为为巨量(OCEANENGINE)，来自巨量的数据
        String sign = "OCEANENGINE";
        GeneralChannelRule rule;
        //根据对应渠道获取相应的具体规则实现类
        if (ChannelRuleEnum.TENCENT.code.equals(sign)) {
            rule = new TencentChannelRule();
        } else if (ChannelRuleEnum.OCEANENGINE.code.equals(sign)) {
            rule = new OceanEngineChannelRule();
        } else {
            //匹配不到,执行其他处理逻辑...
            throw new ServiceException("400", "传参有误");
        }
        //执行对应的数据处理
        rule.process();
    }
}

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

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

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

新增渠道后，修改代码会产生大量的if else，后期维护起来很费劲。为了解决以上两个问题，可以借助枚举类来巧妙优化。

使用枚举处理

1. 下面调整一下枚举类，枚举类中增加一个GeneralChannelRule属性，并且给对应渠道构建对应的GeneralChannelRule实现类，枚举类中新增一个matchRule() 匹配方法。

package com.cyc.basic.test.rule;

import lombok.Getter;

/**
 * @author fei
 */

@Getter
public enum ChannelRuleEnum {
    /**
     * 巨量
     */
    OCEANENGINE("OCEANENGINE"),
    /**
     * 腾讯
     */
    TENCENT("TENCENT");

    public String code;

    public GeneralChannelRule channel;

    ChannelRuleEnum(String code, GeneralChannelRule channel) {
        this.code = code;
        this.channel = channel;
    }

    //匹配规则
    public static ChannelRuleEnum matchRule(String code){
        ChannelRuleEnum[] values = ChannelRuleEnum.values();
        for (ChannelRuleEnum value : values) {
            if(value.code.equals(code)){
                return value;
            }
        }
        return null;
    }

    ChannelRuleEnum(String code) {
        this.code = code;
    }
}

2、改写程序

public class ProcessTestNew {
    public static void main(String[] args) {
        String sign = "TOUTIAO";
        ChannelRuleEnum channelRule = ChannelRuleEnum.matchRule(sign);
        if (Objects.isNull(channelRule)) {
            throw new ServiceException("400", "没有匹配上对应规则");
        }
        GeneralChannelRule rule = channelRule.channel;
        rule.process();
    }
}

解析：通过使用枚举类，在枚举中将 key 与 规则具体实现进行绑定。经过此番优化,

可以减少if -else使得代码更加简介，后期如果需要新增渠道，只需要在编写具体规则实现类并继承GeneralChannelRule抽象类，并在枚举类中新增的对应枚举，而不需要改动原先的任何代码。符合开发封闭原则。