策略模式简单运用

策略模式简单运用

背景是 根据不同的渠道，处理不同的任务，相比if else 可以不用动主逻辑代码，代码可维护性好点，每次处理不同渠道的任务只需要改对应的实现，缺点就是的增加类的数量，核心就是多态。

1. 定义接口

public interface TaskFileHandleService {

    /**
     * 获取任务类型
     * @return
     */
    TaskTypeEnum getTaskType();

    /**
     * 任务文件处理
     * @param task
     */
    void taskFileHandle(Task task);

}

2. 定义宏观类

注入 所有接口实现类，循环匹配即可。

@Service
public class TaskFileHandleStrategy {

    @Autowired
    private List<TaskFileHandleService> taskFileHandleServiceList;

    /**
     * 处理逻辑
     * @param task
     */
    public void handle(Task task){
        for (int i = 0; i < taskFileHandleServiceList.size(); i++) {
            TaskFileHandleService taskFileHandleService = taskFileHandleServiceList.get(i);
            TaskTypeEnum taskType = taskFileHandleService.getTaskType();
            if(Objects.equals(TaskTypeEnum.of(task.getTaskType()),taskType)){
                taskFileHandleService.taskFileHandle(task);
            }
        }
    }
}

3.使用

注入对应的类 taskFileHandleStrategy.handle(task) 即可，主逻辑不用动，每次新加任务,加一个具体实现类即可，也就是所谓面向 接口编程的灵活性。

  @Autowired
  private TaskFileHandleStrategy taskFileHandleStrategy;

  
  @Override
  public void handleTaskJob(List<Task> taskList) {
        if(CollectionUtil.isNotEmpty(taskList)){
            // 根据任务类型解析文件
            taskList.stream().forEach(task -> {
                // 导出任务 线程池 处理
                if(Objects.equals(TaskTypeEnum.DOWNLOAD_AIRCRAFT_CHEF.getValue(),task.getTaskType())){
                    executor.execute(() ->{
                        doTask(task);
                    });
                }else {
                    doTask(task);
                }
            });
        }
    }


 /**
     * 做任务
     * @param task
     */
    private void doTask(Task task){
        try {
            handleTask(task);
        }catch (Exception e){
            log.error("处理任务失败 任务id={} 任务名称={} 任务信息",task.getId(),task.getName(),e);
            task.setFailReason(ExceptionUtil.stacktraceToOneLineString(e,Constants.EXCEPTION_MSG_SIZE));
            handleFailed(task);
        }
    }

   /**
     * 处理任务
     * @param task
     */
    private void handleTask(Task task) {
        Task newTask = new Task();
        newTask.setId(task.getId());
        newTask.setStatus(TaskStatusEnum.PROCESSING.getValue());
        newTask.setStartTime(LocalDateTime.now());
        updateById(newTask);
        taskFileHandleStrategy.handle(task);
        handleSuccess(task);
    }