org.jetlinks.rule.engine.defaults.DefaultRuleEngine 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 lombok.AllArgsConstructor;
import org.jetlinks.rule.engine.api.RuleEngine;
import org.jetlinks.rule.engine.api.scheduler.Scheduler;
import org.jetlinks.rule.engine.api.task.Task;
import org.jetlinks.rule.engine.api.model.RuleModel;
import org.jetlinks.rule.engine.api.worker.Worker;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;

import java.util.function.Function;

@AllArgsConstructor
public class DefaultRuleEngine implements RuleEngine {

    //调度器
    private final Scheduler scheduler;

    public Flux startRule(String instanceId,
                                RuleModel model) {
        return Flux.fromIterable(new ScheduleJobCompiler(instanceId, model).compile())
                .flatMap(scheduler::schedule)
                .collectList()
                .flatMapIterable(Function.identity())
                .flatMap(task -> task.start().thenReturn(task));
    }

    public Flux getTasks(String instanceId) {
        return scheduler.getSchedulingTask(instanceId);
    }

    public Mono shutdown(String instanceId) {
        return scheduler.shutdown(instanceId);
    }

    @Override
    public Flux getWorkers() {
        return scheduler.getWorkers();
    }
}