memcachedテスト
5481 ワード
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;
}
}
}
Note:
: http://code.google.com/p/spcached/wiki/benchmarktool