【策略模式在项目中的应用】

红太狼 2024-03-25 10:11 9阅读 0赞

**需求:**

上传同样的Excel文件，根据入参类型选择对应的实现类来处理这个文件，使用不同的解析方法并存储不同的数据表中。

**分析：**

一共是6种Excel文件，对应六套解析方法，相当于六个模板对应六种解析方式。

方案一：六个文件对应六个接口——大量重复的代码——太low，Pass~~

方案二：一个接口，根据传入的文件类型type选择对应的实现方法，一个type对应一种解析方法——用if或者case判断，不够美观，考虑使用更加优雅的写法~

方案三：考虑使用策略模式，把type和实现类接口以key-value形式存储到Map中，实现类里面有一个共同的接口方法，写六个不同的实现类实现这个接口方法方法，根据传入的type类型，拿到对应的实现类，实现类里面对应具体的业务代码。

**直接上代码:**

**业务接口：**

public interface AchievementImportService {
    
    
        /**
         * 批量导入数据
         * @param file
         */
        void importExcel(String type, MultipartFile file);
    
    }

**实现类：**

@Slf4j
    @Service
    public class AchievementImportServiceImpl implements AchievementImportService {
    
    @Autowired
    Map<String, AchieveImportStrategy> importStrategyMap;
    
       
    @Override
    public void importExcel(String type, MultipartFile file) {
            importStrategyMap.get(TemplateTypeEnum.getServiceByCode(type)).achieveImportExcel(file);
        }
    
    }

**策略模式接口：**

public interface AchieveImportStrategy {
    
        /**
         * 导入excel
         * @param file
         */
        void achieveImportExcel(MultipartFile file);
    }

**拿其中的一个实现类举例：**

//@Service（括号里面是是这个实现类的自定义名称，一共6中不同的实现类）
    @Service(TemplateTypeConstant.COPYRIGHT_ENTRY)
    @Slf4j
    public class CopyrightEntryImportImpl implements AchieveImportStrategy {
    
    
    /**
         * 导入excel
         *
         * @param file
         */
        @Override
        public void achieveImportExcel(MultipartFile file) {
    
            //1.插入一条导入记录
      
    
            //2.对应这个模板文件的具体解析方法（异步）
           
        }
    
    
    }

**控制层（controller）：**

@RestController
    @RequestMapping("xxx/achievement")
    @Api(tags = "批量导入文件")
    @Slf4j
    public class AchievementImportController {
    
    @Resource
    private AchievementImportService achievementImportService;
    
    @RequestMapping(value = "/import", method = RequestMethod.POST)
    public BaseResponse importPatentDeclaration(@RequestParam("file") MultipartFile file, @RequestParam("type") String type) {
            achievementImportService.importExcel(type, file);
            return BaseResponse.success("批量导入中，请稍后查看结果");
        }
    
    }

**总结：**

这样的写法也存在一定的重复代码，但是避免了大量的if-else结构，看上去也美观优雅了一些，以上就是我在实际工作中用到的策略模式的分享。

2022.03.12