grpc
/
grpc-java
mirror of https://github.com/grpc/grpc-java.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

benchmarks: add NETTY_EPOLL as an option for transport testing

This commit is contained in:
Carl Mastrangelo 2016-05-18 11:40:31 -07:00
parent e19848092d
commit 8c7440e141
1 changed files with 37 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
benchmarks/src/jmh/java/io/grpc/benchmarks/TransportBenchmark.java
View File

@ -50,6 +50,9 @@ import io.grpc.netty.NegotiationType;
import io.grpc.netty.NettyChannelBuilder; import io.grpc.netty.NettyChannelBuilder;
import io.grpc.netty.NettyServerBuilder; import io.grpc.netty.NettyServerBuilder;
import io.grpc.okhttp.OkHttpChannelBuilder; import io.grpc.okhttp.OkHttpChannelBuilder;
import io.netty.channel.Channel;
import io.netty.channel.EventLoopGroup;
import io.netty.channel.ServerChannel;
import io.netty.channel.local.LocalAddress; import io.netty.channel.local.LocalAddress;
import io.netty.channel.local.LocalChannel; import io.netty.channel.local.LocalChannel;
import io.netty.channel.local.LocalServerChannel; import io.netty.channel.local.LocalServerChannel;
@ -71,7 +74,7 @@ import java.util.concurrent.TimeUnit;
@State(Scope.Benchmark) @State(Scope.Benchmark)
public class TransportBenchmark { public class TransportBenchmark {
public enum Transport { public enum Transport {
INPROCESS, NETTY, NETTY_LOCAL, OKHTTP INPROCESS, NETTY, NETTY_LOCAL, NETTY_EPOLL, OKHTTP
} }
@Param({"INPROCESS", "NETTY", "NETTY_LOCAL", "OKHTTP"}) @Param({"INPROCESS", "NETTY", "NETTY_LOCAL", "OKHTTP"})
@ -82,6 +85,7 @@ public class TransportBenchmark {
private ManagedChannel channel; private ManagedChannel channel;
private Server server; private Server server;
private BenchmarkServiceGrpc.BenchmarkServiceBlockingStub stub; private BenchmarkServiceGrpc.BenchmarkServiceBlockingStub stub;
private volatile EventLoopGroup groupToShutdown;
@Setup @Setup
public void setUp() throws Exception { public void setUp() throws Exception {
@ -114,6 +118,31 @@ public class TransportBenchmark {
.negotiationType(NegotiationType.PLAINTEXT); .negotiationType(NegotiationType.PLAINTEXT);
break; break;
} }
case NETTY_EPOLL:
{
InetSocketAddress address = new InetSocketAddress("localhost", pickUnusedPort());
// Reflection used since they are only available on linux.
Class<?> groupClass = Class.forName("io.netty.channel.epoll.EpollEventLoopGroup");
EventLoopGroup group = (EventLoopGroup) groupClass.newInstance();
@SuppressWarnings("unchecked")
Class<? extends ServerChannel> serverChannelClass = (Class<? extends ServerChannel>)
Class.forName("io.netty.channel.epoll.EpollServerSocketChannel");
serverBuilder = NettyServerBuilder.forAddress(address)
.bossEventLoopGroup(group)
.workerEventLoopGroup(group)
.channelType(serverChannelClass);
@SuppressWarnings("unchecked")
Class<? extends Channel> channelClass = (Class<? extends Channel>)
Class.forName("io.netty.channel.epoll.EpollSocketChannel");
channelBuilder = NettyChannelBuilder.forAddress(address)
.eventLoopGroup(group)
.channelType(channelClass)
.negotiationType(NegotiationType.PLAINTEXT);
groupToShutdown = group;
break;
}
case OKHTTP: case OKHTTP:
{ {
int port = pickUnusedPort(); int port = pickUnusedPort();
@ -155,6 +184,13 @@ public class TransportBenchmark {
if (!server.isTerminated()) { if (!server.isTerminated()) {
throw new Exception("failed to shut down server"); throw new Exception("failed to shut down server");
} }
if (groupToShutdown != null) {
groupToShutdown.shutdownGracefully(0, 1, TimeUnit.SECONDS);
groupToShutdown.awaitTermination(1, TimeUnit.SECONDS);
if (!groupToShutdown.isTerminated()) {
throw new Exception("failed to shut down event loop group.");
}
}
} }
private SimpleRequest simpleRequest = SimpleRequest.newBuilder() private SimpleRequest simpleRequest = SimpleRequest.newBuilder()