b3d3269b8c
Motivation: Overspecifying interfaces can increase coupling and make refactoring harder down the line. Modification: Places that were specifying fields, variables, and return types as ScheduledFuture, but did not use any features specific to ScheduledFuture, have been changed to specify Future instead. Result: Cleaner code.
157 lines
5.5 KiB
Java
157 lines
5.5 KiB
Java
/*
|
|
* Copyright 2018 The Netty Project
|
|
*
|
|
* The Netty Project licenses this file to you under the Apache License,
|
|
* version 2.0 (the "License"); you may not use this file except in compliance
|
|
* with the License. You may obtain a copy of the License at:
|
|
*
|
|
* https://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
* License for the specific language governing permissions and limitations
|
|
* under the License.
|
|
*/
|
|
package io.netty.microbench.channel.epoll;
|
|
|
|
import io.netty.bootstrap.Bootstrap;
|
|
import io.netty.bootstrap.ServerBootstrap;
|
|
import io.netty.buffer.ByteBuf;
|
|
import io.netty.channel.Channel;
|
|
import io.netty.channel.ChannelDuplexHandler;
|
|
import io.netty.channel.ChannelHandlerContext;
|
|
import io.netty.channel.ChannelInitializer;
|
|
import io.netty.channel.ChannelPromise;
|
|
import io.netty.channel.epoll.EpollEventLoopGroup;
|
|
import io.netty.channel.epoll.EpollServerSocketChannel;
|
|
import io.netty.channel.epoll.EpollSocketChannel;
|
|
import io.netty.microbench.util.AbstractMicrobenchmark;
|
|
import io.netty.util.concurrent.Future;
|
|
import org.openjdk.jmh.annotations.Benchmark;
|
|
import org.openjdk.jmh.annotations.GroupThreads;
|
|
import org.openjdk.jmh.annotations.Setup;
|
|
import org.openjdk.jmh.annotations.TearDown;
|
|
|
|
import java.util.concurrent.TimeUnit;
|
|
|
|
public class EpollSocketChannelBenchmark extends AbstractMicrobenchmark {
|
|
private static final Runnable runnable = new Runnable() {
|
|
@Override
|
|
public void run() { }
|
|
};
|
|
|
|
private EpollEventLoopGroup group;
|
|
private Channel serverChan;
|
|
private Channel chan;
|
|
private ByteBuf abyte;
|
|
private Future<?> future;
|
|
|
|
@Setup
|
|
public void setup() throws Exception {
|
|
group = new EpollEventLoopGroup(1);
|
|
|
|
// add an arbitrary timeout to make the timer reschedule
|
|
future = group.schedule(new Runnable() {
|
|
@Override
|
|
public void run() {
|
|
throw new AssertionError();
|
|
}
|
|
}, 5, TimeUnit.MINUTES);
|
|
serverChan = new ServerBootstrap()
|
|
.channel(EpollServerSocketChannel.class)
|
|
.group(group)
|
|
.childHandler(new ChannelInitializer<Channel>() {
|
|
@Override
|
|
protected void initChannel(Channel ch) {
|
|
ch.pipeline().addLast(new ChannelDuplexHandler() {
|
|
@Override
|
|
public void channelRead(ChannelHandlerContext ctx, Object msg) {
|
|
if (msg instanceof ByteBuf) {
|
|
ctx.writeAndFlush(msg, ctx.voidPromise());
|
|
} else {
|
|
throw new AssertionError();
|
|
}
|
|
}
|
|
});
|
|
}
|
|
})
|
|
.bind(0)
|
|
.sync()
|
|
.channel();
|
|
chan = new Bootstrap()
|
|
.channel(EpollSocketChannel.class)
|
|
.handler(new ChannelInitializer<Channel>() {
|
|
@Override
|
|
protected void initChannel(Channel ch) {
|
|
ch.pipeline().addLast(new ChannelDuplexHandler() {
|
|
|
|
private ChannelPromise lastWritePromise;
|
|
|
|
@Override
|
|
public void channelRead(ChannelHandlerContext ctx, Object msg) {
|
|
if (msg instanceof ByteBuf) {
|
|
|
|
ByteBuf buf = (ByteBuf) msg;
|
|
try {
|
|
if (buf.readableBytes() == 1) {
|
|
lastWritePromise.trySuccess();
|
|
lastWritePromise = null;
|
|
} else {
|
|
throw new AssertionError();
|
|
}
|
|
} finally {
|
|
buf.release();
|
|
}
|
|
} else {
|
|
throw new AssertionError();
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void write(ChannelHandlerContext ctx, Object msg, ChannelPromise promise)
|
|
throws Exception {
|
|
if (lastWritePromise != null) {
|
|
throw new IllegalStateException();
|
|
}
|
|
lastWritePromise = promise;
|
|
super.write(ctx, msg, ctx.voidPromise());
|
|
}
|
|
});
|
|
}
|
|
})
|
|
.group(group)
|
|
.connect(serverChan.localAddress())
|
|
.sync()
|
|
.channel();
|
|
|
|
abyte = chan.alloc().directBuffer(1);
|
|
abyte.writeByte('a');
|
|
}
|
|
|
|
@TearDown
|
|
public void tearDown() throws Exception {
|
|
chan.close().sync();
|
|
serverChan.close().sync();
|
|
future.cancel(true);
|
|
group.shutdownGracefully(0, 0, TimeUnit.SECONDS).sync();
|
|
abyte.release();
|
|
}
|
|
|
|
@Benchmark
|
|
public Object pingPong() throws Exception {
|
|
return chan.pipeline().writeAndFlush(abyte.retainedSlice()).sync();
|
|
}
|
|
|
|
@Benchmark
|
|
public Object executeSingle() throws Exception {
|
|
return chan.eventLoop().submit(runnable).get();
|
|
}
|
|
|
|
@Benchmark
|
|
@GroupThreads(3)
|
|
public Object executeMulti() throws Exception {
|
|
return chan.eventLoop().submit(runnable).get();
|
|
}
|
|
}
|