instrumentedobjecttypenode.java

rule engine drools-2.0-beta-18

package org.drools.reteoo;

import org.drools.FactException;
import org.drools.FactHandle;
import org.drools.spi.ObjectType;

import java.util.ArrayList;
import java.util.List;

public class InstrumentedObjectTypeNode extends ObjectTypeNode
{
    private List assertedObjects;

    private List retractedHandles;

    private List modifiedHandles;

    public InstrumentedObjectTypeNode(ObjectType objectType)
    {
        super( objectType );

        this.assertedObjects = new ArrayList( );
        this.retractedHandles = new ArrayList( );
        this.modifiedHandles = new ArrayList( );
    }

    public void assertObject(FactHandle handle,
                             Object object,
                             WorkingMemoryImpl memory) throws FactException
    {
        super.assertObject( handle, object, memory );

        this.assertedObjects.add( object );
    }

    public void retractObject(FactHandle handle, WorkingMemoryImpl memory) throws FactException
    {
        super.retractObject( handle, memory );

        this.retractedHandles.add( handle );
    }

    public List getAssertedObjects()
    {
        return this.assertedObjects;
    }

    public List getModifiedHandles()
    {
        return this.modifiedHandles;
    }

    public List getRetractedHandles()
    {
        return this.retractedHandles;
    }
}