Metacat

package edu.ucsb.nceas.metacattest;

import java.io.FileReader;

import java.io.IOException;

import java.io.Reader;

import java.util.ArrayList;

import java.util.List;

import java.util.concurrent.ExecutorService;

import java.util.concurrent.Executors;

import java.util.concurrent.TimeUnit;

import edu.ucsb.nceas.metacat.client.Metacat;

import edu.ucsb.nceas.metacat.client.MetacatFactory;

import edu.ucsb.nceas.metacat.client.MetacatInaccessibleException;

import edu.ucsb.nceas.utilities.IOUtil;

import junit.framework.Test;

import junit.framework.TestCase;

import junit.framework.TestSuite;

public class MetaCatQueryPerformanceTest extends TestCase {

	 /**

     * Create a suite of tests to be run together

     */

    public static Test suite()

    {

       TestSuite suite = new TestSuite();

       return suite;

    }

    public static void main(String[] arg)

    {

    	if (arg != null & arg.length == 2)

    	{

    		String metacatURL = arg[0];

    	    String queryFilePath = arg[1];

    	    try {

				//query(metacatURL, queryFilePath);

				multiThreadedQuery(metacatURL, queryFilePath);

			} catch (Exception e) {

				// TODO Auto-generated catch block

				e.printStackTrace();

			}

    	}

    	else

    	{

    		System.out.println("Usage: java MetaCatQueryPerformance metacatURL queryFilePath");

    	}

    }

    public static String query(String metacatURL, String queryFile) throws MetacatInaccessibleException, IOException

    {

        Metacat m = MetacatFactory.createMetacatConnection(metacatURL);;

        FileReader fr = new FileReader(queryFile);

        Reader r = m.query(fr);

        //System.out.println("Starting query...");

        String result = IOUtil.getAsString(r, true);

        System.out.println("Query result:\n" + result);

        return result;

    }

    public static void multiThreadedQuery(String metacatURL, String queryFile) throws InterruptedException {

    	int nThreads = 25;

		ExecutorService executor = Executors.newFixedThreadPool(nThreads);

		List<QueryRunner> tasks = new ArrayList<QueryRunner>();

		// launch threads

		for (int i = 0; i < nThreads; i++) {

			QueryRunner query = new QueryRunner(metacatURL, queryFile, i);

			tasks.add(query);

			executor.execute(query);

		}

		// now wait

		executor.shutdown();

		executor.awaitTermination(1, TimeUnit.HOURS);

		// now show the runtimes

		for (QueryRunner query: tasks) {

			System.out.println("query: " + query.id + " took: " + query.runtime);

		}

    }

}

class QueryRunner implements Runnable{

	String metacatURL;

	String queryFile;

	int id;

	long runtime;

	public QueryRunner(String metacatURL, String queryFile, int id) {

		this.metacatURL = metacatURL;

		this.queryFile = queryFile;

		this.id = id;

	}

	@Override

	public void run() {

		try {

			long startTime = System.currentTimeMillis();

			MetaCatQueryPerformanceTest.query(metacatURL, queryFile);

			long endTime = System.currentTimeMillis();

			this.runtime = endTime - startTime;

		} catch (Exception e) {

			e.printStackTrace();

		}

	}

}