// $Header: /home/cvs/jakarta-jmeter/src/core/org/apache/jmeter/engine/ConvertListeners.java,v 1.10 2004/02/14 03:34:29 sebb Exp $
/*
 * Copyright 2001-2004 The Apache Software Foundation.
 *
 * 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.apache.jmeter.engine;

import java.rmi.RemoteException;
import java.util.Iterator;

import org.apache.jmeter.samplers.RemoteListenerWrapper;
import org.apache.jmeter.samplers.RemoteSampleListener;
import org.apache.jmeter.samplers.RemoteSampleListenerImpl;
import org.apache.jmeter.samplers.RemoteSampleListenerWrapper;
import org.apache.jmeter.samplers.RemoteTestListenerWrapper;
import org.apache.jmeter.samplers.Remoteable;
import org.apache.jmeter.samplers.SampleListener;
import org.apache.jmeter.testelement.TestListener;
import org.apache.jmeter.threads.ThreadGroup;
import org.apache.jorphan.collections.HashTree;
import org.apache.jorphan.collections.HashTreeTraverser;
import org.apache.jorphan.logging.LoggingManager;
import org.apache.log.Logger;

/**
 * @author mstover
 */
public class ConvertListeners implements HashTreeTraverser
{
    transient private static Logger log = LoggingManager.getLoggerForClass();

    /* (non-Javadoc)
     * @see HashTreeTraverser#addNode(Object, HashTree)
     */
    public void addNode(Object node, HashTree subTree)
    {
        if (node instanceof ThreadGroup)
        {
            log.info("num threads = " + ((ThreadGroup) node).getNumThreads());
        }
        Iterator iter = subTree.list().iterator();
        while (iter.hasNext())
        {
            Object item = iter.next();
            if (item instanceof ThreadGroup)
            {
                log.info(
                    "num threads = " + ((ThreadGroup) item).getNumThreads());
            }
            if (item instanceof Remoteable
                && (item instanceof TestListener
                    || item instanceof SampleListener))
            {
                try
                {
                    RemoteSampleListener rtl =
                        new RemoteSampleListenerImpl(item);
                    if (item instanceof TestListener
                        && item instanceof SampleListener)
                    {
                        RemoteListenerWrapper wrap =
                            new RemoteListenerWrapper(rtl);
                        subTree.replace(item, wrap);
                    }
                    else if (item instanceof TestListener)
                    {
                        RemoteTestListenerWrapper wrap =
                            new RemoteTestListenerWrapper(rtl);
                        subTree.replace(item, wrap);
                    }
                    else
                    {
                        RemoteSampleListenerWrapper wrap =
                            new RemoteSampleListenerWrapper(rtl);
                        subTree.replace(item, wrap);
                    }
                }
                catch (RemoteException e)
                {
                    log.error("", e);
                }
            }
        }
    }

    /* (non-Javadoc)
     * @see HashTreeTraverser#subtractNode()
     */
    public void subtractNode()
    {
    }

    /* (non-Javadoc)
     * @see HashTreeTraverser#processPath()
     */
    public void processPath()
    {
    }

}