用java并发测试tokyo cabinet的性能[五四陈手记]

wget http://tokyocabinet.sourceforge.net/tokyocabinet-1.4.28.tar.gz
tar zxvf tokyocabinet-1.4.28.tar.gz
cd tokyocabinet-1.4.28/
./configure
make
make install
cd ../

tar zxvf tokyocabinet-java-1.22.tar.gz
cd tokyocabinet-java-1.22
./configure
make
make install

package test;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.atomic.AtomicLong;

import tokyocabinet.*;

public class BenchMark {

public static class Stat {
private AtomicLong _count = new AtomicLong(0);
private static Stat _instance = new Stat();

public static Stat getInstance() {
return _instance;
}

private Stat() {
_printer = new RatePrinter(_count);
_printer.setDaemon(true);
_printer.start();
}

public void inc() {
_count.incrementAndGet();
}

private RatePrinter _printer;

private static class RatePrinter extends Thread {
private long _last;
private AtomicLong _c;

public RatePrinter(AtomicLong c) {
_c = c;
}

@Override
public void run() {
while (true) {
try {
long current = _c.get();
System.out.println("Rate: " + (current - _last)
+ " req/s");
_last = current;
Thread.sleep(1000L);
} catch (Throwable e) {
e.printStackTrace();
}
}
}
}
}

public static class EchoThread extends Thread {
private TDB tdb;
public EchoThread(String ip,String port, int in, ThreadGroup group){
super(group, "EchoThread-" + in);
// create the object
tdb = new TDB();

// open the database
if(!tdb.open("casket.tct", TDB.OWRITER | TDB.OCREAT)){
int ecode = tdb.ecode();
System.err.println("open error: " + tdb.errmsg(ecode));
}
}

@Override
public void run() {
int index = 0;
while (true) {
try {

String pkey = new Long(tdb.genuid()).toString();
Map cols = new HashMap();
cols.put("name", "mikio"+index);
cols.put("age", "30");
cols.put("lang", "ja,en,c");
if(!tdb.put(pkey, cols)){
int ecode = tdb.ecode();
//System.err.println("put error: " + tdb.errmsg(ecode));
}

//client.insert("Table1", "name"+index, "Standard1:name", ("name"+index).getBytes("UTF-8"), System.currentTimeMillis(), true);
//client.get_column("Table1", "name0", "Standard1:name");
index ++;
Stat.getInstance().inc();
} catch (Throwable e) {
e.printStackTrace();
break;
} finally {
// close the database
if(!tdb.close()){
int ecode = tdb.ecode();
//System.err.println("close error: " + tdb.errmsg(ecode));
}

}
}
}
}

/**
* @param args
* @throws TTransportException
*/
public static void main(String[] args){
if (args.length != 1) {
System.out
.println("Usage: Benchmark ");
System.exit(1);
}

String ip = args[0];
String port = args[0];

Integer concurrent = Integer.valueOf(args[0]);
System.out.println("ip = "+ip+",port = "+port+",concurrent = " + concurrent);

ThreadGroup group = new ThreadGroup("Benchmark");
List
threads = new ArrayList
();
for (int x = 0; x < concurrent; ++x) {
Thread t =new EchoThread(ip,port, x, group);
threads.add(t);
t.start();
}
}

}