org.jetlinks.rule.engine.defaults.ScheduleJobCompiler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of rule-engine-api Show documentation
The newest version!
package org.jetlinks.rule.engine.defaults;

import org.jetlinks.rule.engine.api.model.RuleLink;
import org.jetlinks.rule.engine.api.model.RuleModel;
import org.jetlinks.rule.engine.api.model.RuleNodeModel;
import org.jetlinks.rule.engine.api.scheduler.ScheduleJob;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 调度任务编译器,将规则模型编译成调度任务
 *
 * @author zhouhao
 * @since 1.0.4
 */
public class ScheduleJobCompiler {

    private final Map jobs = new HashMap<>();

    private final String instanceId;

    private final RuleModel model;

    public ScheduleJobCompiler(String instanceId, RuleModel model) {
        this.instanceId = instanceId;
        this.model = model;
    }

    public List compile() {
        for (RuleNodeModel node : model.getNodes()) {
            ScheduleJob job = new ScheduleJob();
            job.setInstanceId(instanceId);
            job.setRuleId(model.getId());
            job.setNodeId(node.getId());
            job.setConfiguration(node.getConfiguration());
            job.setRuleConfiguration(model.getConfiguration());
            job.setModelType(model.getType());
            job.setExecutor(node.getExecutor());
            job.setName(node.getName());
            job.setSchedulingRule(node.getSchedulingRule());
            jobs.put(node.getId(), job);
        }
        for (RuleNodeModel node : model.getNodes()) {
            prepare(node);
        }
        return new ArrayList<>(jobs.values());
    }

    private ScheduleJob getJob(String nodeId) {
        return jobs.get(nodeId);
    }

    private void prepare(RuleNodeModel node) {
        ScheduleJob job = getJob(node.getId());

        {
            List inputs = new ArrayList<>();
            for (RuleLink input : node.getInputs()) {
                inputs.add(input.getSource().getId());
            }
            job.setInputs(inputs);
        }

        {
            for (RuleLink event : node.getEvents()) {
                //事件输入
                getJob(event.getTarget().getId())
                        .getEvents()
                        .add(new ScheduleJob.Event(event.getType(), node.getId()));
                //事件输出
                job.getEventOutputs().add(new ScheduleJob.Event(event.getType(), event.getTarget().getId()));
            }
        }

        {
            List outputs = new ArrayList<>();
            for (RuleLink output : node.getOutputs()) {
                outputs.add(new ScheduleJob.Output(output.getTarget().getId(), output.getCondition()));
            }
            job.setOutputs(outputs);
        }

    }

}