/*
* 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.test.dsl;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import org.drools.core.base.ClassObjectType;
import org.drools.core.common.BetaConstraints;
import org.drools.core.common.DefaultBetaConstraints;
import org.drools.core.common.DoubleBetaConstraints;
import org.drools.core.common.EmptyBetaConstraints;
import org.drools.core.common.QuadroupleBetaConstraints;
import org.drools.core.common.SingleBetaConstraints;
import org.drools.core.common.TripleBetaConstraints;
import org.drools.core.reteoo.JoinNode;
import org.drools.core.reteoo.LeftTupleSource;
import org.drools.core.reteoo.MockObjectSource;
import org.drools.core.reteoo.MockTupleSource;
import org.drools.core.reteoo.ObjectSource;
import org.drools.core.reteoo.ObjectTypeNode;
import org.drools.core.reteoo.builder.BuildContext;
import org.drools.core.rule.Declaration;
import org.drools.core.spi.BetaNodeFieldConstraint;
public class JoinNodeStep
implements
Step {
private ReteTesterHelper reteTesterHelper;
public JoinNodeStep(ReteTesterHelper reteTesterHelper) {
this.reteTesterHelper = reteTesterHelper;
}
public void execute(Map<String, Object> context,
List<String[]> args) {
BuildContext buildContext = (BuildContext) context.get( "BuildContext" );
if ( args.size() != 0 ) {
String[] a = args.get( 0 );
String name = a[0].trim();
String leftInput = a[1].trim();
String rightInput = a[2].trim();
Class cls = null;
LeftTupleSource leftTupleSource;
if ( leftInput.startsWith( "mock" ) ) {
leftTupleSource = new MockTupleSource( buildContext.getNextId() );
} else {
leftTupleSource = (LeftTupleSource) context.get( leftInput );
}
ObjectSource rightObjectSource;
if ( rightInput.startsWith( "mock" ) ) {
String type = rightInput.substring( 5, rightInput.length() -1 );
try {
cls = reteTesterHelper.getTypeResolver().resolveType( type );
} catch ( ClassNotFoundException e ) {
throw new RuntimeException( e );
}
rightObjectSource = new MockObjectSource( buildContext.getNextId() );
} else {
rightObjectSource = (ObjectSource) context.get( rightInput );
ObjectSource source = rightObjectSource;
while ( !( source instanceof ObjectTypeNode ) ) {
source = source.getParentObjectSource();
}
cls = ((ClassObjectType)((ObjectTypeNode)source).getObjectType()).getClassType();
}
List<BetaNodeFieldConstraint> list = new ArrayList<BetaNodeFieldConstraint>();
for ( int i = 1; i < args.size(); i++ ) {
a = args.get( i );
String fieldName = a[0].trim();
String operator = a[1].trim();
String var = a[2].trim();
Declaration declr = (Declaration) context.get( var );
BetaNodeFieldConstraint betaConstraint;
betaConstraint = this.reteTesterHelper.getBoundVariableConstraint( cls,
fieldName,
declr,
operator );
list.add( betaConstraint );
}
BetaConstraints constraints;
switch ( list.size() ) {
case 0:
constraints = new EmptyBetaConstraints();
break;
case 1:
constraints = new SingleBetaConstraints( list.get(0),
buildContext.getKnowledgeBase().getConfiguration() );
break;
case 2:
constraints = new DoubleBetaConstraints( list.toArray( new BetaNodeFieldConstraint[2] ),
buildContext.getKnowledgeBase().getConfiguration() );
break;
case 3:
constraints = new TripleBetaConstraints( list.toArray( new BetaNodeFieldConstraint[2] ),
buildContext.getKnowledgeBase().getConfiguration() );
break;
case 4:
constraints = new QuadroupleBetaConstraints( list.toArray( new BetaNodeFieldConstraint[2] ),
buildContext.getKnowledgeBase().getConfiguration() );
break;
default:
constraints = new DefaultBetaConstraints( list.toArray( new BetaNodeFieldConstraint[2] ),
buildContext.getKnowledgeBase().getConfiguration() );
break;
}
JoinNode joinNode = new JoinNode( buildContext.getNextId(),
leftTupleSource,
rightObjectSource,
constraints,
buildContext );
joinNode.attach(buildContext);
context.put( name,
joinNode );
} else {
throw new IllegalArgumentException( "Cannot arguments " + args );
}
}
}