Examples with AbstractProcessContext org.drools.core.process.AbstractProcessContext used on opensource projects

Example 1 with AbstractProcessContext

use of org.drools.core.process.AbstractProcessContext in project kogito-runtimes by kiegroup.

the class RuleSetNodeInstance method internalTrigger.

@Override
public void internalTrigger(KogitoNodeInstance from, String type) {
    try {
        super.internalTrigger(from, type);
        // if node instance was cancelled, abort
        if (getNodeInstanceContainer().getNodeInstance(getStringId()) == null) {
            return;
        }
        if (!Node.CONNECTION_DEFAULT_TYPE.equals(type)) {
            throw new IllegalArgumentException("A RuleSetNode only accepts default incoming connections!");
        }
        RuleSetNode ruleSetNode = getRuleSetNode();
        KieRuntime kruntime = Optional.ofNullable(getRuleSetNode().getKieRuntime()).orElse(() -> getProcessInstance().getKnowledgeRuntime()).get();
        Map<String, Object> inputs = NodeIoHelper.processInputs(this, varRef -> getVariable(varRef));
        RuleSetNode.RuleType ruleType = ruleSetNode.getRuleType();
        if (ruleType.isDecision()) {
            RuleSetNode.RuleType.Decision decisionModel = (RuleSetNode.RuleType.Decision) ruleType;
            String namespace = resolveExpression(decisionModel.getNamespace());
            String model = resolveExpression(decisionModel.getModel());
            DecisionModel modelInstance = Optional.ofNullable(getRuleSetNode().getDecisionModel()).orElse(() -> new DmnDecisionModel(((KieSession) kruntime).getKieRuntime(DMNRuntime.class), namespace, model)).get();
            // Input Binding
            DMNContext context = DMNJSONUtils.ctx(modelInstance, jsonResolver.resolveAll(inputs));
            DMNResult dmnResult = modelInstance.evaluateAll(context);
            if (dmnResult.hasErrors()) {
                String errors = dmnResult.getMessages(Severity.ERROR).stream().map(Object::toString).collect(Collectors.joining(", "));
                throw new RuntimeException("DMN result errors:: " + errors);
            }
            // Output Binding
            Map<String, Object> outputSet = dmnResult.getContext().getAll();
            NodeIoHelper.processOutputs(this, key -> outputSet.get(key), varName -> this.getVariable(varName));
            triggerCompleted();
        } else if (ruleType.isRuleFlowGroup()) {
            // first set rule flow group
            setRuleFlowGroup(resolveRuleFlowGroup(ruleType.getName()));
            // proceed
            for (Entry<String, Object> entry : inputs.entrySet()) {
                if (FIRE_RULE_LIMIT_PARAMETER.equals(entry.getKey())) {
                    // don't put control parameter for fire limit into working memory
                    continue;
                }
                String inputKey = getRuleFlowGroup() + "_" + getProcessInstance().getStringId() + "_" + entry.getKey();
                factHandles.put(inputKey, kruntime.insert(entry.getValue()));
            }
            if (actAsWaitState()) {
                addRuleSetListener();
                ((InternalAgenda) kruntime.getAgenda()).activateRuleFlowGroup(getRuleFlowGroup(), getProcessInstance().getStringId(), getUniqueId());
            } else {
                int fireLimit = DEFAULT_FIRE_RULE_LIMIT;
                WorkflowProcessInstance processInstance = getProcessInstance();
                if (inputs.containsKey(FIRE_RULE_LIMIT_PARAMETER)) {
                    fireLimit = Integer.parseInt(inputs.get(FIRE_RULE_LIMIT_PARAMETER).toString());
                }
                ((InternalAgenda) kruntime.getAgenda()).activateRuleFlowGroup(getRuleFlowGroup(), processInstance.getStringId(), getUniqueId());
                int fired = ((KieSession) kruntime).fireAllRules(processInstance.getAgendaFilter(), fireLimit);
                if (fired == fireLimit) {
                    throw new RuntimeException("Fire rule limit reached " + fireLimit + ", limit can be set via system property " + FIRE_RULE_LIMIT_PROPERTY + " or via data input of business rule task named " + FIRE_RULE_LIMIT_PARAMETER);
                }
                removeEventListeners();
                retractFacts(kruntime);
                triggerCompleted();
            }
        } else if (ruleType.isRuleUnit()) {
            RuleUnitFactory<RuleUnitData> factory = ruleSetNode.getRuleUnitFactory();
            AbstractProcessContext context = ContextFactory.fromNode(this);
            RuleUnitData model = factory.bind(context);
            RuleUnitInstance<RuleUnitData> instance = factory.unit().createInstance(model);
            instance.fire();
            factory.unbind(context, model);
            triggerCompleted();
        } else {
            throw new UnsupportedOperationException("Unsupported Rule Type: " + ruleType);
        }
    } catch (Exception e) {
        handleException(e);
    }
}