Please wait. This can take some minutes ...
Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
org.drools.core.reteoo.QueryElementNode Maven / Gradle / Ivy
/*
* 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.drools.core.reteoo;
import java.io.IOException;
import java.io.ObjectInput;
import java.io.ObjectOutput;
import java.util.List;
import java.util.Map;
import org.drools.core.RuleBaseConfiguration;
import org.drools.core.base.DroolsQuery;
import org.drools.core.base.InternalViewChangedEventListener;
import org.drools.core.beliefsystem.BeliefSet;
import org.drools.core.beliefsystem.jtms.JTMSBeliefSetImpl.MODE;
import org.drools.core.common.InternalFactHandle;
import org.drools.core.common.InternalWorkingMemory;
import org.drools.core.common.Memory;
import org.drools.core.common.MemoryFactory;
import org.drools.core.common.ObjectStore;
import org.drools.core.common.QueryElementFactHandle;
import org.drools.core.common.TupleSets;
import org.drools.core.common.TupleSetsImpl;
import org.drools.core.common.UpdateContext;
import org.drools.core.definitions.rule.impl.RuleImpl;
import org.drools.core.marshalling.impl.PersisterHelper;
import org.drools.core.marshalling.impl.ProtobufInputMarshaller.QueryElementContext;
import org.drools.core.marshalling.impl.ProtobufInputMarshaller.TupleKey;
import org.drools.core.marshalling.impl.ProtobufMessages;
import org.drools.core.phreak.SegmentUtilities;
import org.drools.core.phreak.StackEntry;
import org.drools.core.reteoo.builder.BuildContext;
import org.drools.core.rule.AbductiveQuery;
import org.drools.core.rule.Declaration;
import org.drools.core.rule.QueryArgument;
import org.drools.core.rule.QueryElement;
import org.drools.core.rule.QueryImpl;
import org.drools.core.spi.PropagationContext;
import org.drools.core.spi.Tuple;
import org.drools.core.util.AbstractBaseLinkedListNode;
public class QueryElementNode extends LeftTupleSource
implements
LeftTupleSinkNode,
MemoryFactory {
private LeftTupleSinkNode previousTupleSinkNode;
private LeftTupleSinkNode nextTupleSinkNode;
protected QueryElement queryElement;
private boolean tupleMemoryEnabled;
protected boolean openQuery;
private boolean dataDriven;
private QueryArgument[] argsTemplate;
public QueryElementNode() {
// for serialization
}
public QueryElementNode(final int id,
final LeftTupleSource tupleSource,
final QueryElement queryElement,
final boolean tupleMemoryEnabled,
final boolean openQuery,
final BuildContext context) {
super(id, context);
setLeftTupleSource(tupleSource);
this.queryElement = queryElement;
this.tupleMemoryEnabled = tupleMemoryEnabled;
this.openQuery = openQuery;
this.dataDriven = context != null && context.getRule().isDataDriven();
initMasks( context, tupleSource );
this.argsTemplate = initArgsTemplate( context );
hashcode = calculateHashCode();
}
private QueryArgument[] initArgsTemplate(BuildContext context) {
ClassLoader classLoader = context.getKnowledgeBase().getRootClassLoader();
QueryArgument[] originalArgs = this.queryElement.getArguments();
QueryArgument[] args = new QueryArgument[originalArgs.length];
for (int i = 0; i < originalArgs.length; i++) {
args[i] = originalArgs[i] == null ? QueryArgument.NULL : originalArgs[i].normalize( classLoader );
}
return args;
}
public void readExternal(ObjectInput in) throws IOException,
ClassNotFoundException {
super.readExternal( in );
queryElement = (QueryElement) in.readObject();
tupleMemoryEnabled = in.readBoolean();
openQuery = in.readBoolean();
dataDriven = in.readBoolean();
this.argsTemplate = (QueryArgument[]) in.readObject();
}
public void writeExternal(ObjectOutput out) throws IOException {
super.writeExternal( out );
out.writeObject( queryElement );
out.writeBoolean( tupleMemoryEnabled );
out.writeBoolean( openQuery );
out.writeBoolean( dataDriven );
out.writeObject( argsTemplate );
}
public void networkUpdated(UpdateContext updateContext) {
this.leftInput.networkUpdated(updateContext);
}
public short getType() {
return NodeTypeEnums.UnificationNode;
}
public boolean isLeftTupleMemoryEnabled() {
return false;
}
public QueryElement getQueryElement() {
return queryElement;
}
public boolean isOpenQuery() {
return openQuery;
}
@SuppressWarnings("unchecked")
public InternalFactHandle createFactHandle(final PropagationContext context,
final InternalWorkingMemory workingMemory,
final LeftTuple leftTuple ) {
ProtobufMessages.FactHandle _handle = null;
if( context.getReaderContext() != null ) {
Map map = (Map) context.getReaderContext().nodeMemories.get( getId() );
if( map != null ) {
_handle = map.get( PersisterHelper.createTupleKey( leftTuple ) ).handle;
}
}
return _handle != null ?
workingMemory.getFactHandleFactory().newFactHandle( _handle.getId(),
null,
_handle.getRecency(),
null,
workingMemory,
workingMemory ) :
workingMemory.getFactHandleFactory().newFactHandle( null,
null,
workingMemory,
workingMemory );
}
public DroolsQuery createDroolsQuery(LeftTuple leftTuple,
InternalFactHandle handle,
StackEntry stackEntry,
final List pmems,
QueryElementNodeMemory qmem,
LeftTupleSink sink,
InternalWorkingMemory workingMemory) {
UnificationNodeViewChangedEventListener collector = createCollector( leftTuple, queryElement.getVariableIndexes(), this.tupleMemoryEnabled );
boolean executeAsOpenQuery = openQuery;
if ( executeAsOpenQuery ) {
// There is no point in doing an open query if the caller is a non-open query.
Object object = leftTuple.get( 0 ).getObject();
if ( object instanceof DroolsQuery && !((DroolsQuery) object).isOpen() ) {
executeAsOpenQuery = false;
}
}
DroolsQuery queryObject = new DroolsQuery( this.queryElement.getQueryName(),
getActualArguments( leftTuple, workingMemory ),
collector,
executeAsOpenQuery,
stackEntry,
pmems,
qmem != null ? qmem.getResultLeftTuples() : null,
qmem,
sink);
collector.setFactHandle( handle );
handle.setObject( queryObject );
leftTuple.setContextObject( handle ); // so it can be retracted later and destroyed
return queryObject;
}
public Object[] getActualArguments( LeftTuple leftTuple, InternalWorkingMemory workingMemory ) {
Object[] args = new Object[argsTemplate.length]; // the actual args, to be created from the template
for (int i = 0; i < argsTemplate.length; i++) {
args[i] = argsTemplate[i].getValue( workingMemory, leftTuple );
}
return args;
}
protected UnificationNodeViewChangedEventListener createCollector( LeftTuple leftTuple, int[] varIndexes, boolean tupleMemoryEnabled ) {
return new UnificationNodeViewChangedEventListener( leftTuple,
varIndexes,
this,
tupleMemoryEnabled );
}
public void setLeftTupleMemoryEnabled(boolean tupleMemoryEnabled) {
this.tupleMemoryEnabled = tupleMemoryEnabled;
}
/**
* Returns the next node
* @return
* The next TupleSinkNode
*/
public LeftTupleSinkNode getNextLeftTupleSinkNode() {
return this.nextTupleSinkNode;
}
/**
* Sets the next node
* @param next
* The next TupleSinkNode
*/
public void setNextLeftTupleSinkNode(final LeftTupleSinkNode next) {
this.nextTupleSinkNode = next;
}
/**
* Returns the previous node
* @return
* The previous TupleSinkNode
*/
public LeftTupleSinkNode getPreviousLeftTupleSinkNode() {
return this.previousTupleSinkNode;
}
/**
* Sets the previous node
* @param previous
* The previous TupleSinkNode
*/
public void setPreviousLeftTupleSinkNode(final LeftTupleSinkNode previous) {
this.previousTupleSinkNode = previous;
}
public static class UnificationNodeViewChangedEventListener
implements
InternalViewChangedEventListener {
protected LeftTuple leftTuple;
protected QueryElementNode node;
protected InternalFactHandle factHandle;
protected int[] variables;
protected boolean tupleMemoryEnabled;
public UnificationNodeViewChangedEventListener(LeftTuple leftTuple,
int[] variables,
QueryElementNode node,
boolean tupleMemoryEnabled) {
this.leftTuple = leftTuple;
this.variables = variables;
this.node = node;
this.tupleMemoryEnabled = tupleMemoryEnabled;
}
public InternalFactHandle getFactHandle() {
return factHandle;
}
public void setFactHandle(InternalFactHandle factHandle) {
this.factHandle = factHandle;
}
public void setVariables(int[] variables) {
this.variables = variables;
}
public void rowAdded(final RuleImpl rule,
LeftTuple resultLeftTuple,
PropagationContext context,
InternalWorkingMemory workingMemory) {
QueryTerminalNode node = resultLeftTuple.getTupleSink();
QueryImpl query = node.getQuery();
Declaration[] decls = node.getRequiredDeclarations();
DroolsQuery dquery = (DroolsQuery) this.factHandle.getObject();
Object[] objects = new Object[ determineResultSize( query, dquery ) ];
Declaration decl;
for (int variable : this.variables) {
decl = decls[variable];
objects[variable] = decl.getValue(workingMemory,
resultLeftTuple.get(decl).getObject());
}
QueryElementFactHandle resultHandle = createQueryResultHandle(context,
workingMemory,
objects);
RightTuple rightTuple = createResultRightTuple(resultHandle, resultLeftTuple, dquery.isOpen());
boolean pass = true;
if ( query.isAbductive() ) {
AbductiveQuery aq = (( AbductiveQuery) query );
int numArgs = aq.getAbducibleArgs().length;
Object[] constructorArgs = new Object[ aq.getAbducibleArgs().length ];
for ( int j = 0; j < numArgs; j++ ) {
int k = aq.mapArgToParam( j );
if ( objects[ k ] != null ) {
constructorArgs[ j ] = objects[ k ];
} else if ( dquery.getElements()[ k ] != null ) {
constructorArgs[ j ] = dquery.getElements()[ k ];
}
}
Object abduced = aq.abduce( constructorArgs );
if ( abduced != null ) {
boolean firstAssertion = true;
ObjectStore store = workingMemory.getObjectStore();
InternalFactHandle handle = store.getHandleForObject( abduced );
if ( handle != null ) {
abduced = handle.getObject();
firstAssertion = false;
} else {
handle = workingMemory.getTruthMaintenanceSystem().insert( abduced,
MODE.POSITIVE.getId(),
query,
(RuleTerminalNodeLeftTuple) resultLeftTuple );
}
BeliefSet bs = handle.getEqualityKey() != null ? handle.getEqualityKey().getBeliefSet() : null;
if ( bs == null ) {
abduced = handle.getObject();
} else {
if ( ! bs.isPositive() ) {
pass = false;
} else {
if ( !firstAssertion ) {
workingMemory.getTruthMaintenanceSystem().insert( abduced,
MODE.POSITIVE.getId(),
query,
(RuleTerminalNodeLeftTuple) resultLeftTuple );
}
}
}
}
objects[ objects.length - 1 ] = abduced;
}
if ( pass ) {
LeftTupleSink sink = dquery.getLeftTupleSink();
LeftTuple childLeftTuple = sink.createLeftTuple( this.leftTuple, rightTuple, sink );
boolean stagedInsertWasEmpty = dquery.getResultLeftTupleSets().addInsert(childLeftTuple);
if ( stagedInsertWasEmpty ) {
dquery.getQueryNodeMemory().setNodeDirtyWithoutNotify();
}
}
}
private int determineResultSize( QueryImpl query, DroolsQuery dquery ) {
int size = dquery.getElements().length;
if (query.isAbductive() && (( AbductiveQuery ) query ).isReturnBound()) {
size++;
}
return size;
}
protected RightTuple createResultRightTuple( QueryElementFactHandle resultHandle, LeftTuple resultLeftTuple, boolean open ) {
RightTuple rightTuple = new RightTupleImpl( resultHandle );
if ( open ) {
rightTuple.setBlocked( resultLeftTuple );
resultLeftTuple.setContextObject( rightTuple );
}
rightTuple.setPropagationContext( resultLeftTuple.getPropagationContext() );
return rightTuple;
}
@SuppressWarnings("unchecked")
protected QueryElementFactHandle createQueryResultHandle(PropagationContext context,
InternalWorkingMemory workingMemory,
Object[] objects) {
ProtobufMessages.FactHandle _handle = null;
if( context.getReaderContext() != null ) {
Map map = (Map) context.getReaderContext().nodeMemories.get( node.getId() );
if( map != null ) {
QueryElementContext _context = map.get( PersisterHelper.createTupleKey( leftTuple ) );
if( _context != null ) {
_handle = _context.results.removeFirst();
}
}
}
return _handle != null ?
new QueryElementFactHandle( objects,
_handle.getId(),
_handle.getRecency() ) :
new QueryElementFactHandle( objects,
workingMemory.getFactHandleFactory().getNextId(),
workingMemory.getFactHandleFactory().getNextRecency() );
}
public void rowRemoved(final RuleImpl rule,
final LeftTuple resultLeftTuple,
final PropagationContext context,
final InternalWorkingMemory workingMemory) {
RightTuple rightTuple = (RightTuple) resultLeftTuple.getContextObject();
rightTuple.setBlocked( null );
resultLeftTuple.setContextObject( null );
DroolsQuery query = (DroolsQuery) this.factHandle.getObject();
TupleSets leftTuples = query.getResultLeftTupleSets();
LeftTuple childLeftTuple = rightTuple.getFirstChild();
if (childLeftTuple.isStagedOnRight()) {
( (SubnetworkTuple) childLeftTuple ).moveStagingFromRightToLeft();
} else {
switch ( childLeftTuple.getStagedTypeForQueries() ) {
// handle clash with already staged entries
case LeftTuple.INSERT:
leftTuples.removeInsert( childLeftTuple );
return;
case LeftTuple.UPDATE:
leftTuples.removeUpdate( childLeftTuple );
break;
}
}
leftTuples.addDelete(childLeftTuple);
}
public void rowUpdated(final RuleImpl rule,
final LeftTuple resultLeftTuple,
final PropagationContext context,
final InternalWorkingMemory workingMemory) {
RightTuple rightTuple = (RightTuple) resultLeftTuple.getContextObject();
if ( rightTuple.getMemory() != null ) {
// Already sheduled as an insert
return;
}
rightTuple.setBlocked( null );
resultLeftTuple.setContextObject( null );
// We need to recopy everything back again, as we don't know what has or hasn't changed
QueryTerminalNode node = resultLeftTuple.getTupleSink();
Declaration[] decls = node.getRequiredDeclarations();
InternalFactHandle rootHandle = resultLeftTuple.get( 0 );
DroolsQuery dquery = (DroolsQuery) rootHandle.getObject();
Object[] objects = new Object[dquery.getElements().length];
Declaration decl;
for (int variable : this.variables) {
decl = decls[variable];
objects[variable] = decl.getValue(workingMemory,
resultLeftTuple.get(decl).getObject());
}
QueryElementFactHandle handle = (QueryElementFactHandle) rightTuple.getFactHandle();
handle.setRecency(workingMemory.getFactHandleFactory().getNextRecency());
handle.setObject( objects );
if ( dquery.isOpen() ) {
rightTuple.setBlocked( resultLeftTuple );
resultLeftTuple.setContextObject( rightTuple );
}
TupleSets leftTuples = dquery.getResultLeftTupleSets();
LeftTuple childLeftTuple = rightTuple.getFirstChild();
switch ( childLeftTuple.getStagedTypeForQueries() ) {
// handle clash with already staged entries
case LeftTuple.INSERT :
leftTuples.removeInsert( childLeftTuple );
break;
case LeftTuple.UPDATE :
leftTuples.removeUpdate( childLeftTuple );
break;
}
leftTuples.addUpdate( childLeftTuple );
}
public List getResults() {
throw new UnsupportedOperationException( getClass().getCanonicalName() + " does not support the getResults() method." );
}
public LeftTuple getLeftTuple() {
return leftTuple;
}
}
public LeftTuple createLeftTuple(InternalFactHandle factHandle,
Sink sink,
boolean leftTupleMemoryEnabled) {
return new QueryElementNodeLeftTuple( factHandle,
sink,
leftTupleMemoryEnabled );
}
public LeftTuple createLeftTuple(final InternalFactHandle factHandle,
final LeftTuple leftTuple,
final Sink sink) {
return new QueryElementNodeLeftTuple(factHandle,leftTuple, sink );
}
public LeftTuple createLeftTuple(LeftTuple leftTuple,
Sink sink,
PropagationContext pctx, boolean leftTupleMemoryEnabled) {
return new QueryElementNodeLeftTuple( leftTuple,
sink,
pctx,
leftTupleMemoryEnabled );
}
public LeftTuple createLeftTuple(LeftTuple leftTuple,
RightTuple rightTuple,
Sink sink) {
return new QueryElementNodeLeftTuple( leftTuple,
rightTuple,
sink );
}
public LeftTuple createLeftTuple(LeftTuple leftTuple,
RightTuple rightTuple,
LeftTuple currentLeftChild,
LeftTuple currentRightChild,
Sink sink,
boolean leftTupleMemoryEnabled) {
return new QueryElementNodeLeftTuple( leftTuple,
rightTuple,
currentLeftChild,
currentRightChild,
sink,
leftTupleMemoryEnabled );
}
private int calculateHashCode() {
final int prime = 31;
int result = super.hashCode();
result = prime * result + (openQuery ? 1231 : 1237);
result = prime * result + ((queryElement == null) ? 0 : queryElement.hashCode());
result = prime * result + ((leftInput == null) ? 0 : leftInput.hashCode());
return result;
}
@Override
public boolean equals(Object object) {
return this == object ||
( internalEquals( object ) && this.leftInput.thisNodeEquals( ((QueryElementNode)object).leftInput ) );
}
@Override
protected boolean internalEquals( Object object ) {
if ( object == null || !(object instanceof QueryElementNode) || this.hashCode() != object.hashCode() ) {
return false;
}
QueryElementNode other = (QueryElementNode) object;
if ( openQuery != other.openQuery ) return false;
if ( !openQuery && dataDriven != other.dataDriven ) return false;
if ( queryElement == null ) {
if ( other.queryElement != null ) return false;
} else if ( !queryElement.equals( other.queryElement ) ) return false;
return true;
}
public QueryElementNodeMemory createMemory(RuleBaseConfiguration config, InternalWorkingMemory wm) {
return new QueryElementNodeMemory(this);
}
public static class QueryElementNodeMemory extends AbstractBaseLinkedListNode implements SegmentNodeMemory {
private QueryElementNode node;
private SegmentMemory smem;
private SegmentMemory querySegmentMemory;
private TupleSets resultLeftTuples;
private long nodePosMaskBit;
public QueryElementNodeMemory(QueryElementNode node) {
this.node = node;
// if there is only one sink there is no split and then no smem staging and no normalization
// otherwise it uses special tuplset with alternative linking fields (rightParentPrev/Next)
this.resultLeftTuples = node.getSinkPropagator().size() > 1 ?
new QueryTupleSets() :
new TupleSetsImpl();
}
public QueryElementNode getNode() {
return this.node;
}
public short getNodeType() {
return NodeTypeEnums.QueryElementNode;
}
public void setSegmentMemory(SegmentMemory smem) {
this.smem = smem;
}
public SegmentMemory getSegmentMemory() {
return smem;
}
public SegmentMemory getQuerySegmentMemory() {
return querySegmentMemory;
}
public void setQuerySegmentMemory(SegmentMemory querySegmentMemory) {
this.querySegmentMemory = querySegmentMemory;
}
public TupleSets getResultLeftTuples() {
return resultLeftTuples;
}
public void correctMemoryOnSinksChanged(TerminalNode removingTN) {
if (resultLeftTuples instanceof QueryTupleSets ) {
if (!SegmentUtilities.isTipNode( node, removingTN )) {
// a sink has been removed and now there is no longer a split
TupleSetsImpl newTupleSets = new TupleSetsImpl();
this.resultLeftTuples.addTo( newTupleSets );
this.resultLeftTuples = newTupleSets;
}
} else {
if (SegmentUtilities.isTipNode( node, removingTN )) {
// a sink has been added and now there is a split
TupleSetsImpl newTupleSets = new QueryTupleSets();
this.resultLeftTuples.addTo( newTupleSets );
this.resultLeftTuples = newTupleSets;
}
}
}
public long getNodePosMaskBit() {
return nodePosMaskBit;
}
public void setNodePosMaskBit(long segmentPos) {
this.nodePosMaskBit = segmentPos;
}
public void setNodeDirtyWithoutNotify() {
smem.updateDirtyNodeMask( nodePosMaskBit );
}
public void setNodeCleanWithoutNotify() {
smem.updateCleanNodeMask( nodePosMaskBit );
}
public void reset() {
resultLeftTuples.resetAll();
}
public static class QueryTupleSets extends TupleSetsImpl {
@Override
protected LeftTuple getPreviousTuple( LeftTuple tuple ) {
return tuple.getRightParentPrevious();
}
@Override
protected void setPreviousTuple( LeftTuple tuple, LeftTuple stagedPrevious ) {
tuple.setRightParentPrevious( stagedPrevious );
}
@Override
protected LeftTuple getNextTuple( LeftTuple tuple ) {
return tuple.getRightParentNext();
}
@Override
protected void setNextTuple( LeftTuple tuple, LeftTuple stagedNext ) {
tuple.setRightParentNext( stagedNext );
}
@Override
protected void setStagedType( LeftTuple tuple, short type ) {
tuple.setStagedTypeForQueries( type );
}
@Override
protected short getStagedType( LeftTuple tuple ) {
return tuple.getStagedTypeForQueries();
}
public void addTo(TupleSets tupleSets) {
addAllInsertsTo( tupleSets );
addAllDeletesTo( tupleSets );
addAllUpdatesTo( tupleSets );
}
private void addAllInsertsTo( TupleSets tupleSets ) {
LeftTuple leftTuple = getInsertFirst();
while (leftTuple != null) {
LeftTuple next = getNextTuple( leftTuple );
clear( leftTuple );
tupleSets.addInsert( leftTuple );
leftTuple = next;
}
setInsertFirst( null );
}
private void addAllUpdatesTo( TupleSets tupleSets ) {
LeftTuple leftTuple = getUpdateFirst();
while (leftTuple != null) {
LeftTuple next = getNextTuple( leftTuple );
clear( leftTuple );
tupleSets.addUpdate( leftTuple );
leftTuple = next;
}
setUpdateFirst( null );
}
private void addAllDeletesTo( TupleSets tupleSets ) {
LeftTuple leftTuple = getDeleteFirst();
while (leftTuple != null) {
LeftTuple next = getNextTuple( leftTuple );
clear( leftTuple );
tupleSets.addDelete( leftTuple );
leftTuple = next;
}
setDeleteFirst( null );
}
private void clear( LeftTuple leftTuple ) {
setStagedType( leftTuple, Tuple.NONE );
setPreviousTuple( leftTuple, null );
setNextTuple( leftTuple, null );
}
}
}
@Override
public ObjectTypeNode getObjectTypeNode() {
return leftInput.getObjectTypeNode();
}
@Override
public LeftTuple createPeer(LeftTuple original) {
QueryElementNodeLeftTuple peer = new QueryElementNodeLeftTuple();
peer.initPeer((BaseLeftTuple) original, this);
original.setPeer(peer);
return peer;
}
public String toString() {
return "[" + this.getClass().getSimpleName() + "(" + this.id + ", " + queryElement.getQueryName() + ")]";
}
public void attach( BuildContext context ) {
this.leftInput.addTupleSink( this, context );
}
protected boolean doRemove(RuleRemovalContext context,
ReteooBuilder builder,
InternalWorkingMemory[] workingMemories) {
if (!isInUse()) {
getLeftTupleSource().removeTupleSink(this);
return true;
}
return false;
}
}