使用策略模式+工厂模式干掉代码中过多的if-else

Posted 2022-02-03 赵侠客

过多if-else项目背景

如果一开始就知道现在的业务需要，大部分人都不会在代码里添加过多的if-else判断的，烂代码基本都是刚开始写代码时并没有太多的需求，随着期需求不断的修改增加，开发时间也较的紧张，代码往往都是怎么快速怎么写。当然多写一个if-else比使用各种设计模式肯定来的更快速了，这也就导致项目代码慢慢变得臃肿，难以维护的主要原因。在有空闲时间的情况下就可以给以前的代码做一次手术了。
先看本次未优化前的代码：

    @Override
    public MMediaInfo copyToLibType(UserQiniuDTO userQiniuDTO, Long mediaId, Integer libType) 
        MMediaInfo mMediaInfo = new MMediaInfo();
        LibTypeEnum libTypeEnum = LibTypeEnum.valueof(libType);
        if (libTypeEnum.getFileType().equals(FileType.VIDEO)) 
		 	 mMediaInfo = mvideoInfoService.copyToLibType(userQiniuDTO, mediaId, libType);
        
        if (libTypeEnum.getFileType().equals(FileType.AUDIO)) 
            mMediaInfo = mAudioInfoService.copyToLibType(userQiniuDTO, mediaId, libType);
        
        if (libTypeEnum.getFileType().equals(FileType.PICTURE)) 
             mMediaInfo = mImgInfoService.copyToLibType(userQiniuDTO, mediaId, libType);
        
        if (libTypeEnum.getFileType().equals(FileType.FILE)) 
            mMediaInfo= mFileInfoService.copyToLibType(userQiniuDTO, mediaId, libType);
        
        return mMediaInfo;
    

之所有会有些段代码，主要原因是项目前期，需求中只有 视频/音频的 ，并且音频操作接口，逻辑都是分开的，后面又增加了图片/文件需求，并且要求视频/音频/文件/图片 能合并在同一接口处理， 这就导致了代码中很多操作有大量 if-else判断
再看优化后的代码：

    @Override
    public MMediaInfo copyToLibType(UserQiniuDTO userQiniuDTO, Long mediaId, Integer libType) 
        LibTypeEnum libTypeEnum = LibTypeEnum.valueof(libType);
        IFileService iFileService = FileFactory.getFileService(libTypeEnum.getFileType());
        return   iFileService.copyToLibType(userQiniuDTO, mediaId, libType);
    

此断代码看起来就清爽多了，实现的基本思路就是使用了 策略模式+工厂模式，代码通过不同的文件类型返回对应的实现类来实现复制逻辑

基本视频步骤

1. 创建IFileService接口

public interface IFileService 
    Integer getFileType();
    MMediaInfo copyToLibType(UserQiniuDTO userQiniuDTO, Long mediaId, Integer libType);

2. 视频/音频/图片/文件类分别视频IFileService接口

public class MVideoInfoServiceImpl extends ServiceImpl<MvideoInfoMapper, MvideoInfo> implements MvideoInfoService, IFileService 
    
    @Override
    public Integer getFileType() 
        return FileType.VIDEO;
    
    @Override
    public MMediaInfo copyToLibType(UserQiniuDTO userQiniuDTO, Long id, Integer libType) 
 		//复制视频逻辑
 		
        return mMediaInfo;
    

3. 创建FileFactory工厂类

@Component
public class FileFactory implements ApplicationContextAware 
 
    private static Map<Integer, IFileService> fileServiceMap;
 
    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException 
        Map<String, IFileService> map = applicationContext.getBeansOfType(IFileService.class);
        fileServiceMap = new HashMap<>();
        map.forEach((key, value) -> fileServiceMap.put(value.getFileType(), value));
    
 
    public static <T extends IFileService> T getFileService(Integer fileType) 
        return (T)fileServiceMap.get(fileType);
    
 

总结

copyToLibType只是优化的一个方式，IFileService中还有很多 如 移动/删除 等操作的代码也可以干掉过多的if-else了。