/**
* Copyright 2010 JBoss Inc
*
* 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.base.mvel;
import java.io.Externalizable;
import java.io.IOException;
import java.io.ObjectInput;
import java.io.ObjectOutput;
import java.io.Serializable;
import java.util.HashMap;
import java.util.Map;
import java.util.Map.Entry;
import org.drools.common.InternalFactHandle;
import org.drools.rule.Declaration;
import org.drools.rule.MVELDialectRuntimeData;
import org.drools.rule.Package;
import org.drools.WorkingMemory;
import org.drools.spi.Accumulator;
import org.drools.spi.Tuple;
import org.mvel2.MVEL;
import org.mvel2.compiler.CompiledExpression;
/**
* An MVEL accumulator implementation
*
* @author etirelli
*/
public class MVELAccumulator
implements
MVELCompileable,
Accumulator,
Externalizable {
private static final long serialVersionUID = 510l;
MVELCompilationUnit initUnit;
MVELCompilationUnit actionUnit;
MVELCompilationUnit reverseUnit;
MVELCompilationUnit resultUnit;
private DroolsMVELFactory initPrototype;
private DroolsMVELFactory actionPrototype;
private DroolsMVELFactory resultPrototype;
private DroolsMVELFactory reversePrototype;
private Serializable init;
private Serializable action;
private Serializable reverse;
private Serializable result;
public MVELAccumulator() {
}
public MVELAccumulator(final MVELCompilationUnit initUnit,
final MVELCompilationUnit actionUnit,
final MVELCompilationUnit reverseUnit,
final MVELCompilationUnit resultUnit) {
super();
this.initUnit = initUnit;
this.actionUnit = actionUnit;
this.reverseUnit = reverseUnit;
this.resultUnit = resultUnit;
}
public void readExternal(ObjectInput in) throws IOException,
ClassNotFoundException {
initUnit = (MVELCompilationUnit) in.readObject();
actionUnit = (MVELCompilationUnit) in.readObject();
reverseUnit = (MVELCompilationUnit) in.readObject();
resultUnit = (MVELCompilationUnit) in.readObject();
}
public void writeExternal(ObjectOutput out) throws IOException {
out.writeObject( initUnit );
out.writeObject( actionUnit );
out.writeObject( reverseUnit );
out.writeObject( resultUnit );
}
public void compile(ClassLoader classLoader) {
init = initUnit.getCompiledExpression( classLoader );
action = actionUnit.getCompiledExpression( classLoader );
result = resultUnit.getCompiledExpression( classLoader );
initPrototype = initUnit.getFactory();
actionPrototype = actionUnit.getFactory();
if ( reverseUnit != null ) {
reverse = reverseUnit.getCompiledExpression( classLoader );
reversePrototype = reverseUnit.getFactory();
}
resultPrototype = resultUnit.getFactory();
}
/* (non-Javadoc)
* @see org.drools.spi.Accumulator#createContext()
*/
public Serializable createContext() {
Map<Integer, Map<String, Object>> shadow = null;
if ( this.reverse != null ) {
shadow = new HashMap<Integer, Map<String, Object>>();
}
return new MVELAccumulatorContext( new HashMap(),
shadow );
}
/* (non-Javadoc)
* @see org.drools.spi.Accumulator#init(java.lang.Object, org.drools.spi.Tuple, org.drools.rule.Declaration[], org.drools.WorkingMemory)
*/
public void init(Object workingMemoryContext,
Object context,
Tuple leftTuple,
Declaration[] declarations,
WorkingMemory workingMemory) throws Exception {
DroolsMVELFactory factory = (DroolsMVELFactory) ((MVELFactoryContext)workingMemoryContext).initPrototype;
Package pkg = workingMemory.getRuleBase().getPackage( "MAIN" );
if ( pkg != null ) {
MVELDialectRuntimeData data = (MVELDialectRuntimeData) pkg.getDialectRuntimeRegistry().getDialectData( "mvel" );
factory.setNextFactory( data.getFunctionFactory() );
}
factory.setContext( leftTuple,
null,
null,
workingMemory,
((MVELAccumulatorContext) context).variables );
MVEL.executeExpression( this.init,
null,
factory );
}
/* (non-Javadoc)
* @see org.drools.spi.Accumulator#accumulate(java.lang.Object, org.drools.spi.Tuple, org.drools.common.InternalFactHandle, org.drools.rule.Declaration[], org.drools.rule.Declaration[], org.drools.WorkingMemory)
*/
public void accumulate(Object workingMemoryContext,
Object context,
Tuple leftTuple,
InternalFactHandle handle,
Declaration[] declarations,
Declaration[] innerDeclarations,
WorkingMemory workingMemory) throws Exception {
DroolsMVELFactory factory = (DroolsMVELFactory) ((MVELFactoryContext)workingMemoryContext).actionPrototype;
factory.setContext( leftTuple,
null,
handle.getObject(),
workingMemory,
((MVELAccumulatorContext) context).variables );
if ( reverse != null ) {
// SNAPSHOT variable values
takeSnapshot( context,
factory,
handle );
}
MVEL.executeExpression( this.action,
null,
factory );
}
private void takeSnapshot(Object context,
DroolsMVELFactory factory,
InternalFactHandle handle) {
DroolsMVELShadowFactory shad = (DroolsMVELShadowFactory) factory;
Map<String, Object> varsMap = ((MVELAccumulatorContext) context).shadow.get( Integer.valueOf( handle.getId() ) );
if ( varsMap == null ) {
varsMap = new HashMap<String, Object>();
((MVELAccumulatorContext) context).shadow.put( Integer.valueOf( handle.getId() ),
varsMap );
}
for ( String var : shad.getShadowVariables() ) {
varsMap.put( var,
shad.getVariableResolver( var ).getValue() );
}
}
public void reverse(Object workingMemoryContext,
Object context,
Tuple leftTuple,
InternalFactHandle handle,
Declaration[] declarations,
Declaration[] innerDeclarations,
WorkingMemory workingMemory) throws Exception {
DroolsMVELFactory factory = (DroolsMVELFactory) ((MVELFactoryContext)workingMemoryContext).reversePrototype;
factory.setContext( leftTuple,
null,
handle.getObject(),
workingMemory,
((MVELAccumulatorContext) context).variables );
// set shadow values overriding actual values
// ALSO, since reverse() is called, we know the factory is a shadow factory
((DroolsMVELShadowFactory) factory).setShadowValues( ((MVELAccumulatorContext) context).shadow.get( Integer.valueOf( handle.getId() ) ) );
MVEL.executeExpression( this.reverse,
null,
factory );
// cleaning up shadow values map
((DroolsMVELShadowFactory) factory).setShadowValues( null );
((MVELAccumulatorContext) context).shadow.remove( Integer.valueOf( handle.getId() ) );
}
/* (non-Javadoc)
* @see org.drools.spi.Accumulator#getResult(java.lang.Object, org.drools.spi.Tuple, org.drools.rule.Declaration[], org.drools.WorkingMemory)
*/
public Object getResult(Object workingMemoryContext,
Object context,
Tuple leftTuple,
Declaration[] declarations,
WorkingMemory workingMemory) throws Exception {
DroolsMVELFactory factory = (DroolsMVELFactory) ((MVELFactoryContext)workingMemoryContext).resultPrototype;
factory.setContext( leftTuple,
null,
null,
workingMemory,
((MVELAccumulatorContext) context).variables );
Map<String, Object> vars = ((MVELAccumulatorContext) context).variables;
final Object result = MVEL.executeExpression( this.result,
null,
factory );
return result;
}
public boolean supportsReverse() {
return this.reverse != null;
}
public Object createWorkingMemoryContext() {
return new MVELFactoryContext( this );
}
private static class MVELFactoryContext {
public DroolsMVELFactory initPrototype;
public DroolsMVELFactory actionPrototype;
public DroolsMVELFactory resultPrototype;
public DroolsMVELFactory reversePrototype;
public MVELFactoryContext(MVELAccumulator acc) {
initPrototype = (DroolsMVELFactory) acc.initPrototype.clone();
actionPrototype = (DroolsMVELFactory) acc.actionPrototype.clone();
resultPrototype = (DroolsMVELFactory)acc.resultPrototype.clone();
if ( acc.reversePrototype != null ) {
reversePrototype =(DroolsMVELFactory) acc.reversePrototype.clone();
}
for ( Entry<String, Class> var : ((CompiledExpression)acc.init).getParserContext().getVariables().entrySet() ) {
actionPrototype.addResolver( var.getKey(), new LocalVariableResolver(actionPrototype, var.getKey()) );
resultPrototype.addResolver( var.getKey(), new LocalVariableResolver(resultPrototype, var.getKey()) );
if ( acc.reversePrototype != null ) {
reversePrototype.addResolver( var.getKey(), new LocalVariableResolver(reversePrototype, var.getKey()) );
}
}
}
}
private static class MVELAccumulatorContext
implements
Serializable {
private static final long serialVersionUID = 510l;
public final Map variables;
public final Map<Integer, Map<String, Object>> shadow;
public MVELAccumulatorContext(final Map variables,
final Map<Integer, Map<String, Object>> shadow) {
this.variables = variables;
this.shadow = shadow;
}
}
}