package com.dianping.cat.message.spi.codec;
import org.jboss.netty.buffer.ChannelBuffer;
import org.jboss.netty.buffer.DynamicChannelBuffer;
import org.junit.Test;
import com.dianping.cat.message.CatTestCase;
import com.dianping.cat.message.Message;
import com.dianping.cat.message.internal.MockMessageBuilder;
import com.dianping.cat.message.spi.MessageCodec;
import com.dianping.cat.message.spi.MessageTree;
import com.dianping.cat.message.spi.internal.DefaultMessageTree;
public class MessageCodecPerformanceTest extends CatTestCase {
public static final String ID = "plain-text";
@Test
public void testCodePerformance() throws Exception {
MessageCodec codec = lookup(MessageCodec.class, ID);
MessageTree tree = buildMessage();
ChannelBuffer buf = new DynamicChannelBuffer(10240);
codec.encode(tree, buf);
int count = 5000000;
for (int i = 0; i < count; i++) {
buf.markReaderIndex();
// read the size of the message
buf.readInt();
DefaultMessageTree result = (DefaultMessageTree) codec.decode(buf);
buf.resetReaderIndex();
result.setBuffer(buf);
}
}
public MessageTree buildMessage() {
Message message = new MockMessageBuilder() {
@Override
public MessageHolder define() {
TransactionHolder t = t("WEB CLUSTER", "GET", 112819) //
.at(1348374838231L) //
.after(1300).child(t("QUICKIE SERVICE", "gimme_stuff", 1571)) //
.after(100).child(e("SERVICE", "event1")) //
.after(100).child(h("SERVICE", "heartbeat1")) //
.after(100).child(t("WEB SERVER", "GET", 109358) //
.after(1000).child(t("SOME SERVICE", "get", 4345) //
.after(4000).child(t("MEMCACHED", "Get", 279))) //
.mark().after(200).child(t("MEMCACHED", "Inc", 319)) //
.reset().after(500).child(t("BIG ASS SERVICE", "getThemDatar", 97155) //
.after(1000).mark().child(t("SERVICE", "getStuff", 3760)) //
.reset().child(t("DATAR", "findThings", 94537)) //
.after(200).child(t("THINGIE", "getMoar", 1435)) //
) //
.after(100).mark().child(t("OTHER DATA SERVICE", "get", 4394) //
.after(1000).mark().child(t("MEMCACHED", "Get", 378)) //
.reset().child(t("MEMCACHED", "Get", 3496)) //
) //
.reset().child(t("FINAL DATA SERVICE", "get", 4394) //
.after(1000).mark().child(t("MEMCACHED", "Get", 386)) //
.reset().child(t("MEMCACHED", "Get", 322)) //
.reset().child(t("MEMCACHED", "Get", 322)) //
) //
) //
;
return t;
}
}.build();
MessageTree tree = new DefaultMessageTree();
tree.setDomain("cat");
tree.setHostName("test");
tree.setIpAddress("test");
tree.setThreadGroupName("test");
tree.setThreadId("test");
tree.setThreadName("test");
tree.setMessage(message);
return tree;
}
}