/**
* Copyright (C) 2009-2012 Barchart, Inc. <http://www.barchart.com/>
*
* All rights reserved. Licensed under the OSI BSD License.
*
* http://www.opensource.org/licenses/bsd-license.php
*/
package bench.poll_wait;
import java.nio.IntBuffer;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicBoolean;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import util.ConsoleReporterUDT;
import com.barchart.udt.EpollUDT;
import com.barchart.udt.ExceptionUDT;
import com.barchart.udt.SocketUDT;
import com.barchart.udt.TypeUDT;
import com.barchart.udt.util.HelpUDT;
import com.yammer.metrics.Metrics;
import com.yammer.metrics.core.Timer;
import com.yammer.metrics.core.TimerContext;
/**
* poll in two threads
*/
public class BenchPollTwo extends SocketUDT {
public BenchPollTwo(final TypeUDT type) throws ExceptionUDT {
super(type);
}
static final Logger log = LoggerFactory.getLogger(BenchPollTwo.class);
static final int time = 60 * 1000;
static {
Metrics.newCounter( //
BenchPollTwo.class, "benchmark duration").inc(time);
}
static final Timer pollTimeOne = Metrics.newTimer(BenchPollTwo.class,
"poll time one", TimeUnit.MICROSECONDS, TimeUnit.SECONDS);
static final Timer pollTimeTwo = Metrics.newTimer(BenchPollTwo.class,
"poll time two", TimeUnit.MICROSECONDS, TimeUnit.SECONDS);
public static void main(final String[] args) throws Exception {
log.info("init");
final SocketUDT socket = new SocketUDT(TypeUDT.DATAGRAM);
final int socketID = socket.id();
final int epollID = BenchPollTwo.epollCreate0();
BenchPollTwo.epollAdd0(epollID, socketID, EpollUDT.Opt.BOTH.code);
final AtomicBoolean isOn = new AtomicBoolean(true);
final Runnable epollTaskOne = new Runnable() {
final IntBuffer readBuffer = HelpUDT.newDirectIntBufer(1024);
final IntBuffer writeBuffer = HelpUDT.newDirectIntBufer(1024);
final IntBuffer sizeBuffer = HelpUDT.newDirectIntBufer(1024);
@Override
public void run() {
try {
while (isOn.get()) {
runCore();
}
} catch (final Exception e) {
log.error("", e);
}
}
void runCore() throws Exception {
final TimerContext timer = pollTimeOne.time();
SocketUDT.epollWait0( //
epollID, readBuffer, writeBuffer, sizeBuffer, 0);
timer.stop();
}
};
final Runnable epollTaskTwo = new Runnable() {
final IntBuffer readBuffer = HelpUDT.newDirectIntBufer(1024);
final IntBuffer writeBuffer = HelpUDT.newDirectIntBufer(1024);
final IntBuffer sizeBuffer = HelpUDT.newDirectIntBufer(1024);
@Override
public void run() {
try {
while (isOn.get()) {
runCore();
}
} catch (final Exception e) {
log.error("", e);
}
}
void runCore() throws Exception {
final TimerContext timer = pollTimeTwo.time();
SocketUDT.epollWait0( //
epollID, readBuffer, writeBuffer, sizeBuffer, 0);
timer.stop();
}
};
final ExecutorService executor = Executors.newFixedThreadPool(2);
executor.submit(epollTaskOne);
executor.submit(epollTaskTwo);
ConsoleReporterUDT.enable(3, TimeUnit.SECONDS);
Thread.sleep(time);
isOn.set(false);
Thread.sleep(1 * 1000);
executor.shutdownNow();
Metrics.defaultRegistry().shutdown();
BenchPollTwo.epollRemove0(epollID, socketID);
BenchPollTwo.epollRelease0(epollID);
log.info("done");
}
}