量化交易系统任务框架的演化之路（4）用Push方式解决任务之间的依赖

在上一篇文章《量化交易系统任务框架的演化之路（3）基于多状态的任务依赖解决》中提供了一种利用数据库作为状态数据交换媒介的解决方案，通过这种方法，实现了依赖和被依赖的任务之间解耦，但是问题也十分明显，就是轮询被依赖任务的状态的做法会带来无谓的计算资源的浪费，一旦任务过多，无论是对数据库还是对任务系统本身都会造成一定压力。我们称把这种方法称为Pull，也就是拉取的思路；那么与之对应的就是推送（Push）的解决方法，依赖任务不再主动查询被依赖任务的状态，而是等待被依赖任务完成后，主动将状态推送给依赖任务，然后依赖任务才开始执行。
可是仔细思考一下，就会发现一个问题，如果任务很多，并且依赖关系很复杂，那么被依赖任务，要是想搞明白都谁依赖了它，也会十分困难，并且从项目开发的时间维度来看，多数依赖任务的开发是在被依赖任务的上线之后。因此如果依赖关系的管理放在被依赖任务处，就会让代码的复杂度提高，把松耦合的架构变成了一段乱麻。
那么应该如何解决呢？
在系列的第二篇文章中，引入了TaskManager，它负责任务的加载、启动和停止，我们只需要给它增加一个功能，维护任务之间的依赖关系。先看下时序图：

通过这个时序图，我们应该可以很直观看清楚它们之间的逻辑关系了。那么接下来就看看代码应该怎么处理。

在任务的注解上，增加一个dependecy：

@Target({ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
@Component
public @interface QuantManagedTask {
    String name();
    String dependency() "";
}

管理容器增加对依赖的管理（只显示了新增代码），

@Component
public class QuantTaskManager implements BeanPostProcessor{
    protected Map<String, List<QuantTask>> nameDependencyTaskListMap = new HashMap<>(); 

    @Override
    public Object postProcessAfterInitialization(Object bean, String s) throws BeansException {
        if(bean instanceof QuantTask) {
            try {
                QuantTask task = (QuantTask) bean;

                QuantManagedTask quantManagedTask = AopUtils.getTargetClass(bean).getAnnotation(QuantManagedTask.class);
                String name = quantManagedTask.name();
                nameTaskMap.put(name, task);
                task.setName(name);
                // 获取被依赖的任务的名称
                String dependency = quantManagedTask.dependency();
                // 创建被依赖任务的name和依赖任务列表的map
                if(StringUtils.isNotBlank(dependency)){
                    if(!nameDependencyTaskListMap.containsKey(dependency)) {
                        nameDependencyTaskListMap.put(dependency, new ArrayList())
                    }
                    nameDependencyTaskListMap.get(dependency).add(task);
                }
            }catch (Exception e){
                e.printStackTrace();
            }
        }
        return bean;
    }

    /**
     * 任务执行结束需要调用的方法，通知启动后续任务
     */
    public void postExecution(String name){
        if(nameDependencyTaskListMap.keySet().contains(name)){
            List<QuantTask> taskList = nameDependencyTaskListMap.get(name);
            taskList -> forEach(QuantTask::start);
        }
    }
}

任务的基类中，增加状态通知，只需要改造stop方法。

    private String name;

    public void setName(String name){
        this.name = name;
    }
    /**
     * 停止当前任务，改变状态
     */
    public void stop(){
        isRunning = false;
        TaskManager taskManager = ApplicationContext.getBean("taskManager");
        taskManager.postExecution(name);
    }

那么任务定义时的注解就需要改为：

@QuantManagedTask(name="sample_task", dependency="task_a")

至此，我们通过四篇文章，实现了一个完整的任务计算框架，应该可以满足大部分量化交易系统的需求。不过在开发量化交易系统的过程中，随着策略的不断丰富，计算任务也在不断增多，那么如何提高计算性能，就是一个亟待解决的问题了，在后面的一篇文章中，我们将讨论这个问题。