/*
* Copyright 2005 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.core.reteoo;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.Externalizable;
import java.io.IOException;
import java.io.ObjectInput;
import java.io.ObjectInputStream;
import java.io.ObjectOutput;
import java.util.Arrays;
import java.util.Collection;
import java.util.Collections;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
import org.drools.core.FactException;
import org.drools.core.FactHandle;
import org.drools.core.RuleBaseConfiguration;
import org.drools.core.SessionConfiguration;
import org.drools.core.StatefulSession;
import org.drools.core.StatelessSession;
import org.drools.core.common.AbstractWorkingMemory;
import org.drools.core.common.AbstractWorkingMemory.WorkingMemoryReteAssertAction;
import org.drools.core.common.DefaultFactHandle;
import org.drools.core.common.DroolsObjectInput;
import org.drools.core.common.DroolsObjectInputStream;
import org.drools.core.common.DroolsObjectOutputStream;
import org.drools.core.common.InternalFactHandle;
import org.drools.core.common.InternalRuleBase;
import org.drools.core.common.InternalWorkingMemory;
import org.drools.core.common.ProjectClassLoader;
import org.drools.core.common.RuleBasePartitionId;
import org.drools.core.common.UpgradableReentrantReadWriteLock;
import org.drools.core.common.WorkingMemoryFactory;
import org.drools.core.event.RuleBaseEventListener;
import org.drools.core.impl.EnvironmentFactory;
import org.drools.core.impl.KnowledgeBaseImpl;
import org.drools.core.impl.StatefulKnowledgeSessionImpl;
import org.drools.core.management.DroolsManagementAgent;
import org.drools.core.rule.EntryPointId;
import org.drools.core.rule.InvalidPatternException;
import org.drools.core.rule.Package;
import org.drools.core.rule.Rule;
import org.drools.core.rule.TypeDeclaration;
import org.drools.core.rule.WindowDeclaration;
import org.drools.core.spi.FactHandleFactory;
import org.drools.core.spi.PropagationContext;
import org.kie.api.conf.EventProcessingOption;
import org.kie.api.io.Resource;
import org.kie.api.marshalling.Marshaller;
import org.kie.internal.marshalling.MarshallerFactory;
import org.kie.api.marshalling.ObjectMarshallingStrategy;
import org.kie.api.runtime.Environment;
import org.kie.api.runtime.KieSession;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Queue;
import java.util.concurrent.ConcurrentLinkedQueue;
import java.util.concurrent.locks.ReentrantLock;
import org.drools.core.RuleBase;
import org.drools.core.RuntimeDroolsException;
import org.drools.core.PackageIntegrationException;
import org.drools.core.base.ClassFieldAccessorCache;
import org.drools.core.factmodel.ClassDefinition;
import org.drools.core.reteoo.SegmentMemory.Prototype;
import org.drools.core.util.ObjectHashSet;
import org.drools.core.util.TripleStore;
import org.drools.core.event.RuleBaseEventSupport;
import org.drools.core.factmodel.traits.TraitRegistry;
import org.drools.core.rule.DialectRuntimeRegistry;
import org.drools.core.rule.Function;
import org.drools.core.rule.ImportDeclaration;
import org.drools.core.rule.JavaDialectRuntimeData;
import org.kie.api.definition.process.Process;
import org.kie.api.definition.type.FactType;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.util.concurrent.CopyOnWriteArrayList;
import java.util.concurrent.atomic.AtomicInteger;
import static org.drools.core.common.ProjectClassLoader.createProjectClassLoader;
import static org.drools.core.util.BitMaskUtil.isSet;
/**
* Implementation of <code>RuleBase</code>.
*
*
* @version $Id: RuleBaseImpl.java,v 1.5 2005/08/14 22:44:12 mproctor Exp $
*/
public class ReteooRuleBase
implements
InternalRuleBase,
Externalizable {
protected static transient Logger logger = LoggerFactory.getLogger(ReteooRuleBase.class);
/**
* DO NOT CHANGE BELOW SERIAL_VERSION_ID UNLESS YOU ARE CHANGING DROOLS VERSION
* SERIAL_VERSION_ID=320 stands for version 3.2.0
*/
private static final long serialVersionUID = 510l;
public Set<EntryPointNode> addedEntryNodeCache;
public Set<EntryPointNode> removedEntryNodeCache;
// ------------------------------------------------------------
// Instance members
// ------------------------------------------------------------
private String id;
private final AtomicInteger workingMemoryCounter = new AtomicInteger(0);
private RuleBaseConfiguration config;
protected Map<String, Package> pkgs;
private Map<String, Process> processes;
private transient ClassLoader rootClassLoader;
/**
* The fact handle factory.
*/
private FactHandleFactory factHandleFactory;
private transient Map<String, Class<?>> globals;
private final transient Queue<DialectRuntimeRegistry> reloadPackageCompilationData = new ConcurrentLinkedQueue<DialectRuntimeRegistry>();
private RuleBaseEventSupport eventSupport = new RuleBaseEventSupport(this);
private transient ObjectHashSet statefulSessions;
// lock for entire rulebase, used for dynamic updates
private final UpgradableReentrantReadWriteLock lock = new UpgradableReentrantReadWriteLock();
/**
* This lock is used when adding to, or reading the <field>statefulSessions</field>
*/
private final ReentrantLock statefulSessionLock = new ReentrantLock();
private int additionsSinceLock;
private int removalsSinceLock;
private transient Map<String, TypeDeclaration> classTypeDeclaration;
private List<RuleBasePartitionId> partitionIDs;
private ClassFieldAccessorCache classFieldAccessorCache;
/** The root Rete-OO for this <code>RuleBase</code>. */
private transient Rete rete;
private ReteooBuilder reteooBuilder;
private transient Map<Integer, SegmentMemory.Prototype> segmentProtos = new ConcurrentHashMap<Integer, Prototype>();
private KieComponentFactory kieComponentFactory;
/**
* Default constructor - for Externalizable. This should never be used by a user, as it
* will result in an invalid state for the instance.
*/
public ReteooRuleBase() {
}
/**
* Construct.
*
* @param id The rete network.
*/
public ReteooRuleBase(final String id,
final RuleBaseConfiguration config) {
this.config = (config != null) ? config : new RuleBaseConfiguration();
this.config.makeImmutable();
if ( this.config.isPhreakEnabled() ) {
logger.debug("Starting Engine in PHREAK mode");
} else {
logger.debug("Starting Engine in RETEOO mode");
}
createRulebaseId(id);
this.rootClassLoader = this.config.getClassLoader();
this.pkgs = new HashMap<String, Package>();
this.processes = new HashMap<String, Process>();
this.globals = new HashMap<String, Class<?>>();
this.statefulSessions = new ObjectHashSet();
this.classTypeDeclaration = new HashMap<String, TypeDeclaration>();
this.partitionIDs = new CopyOnWriteArrayList<RuleBasePartitionId>();
this.classFieldAccessorCache = new ClassFieldAccessorCache(this.rootClassLoader);
kieComponentFactory = getConfiguration().getComponentFactory();
this.factHandleFactory = kieComponentFactory.getFactHandleFactoryService();
kieComponentFactory.getTraitFactory().setRuleBase(this);
kieComponentFactory.getTripleStore().setId(id);
setupRete();
if (config != null && config.isMBeansEnabled()) {
DroolsManagementAgent.getInstance().registerKnowledgeBase(this);
}
}
public int nextWorkingMemoryCounter() {
return this.workingMemoryCounter.getAndIncrement();
}
public int getWorkingMemoryCounter() {
return this.workingMemoryCounter.get();
}
private void createRulebaseId(final String id) {
if (id != null) {
this.id = id;
} else {
String key = "";
if (config.isMBeansEnabled()) {
DroolsManagementAgent agent = DroolsManagementAgent.getInstance();
key = String.valueOf(agent.getNextKnowledgeBaseId());
}
this.id = "default" + key;
}
}
// ------------------------------------------------------------
// Instance methods
// ------------------------------------------------------------
/**
* Handles the read serialization of the Package. Patterns in Rules may reference generated data which cannot be serialized by default methods.
* The Package uses PackageCompilationData to hold a reference to the generated bytecode; which must be restored before any Rules.
* A custom ObjectInputStream, able to resolve classes against the bytecode in the PackageCompilationData, is used to restore the Rules.
*/
public void readExternal(final ObjectInput in) throws IOException,
ClassNotFoundException {
// PackageCompilationData must be restored before Rules as it has the ClassLoader needed to resolve the generated code references in Rules
DroolsObjectInput droolsStream;
boolean isDrools = in instanceof DroolsObjectInputStream;
boolean wasDrools = in.readBoolean();
if (wasDrools && !isDrools) {
throw new IllegalArgumentException("The knowledge base was serialized using a DroolsObjectOutputStream. A DroolsObjectInputStream is required for deserialization.");
}
if (isDrools) {
droolsStream = (DroolsObjectInput) in;
} else {
ByteArrayInputStream bytes = new ByteArrayInputStream((byte[]) in.readObject());
droolsStream = new DroolsObjectInputStream(bytes);
}
boolean classLoaderCacheEnabled = droolsStream.readBoolean();
Map<String, byte[]> store = (Map<String, byte[]>) droolsStream.readObject();
this.rootClassLoader = createProjectClassLoader(droolsStream.getParentClassLoader(), store);
droolsStream.setClassLoader(this.rootClassLoader);
droolsStream.setRuleBase(this);
this.classFieldAccessorCache = new ClassFieldAccessorCache(this.rootClassLoader);
this.config = (RuleBaseConfiguration) droolsStream.readObject();
this.config.setClassLoader(droolsStream.getParentClassLoader());
kieComponentFactory = getConfiguration().getComponentFactory();
this.pkgs = (Map<String, Package>) droolsStream.readObject();
for (Package pkg : this.pkgs.values()) {
pkg.getDialectRuntimeRegistry().onAdd(this.rootClassLoader);
}
// PackageCompilationData must be restored before Rules as it has the ClassLoader needed to resolve the generated code references in Rules
this.id = (String) droolsStream.readObject();
this.workingMemoryCounter.set(droolsStream.readInt());
this.processes = (Map<String, Process>) droolsStream.readObject();
Class cls = null;
try {
cls = droolsStream.getParentClassLoader().loadClass(droolsStream.readUTF());
this.factHandleFactory = (FactHandleFactory) cls.newInstance();
} catch (InstantiationException e) {
DroolsObjectInputStream.newInvalidClassException(cls,
e);
} catch (IllegalAccessException e) {
DroolsObjectInputStream.newInvalidClassException(cls,
e);
}
for (Package pkg : this.pkgs.values()) {
pkg.getDialectRuntimeRegistry().onBeforeExecute();
pkg.getClassFieldAccessorStore().setClassFieldAccessorCache(this.classFieldAccessorCache);
pkg.getClassFieldAccessorStore().wire();
}
this.populateTypeDeclarationMaps();
// read globals
Map<String, String> globs = (Map<String, String>) droolsStream.readObject();
populateGlobalsMap(globs);
this.partitionIDs = (List<RuleBasePartitionId>) droolsStream.readObject();
this.eventSupport = (RuleBaseEventSupport) droolsStream.readObject();
this.eventSupport.setRuleBase(this);
this.statefulSessions = new ObjectHashSet();
this.reteooBuilder = (ReteooBuilder) droolsStream.readObject();
this.reteooBuilder.setRuleBase(this);
this.rete = (Rete) droolsStream.readObject();
if (!isDrools) {
droolsStream.close();
}
this.getConfiguration().getComponentFactory().getTraitFactory().setRuleBase(this);
}
/**
* Handles the write serialization of the Package. Patterns in Rules may reference generated data which cannot be serialized by default methods.
* The Package uses PackageCompilationData to hold a reference to the generated bytecode. The generated bytecode must be restored before any Rules.
*/
public void writeExternal(final ObjectOutput out) throws IOException {
ObjectOutput droolsStream;
boolean isDrools = out instanceof DroolsObjectOutputStream;
ByteArrayOutputStream bytes;
out.writeBoolean( isDrools );
if (isDrools) {
droolsStream = out;
bytes = null;
} else {
bytes = new ByteArrayOutputStream();
droolsStream = new DroolsObjectOutputStream(bytes);
}
// must write this option first in order to properly deserialize later
droolsStream.writeBoolean(this.config.isClassLoaderCacheEnabled());
droolsStream.writeObject(((ProjectClassLoader) rootClassLoader).getStore());
droolsStream.writeObject(this.config);
droolsStream.writeObject(this.pkgs);
// Rules must be restored by an ObjectInputStream that can resolve using a given ClassLoader to handle seaprately by storing as
// a byte[]
droolsStream.writeObject(this.id);
droolsStream.writeInt(this.workingMemoryCounter.get());
droolsStream.writeObject(this.processes);
droolsStream.writeUTF(this.factHandleFactory.getClass().getName());
droolsStream.writeObject(buildGlobalMapForSerialization());
droolsStream.writeObject(this.partitionIDs);
this.eventSupport.removeEventListener(RuleBaseEventListener.class);
droolsStream.writeObject(this.eventSupport);
droolsStream.writeObject(this.reteooBuilder);
droolsStream.writeObject(this.rete);
if (!isDrools) {
droolsStream.flush();
droolsStream.close();
bytes.close();
out.writeObject(bytes.toByteArray());
}
}
private Map<String, String> buildGlobalMapForSerialization() {
Map<String, String> gl = new HashMap<String, String>();
for (Map.Entry<String, Class<?>> entry : this.globals.entrySet()) {
gl.put(entry.getKey(),
entry.getValue().getName());
}
return gl;
}
/**
* globals class types must be re-wired after serialization
*
* @param globs
* @throws ClassNotFoundException
*/
private void populateGlobalsMap(Map<String, String> globs) throws ClassNotFoundException {
this.globals = new HashMap<String, Class<?>>();
for (Map.Entry<String, String> entry : globs.entrySet()) {
this.globals.put(entry.getKey(),
this.rootClassLoader.loadClass(entry.getValue()));
}
}
/**
* type classes must be re-wired after serialization
*
* @throws ClassNotFoundException
*/
private void populateTypeDeclarationMaps() throws ClassNotFoundException {
// FIXME: readLock
this.classTypeDeclaration = new HashMap<String, TypeDeclaration>();
for (Package pkg : this.pkgs.values()) {
for (TypeDeclaration type : pkg.getTypeDeclarations().values()) {
type.setTypeClass(this.rootClassLoader.loadClass(type.getTypeClassName()));
this.classTypeDeclaration.put(type.getTypeClassName(),
type);
}
}
}
/**
* @return the id
*/
public String getId() {
return this.id;
}
public RuleBaseConfiguration getConfig() {
return config;
}
/**
* @see RuleBase
*/
public StatefulSession newStatefulSession() {
return newStatefulSession(SessionConfiguration.getDefaultInstance(),
EnvironmentFactory.newEnvironment());
}
public void disposeStatefulSession(final StatefulSession statefulSession) {
statefulSessionLock.lock();
try {
this.statefulSessions.remove(statefulSession);
for (Object listener : statefulSession.getRuleBaseUpdateListeners()) {
this.removeEventListener((RuleBaseEventListener) listener);
}
} finally {
statefulSessionLock.unlock();
}
}
/**
* @see RuleBase
*/
public FactHandleFactory getFactHandleFactory() {
return this.factHandleFactory;
}
public FactHandleFactory newFactHandleFactory() {
return this.factHandleFactory.newInstance();
}
public FactHandleFactory newFactHandleFactory(int id,
long counter) {
return this.factHandleFactory.newInstance(id,
counter);
}
public Process[] getProcesses() {
readLock();
try {
return this.processes.values().toArray(new Process[this.processes.size()]);
} finally {
readUnlock();
}
}
public Package[] getPackages() {
readLock();
try {
return this.pkgs.values().toArray(new Package[this.pkgs.size()]);
} finally {
readUnlock();
}
}
// FIXME: this returns the live map!
public Map<String, Package> getPackagesMap() {
return this.pkgs;
}
public Map<String, Class<?>> getGlobals() {
return this.globals;
}
public int getAdditionsSinceLock() {
return additionsSinceLock;
}
public int getRemovalsSinceLock() {
return removalsSinceLock;
}
public void lock() {
// The lock is reentrant, so we need additional magic here to skip
// notifications for locked if this thread already has locked it.
boolean firstLock = !this.lock.isWriteLockedByCurrentThread();
if (firstLock) {
this.eventSupport.fireBeforeRuleBaseLocked();
}
// Always lock to increase the counter
this.lock.writeLock();
if ( firstLock ) {
this.additionsSinceLock = 0;
this.removalsSinceLock = 0;
this.eventSupport.fireAfterRuleBaseLocked();
}
}
public void unlock() {
boolean lastUnlock = this.lock.getWriteHoldCount() == 1;
if (lastUnlock) {
this.eventSupport.fireBeforeRuleBaseUnlocked();
}
this.lock.writeUnlock();
if ( lastUnlock ) {
this.eventSupport.fireAfterRuleBaseUnlocked();
}
}
public void readLock() {
this.lock.readLock();
}
public void readUnlock() {
this.lock.readUnlock();
}
/**
* Add a <code>Package</code> to the network. Iterates through the
* <code>Package</code> adding Each individual <code>Rule</code> to the
* network. Before update network each referenced <code>WorkingMemory</code>
* is locked.
*
* @param newPkgs The package to add.
*/
public void addPackages( final Collection<Package> newPkgs ) {
List<Package> clonedPkgs = new ArrayList<Package>();
for (Package newPkg : newPkgs) {
clonedPkgs.add(newPkg.deepCloneIfAlreadyInUse(rootClassLoader));
}
lock();
try {
// we need to merge all byte[] first, so that the root classloader can resolve classes
for (Package newPkg : clonedPkgs) {
newPkg.checkValidity();
this.additionsSinceLock++;
this.eventSupport.fireBeforePackageAdded( newPkg );
if ( newPkg.hasTraitRegistry() ) {
getTraitRegistry().merge( newPkg.getTraitRegistry() );
}
Package pkg = this.pkgs.get( newPkg.getName() );
if ( pkg == null ) {
pkg = new Package( newPkg.getName() );
// @TODO we really should have a single root cache
pkg.setClassFieldAccessorCache( this.classFieldAccessorCache );
pkgs.put( pkg.getName(),
pkg );
}
// first merge anything related to classloader re-wiring
pkg.getDialectRuntimeRegistry().merge( newPkg.getDialectRuntimeRegistry(),
this.rootClassLoader,
true );
}
// Add all Type Declarations, this has to be done first incase packages cross reference each other during build process.
for ( Package newPkg : clonedPkgs ) {
// we have to do this before the merging, as it does some classloader resolving
String lastType = null;
try {
// Add the type declarations to the RuleBase
if ( newPkg.getTypeDeclarations() != null ) {
JavaDialectRuntimeData runtime = ((JavaDialectRuntimeData) newPkg.getDialectRuntimeRegistry().getDialectData( "java" ));
// add type declarations
for ( TypeDeclaration newDecl : newPkg.getTypeDeclarations().values() ) {
lastType = newDecl.getTypeClassName();
TypeDeclaration typeDeclaration = this.classTypeDeclaration.get( newDecl.getTypeClassName() );
if ( typeDeclaration == null ) {
String className = newDecl.getTypeClassName();
byte [] def = runtime.getClassDefinition(JavaDialectRuntimeData.convertClassToResourcePath(className));
Class<?> definedKlass = registerAndLoadTypeDefinition( className, def );
if ( definedKlass == null && typeDeclaration.isNovel() ) {
throw new RuntimeException( "Registering null bytes for class " + className );
}
if (newDecl.getTypeClassDef() == null) {
newDecl.setTypeClassDef( new ClassDefinition() );
}
newDecl.getTypeClassDef().setDefinedClass( definedKlass );
newDecl.setTypeClass( definedKlass );
this.classTypeDeclaration.put( className, newDecl );
typeDeclaration = newDecl;
} else {
Class<?> definedKlass = typeDeclaration.getTypeClass();
newDecl.getTypeClassDef().setDefinedClass( definedKlass );
newDecl.setTypeClass( definedKlass );
mergeTypeDeclarations( typeDeclaration,
newDecl );
}
// update existing OTNs
updateDependentTypes( newPkg,
typeDeclaration );
}
}
} catch (ClassNotFoundException e) {
throw new RuntimeDroolsException( "unable to resolve Type Declaration class '" + lastType + "'", e );
}
}
// now iterate again, this time onBeforeExecute will handle any wiring or cloader re-creating that needs to be done as part of the merge
for (Package newPkg : clonedPkgs) {
Package pkg = this.pkgs.get( newPkg.getName() );
// this needs to go here, as functions will set a java dialect to dirty
if (newPkg.getFunctions() != null) {
for (Map.Entry<String, Function> entry : newPkg.getFunctions().entrySet()) {
pkg.addFunction( entry.getValue() );
}
}
pkg.getDialectRuntimeRegistry().onBeforeExecute();
// with the classloader recreated for all byte[] classes, we should now merge and wire any new accessors
pkg.getClassFieldAccessorStore().merge( newPkg.getClassFieldAccessorStore() );
}
for (Package newPkg : clonedPkgs) {
Package pkg = this.pkgs.get( newPkg.getName() );
// now merge the new package into the existing one
mergePackage( pkg,
newPkg );
// add the window declarations to the kbase
for( WindowDeclaration window : newPkg.getWindowDeclarations().values() ) {
addWindowDeclaration( newPkg,
window );
}
// add entry points to the kbase
for (String id : newPkg.getEntryPointIds()) {
addEntryPoint( id );
}
// add the rules to the RuleBase
for ( Rule rule : newPkg.getRules() ) {
addRule( newPkg, rule );
}
// add the flows to the RuleBase
if ( newPkg.getRuleFlows() != null ) {
final Map<String, org.kie.api.definition.process.Process> flows = newPkg.getRuleFlows();
for ( org.kie.api.definition.process.Process process : flows.values() ) {
// XXX: we could take the lock inside addProcess() out, but OTOH: this is what the VM is supposed to do ...
addProcess( process );
}
}
this.eventSupport.fireAfterPackageAdded( newPkg );
}
} finally {
unlock();
}
}
public Class<?> registerAndLoadTypeDefinition( String className, byte[] def ) throws ClassNotFoundException {
try {
return this.rootClassLoader.loadClass( className );
} catch (ClassNotFoundException e) {
if (def != null && rootClassLoader instanceof ProjectClassLoader) {
return ((ProjectClassLoader)rootClassLoader).defineClass(className, def);
}
throw e;
}
}
protected void updateDependentTypes( Package newPkg,
TypeDeclaration typeDeclaration ) {
// update OTNs
if( this.getConfiguration().getEventProcessingMode().equals( EventProcessingOption.STREAM ) ) {
// if the expiration for the type was set, then add 1, otherwise return -1
long exp = typeDeclaration.getExpirationOffset() > -1 ? typeDeclaration.getExpirationOffset() + 1 : -1;
// if we are running in STREAM mode, update expiration offset
for( EntryPointNode ep : this.rete.getEntryPointNodes().values() ) {
for( ObjectTypeNode node : ep.getObjectTypeNodes().values() ) {
if( node.isAssignableFrom( typeDeclaration.getObjectType() ) ) {
node.setExpirationOffset( Math.max( node.getExpirationOffset(), exp ) );
}
}
}
}
}
private void mergeTypeDeclarations( TypeDeclaration existingDecl,
TypeDeclaration newDecl ) {
existingDecl.addRedeclaration(newDecl);
if ( ! nullSafeEquals( existingDecl.getFormat(),
newDecl.getFormat() ) ||
! nullSafeEquals( existingDecl.getObjectType(),
newDecl.getObjectType() ) ||
! nullSafeEquals( existingDecl.getTypeClassName(),
newDecl.getTypeClassName() ) ||
! nullSafeEquals( existingDecl.getTypeName(),
newDecl.getTypeName() ) ) {
throw new RuntimeDroolsException( "Unable to merge Type Declaration for class '" + existingDecl.getTypeName() +
"'" );
}
existingDecl.setDurationAttribute( mergeLeft( existingDecl.getTypeName(),
"Unable to merge @duration attribute for type declaration of class:",
existingDecl.getDurationAttribute(),
newDecl.getDurationAttribute(),
true,
false ) );
existingDecl.setDynamic( mergeLeft( existingDecl.getTypeName(),
"Unable to merge @propertyChangeSupport (a.k.a. dynamic) attribute for type declaration of class:",
existingDecl.isDynamic(),
newDecl.isDynamic(),
true,
false ) );
existingDecl.setPropertyReactive( mergeLeft(existingDecl.getTypeName(),
"Unable to merge @propertyReactive attribute for type declaration of class:",
existingDecl.isPropertyReactive(),
newDecl.isPropertyReactive(),
true,
false) );
existingDecl.setExpirationOffset( Math.max( existingDecl.getExpirationOffset(),
newDecl.getExpirationOffset() ) );
if ( newDecl.getNature().equals( TypeDeclaration.Nature.DEFINITION ) && newDecl.isNovel() ) {
// At this point, the definitions must be equivalent.
// So the only illegal case is a novel definition of an already existing type
existingDecl.setNovel( mergeLeft( existingDecl.getTypeName(),
"Unable to merge @novel attribute for type declaration of class:",
existingDecl.isNovel(),
newDecl.isNovel(),
true,
false ) );
}
if ( newDecl.getNature().equals( TypeDeclaration.Nature.DEFINITION ) || existingDecl.getResource() == null ) {
existingDecl.setResource( mergeLeft( existingDecl.getTypeName(),
"Unable to merge resource attribute for type declaration of class:",
existingDecl.getResource(),
newDecl.getResource(),
true,
true ) );
}
existingDecl.setRole( mergeLeft( existingDecl.getTypeName(),
"Unable to merge @role attribute for type declaration of class:",
isSet(existingDecl.getSetMask(), TypeDeclaration.ROLE_BIT)
&& newDecl.getRole() != TypeDeclaration.Role.FACT
? existingDecl.getRole() : null,
newDecl.getRole(),
true,
false ) );
existingDecl.setTimestampAttribute( mergeLeft( existingDecl.getTypeName(),
"Unable to merge @timestamp attribute for type declaration of class:",
existingDecl.getTimestampAttribute(),
newDecl.getTimestampAttribute(),
true,
false ) );
existingDecl.setTypesafe( mergeLeft(existingDecl.getTypeName(),
"Unable to merge @typesafe attribute for type declaration of class:",
existingDecl.isTypesafe(),
newDecl.isTypesafe(),
true,
false ) );
}
private <T> T mergeLeft( String typeClass,
String errorMsg,
T leftVal,
T rightVal,
boolean errorOnDiff,
boolean override ) {
T newValue = leftVal;
if ( ! nullSafeEquals( leftVal,
rightVal ) ) {
if ( leftVal == null && rightVal != null ) {
newValue = rightVal;
} else if ( leftVal != null && rightVal != null ) {
if ( override ) {
newValue = rightVal;
} else {
if ( errorOnDiff ) {
throw new RuntimeDroolsException( errorMsg + " '" + typeClass + "'" );
} else {
// do nothing, just use the left value
}
}
}
}
return newValue;
}
private boolean nullSafeEquals( Object o1,
Object o2 ) {
return ( o1 == null ) ? o2 == null : o1.equals( o2 );
}
/**
* Merge a new package with an existing package.
* Most of the work is done by the concrete implementations,
* but this class does some work (including combining imports, compilation data, globals,
* and the actual Rule objects into the package).
*/
private void mergePackage( final Package pkg,
final Package newPkg ) {
// Merge imports
final Map<String, ImportDeclaration> imports = pkg.getImports();
imports.putAll(newPkg.getImports());
String lastIdent = null;
String lastType = null;
try {
// merge globals
if (newPkg.getGlobals() != null && newPkg.getGlobals() != Collections.EMPTY_MAP) {
Map<String, String> globals = pkg.getGlobals();
// Add globals
for (final Map.Entry<String, String> entry : newPkg.getGlobals().entrySet()) {
final String identifier = entry.getKey();
final String type = entry.getValue();
lastIdent = identifier;
lastType = type;
if (globals.containsKey( identifier ) && !globals.get( identifier ).equals( type )) {
throw new PackageIntegrationException( pkg );
} else {
pkg.addGlobal( identifier,
this.rootClassLoader.loadClass( type ) );
// this isn't a package merge, it's adding to the rulebase, but I've put it here for convienience
this.globals.put( identifier,
this.rootClassLoader.loadClass( type ) );
}
}
}
} catch (ClassNotFoundException e) {
throw new RuntimeDroolsException( "Unable to resolve class '" + lastType +
"' for global '" + lastIdent + "'" );
}
// merge entry point declarations
if (newPkg.getEntryPointIds() != null) {
for (String ep : newPkg.getEntryPointIds()) {
pkg.addEntryPointId( ep );
}
}
// merge the type declarations
if (newPkg.getTypeDeclarations() != null) {
// add type declarations
for (TypeDeclaration type : newPkg.getTypeDeclarations().values()) {
// @TODO should we allow overrides? only if the class is not in use.
if (!pkg.getTypeDeclarations().containsKey( type.getTypeName() )) {
// add to package list of type declarations
pkg.addTypeDeclaration( type );
}
}
}
// merge window declarations
if ( newPkg.getWindowDeclarations() != null ) {
// add window declarations
for ( WindowDeclaration window : newPkg.getWindowDeclarations().values() ) {
if ( !pkg.getWindowDeclarations().containsKey( window.getName() ) ||
pkg.getWindowDeclarations().get( window.getName() ).equals( window ) ) {
pkg.addWindowDeclaration( window );
} else {
throw new RuntimeDroolsException( "Unable to merge two conflicting window declarations for window named: "+window.getName() );
}
}
}
//Merge rules into the RuleBase package
//as this is needed for individual rule removal later on
for (final Rule newRule : newPkg.getRules()) {
// remove the rule if it already exists
if (pkg.getRule(newRule.getName()) != null) {
removeRule( pkg, pkg.getRule(newRule.getName()) );
}
pkg.addRule(newRule);
}
//Merge The Rule Flows
if (newPkg.getRuleFlows() != null) {
for (Process flow : newPkg.getRuleFlows().values()) {
pkg.addProcess(flow);
}
}
}
protected void setupRete() {
this.rete = new Rete( this );
this.reteooBuilder = new ReteooBuilder( this );
// always add the default entry point
EntryPointNode epn = kieComponentFactory.getNodeFactoryService().buildEntryPointNode( this.reteooBuilder.getIdGenerator().getNextId(),
RuleBasePartitionId.MAIN_PARTITION,
this.getConfiguration().isMultithreadEvaluation(),
this.rete,
EntryPointId.DEFAULT );
epn.attach();
}
public void registerAddedEntryNodeCache(EntryPointNode node) {
if (addedEntryNodeCache == null) addedEntryNodeCache = new HashSet<EntryPointNode>();
addedEntryNodeCache.add(node);
}
public Set<EntryPointNode> getAddedEntryNodeCache() {
return addedEntryNodeCache;
}
public void registeRremovedEntryNodeCache(EntryPointNode node) {
if (removedEntryNodeCache == null) removedEntryNodeCache = new HashSet<EntryPointNode>();
removedEntryNodeCache.add(node);
}
public Set<EntryPointNode> getRemovedEntryNodeCache() {
return removedEntryNodeCache;
}
/**
* Retrieve the Rete-OO network for this <code>RuleBase</code>.
*
* @return The RETE-OO network.
*/
public Rete getRete() {
return this.rete;
}
public ReteooBuilder getReteooBuilder() {
return this.reteooBuilder;
}
/**
* Assert a fact object.
*
* @param handle
* The handle.
* @param object
* The fact.
* @param workingMemory
* The working-memory.
*
* @throws org.drools.core.FactException
* If an error occurs while performing the assertion.
*/
public void assertObject(final FactHandle handle,
final Object object,
final PropagationContext context,
final InternalWorkingMemory workingMemory) throws FactException {
getRete().assertObject( (DefaultFactHandle) handle,
context,
workingMemory );
}
/**
* Retract a fact object.
*
* @param handle
* The handle.
* @param workingMemory
* The working-memory.
*
* @throws org.drools.core.FactException
* If an error occurs while performing the retraction.
*/
public void retractObject(final FactHandle handle,
final PropagationContext context,
final AbstractWorkingMemory workingMemory) throws FactException {
getRete().retractObject( (InternalFactHandle) handle,
context,
workingMemory );
}
public StatefulSession newStatefulSession(boolean keepReference) {
SessionConfiguration config = new SessionConfiguration();
config.setKeepReference( keepReference );
return newStatefulSession( config,
EnvironmentFactory.newEnvironment() );
}
public StatefulSession newStatefulSession(java.io.InputStream stream) {
return newStatefulSession( stream,
true );
}
public StatefulSession newStatefulSession(java.io.InputStream stream,
boolean keepReference) {
return newStatefulSession( stream,
keepReference,
SessionConfiguration.getDefaultInstance() );
}
public StatefulSession newStatefulSession(java.io.InputStream stream,
boolean keepReference,
SessionConfiguration conf) {
StatefulSession session = null;
try {
readLock();
try {
// first unwrap the byte[]
ObjectInputStream ois = new ObjectInputStream( stream );
// standard serialisation would have written the statateful session instance info to the stream first
// so we read it, but we don't need it, so just ignore.
AbstractWorkingMemory rsession = (AbstractWorkingMemory) ois.readObject();
// now unmarshall that byte[]
ByteArrayInputStream bais = new ByteArrayInputStream( rsession.bytes );
Marshaller marshaller = MarshallerFactory.newMarshaller(new KnowledgeBaseImpl(this), new ObjectMarshallingStrategy[]{MarshallerFactory.newSerializeMarshallingStrategy()});
Environment environment = EnvironmentFactory.newEnvironment();
KieSession ksession = marshaller.unmarshall( bais,
conf,
environment );
session = (StatefulSession) ((StatefulKnowledgeSessionImpl) ksession).session;
if ( keepReference ) {
addStatefulSession(session);
for (Object listener : session.getRuleBaseUpdateListeners()) {
addEventListener((RuleBaseEventListener) listener);
}
}
bais.close();
} finally {
readUnlock();
}
} catch ( Exception e ) {
throw new RuntimeException( "Unable to unmarshall session",
e );
} finally {
try {
stream.close();
} catch ( IOException e ) {
throw new RuntimeException( "Unable to close stream", e );
}
}
return session;
}
public StatefulSession newStatefulSession(SessionConfiguration sessionConfig,
Environment environment) {
if ( sessionConfig == null ) {
sessionConfig = SessionConfiguration.getDefaultInstance();
}
if ( environment == null ) {
environment = EnvironmentFactory.newEnvironment();
}
return newStatefulSession( nextWorkingMemoryCounter(),
sessionConfig,
environment );
}
StatefulSession newStatefulSession(int id,
final SessionConfiguration sessionConfig,
final Environment environment) {
if ( this.getConfiguration().isSequential() ) {
throw new RuntimeException( "Cannot have a stateful rule session, with sequential configuration set to true" );
}
readLock();
try {
WorkingMemoryFactory wmFactory = kieComponentFactory.getWorkingMemoryFactory();
AbstractWorkingMemory session = ( AbstractWorkingMemory ) wmFactory.createWorkingMemory( id, this,
sessionConfig, environment );
new StatefulKnowledgeSessionImpl(session);
if ( sessionConfig.isKeepReference() ) {
addStatefulSession(session);
for (Object listener : session.getRuleBaseUpdateListeners()) {
addEventListener((RuleBaseEventListener) listener);
}
}
return session;
} finally {
readUnlock();
}
}
public StatelessSession newStatelessSession() {
synchronized ( this.pkgs ) {
return new ReteooStatelessSession( this );
}
}
public int getNodeCount() {
// may start in 0
return this.reteooBuilder.getIdGenerator().getLastId() + 1;
}
public void addPackages(Package[] pkgs) {
addPackages( Arrays.asList(pkgs) );
}
public void addPackage(final Package newPkg) {
addPackages( Collections.singleton( newPkg ) );
}
public void registerSegmentPrototype(LeftTupleSource tupleSource, SegmentMemory smem) {
segmentProtos.put(tupleSource.getId(), smem.asPrototype());
}
public void invalidateSegmentPrototype(LeftTupleSource tupleSource) {
segmentProtos.remove(tupleSource.getId());
LeftTupleSinkPropagator sinkProp = tupleSource.getSinkPropagator();
for (LeftTupleSinkNode sink = (LeftTupleSinkNode) sinkProp.getFirstLeftTupleSink(); sink != null; sink = sink.getNextLeftTupleSinkNode()) {
if (sink instanceof LeftTupleSource) {
invalidateSegmentPrototype((LeftTupleSource)sink);
}
}
}
public SegmentMemory getSegmentFromPrototype(InternalWorkingMemory wm, LeftTupleSource tupleSource) {
SegmentMemory.Prototype proto = segmentProtos.get(tupleSource.getId());
if (proto == null) {
return null;
}
return proto.newSegmentMemory(wm);
}
private static class TypeDeclarationCandidate {
public TypeDeclaration candidate = null;
public int score = Integer.MAX_VALUE;
}
public TypeDeclaration getTypeDeclaration( Class<?> clazz ) {
TypeDeclaration typeDeclaration = this.classTypeDeclaration.get( clazz.getName() );
if (typeDeclaration == null) {
// check super classes and keep a score of how up in the hierarchy is there a declaration
TypeDeclarationCandidate candidate = checkSuperClasses( clazz );
// now check interfaces
candidate = checkInterfaces( clazz,
candidate,
1 );
if (candidate != null) {
typeDeclaration = candidate.candidate;
}
}
return typeDeclaration;
}
private TypeDeclarationCandidate checkSuperClasses( Class<?> clazz ) {
TypeDeclaration typeDeclaration = null;
Class<?> current = clazz.getSuperclass();
int score = 0;
while ( typeDeclaration == null && current != null ) {
score++;
typeDeclaration = this.classTypeDeclaration.get( current.getName() );
current = current.getSuperclass();
}
TypeDeclarationCandidate candidate = null;
if ( typeDeclaration != null ) {
candidate = new TypeDeclarationCandidate();
candidate.candidate = typeDeclaration;
candidate.score = score;
}
return candidate;
}
private TypeDeclarationCandidate checkInterfaces( Class<?> clazz,
TypeDeclarationCandidate baseline,
int level ) {
TypeDeclarationCandidate candidate = null;
if (baseline == null || level < baseline.score) {
// search
TypeDeclaration typeDeclaration;
for (Class<?> ifc : clazz.getInterfaces()) {
typeDeclaration = this.classTypeDeclaration.get( ifc.getName() );
if (typeDeclaration != null) {
candidate = new TypeDeclarationCandidate();
candidate.candidate = typeDeclaration;
candidate.score = level;
break;
} else {
candidate = checkInterfaces( ifc,
baseline,
level + 1 );
}
}
} else {
candidate = baseline;
}
return candidate;
}
public Collection<TypeDeclaration> getTypeDeclarations() {
return this.classTypeDeclaration.values();
}
public void addRule( final Package pkg,
final Rule rule ) throws InvalidPatternException {
lock();
try {
this.eventSupport.fireBeforeRuleAdded( pkg,
rule );
// if ( !rule.isValid() ) {
// throw new IllegalArgumentException( "The rule called " + rule.getName() + " is not valid. Check for compile errors reported." );
// }
addRule( rule );
this.eventSupport.fireAfterRuleAdded( pkg,
rule );
} finally {
unlock();
}
}
protected void addEntryPoint( final Package pkg,
final String id ) throws InvalidPatternException {
lock();
try {
addEntryPoint(id);
} finally {
unlock();
}
}
protected void addRule(final Rule rule) throws InvalidPatternException {
// This adds the rule. ReteBuilder has a reference to the WorkingMemories and will propagate any existing facts.
this.reteooBuilder.addRule(rule);
}
protected void addEntryPoint(final String id) throws InvalidPatternException {
// This adds the entry point. ReteBuilder has a reference to the WorkingMemories and will propagate any existing facts.
this.reteooBuilder.addEntryPoint(id);
}
public void addWindowDeclaration( final Package pkg,
final WindowDeclaration window ) throws InvalidPatternException {
lock();
try {
addWindowDeclaration( window );
} finally {
unlock();
}
}
protected void addWindowDeclaration(final WindowDeclaration window) throws InvalidPatternException {
// This adds the named window. ReteBuilder has a reference to the WorkingMemories and will propagate any existing facts.
this.reteooBuilder.addNamedWindow(window);
}
public void removePackage( final String packageName ) {
lock();
try {
final Package pkg = this.pkgs.get( packageName );
if (pkg == null) {
throw new IllegalArgumentException( "Package name '" + packageName +
"' does not exist for this Rule Base." );
}
this.removalsSinceLock++;
this.eventSupport.fireBeforePackageRemoved( pkg );
for (Rule rule : pkg.getRules()) {
removeRule( pkg, rule );
}
// getting the list of referenced globals
final Set<String> referencedGlobals = new HashSet<String>();
for (Package pkgref : this.pkgs.values()) {
if (pkgref != pkg) {
referencedGlobals.addAll( pkgref.getGlobals().keySet() );
}
}
// removing globals declared inside the package that are not shared
for (String globalName : pkg.getGlobals().keySet()) {
if (!referencedGlobals.contains( globalName )) {
this.globals.remove( globalName );
}
}
//and now the rule flows
for ( String processName : new ArrayList<String>(pkg.getRuleFlows().keySet()) ) {
removeProcess( processName );
}
// removing the package itself from the list
this.pkgs.remove( pkg.getName() );
pkg.getDialectRuntimeRegistry().onRemove();
//clear all members of the pkg
pkg.clear();
this.eventSupport.fireAfterPackageRemoved( pkg );
} finally {
unlock();
}
}
public void removeQuery( final String packageName,
final String ruleName ) {
removeRule(packageName,
ruleName);
}
public void removeRule( final String packageName,
final String ruleName ) {
lock();
try {
final Package pkg = this.pkgs.get( packageName );
if (pkg == null) {
throw new IllegalArgumentException( "Package name '" + packageName +
"' does not exist for this Rule Base." );
}
final Rule rule = pkg.getRule( ruleName );
if (rule == null) {
throw new IllegalArgumentException( "Rule name '" + ruleName +
"' does not exist in the Package '" +
packageName +
"'." );
}
this.removalsSinceLock++;
removeRule( pkg,
rule );
pkg.removeRule( rule );
addReloadDialectDatas( pkg.getDialectRuntimeRegistry() );
} finally {
unlock();
}
}
/**
* Notify listeners and sub-classes about imminent removal of a rule from a package.
*
* @param pkg
* @param rule
*/
// FIXME: removeTerminalNode(String, String) and removeTerminalNode(Package, Rule) do totally different things!
public void removeRule( final Package pkg,
final Rule rule ) {
lock();
try {
this.eventSupport.fireBeforeRuleRemoved(pkg,
rule);
removeRule(rule);
this.eventSupport.fireAfterRuleRemoved(pkg,
rule);
} finally {
unlock();
}
}
protected void removeRule(final Rule rule) {
this.reteooBuilder.removeRule( rule );
}
public void removeFunction( final String packageName,
final String functionName ) {
lock();
try {
final Package pkg = this.pkgs.get( packageName );
if (pkg == null) {
throw new IllegalArgumentException( "Package name '" + packageName +
"' does not exist for this Rule Base." );
}
if (!pkg.getFunctions().containsKey( functionName )) {
throw new IllegalArgumentException( "function name '" + packageName +
"' does not exist in the Package '" +
packageName +
"'." );
}
removeFunction( pkg,
functionName );
pkg.removeFunction( functionName );
addReloadDialectDatas( pkg.getDialectRuntimeRegistry() );
} finally {
unlock();
}
}
/**
* Handle function removal.
*
* This method is intended for sub-classes, and called after the
* {@link RuleBaseEventListener#beforeRuleRemoved(org.drools.core.event.BeforeRuleRemovedEvent) before-rule-removed}
* event is fired, and before the function is physically removed from the package.
*
* This method is called with the rulebase lock held.
* @param functionName
*/
protected/* abstract */void removeFunction( String functionName ) {
// Nothing in default.
}
/**
* Notify listeners and sub-classes about imminent removal of a function from a package.
*
* This method is called with the rulebase lock held.
* @param pkg
* @param functionName
*/
private void removeFunction( final Package pkg,
final String functionName ) {
this.eventSupport.fireBeforeFunctionRemoved( pkg,
functionName );
removeFunction( functionName );
this.eventSupport.fireAfterFunctionRemoved( pkg,
functionName );
}
public void addProcess( final Process process ) {
// XXX: could use a synchronized(processes) here.
this.eventSupport.fireBeforeProcessAdded( process );
lock();
try {
this.processes.put( process.getId(),
process );
} finally {
unlock();
}
this.eventSupport.fireAfterProcessAdded( process );
}
public void removeProcess( final String id ) {
Process process = this.processes.get( id );
if (process == null) {
throw new IllegalArgumentException( "Process '" + id + "' does not exist for this Rule Base." );
}
this.eventSupport.fireBeforeProcessRemoved( process );
lock();
try {
this.processes.remove( id );
this.pkgs.get(process.getPackageName()).removeRuleFlow(id);
} finally {
unlock();
}
this.eventSupport.fireAfterProcessRemoved( process );
}
public Process getProcess( final String id ) {
readLock();
try {
return this.processes.get( id );
} finally {
readUnlock();
}
}
public void addStatefulSession( final StatefulSession statefulSession ) {
statefulSessionLock.lock();
try {
this.statefulSessions.add( statefulSession );
} finally {
statefulSessionLock.unlock();
}
}
public Package getPackage( final String name ) {
return this.pkgs.get( name );
}
public StatefulSession[] getStatefulSessions() {
statefulSessionLock.lock();
try {
final StatefulSession[] copyOfSessions = new StatefulSession[this.statefulSessions.size()];
this.statefulSessions.toArray( copyOfSessions );
return copyOfSessions;
} finally {
statefulSessionLock.unlock();
}
}
public InternalWorkingMemory[] getWorkingMemories() {
statefulSessionLock.lock();
try {
final InternalWorkingMemory[] copyOfMemories = new InternalWorkingMemory[this.statefulSessions.size()];
this.statefulSessions.toArray( copyOfMemories );
return copyOfMemories;
} finally {
statefulSessionLock.unlock();
}
}
public RuleBaseConfiguration getConfiguration() {
if ( this.config == null ) {
this.config = new RuleBaseConfiguration();
}
return this.config;
}
public ClassLoader getRootClassLoader() {
return this.rootClassLoader;
}
public void executeQueuedActions() {
DialectRuntimeRegistry registry;
while (( registry = reloadPackageCompilationData.poll() ) != null) {
registry.onBeforeExecute();
}
}
public RuleBasePartitionId createNewPartitionId() {
RuleBasePartitionId p;
synchronized (this.partitionIDs) {
p = new RuleBasePartitionId( "P-" + this.partitionIDs.size() );
this.partitionIDs.add( p );
}
return p;
}
public List<RuleBasePartitionId> getPartitionIds() {
// this returns an unmodifiable CopyOnWriteArrayList, so should be safe for concurrency
return Collections.unmodifiableList( this.partitionIDs );
}
public void addEventListener( final RuleBaseEventListener listener ) {
// no need for synchonization or locking because eventSupport is thread-safe
this.eventSupport.addEventListener( listener );
}
public void removeEventListener( final RuleBaseEventListener listener ) {
// no need for synchonization or locking because eventSupport is thread-safe
this.eventSupport.removeEventListener( listener );
}
public List<RuleBaseEventListener> getRuleBaseEventListeners() {
// no need for synchonization or locking because eventSupport is thread-safe
return this.eventSupport.getEventListeners();
}
public boolean isEvent( Class<?> clazz ) {
readLock();
try {
for (Package pkg : this.pkgs.values()) {
if (pkg.isEvent( clazz )) {
return true;
}
}
return false;
} finally {
readUnlock();
}
}
public FactType getFactType( final String name ) {
readLock();
try {
for (Package pkg : this.pkgs.values()) {
FactType type = pkg.getFactType( name );
if (type != null) {
return type;
}
}
return null;
} finally {
readUnlock();
}
}
private void addReloadDialectDatas( DialectRuntimeRegistry registry ) {
this.reloadPackageCompilationData.offer( registry );
}
public static interface RuleBaseAction
extends
Externalizable {
public void execute( InternalRuleBase ruleBase );
}
public ClassFieldAccessorCache getClassFieldAccessorCache() {
return this.classFieldAccessorCache;
}
public Set<String> getEntryPointIds() {
Set<String> entryPointIds = new HashSet<String>();
for (Package pkg : this.pkgs.values()) {
entryPointIds.addAll( pkg.getEntryPointIds() );
}
return entryPointIds;
}
public TripleStore getTripleStore() {
return this.getConfiguration().getComponentFactory().getTripleStore();
}
public TraitRegistry getTraitRegistry() {
return this.getConfiguration().getComponentFactory().getTraitRegistry();
}
public void removeObjectsGeneratedFromResource(Resource resource) {
for (Package pkg : pkgs.values()) {
List<Rule> rulesToBeRemoved = pkg.removeRulesGeneratedFromResource(resource);
for (Rule rule : rulesToBeRemoved) {
removeRule(rule);
}
List<Function> functionsToBeRemoved = pkg.removeFunctionsGeneratedFromResource(resource);
for (Function function : functionsToBeRemoved) {
removeFunction(function.getName());
}
}
}
}