Examples with VariableAugmentor io.automatiko.engine.api.workflow.VariableAugmentor used on opensource projects

Example 1 with VariableAugmentor

use of io.automatiko.engine.api.workflow.VariableAugmentor in project automatiko-engine by automatiko-io.

the class AbstractProtobufProcessInstanceMarshaller method readVariableScope.

protected void readVariableScope(MarshallerReaderContext context, Process process, WorkflowProcessInstanceImpl processInstance, AutomatikoMessages.ProcessInstance _instance) throws IOException {
    if (_instance.getVariableCount() > 0) {
        VariableScope variableScope = (VariableScope) ((io.automatiko.engine.workflow.base.core.Process) process).getDefaultContext(VariableScope.VARIABLE_SCOPE);
        VariableScopeInstance variableScopeInstance = (VariableScopeInstance) processInstance.getContextInstance(variableScope);
        for (AutomatikoMessages.Variable _variable : _instance.getVariableList()) {
            try {
                Object _value = ProtobufProcessMarshaller.unmarshallVariableValue(context, _variable);
                if ((boolean) context.env.getOrDefault("_import_", false)) {
                    VariableInitializer initializer = ((ProcessRuntimeServiceProvider) context.env.get("_services_")).getVariableInitializer();
                    for (VariableAugmentor augmentor : initializer.augmentors()) {
                        Variable var = variableScope.findVariable(_variable.getName());
                        if (augmentor.accept(var, _value)) {
                            _value = augmentor.augmentOnCreate(process.getId(), process.getVersion(), _instance.getId(), var, _value);
                        }
                    }
                }
                variableScopeInstance.internalSetVariable(_variable.getName(), _value);
            } catch (ClassNotFoundException e) {
                throw new IllegalArgumentException("Could not reload variable " + _variable.getName());
            }
        }
    }
}

Example 2 with VariableAugmentor

use of io.automatiko.engine.api.workflow.VariableAugmentor in project automatiko-engine by automatiko-io.

the class VariableScopeInstance method setVariable.

@SuppressWarnings("unchecked")
public void setVariable(NodeInstance nodeInstance, String name, Object value) {
    if (name == null) {
        throw new IllegalArgumentException("The name of a variable may not be null!");
    }
    Variable var = getVariableScope().findVariable(name);
    if (var != null) {
        name = var.getName();
    }
    Object oldValue = getVariable(name);
    if (oldValue == null) {
        if (value == null) {
            return;
        }
    }
    // check if variable that is being set is readonly and has already been set
    if (oldValue != null && getVariableScope().isReadOnly(name)) {
        throw new VariableViolationException(getProcessInstance().getId(), name, "Variable '" + name + "' is already set and is marked as read only");
    }
    // in case variable is marked as notnull (via tag) then null values should be ignored
    if (value == null && getVariableScope().isNullable(name)) {
        return;
    }
    // in case variable is versioned store the old value into versioned variable by variable name
    if (var != null && var.hasTag(Variable.VERSIONED_TAG)) {
        Map<String, List<Object>> versions = (Map<String, List<Object>>) variables.computeIfAbsent(VariableScope.VERSIONED_VARIABLES, key -> new ConcurrentHashMap<>());
        List<Object> varVersions = versions.computeIfAbsent(name, k -> new ArrayList<>());
        int versionLimit = Integer.parseInt(var.getMetaData().getOrDefault(Variable.VAR_VERSIONS_LIMIT, "10").toString());
        if (oldValue != null) {
            varVersions.add(oldValue);
            // and remove the oldest if exceeding
            if (varVersions.size() > versionLimit) {
                varVersions.remove(0);
            }
        }
    }
    if (getProcessInstance().getProcessRuntime().getVariableInitializer() != null) {
        for (VariableAugmentor augmentor : getProcessInstance().getProcessRuntime().getVariableInitializer().augmentors()) {
            if (augmentor.accept(var, value)) {
                // run any of the available augmentors on the value
                if (oldValue != null) {
                    value = augmentor.augmentOnUpdate(getProcessInstance().getProcess().getId(), getProcessInstance().getProcess().getVersion(), getProcessInstance().getId(), var, value);
                } else if (value == null) {
                    augmentor.augmentOnDelete(getProcessInstance().getProcess().getId(), getProcessInstance().getProcess().getVersion(), getProcessInstance().getId(), var, oldValue);
                } else {
                    value = augmentor.augmentOnCreate(getProcessInstance().getProcess().getId(), getProcessInstance().getProcess().getVersion(), getProcessInstance().getId(), var, value);
                }
            }
        }
    }
    ProcessInstance processInstance = getProcessInstance();
    if (nodeInstance != null) {
        processInstance = nodeInstance.getProcessInstance();
    }
    ProcessEventSupport processEventSupport = ((InternalProcessRuntime) getProcessInstance().getProcessRuntime()).getProcessEventSupport();
    processEventSupport.fireBeforeVariableChanged((variableIdPrefix == null ? "" : variableIdPrefix + ":") + name, (variableInstanceIdPrefix == null ? "" : variableInstanceIdPrefix + ":") + name, oldValue, value, getVariableScope().tags(name), processInstance, nodeInstance, getProcessInstance().getProcessRuntime());
    internalSetVariable(name, value);
    processEventSupport.fireAfterVariableChanged((variableIdPrefix == null ? "" : variableIdPrefix + ":") + name, (variableInstanceIdPrefix == null ? "" : variableInstanceIdPrefix + ":") + name, oldValue, value, getVariableScope().tags(name), processInstance, nodeInstance, getProcessInstance().getProcessRuntime());
    processInstance.signalEvent("variableChanged", value);
}