public void run(URI uri) throws Exception {
int port = uri.getPort() > 0 ? uri.getPort() : BaseTestConfig.HTTPS_PORT;
String host = uri.getHost();
// Configure SSL context if necessary.
final SslContext sslCtx;
sslCtx =
SslContextBuilder.forClient().trustManager(InsecureTrustManagerFactory.INSTANCE).build();
// Configure the client.
EventLoopGroup group = new NioEventLoopGroup();
httpsInitializer = generateHttpsInitializer(sslCtx);
try {
Bootstrap b = new Bootstrap();
b.option(ChannelOption.CONNECT_TIMEOUT_MILLIS, 10000);
b.group(group).channel(NioSocketChannel.class).handler(httpsInitializer);
// Make the connection attempt.
startTime = System.nanoTime();
Channel ch = b.connect(host, port).sync().channel();
localAddress = ch.localAddress();
remoteAddress = ch.remoteAddress();
sendRequests(uri.toURL(), ch);
// Wait for the server to close the connection.
ch.closeFuture().sync();
endTime = System.nanoTime();
long duration = endTime - startTime;
logger.info(String.format("connection duration: %,dns (%d)", duration, duration));
} finally {
// Shut down executor threads to exit.
group.shutdownGracefully();
}
}
private static SslContext getSslContext() {
try {
return SslContextBuilder.forClient()
.trustManager(InsecureTrustManagerFactory.INSTANCE)
.build();
} catch (SSLException e) {
return null;
}
}
public static void main(String[] args) throws Exception {
// Configure SSL.
final SslContext sslCtx =
SslContextBuilder.forClient().trustManager(InsecureTrustManagerFactory.INSTANCE).build();
EventLoopGroup group = new NioEventLoopGroup();
try {
Bootstrap bootstrap =
new Bootstrap()
.group(group)
.channel(NioSocketChannel.class)
.handler(new SimpleNettyClientInitializer(sslCtx));
// Start the connection attempt.
Channel ch = bootstrap.connect(HOST, PORT).sync().channel();
// Read commands from the stdin.
ChannelFuture lastWriteFuture = null;
BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
for (; ; ) {
String line = in.readLine();
if (line == null) {
break;
}
// Sends the received line to the server.
lastWriteFuture = ch.writeAndFlush(line);
// If user typed the 'bye' command, wait until the server closes
// the connection.
if ("bye".equals(line.toLowerCase())) {
ch.closeFuture().sync();
break;
}
}
// Wait until all messages are flushed before closing the channel.
if (lastWriteFuture != null) {
lastWriteFuture.sync();
}
} finally {
// The connection is closed automatically on shutdown.
group.shutdownGracefully();
}
}
@Override
protected void serviceInit() throws Exception {
final SslContext sslCtx;
if (config.isSsl()) {
sslCtx =
SslContextBuilder.forClient().trustManager(InsecureTrustManagerFactory.INSTANCE).build();
} else {
sslCtx = null;
}
Class clazz;
if (config.isUseEpoll()) {
group = new EpollEventLoopGroup(config.getChildNioEventThreads());
clazz = EpollSocketChannel.class;
} else {
group = new NioEventLoopGroup(config.getChildNioEventThreads());
clazz = NioSocketChannel.class;
}
bootstrap = new Bootstrap();
bootstrap.option(ChannelOption.SO_KEEPALIVE, true);
bootstrap.option(ChannelOption.WRITE_BUFFER_HIGH_WATER_MARK, 32 * 1024);
bootstrap.option(ChannelOption.WRITE_BUFFER_LOW_WATER_MARK, 8 * 1024);
bootstrap.option(ChannelOption.SO_LINGER, config.getSoLinger());
bootstrap.option(ChannelOption.RCVBUF_ALLOCATOR, AdaptiveRecvByteBufAllocator.DEFAULT);
bootstrap
.group(group)
.channel(clazz)
.handler(
new ChannelInitializer<SocketChannel>() {
@Override
public void initChannel(SocketChannel ch) throws Exception {
ChannelPipeline p = ch.pipeline();
if (sslCtx != null) {
p.addLast(sslCtx.newHandler(ch.alloc(), config.getHost(), config.getPort()));
}
p.addLast(
new MsgPackEncoder(),
new MsgPackDecoder(config.getPayload()),
new ClientHandler(ClientProxy.this));
}
});
}
public NettyChannel connect() throws Exception {
if (ssl) {
this.sslCtx =
SslContextBuilder.forClient().trustManager(InsecureTrustManagerFactory.INSTANCE).build();
}
group = new NioEventLoopGroup();
Bootstrap b = new Bootstrap();
b.group(group)
.channel(NioSocketChannel.class)
.option(ChannelOption.TCP_NODELAY, true)
.option(ChannelOption.SO_TIMEOUT, socketTimeout)
.option(ChannelOption.CONNECT_TIMEOUT_MILLIS, connectTimeout)
.handler(
new ChannelInitializer<SocketChannel>() {
@Override
public void initChannel(SocketChannel ch) throws Exception {
channel = new NettyChannel(ch, callbackExecutor);
channel.setMessagesReceiver(receiver);
if (ssl) {
ch.pipeline().addLast(sslCtx.newHandler(ch.alloc(), host, port));
}
ch.pipeline().addLast("lengthprepender", new LengthFieldPrepender(4));
ch.pipeline()
.addLast(
"lengthbaseddecoder",
new LengthFieldBasedFrameDecoder(Integer.MAX_VALUE, 0, 4, 0, 4));
//
ch.pipeline().addLast("messageencoder", new DataMessageEncoder());
ch.pipeline().addLast("messagedecoder", new DataMessageDecoder());
ch.pipeline().addLast(new InboundMessageHandler(channel));
}
});
ChannelFuture f = b.connect(host, port).sync();
socketchannel = f.channel();
return channel;
}