net.sf.jabb.util.test.RateTestUtility Maven / Gradle / Ivy

Go to download
/**
 * 
 */
package net.sf.jabb.util.test;

import java.text.DecimalFormat;
import java.util.LinkedList;
import java.util.List;
import java.util.concurrent.ConcurrentLinkedQueue;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.Future;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.function.LongConsumer;
import java.util.function.LongUnaryOperator;

/**
 * @author James Hu
 *
 */
public class RateTestUtility {
	
	public static LongConsumer emptyLoop = endTime -> {
		while(System.currentTimeMillis() < endTime){}
	};

	public static void doRateTest(ExecutorService threadPool, int numThreads, 
			int warmUpPeriod, TimeUnit warmUpPeriodUnit, LongConsumer warmUpConsumer,
			int testPeriod, TimeUnit testPeriodUnit, LongUnaryOperator testFunction
			) throws Exception{
		doRateTest(null, threadPool, numThreads, warmUpPeriod, warmUpPeriodUnit, warmUpConsumer,
				testPeriod, testPeriodUnit, testFunction);
	}
	public static void doRateTest(int numThreads, 
			int warmUpPeriod, TimeUnit warmUpPeriodUnit, LongConsumer warmUpConsumer,
			int testPeriod, TimeUnit testPeriodUnit, LongUnaryOperator testFunction
			) throws Exception{
		ThreadPoolExecutor threadPool = new ThreadPoolExecutor(numThreads, numThreads, 1L, TimeUnit.MINUTES, new LinkedBlockingQueue());
		threadPool.allowCoreThreadTimeOut(true);
		doRateTest(null, threadPool, numThreads, warmUpPeriod, warmUpPeriodUnit, warmUpConsumer,
				testPeriod, testPeriodUnit, testFunction);
		threadPool.shutdown();
		threadPool.awaitTermination(1, TimeUnit.MINUTES);
		threadPool.shutdownNow();
	}
	
	public static double doRateTest(String title, int numThreads, 
			int warmUpPeriod, TimeUnit warmUpPeriodUnit, LongConsumer warmUpConsumer,
			int testPeriod, TimeUnit testPeriodUnit, LongUnaryOperator testFunction
			) throws Exception{
		ThreadPoolExecutor threadPool = new ThreadPoolExecutor(numThreads, numThreads, 1L, TimeUnit.MINUTES, new LinkedBlockingQueue());
		threadPool.allowCoreThreadTimeOut(true);
		double result = doRateTest(title, threadPool, numThreads, warmUpPeriod, warmUpPeriodUnit, warmUpConsumer,
				testPeriod, testPeriodUnit, testFunction);
		threadPool.shutdown();
		threadPool.awaitTermination(1, TimeUnit.MINUTES);
		threadPool.shutdownNow();
		return result;
	}
		
	public static double doRateTest(String title, ExecutorService threadPool, int numThreads, 
			int warmUpPeriod, TimeUnit warmUpPeriodUnit, LongConsumer warmUpConsumer,
			int testPeriod, TimeUnit testPeriodUnit, LongUnaryOperator testFunction
			) throws Exception{
		LongConsumer warmUpFunction = warmUpConsumer != null? 
				warmUpConsumer : endTime -> testFunction.applyAsLong(endTime);
		
		long start = System.currentTimeMillis() + TimeUnit.MILLISECONDS.convert(warmUpPeriod, warmUpPeriodUnit);
		long end = start + TimeUnit.MILLISECONDS.convert(testPeriod, testPeriodUnit);
		
		List> futures = new LinkedList<>();
		for (int i = 0; i < numThreads; i ++){
			futures.add(threadPool.submit(()->{
				long total = 0;
				for (long now = System.currentTimeMillis(); now < end; now = System.currentTimeMillis()){
					if (now < start){ // warm up period
						warmUpFunction.accept(start);
					}else{
						total += testFunction.applyAsLong(end);
					}
				}
				return total;
			}));
		}
		
		final AtomicBoolean errorHappened = new AtomicBoolean(false);
		long total = futures.stream().mapToLong(f -> {
			try{
				return f.get();
			}catch(Exception e){
				e.printStackTrace();
				errorHappened.set(true);
				return 0;
			}
		}).sum();
		
		double rate = (double)total/TimeUnit.SECONDS.convert(testPeriod, testPeriodUnit);
		if (title != null){
			System.out.println("Rate of " + title + " is " + formatDouble("###,###.####", rate) + " per second" );
		}
		if (errorHappened.get()){
			System.out.println("Error happened during the test.");
		}
		return rate;
	}
	
	static public String formatDouble(String pattern, double value) {
		DecimalFormat myFormatter = new DecimalFormat(pattern);
		String output = myFormatter.format(value);
		return output;
	}

}