Examples with EntryPointNode org.drools.core.reteoo.EntryPointNode used on opensource projects

Example 21 with EntryPointNode

use of org.drools.core.reteoo.EntryPointNode in project drools by kiegroup.

the class BaseNode method getSinks.

public Sink[] getSinks() {
    Sink[] sinks = null;
    if (this instanceof EntryPointNode) {
        EntryPointNode source = (EntryPointNode) this;
        Collection<ObjectTypeNode> otns = source.getObjectTypeNodes().values();
        sinks = otns.toArray(new Sink[otns.size()]);
    } else if (this instanceof ObjectSource) {
        ObjectSource source = (ObjectSource) this;
        sinks = source.getObjectSinkPropagator().getSinks();
    } else if (this instanceof LeftTupleSource) {
        LeftTupleSource source = (LeftTupleSource) this;
        sinks = source.getSinkPropagator().getSinks();
    }
    return sinks;
}

Example 22 with EntryPointNode

use of org.drools.core.reteoo.EntryPointNode in project drools by kiegroup.

the class BuildUtils method attachNode.

/**
 * Attaches a node into the network. If a node already exists that could
 * substitute, it is used instead.
 *
 * @param context
 *            The current build context
 * @param candidate
 *            The node to attach.
 *
 * @return the actual attached node that may be the one given as parameter
 *         or eventually one that was already in the cache if sharing is enabled
 */
public <T extends BaseNode> T attachNode(BuildContext context, T candidate) {
    BaseNode node = null;
    RuleBasePartitionId partition = null;
    if (candidate.getType() == NodeTypeEnums.EntryPointNode) {
        // entry point nodes are always shared
        node = context.getKnowledgeBase().getRete().getEntryPointNode(((EntryPointNode) candidate).getEntryPoint());
        // all EntryPointNodes belong to the main partition
        partition = RuleBasePartitionId.MAIN_PARTITION;
    } else if (candidate.getType() == NodeTypeEnums.ObjectTypeNode) {
        // object type nodes are always shared
        Map<ObjectType, ObjectTypeNode> map = context.getKnowledgeBase().getRete().getObjectTypeNodes(context.getCurrentEntryPoint());
        if (map != null) {
            ObjectTypeNode otn = map.get(((ObjectTypeNode) candidate).getObjectType());
            if (otn != null) {
                // adjusting expiration offset
                otn.mergeExpirationOffset((ObjectTypeNode) candidate);
                node = otn;
            }
        }
        // all ObjectTypeNodes belong to the main partition
        partition = RuleBasePartitionId.MAIN_PARTITION;
    } else if (isSharingEnabledForNode(context, candidate)) {
        if ((context.getTupleSource() != null) && NodeTypeEnums.isLeftTupleSink(candidate)) {
            node = context.getTupleSource().getSinkPropagator().getMatchingNode(candidate);
        } else if ((context.getObjectSource() != null) && NodeTypeEnums.isObjectSink(candidate)) {
            node = context.getObjectSource().getObjectSinkPropagator().getMatchingNode(candidate);
        } else {
            throw new RuntimeException("This is a bug on node sharing verification. Please report to development team.");
        }
    }
    if (node != null && !areNodesCompatibleForSharing(context, node, candidate)) {
        node = null;
    }
    if (node == null) {
        // only attach() if it is a new node
        node = candidate;
        // new node, so it must be labeled
        if (partition == null) {
            // if it does not has a predefined label
            if (context.getPartitionId() == null) {
                // if no label in current context, create one
                context.setPartitionId(context.getKnowledgeBase().createNewPartitionId());
            }
            partition = context.getPartitionId();
        }
        // set node whit the actual partition label
        node.setPartitionId(context, partition);
        node.attach(context);
        // adds the node to the context list to track all added nodes
        context.getNodes().add(node);
    } else {
        // shared node found
        mergeNodes(node, candidate);
        // undo previous id assignment
        context.releaseId(candidate);
        if (partition == null && context.getPartitionId() == null) {
            partition = node.getPartitionId();
            // if no label in current context, create one
            context.setPartitionId(partition);
        }
    }
    node.addAssociation(context, context.getRule());
    return (T) node;
}

Example 23 with EntryPointNode

use of org.drools.core.reteoo.EntryPointNode in project drools by kiegroup.

the class EntryPointBuilder method build.

/* (non-Javadoc)
     * @see org.kie.reteoo.builder.ReteooComponentBuilder#build(org.kie.reteoo.builder.BuildContext, org.kie.reteoo.builder.BuildUtils, org.kie.rule.RuleConditionElement)
     */
public void build(BuildContext context, BuildUtils utils, RuleConditionElement rce) {
    final EntryPointId entry = (EntryPointId) rce;
    context.setCurrentEntryPoint(entry);
    EntryPointNode epn = context.getKnowledgeBase().getRete().getEntryPointNode(entry);
    if (epn == null) {
        NodeFactory nFactory = context.getComponentFactory().getNodeFactoryService();
        context.setObjectSource(utils.attachNode(context, nFactory.buildEntryPointNode(context.getNextId(), context.getKnowledgeBase().getRete(), context)));
    } else {
        context.setObjectSource(epn);
    }
}