Examples with TimeBasedScheduleInfo org.smartdata.model.rule.TimeBasedScheduleInfo used on opensource projects

Example 6 with TimeBasedScheduleInfo

use of org.smartdata.model.rule.TimeBasedScheduleInfo in project SSM by Intel-bigdata.

the class RuleExecutor method run.

@Override
public void run() {
    long startCheckTime = System.currentTimeMillis();
    if (exited) {
        exitSchedule();
    }
    if (!tr.getTbScheduleInfo().isExecutable(startCheckTime)) {
        return;
    }
    List<RuleExecutorPlugin> plugins = RuleExecutorPluginManager.getPlugins();
    long rid = ctx.getRuleId();
    try {
        if (ruleManager.isClosed()) {
            exitSchedule();
        }
        long endCheckTime;
        int numCmdSubmitted = 0;
        List<String> files = new ArrayList<>();
        RuleInfo info = ruleManager.getRuleInfo(rid);
        boolean doExec = true;
        for (RuleExecutorPlugin plugin : plugins) {
            doExec &= plugin.preExecution(info, tr);
            if (!doExec) {
                break;
            }
        }
        RuleState state = info.getState();
        if (exited || state == RuleState.DELETED || state == RuleState.FINISHED || state == RuleState.DISABLED) {
            exitSchedule();
        }
        TimeBasedScheduleInfo scheduleInfo = tr.getTbScheduleInfo();
        if (!scheduleInfo.isOnce() && scheduleInfo.getEndTime() != TimeBasedScheduleInfo.FOR_EVER) {
            boolean befExit = false;
            if (scheduleInfo.isOneShot()) {
                // The subScheduleTime is set in triggering time.
                if (scheduleInfo.getSubScheduleTime() > scheduleInfo.getEndTime()) {
                    befExit = true;
                }
            } else if (startCheckTime - scheduleInfo.getEndTime() > 0) {
                befExit = true;
            }
            if (befExit) {
                LOG.info("Rule " + ctx.getRuleId() + " exit rule executor due to time passed");
                ruleManager.updateRuleInfo(rid, RuleState.FINISHED, startCheckTime, 0, 0);
                exitSchedule();
            }
        }
        if (doExec) {
            files = executeFileRuleQuery();
            if (exited) {
                exitSchedule();
            }
        }
        endCheckTime = System.currentTimeMillis();
        if (doExec) {
            for (RuleExecutorPlugin plugin : plugins) {
                files = plugin.preSubmitCmdlet(info, files);
            }
            numCmdSubmitted = submitCmdlets(info, files);
        }
        ruleManager.updateRuleInfo(rid, null, startCheckTime, 1, numCmdSubmitted);
        long endProcessTime = System.currentTimeMillis();
        if (endProcessTime - startCheckTime > 2000 || LOG.isDebugEnabled()) {
            LOG.warn("Rule " + ctx.getRuleId() + " execution took " + (endProcessTime - startCheckTime) + "ms. QueryTime = " + (endCheckTime - startCheckTime) + "ms, SubmitTime = " + (endProcessTime - endCheckTime) + "ms, fileNum = " + numCmdSubmitted + ".");
        }
        if (scheduleInfo.isOneShot()) {
            ruleManager.updateRuleInfo(rid, RuleState.FINISHED, startCheckTime, 0, 0);
            exitSchedule();
        }
        if (endProcessTime + scheduleInfo.getBaseEvery() > scheduleInfo.getEndTime()) {
            LOG.info("Rule " + ctx.getRuleId() + " exit rule executor due to finished");
            ruleManager.updateRuleInfo(rid, RuleState.FINISHED, startCheckTime, 0, 0);
            exitSchedule();
        }
        if (exited) {
            exitSchedule();
        }
    } catch (IOException e) {
        LOG.error("Rule " + ctx.getRuleId() + " exception", e);
    }
}