org.jbpm.process.builder.ExtendedNodeBuilder Maven / Gradle / Ivy

Go to download
/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */
package org.jbpm.process.builder;

import java.util.Collection;
import java.util.List;
import java.util.Map;

import org.drools.compiler.rule.builder.PackageBuildContext;
import org.drools.drl.ast.descr.ActionDescr;
import org.drools.drl.ast.descr.ProcessDescr;
import org.jbpm.process.builder.dialect.ProcessDialect;
import org.jbpm.process.builder.dialect.ProcessDialectRegistry;
import org.jbpm.workflow.core.DroolsAction;
import org.jbpm.workflow.core.impl.DataAssociation;
import org.jbpm.workflow.core.impl.DroolsConsequenceAction;
import org.jbpm.workflow.core.impl.ExtendedNodeImpl;
import org.jbpm.workflow.core.impl.NodeImpl;
import org.jbpm.workflow.core.node.Assignment;
import org.kie.api.definition.process.Node;
import org.kie.api.definition.process.Process;

public class ExtendedNodeBuilder
        implements
        ProcessNodeBuilder {

    public void build(Process process,
            ProcessDescr processDescr,
            ProcessBuildContext context,
            Node node) {
        ExtendedNodeImpl extendedNode = (ExtendedNodeImpl) node;
        for (String type : extendedNode.getActionTypes()) {
            List actions = extendedNode.getActions(type);
            if (actions != null) {
                for (DroolsAction droolsAction : actions) {
                    buildAction(droolsAction, context, (NodeImpl) node);
                }
            }
        }
    }

    protected void buildAction(DroolsAction droolsAction, ProcessBuildContext context, NodeImpl node) {
        DroolsConsequenceAction action = (DroolsConsequenceAction) droolsAction;
        ActionDescr actionDescr = new ActionDescr();
        actionDescr.setText(action.getConsequence());
        actionDescr.setResource(context.getProcessDescr().getResource());

        ProcessDialect dialect = ProcessDialectRegistry.getDialect(action.getDialect());
        dialect.getActionBuilder().build(context, action, actionDescr, node);
    }

    protected void buildDataAssociation(PackageBuildContext context, Collection dataAssociations, Map parameters) {

        for (DataAssociation dataAssociation : dataAssociations) {
            List assignments = dataAssociation.getAssignments();
            if (assignments != null) {
                for (Assignment assignment : assignments) {
                    if (assignment.getDialect() != null) {
                        ProcessDialect processDialect = ProcessDialectRegistry.getDialect(assignment.getDialect());
                        processDialect.getAssignmentBuilder().build(context,
                                assignment,
                                dataAssociation.getSources(),
                                dataAssociation.getTarget());
                    }
                }
            }
        }
    }
}