package org.jacorb.orb.portableInterceptor;

import org.omg.PortableInterceptor.*;

import java.util.*;
import org.jacorb.orb.LocalityConstrainedObject;

public abstract class RecursionAwareCI
    extends LocalityConstrainedObject 
    implements ClientRequestInterceptor
{
    private Hashtable thread_stacks = null;
    
    private Hashtable ignore_operations = null;

    /**
     * @param ignore_special_ops If set to true, calls to
     * methods from the CORBA.Object interface like _is_a 
     * will be ignored.
     */
    public RecursionAwareCI( boolean ignore_special_ops ) 
    {
        thread_stacks = new Hashtable();
        ignore_operations = new Hashtable();

        if ( ignore_special_ops )
        {
            ignore_operations.put( "_is_a", "" );
            ignore_operations.put( "_get_interface", "" );
            ignore_operations.put( "_non_existent", "" );            
            ignore_operations.put( "_get_policy", "" );
            ignore_operations.put( "_get_domain_managers", "" );
            ignore_operations.put( "_set_policy_overrides", "" );
        }
    }

    public void addIgnoreOperation( String operation_name )
    {
        ignore_operations.put( operation_name, operation_name );
    }

    private boolean enterCall( String operation )
    {
        if ( ignore_operations.containsKey( operation ) )
            return false;

        Thread current = Thread.currentThread();

        if ( thread_stacks.containsKey( current ))
            return false;

        thread_stacks.put( current, current );

        return true;
    }

    private void exitCall()
    {
        thread_stacks.remove( Thread.currentThread() );
    }

    // implementation InterceptorOperations interface
    public final void send_request( ClientRequestInfo ri ) 
        throws ForwardRequest
    {
        if( enterCall( ri.operation() ))
        {
            try
            {
                do_send_request( ri );
            }
            finally
            {
                exitCall();
            }
        }
    }

    public final void send_poll(ClientRequestInfo ri)
    {
        if( enterCall( ri.operation() ))
        {
            try
            {
                do_send_poll( ri );
            }
            finally
            {
                exitCall();
            }
        }
    }

    public final void receive_reply(ClientRequestInfo ri)
    {	
        if( enterCall( ri.operation() ))
        {
            try
            {
                do_receive_reply( ri );
            }
            finally
            {
                exitCall();
            }	
        }
    }

    public final void receive_exception(ClientRequestInfo ri) 
        throws ForwardRequest
    {
        if( enterCall( ri.operation() ))
        {
            try
            {
                do_receive_exception( ri );
            }
            finally
            {
                exitCall();
            }	
        }
    }

    public final void receive_other(ClientRequestInfo ri) 
        throws ForwardRequest
    {
        if( enterCall( ri.operation() ))
        {
            try
            {
                do_receive_other( ri );
            }
            finally
            {
                exitCall();
            }	
        }
    }

    public abstract void do_send_request( ClientRequestInfo ri ) 
        throws ForwardRequest;

    public abstract void do_send_poll(ClientRequestInfo ri);

    public abstract void do_receive_reply(ClientRequestInfo ri);

    public abstract void do_receive_exception(ClientRequestInfo ri)
        throws ForwardRequest;

    public abstract void do_receive_other(ClientRequestInfo ri) 
        throws ForwardRequest;
}











