A java class for benchmark memcached, which uses threads to execute set and get operations.

Introduction

This class bases on java_memcached 1.3.2.

Details

package com.danga.MemCached.test;

import com.danga.MemCached.*;
import java.util.*;

public class MemCachedThreadBench {

	private static class WorkerStat {
		public int start, runs;
		public long setterTime, getterTime;

		public WorkerStat() {
			start = runs = 0;
			setterTime = getterTime = 0;
		}
	}

	public static void main(String[] args) throws Exception {

		if( args.length != 4 ) {
			System.out.println( "Usage: java " + MemCachedThreadBench.class.getName() 
					+ " <runs> <start> <port> <threads>" );
			System.exit(1);   
		}

		int runs = Integer.parseInt(args[0]);
		int start = Integer.parseInt(args[1]);
		String[] serverlist = { "127.0.0.1:" + args[2] };
		int threads = Integer.parseInt( args[3] );

		SockIOPool pool = SockIOPool.getInstance();
		pool.setServers(serverlist);

		pool.setInitConn( threads );
		pool.setMinConn( threads );
		pool.setMaxConn(500);
		pool.setMaintSleep(30);

		pool.setNagle(false);
		pool.initialize();

		WorkerStat [] statArray = new WorkerStat[ threads ];
		Thread [] threadArray = new Thread[ threads ];

		WorkerStat mainStat = new WorkerStat();
		mainStat.runs = runs * threads;

		long begin = System.currentTimeMillis();

		for (int i = 0; i < threads; i++) {
			statArray[i] = new WorkerStat();
			statArray[i].start = start + i * runs;
			statArray[i].runs = runs;
			threadArray[i] = new SetterThread( statArray[i] );
			threadArray[i].start();
		}

		for( int i = 0; i < threads; i++ ) {
			threadArray[i].join();
		}

		mainStat.setterTime = System.currentTimeMillis() - begin;

		begin = System.currentTimeMillis();

		for (int i = 0; i < threads; i++) {
			threadArray[i] = new GetterThread( statArray[i] );
			threadArray[i].start();
		}

		for( int i = 0; i < threads; i++ ) {
			threadArray[i].join();
		}

		mainStat.getterTime = System.currentTimeMillis() - begin;

		SockIOPool.getInstance().shutDown();

		WorkerStat totalStat = new WorkerStat();

		System.out.println( "Thread\tstart\truns\tset time(ms)\tget time(ms)" );
		for( int i = 0; i < threads; i++ ) {
			System.out.println( "" + i + "\t" + statArray[i].start + "\t" +
				statArray[i].runs + "\t" + statArray[i].setterTime + "\t\t"
				+ statArray[i].getterTime );

			totalStat.runs = totalStat.runs + statArray[i].runs;
			totalStat.setterTime = totalStat.setterTime + statArray[i].setterTime;
			totalStat.getterTime = totalStat.getterTime + statArray[i].getterTime;
		}

		System.out.println( "\nAvg\t\t" + runs + "\t" + totalStat.setterTime / threads
			+ "\t\t" + totalStat.getterTime / threads );

		System.out.println( "\nTotal\t\t" + totalStat.runs + "\t"
			+ totalStat.setterTime + "\t\t" + totalStat.getterTime );
		System.out.println( "\tReqPerSecond\tset - " + 1000 * totalStat.runs / totalStat.setterTime
			+ "\tget - " + 1000 * totalStat.runs / totalStat.getterTime );

		System.out.println( "\nMain\t\t" + mainStat.runs + "\t"
			+ mainStat.setterTime + "\t\t" + mainStat.getterTime );
		System.out.println( "\tReqPerSecond\tset - " + 1000 * mainStat.runs / mainStat.setterTime
			+ "\tget - " + 1000 * mainStat.runs / mainStat.getterTime );
	}

	private static class SetterThread extends Thread {   
		private WorkerStat stat;

		SetterThread( WorkerStat stat ) {
			this.stat = stat;
		}

		public void run() {   
			MemCachedClient mc = new MemCachedClient();
			mc.setCompressEnable(false);

			String keyBase = "testKey";
			String object = "This is a test of an object blah blah es, "
				+ "serialization does not seem to slow things down so much.  "
				+ "The gzip compression is horrible horrible performance, "
				+ "so we only use it for very large objects.  "
				+ "I have not done any heavy benchmarking recently";

			long begin = System.currentTimeMillis();
			for (int i = stat.start; i < stat.start + stat.runs; i++) {
				mc.set( "" + i + keyBase, object);
			}
			long end = System.currentTimeMillis();

			stat.setterTime = end - begin;
		}
	}

	private static class GetterThread extends Thread {   
		private WorkerStat stat;

		GetterThread( WorkerStat stat ) {
			this.stat = stat;
		}

		public void run() {   
			MemCachedClient mc = new MemCachedClient();
			mc.setCompressEnable(false);

			String keyBase = "testKey";

			long begin = System.currentTimeMillis();
			for (int i = stat.start; i < stat.start + stat.runs; i++) {
				String str = (String) mc.get( "" + i + keyBase );
			}
			long end = System.currentTimeMillis();

			stat.getterTime = end - begin;
		}
	}
}