PerformanceTestWrapper.java

/*
 * JBoss, Home of Professional Open Source
 * Copyright 2008, Red Hat Middleware LLC, and individual contributors
 * as indicated by the @author tags.
 * See the copyright.txt in the distribution for a
 * full listing of individual contributors.
 * This copyrighted material is made available to anyone wishing to use,
 * modify, copy, or redistribute it subject to the terms and conditions
 * of the GNU Lesser General Public License, v. 2.1.
 * This program is distributed in the hope that it will be useful, but WITHOUT A
 * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A
 * PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more details.
 * You should have received a copy of the GNU Lesser General Public License,
 * v.2.1 along with this distribution; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
 * MA  02110-1301, USA.
 * 
 * (C) 2008,
 * @author JBoss Inc.
 */
package org.jboss.jbossts.qa.performance;

import com.arjuna.orbportability.ORB;
import com.arjuna.orbportability.OA;

import java.util.ArrayList;

import org.jboss.jbossts.qa.Utils.ServerIORStore;

public class PerformanceTestWrapper
{
    private final static String ADD_SERVICE_NAME = "-service";
    private final static String DISPLAY_GRAPH_PARAMETER = "-displaygraph";
    private final static String DISABLE_ORB_PARAMETER = "-disableorb";
    private final static String CSV_FILENAME_PARAMETER = "-csv";
    private final static String CLASSNAME_PARAMETER = "-classname";
    private final static String NUMBER_OF_THREADS_PARAMETER = "-threads";
    private final static String NUMBER_OF_THREADS_MIN_PARAMETER = "-minthreads";
    private final static String NUMBER_OF_THREADS_MAX_PARAMETER = "-maxthreads";
    private final static String NUMBER_OF_ITERATIONS_PARAMETER = "-iterations";
    private final static String XDATA_PARAMETER = "-xdata";
    private final static String YDATA_PARAMETER = "-ydata";

    private final static int DEFAULT_NUMBER_OF_THREADS = 5;
    private final static int DEFAULT_NUMBER_OF_ITERATIONS = 100;

    /**
     * The main test method which must assert either a pass or a fail.
     */
    public static void main(String[] args)
    {
        String classname = null;
        int minNumberOfThreads = DEFAULT_NUMBER_OF_THREADS;
        int maxNumberOfThreads = DEFAULT_NUMBER_OF_THREADS;
        int numberOfIterations = DEFAULT_NUMBER_OF_ITERATIONS;
        boolean displayGraph = false;
        int xData = -1;
        int yData = -1;
        String csvFilename = null;
        ArrayList configList = new ArrayList();
        boolean disableOrb = false;

        try
        {
            if ( args.length > 0 )
            {
                for (int count=0;count<args.length;count++)
                {
                    if ( args[count].equals( CLASSNAME_PARAMETER ) )
                    {
                        classname = args[count + 1];
                    }

                    if ( args[count].equals( CSV_FILENAME_PARAMETER ) )
                    {
                        csvFilename = args[count + 1];
                    }

                    if ( args[count].equals( DISPLAY_GRAPH_PARAMETER ) )
                    {
                        displayGraph = true;
                    }

                    if ( args[count].equals( NUMBER_OF_THREADS_PARAMETER ) )
                    {
                        maxNumberOfThreads = minNumberOfThreads = Integer.parseInt( args[count + 1] );
                    }

                    if ( args[count].equals( NUMBER_OF_THREADS_MIN_PARAMETER ) )
                    {
                        minNumberOfThreads = Integer.parseInt( args[count + 1] );
                    }

                    if ( args[count].equals( NUMBER_OF_THREADS_MAX_PARAMETER ) )
                    {
                        maxNumberOfThreads = Integer.parseInt( args[count + 1] );
                    }

                    if ( args[count].equals( NUMBER_OF_ITERATIONS_PARAMETER ) )
                    {
                        numberOfIterations = Integer.parseInt( args[count + 1] );
                    }

                    if ( args[count].equals( DISABLE_ORB_PARAMETER ) )
                    {
                        disableOrb = true;
                    }

                    if ( args[count].equals( XDATA_PARAMETER ) )
                    {
                        xData = PerformanceFramework.parseDataType( args[count + 1] );

                        if ( xData == -1 )
                        {
                            System.err.println("Error - specified x-data parameter is invalid");
                            assertFailure();
                        }
                        else
                        {
                            System.out.println("X-Data set to '"+args[count + 1]+"'");
                        }
                    }

                    if ( args[count].equals( YDATA_PARAMETER ) )
                    {
                        yData = PerformanceFramework.parseDataType( args[count + 1] );

                        if ( yData == -1 )
                        {
                            System.err.println("Error - specified y-data parameter is invalid");
                            assertFailure();
                        }
                        else
                        {
                            System.out.println("Y-Data set to '"+args[count + 1]+"'");
                        }
                    }

                    if ( args[count].equals( ADD_SERVICE_NAME ) )
                    {
                        System.out.println("Added Service config: '"+ args[count + 1] +"'");


                        configList.add( ServerIORStore.loadIOR( args[count + 1]) );
                    }
                }

                if ( classname != null )
                {
                    System.out.println("            Classname: "+classname);
                    System.out.println("Min Number of Threads: "+minNumberOfThreads);
                    System.out.println("Max Number of Threads: "+maxNumberOfThreads);
                    System.out.println(" Number of Iterations: "+numberOfIterations);

                    if ( !disableOrb )
                    {
                        try
                        {
                            /**
                             * Retrieve ORB and OA references, intialise them
                             * and then set the OTSManager ORB and OA properties
                             */
                            ORB orb = ORB.getInstance( PerformanceFramework.ORB_INSTANCE_NAME );
                            OA oa = OA.getRootOA( orb );

                            orb.initORB(args, null);
                            oa.initOA(args);
                        }
                        catch (Exception e)
                        {
                            e.printStackTrace(System.err);
                            assertFailure();
                        }
                    }

                    PerformanceFramework pf = new PerformanceFramework( args );

                    pf.setDisplayGraph(displayGraph);
                    pf.setCSVFilename(csvFilename);

                    if ( xData != -1 )
                    {
                        pf.setXData( xData );
                    }

                    if ( yData != -1 )
                    {
                        pf.setYData( yData );
                    }

                    pf.setNumberOfIterations(numberOfIterations);
                    pf.setMinimumNumberOfThreads(minNumberOfThreads);
                    pf.setMaximumNumberOfThreads(maxNumberOfThreads);

                    String[] configs = new String[ configList.size() ];
                    configList.toArray(configs);

                    if ( !pf.performTest( classname, configs ) )
                    {
                        //assertSuccess();
                    }
                    else
                    {
                        assertFailure();
                    }
                }
                else
                {
                    System.out.println("Parameter '"+CLASSNAME_PARAMETER+"' not specified");
                    assertFailure();
                }
            }
            else
            {
                System.out.println("No parameters passed");
                assertFailure();
            }
        }
        catch (Exception e)
        {
            e.printStackTrace(System.err);
            assertFailure();
        }
    }

    private static void assertFailure() {
        System.out.println("Failed");
    }
}