// $Header: /home/cvs/jakarta-jmeter/src/core/org/apache/jmeter/engine/ClientJMeterEngine.java,v 1.14 2004/02/12 23:59:01 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.net.MalformedURLException;
import java.rmi.Naming;
import java.rmi.NotBoundException;
import java.rmi.RemoteException;

import org.apache.jmeter.testelement.TestListener;
import org.apache.jorphan.collections.HashTree;
import org.apache.jorphan.collections.SearchByClass;
import org.apache.jorphan.logging.LoggingManager;
import org.apache.log.Logger;


/**
 * @author     unattributed
 * @version    $Revision: 1.14 $ Updated on: $Date: 2004/02/12 23:59:01 $
 */
public class ClientJMeterEngine implements JMeterEngine,Runnable
{
    transient private static Logger log = LoggingManager.getLoggerForClass();
    RemoteJMeterEngine remote;
    HashTree test;
    SearchByClass testListeners;
    ConvertListeners sampleListeners;
    private String host;

    public ClientJMeterEngine(String host)
        throws MalformedURLException, NotBoundException, RemoteException
    {
        this((RemoteJMeterEngine) Naming.lookup("//" + host + "/JMeterEngine"));
        this.host = host;
    }

    public ClientJMeterEngine(RemoteJMeterEngine remote)
    {
        this.remote = remote;
    }

    protected HashTree getTestTree()
    {
        return test;
    }

    public void configure(HashTree testTree)
    {
        test = testTree;
    }

    public void setHost(String host)
    {
        this.host = host;
    }

    public void runTest()
    {
        log.info("about to run remote test");
        new Thread(this).start();
        log.info("done initiating run command");
    }

    public void stopTest()
    {
        try
        {
            remote.stopTest();
        }
        catch (Exception ex)
        {
            log.error("", ex);
        }
    }

    public void reset()
    {
        try
        {
            remote.reset();
        }
        catch (Exception ex)
        {
            log.error("", ex);
        }
    }

    /* (non-Javadoc)
     * @see java.lang.Runnable#run()
     */
    public void run()
    {
        log.info("running clientengine run method");
        testListeners = new SearchByClass(TestListener.class);
        getTestTree().traverse(testListeners);
        sampleListeners = new ConvertListeners();
        
        //TODO this is a temporary fix - see bug 23487 
        try {
            getTestTree().traverse(sampleListeners);
        }
        catch(IndexOutOfBoundsException e)
        {
        	log.warn("Error replacing sample listeners",e);
        }
        
        try
        {
            remote.setHost(host);
            log.info("sent host ="+host);
            remote.configure(test);
            log.info("sent test");
            remote.runTest();
            log.info("sent run command");
        }
        catch(Exception ex)
        {
            log.error("",ex);
        }
    }

    /* (non-Javadoc)
     * @see org.apache.jmeter.engine.JMeterEngine#exit()
     */
    public void exit()
    {
    	try
        {
            remote.exit();
        }
        catch (RemoteException e)
        {
        	log.warn("Could not perform remote exit: "+e.toString());
        }
    }
}