xyz.erupt.job.service.EruptJobFetch Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of erupt-job Show documentation
Task scheduling module
The newest version!
package xyz.erupt.job.service;

import lombok.SneakyThrows;
import org.springframework.core.type.filter.AssignableTypeFilter;
import org.springframework.core.type.filter.TypeFilter;
import xyz.erupt.annotation.fun.ChoiceFetchHandler;
import xyz.erupt.annotation.fun.ChoiceTrigger;
import xyz.erupt.annotation.fun.VLModel;
import xyz.erupt.core.annotation.EruptHandlerNaming;
import xyz.erupt.core.service.EruptApplication;
import xyz.erupt.core.util.EruptSpringUtil;
import xyz.erupt.job.handler.EruptJobHandler;
import xyz.erupt.job.model.EruptJob;
import xyz.erupt.linq.lambda.LambdaSee;

import java.util.*;

/**
 * @author YuePeng
 * date 2021/2/27 22:46
 */
public class EruptJobFetch implements ChoiceFetchHandler, ChoiceTrigger {

    private static List loadedJobHandler;

    @Override
    public synchronized List fetch(String[] params) {
        if (null == loadedJobHandler) {
            loadedJobHandler = new ArrayList<>();
            EruptSpringUtil.scannerPackage(EruptApplication.getScanPackage(), new TypeFilter[]{new AssignableTypeFilter(EruptJobHandler.class)}, clazz -> {
                EruptHandlerNaming eruptHandlerNaming = clazz.getAnnotation(EruptHandlerNaming.class);
                if (null == eruptHandlerNaming) {
                    loadedJobHandler.add(new VLModel(clazz.getName(), ((EruptJobHandler) EruptSpringUtil.getBean(clazz)).name(), clazz.getName()));
                } else {
                    loadedJobHandler.add(new VLModel(clazz.getName(), eruptHandlerNaming.value(), clazz.getName()));
                }
            });
        }
        return loadedJobHandler;
    }

    @Override
    @SneakyThrows
    public Map trigger(Object value, String[] params) {
        for (VLModel vl : loadedJobHandler) {
            if (vl.getValue().equals(value)) {
                Map map = new HashMap<>();
                EruptJobHandler jobHandler = EruptSpringUtil.getBeanByPath(vl.getDesc(), EruptJobHandler.class);
                map.put(LambdaSee.field(EruptJob::getName), vl.getLabel());
                if (null != jobHandler.param()) {
                    map.put(LambdaSee.field(EruptJob::getHandlerParam), jobHandler.param());
                }
                if (null != jobHandler.cron()) {
                    map.put(LambdaSee.field(EruptJob::getCron), jobHandler.cron());
                }
                return map;
            }
        }
        return Collections.emptyMap();
    }
}