0
点赞
收藏
分享

微信扫一扫

深度解析：如何替换掉代码中的ifelse，我女朋友看完都会了！

亿奇学 2021-09-30 阅读 18
日记本

代码实现

代码实现如下（注意下面所有的代码都不能直接运行，只是关键逻辑部分的示例代码）：

烂代码示例

我们有一个短信发送类：SmsSendService，里面有一个send方法发送短信
SmsSendService.java

public class SmsSendService{
    /**
     * @Param phoneNo 手机号
     * @Param content 短信内容
     */
    public void send(String phoneNo,String content){
        //从配置中读取 短信渠道
        String channelType=config.getChannelType();
        
        //如果是短信渠道A，则调用渠道A的api发送
        if(Objects.equals(channelType,"CHANNEL_A")){
            System.out.println("通过短信渠道A发送短信");
        }
        //如果是短信渠道B，则调用渠道B的api发送
        else if(Objects.equals(channelType,"CHANNEL_B")){
            System.out.println("通过短信渠道B发送短信");
        }
    }
}

如果某天增加了一个短信渠道C，那么接着追加一个”else if…"

//... 此处省略部分代码 ...

//从配置中读取 短信渠道
String channelType=config.getChannelType();
//如果是短信渠道A，则调用渠道A的api发送
if(Objects.equals(channelType,"CHANNEL_A")){
    System.out.println("通过短信渠道A发送短信");
}
//如果是短信渠道B，则调用渠道B的api发送
else if(Objects.equals(channelType,"CHANNEL_B")){
    System.out.println("通过短信渠道B发送短信");
}
//ADD: 如果是短信渠道C，则调用渠道C的api发送
else if(Objects.equals(channelType,"CHANNEL_C")){
    System.out.println("通过短信渠道C发送短信");
}

//... 此处省略部分代码 ...

接下来我们把代码优化一下：

优化代码1

定义一个短信渠道的接口 SmsChannelService，所有的短信渠道API都实现该接口；
短信渠道接口 SmsChannelService.java

public interface SmsChannelService{
    //发送短信
    void send(String phoneNo,String content);
}

短信渠道A SmsChannelServiceImplA.java

public class SmsChannelServiceImplA implements SmsChannelService {
    public void send(String phoneNo, String content) {
        System.out.println("通过短信渠道A发送短信");
    }
}

短信渠道B SmsChannelServiceImplB.java

public class SmsChannelServiceImplB implements SmsChannelService {
    public void send(String phoneNo, String content) {
        System.out.println("通过短信渠道B发送短信");
    }
}

通过工厂类来初始化所有短信渠道service
SmsChannelFactory.java

public class SmsChannelFactory {
    private Map<String,SmsChannelService> serviceMap;

    //初始化工厂，将所有的短信渠道Service放入Map中
    public SmsChannelFactory(){
        //渠道类型为 key , 对应的服务类为value ：
        serviceMap=new HashMap<String, SmsChannelService>(2);
        serviceMap.put("CHANNEL_A",new SmsChannelServiceImplA());
        serviceMap.put("CHANNEL_B",new SmsChannelServiceImplB());
    }

    //根据短信渠道类型获得对应渠道的Service
    public SmsChannelService buildService(String channelType){
        return serviceMap.get(channelType);
    }
}

在原来的SmsSendService中调用不同短信渠道的接口。
原来的 SmsSendService 类优化如下

public class SmsSendService {

    private SmsChannelFactory smsChannelFactory;

    public SmsSendService(){
        smsChannelFactory=new SmsChannelFactory();
    }

    public void send(String phoneNo,String content){
        //从配置中读取 短信渠道
        String channelType=config.getChannelType();
        //获取渠道类型对应的服务类
        SmsChannelService channelService=smsChannelFactory.buildService(channelType);
        //发送短信
        channelService.send(phoneNo,content);
    }

}

增加短信渠道C的实现 SmsChannelServiceImplC.java

public class SmsChannelServiceImplC implements SmsChannelService {
    public void send(String phoneNo, String content) {
        System.out.println("通过短信渠道C发送短信");
    }
}

修改工厂类 SmsChannelFactory.java

public class SmsChannelFactory {
    private Map<String,SmsChannelService> serviceMap;

    //初始化 serviceMap ，将所有的短信渠道Service放入Map中
    public SmsChannelFactory(){
        //渠道类型为 key , 对应的服务类为value ：
        serviceMap=new HashMap<String, SmsChannelService>(3);
        serviceMap.put("CHANNEL_A",new SmsChannelServiceImplA());
        serviceMap.put("CHANNEL_B",new SmsChannelServiceImplB());
        //ADD 增加一行 SmsChannelServiceImplC 的初始化代码 
        serviceMap.put("CHANNEL_C",new SmsChannelServiceImplC());
    }

    //根据渠道类型构建短信渠道Service
    public SmsChannelService buildService(String channelType){
        return serviceMap.get(channelType);
    }
}

我们通过使用spring的依赖注入进一步优化代码：

优化代码2

SmsChannelService 接口增加 getChannelType() 方法，这一步很关键。

public interface SmsChannelService {
    //发送短信
    void send(String phoneNo,String content);
    //关键：增加getChannelType()方法，子类实现这个方法用于标识出渠道类型
    String getChannelType();
}

子类增加该方法的实现，并加上 @Service 注解，使其让spring容器管理起来
SmsChannelServiceImplA.java

@Service
public class SmsChannelServiceImplA implements SmsChannelService {
    public void send(String phoneNo, String content) {
        System.out.println("通过短信渠道A发送短信");
    }
    //关键：增加 getChannelType() 实现
    public String getChannelType() {
        return "CHANNEL_A";
    }
}

SmsChannelServiceImplB.java

@Service
public class SmsChannelServiceImplB implements SmsChannelService {
    public void send(String phoneNo, String content) {
        System.out.println("通过短信渠道B发送短信");
    }
    //关键：增加 getChannelType() 实现
    public String getChannelType() {
        return "CHANNEL_B";
    }
}

修改 SmsChannelFactory 类： 这一步也很关键。
SmsChannelFactory.java

@Service
public class SmsChannelFactory {

    private Map<String,SmsChannelService> serviceMap;
    
    /*注入：通过spring容器将所有实现 SmsChannelService 接口的类的实例注入到 serviceList 中*/
    @Autowired
    private List<SmsChannelService> serviceList;

    /*通过 @PostConstruct 注解，在 SmsChannelFactory 实例化后，来初始化 serviceMap */
    @PostConstruct
    private void init(){
        if(CollectionUtils.isEmpty(serviceList)){
            return ;
        }
        serviceMap=new HashMap<String, SmsChannelService>(serviceList.size());
        //将 serviceList 转换为 serviceMap
        for (SmsChannelService channelService : serviceList) {
            String channelType=channelService.getChannelType();
            //重复性校验，避免不同实现类的 getChannelType() 方法返回同一个值。
            if(serviceMap.get(channelType)!=null){
                throw new RuntimeException("同一个短信渠道只能有一个实现类");
            }
            /*渠道类型为 key , 对应的服务类为value ：
            与“优化代码1”中的通过手工设置“CHANNEL_A"、"CHANNEL_B"相比，
            这种方式更加自动化，后续在增加“CHANNEL_C"无需再改此处代码*/
            serviceMap.put(channelType,channelService);
        }
    }

    //根据渠道类型获取对应短信渠道的Service
    public SmsChannelService buildService(String channelType){
        return serviceMap.get(channelType);
    }
}

SmsSendService 加上 @Service 注解。通过 @Autowired 注入 SmsChannelFactory
SmsSendService.java

@Service
public class SmsSendService {

    @Autowired
    private SmsChannelFactory smsChannelFactory;

    public void send(String phoneNo,String content){
        //从配置中读取短信渠道类型
        String channelType=config.getChannelType();
        //构建渠道类型对应的服务类
        SmsChannelService channelService=smsChannelFactory.buildService(channelType);
        //发送短信
        channelService.send(phoneNo,content);
    }

}

SmsChannelServiceImplC.java

@Service
public class SmsChannelServiceImplC implements SmsChannelService {
    public void send(String phoneNo, String content) {
        System.out.println("通过短信渠道C发送短信");
    }

    public String getChannelType() {
        return "CHANNEL_C";
    }
}

总结
通过上述优化很好的去掉了 “if else" ，再也不会出现”又臭又长“像”卫生卷纸"一样的代码了，而且完全遵循”开闭原则"。
spring是个好东西，关键看你怎么用。

看完有什么不懂的欢迎在下方留言评论，记得点个赞哦！

举报

相关推荐

Android 如何实现气泡选择动画，食堂大妈看完都会了

架构移动开发android

如果我是你的女朋友(贴图)

时间段云平台云计算

【Nginx】如何实现Nginx的高可用负载均衡？看完我也会了！！

随笔

热度直逼TensorFlow的深度学习框架，我用它画下女朋友最美的脸

深度学习神经网络卷积神经网络JavaScript前端开发

女朋友要我讲解@Controller注解的原理，真是难为我了

数据库pythonjavawindows服务器Python后端开发

我背着女朋友，用 Python 偷偷抓取了她的行踪

Python编程网赚待管理python前端UI参考蜂采前端

mac如何恢复被同名替换掉的文件夹 mac文件被替换如何恢复

功能测试

深度分析：mybatis的底层实现原理，看完你学会了吗？

日记本

学完Python面向对象，我觉的我有女朋友了!哈哈哈.....

面向对象类Python进阶Python基础父类Python编程语言

女朋友看了我写的项目，说太LOW了，于是我撸了一天

随笔Java 杂谈Spring-...开发技巧
0 条评论