org.drools.xml.processes.ConnectionHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of drools-compiler
There is a newer version: 7.9.0.Final
package org.drools.xml.processes;

import java.util.HashSet;

import org.drools.definition.process.Node;
import org.drools.workflow.core.Connection;
import org.drools.workflow.core.NodeContainer;
import org.drools.workflow.core.impl.ConnectionImpl;
import org.drools.workflow.core.impl.NodeImpl;
import org.drools.xml.BaseAbstractHandler;
import org.drools.xml.ExtensibleXmlParser;
import org.drools.xml.Handler;
import org.xml.sax.Attributes;
import org.xml.sax.SAXException;
import org.xml.sax.SAXParseException;

public class ConnectionHandler extends BaseAbstractHandler implements Handler {

	public ConnectionHandler() {
		if ((this.validParents == null) && (this.validPeers == null)) {
			this.validParents = new HashSet>();
			this.validParents.add(NodeContainer.class);
			this.validPeers = new HashSet>();
			this.validPeers.add(null);
			this.validPeers.add(Connection.class);
			this.allowNesting = false;
		}
	}

	public Object start(final String uri, final String localName,
			final Attributes attrs, final ExtensibleXmlParser parser)
			throws SAXException {
		parser.startElementBuilder(localName, attrs);

		String fromId = attrs.getValue("from");
		emptyAttributeCheck(localName, "from", fromId, parser);
		String toId = attrs.getValue("to");
		emptyAttributeCheck(localName, "to", toId, parser);
		String bendpoints = attrs.getValue("bendpoints");

		String fromType = attrs.getValue("fromType");
		if (fromType == null || fromType.trim().length() == 0) {
			fromType = NodeImpl.CONNECTION_DEFAULT_TYPE;
		}
		String toType = attrs.getValue("toType");
		if (toType == null || toType.trim().length() == 0) {
			toType = NodeImpl.CONNECTION_DEFAULT_TYPE;
		}

		NodeContainer nodeContainer = (NodeContainer) parser.getParent();
		Node fromNode = nodeContainer.getNode(new Long(fromId));
		Node toNode = nodeContainer.getNode(new Long(toId));

		if (fromNode == null) {
			throw new SAXParseException(
				"Node '" + fromId + "' cannot be found", parser.getLocator());
		}
		if (toNode == null) {
			throw new SAXParseException("Node '" + toId + "' cannot be found", parser.getLocator());
		}

		ConnectionImpl connection = new ConnectionImpl(fromNode, fromType, toNode, toType);
		connection.setMetaData("bendpoints", bendpoints);

		return connection;
	}

	public Object end(final String uri, final String localName, final ExtensibleXmlParser parser) throws SAXException {
		return parser.getCurrent();
	}

	public Class generateNodeFor() {
		return Connection.class;
	}

}