/** * Copyright 2010 Red Hat, Inc. and/or its affiliates. * * Licensed 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.bpmn2.xml; import java.util.HashSet; import java.util.List; import java.util.Map; import org.drools.core.process.core.datatype.DataType; import org.drools.core.process.core.datatype.impl.type.BooleanDataType; import org.drools.core.process.core.datatype.impl.type.FloatDataType; import org.drools.core.process.core.datatype.impl.type.IntegerDataType; import org.drools.core.process.core.datatype.impl.type.ObjectDataType; import org.drools.core.process.core.datatype.impl.type.StringDataType; import org.drools.core.xml.BaseAbstractHandler; import org.drools.core.xml.ExtensibleXmlParser; import org.drools.core.xml.Handler; import org.jbpm.bpmn2.core.ItemDefinition; import org.jbpm.bpmn2.core.SequenceFlow; import org.jbpm.compiler.xml.ProcessBuildData; import org.jbpm.process.core.ContextContainer; import org.jbpm.process.core.context.variable.Variable; import org.jbpm.process.core.context.variable.VariableScope; import org.jbpm.workflow.core.Node; import org.xml.sax.Attributes; import org.xml.sax.SAXException; public class DataObjectHandler extends BaseAbstractHandler implements Handler { public DataObjectHandler() { initValidParents(); initValidPeers(); this.allowNesting = false; } protected void initValidParents() { this.validParents = new HashSet<Class<?>>(); this.validParents.add(ContextContainer.class); } protected void initValidPeers() { this.validPeers = new HashSet<Class<?>>(); this.validPeers.add(null); this.validPeers.add(Variable.class); this.validPeers.add(Node.class); this.validPeers.add(SequenceFlow.class); } @SuppressWarnings("unchecked") public Object start(final String uri, final String localName, final Attributes attrs, final ExtensibleXmlParser parser) throws SAXException { parser.startElementBuilder(localName, attrs); final String id = attrs.getValue("id"); final String itemSubjectRef = attrs.getValue("itemSubjectRef"); Object parent = parser.getParent(); if (parent instanceof ContextContainer) { ContextContainer contextContainer = (ContextContainer) parent; VariableScope variableScope = (VariableScope) contextContainer.getDefaultContext(VariableScope.VARIABLE_SCOPE); List variables = variableScope.getVariables(); Variable variable = new Variable(); variable.setMetaData("DataObject", "true"); variable.setName(id); // retrieve type from item definition DataType dataType = new ObjectDataType(); Map<String, ItemDefinition> itemDefinitions = (Map<String, ItemDefinition>) ((ProcessBuildData) parser.getData()).getMetaData("ItemDefinitions"); if (itemDefinitions != null) { ItemDefinition itemDefinition = itemDefinitions.get(itemSubjectRef); if (itemDefinition != null) { String structureRef = itemDefinition.getStructureRef(); if ("java.lang.Boolean".equals(structureRef) || "Boolean".equals(structureRef)) { dataType = new BooleanDataType(); } else if ("java.lang.Integer".equals(structureRef) || "Integer".equals(structureRef)) { dataType = new IntegerDataType(); } else if ("java.lang.Float".equals(structureRef) || "Float".equals(structureRef)) { dataType = new FloatDataType(); } else if ("java.lang.String".equals(structureRef) || "String".equals(structureRef)) { dataType = new StringDataType(); } else if ("java.lang.Object".equals(structureRef) || "Object".equals(structureRef)) { // use FQCN of Object dataType = new ObjectDataType("java.lang.Object"); } else { dataType = new ObjectDataType(structureRef, parser.getClassLoader()); } } } variable.setType(dataType); variables.add(variable); return variable; } return new Variable(); } public Object end(final String uri, final String localName, final ExtensibleXmlParser parser) throws SAXException { parser.endElementBuilder(); return parser.getCurrent(); } public Class<?> generateNodeFor() { return Variable.class; } }