0
点赞
收藏
分享

微信扫一扫

基于netty4的TCP短连接测试

得一道人 2022-12-07 阅读 156


本文中的代码做了一定优化,但是还不是很完全,欢迎指正


工程结构图如下:



TcpServer.java





package com.lin.netty4.tcp;

import io.netty.bootstrap.ServerBootstrap;
import io.netty.channel.ChannelInitializer;
import io.netty.channel.ChannelPipeline;
import io.netty.channel.EventLoopGroup;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.SocketChannel;
import io.netty.channel.socket.nio.NioServerSocketChannel;
import io.netty.handler.codec.LengthFieldBasedFrameDecoder;
import io.netty.handler.codec.LengthFieldPrepender;

import org.apache.log4j.Logger;

public class TcpServer {
private static final Logger logger = Logger.getLogger(TcpServer.class);
private static final String IP = "127.0.0.1";
private static final int PORT = 9999;
/**用于分配处理业务线程的线程组个数 */
protected static final int BIZGROUPSIZE = Runtime.getRuntime().availableProcessors()*2; //默认
/** 业务出现线程大小*/
protected static final int BIZTHREADSIZE = 1000;
private static final EventLoopGroup bossGroup = new NioEventLoopGroup(BIZGROUPSIZE);
private static final EventLoopGroup workerGroup = new NioEventLoopGroup(BIZTHREADSIZE);
protected static void run() throws Exception {
ServerBootstrap b = new ServerBootstrap();
b.group(bossGroup, workerGroup);
b.channel(NioServerSocketChannel.class);
b.childHandler(new ChannelInitializer<SocketChannel>() {
@Override
public void initChannel(SocketChannel ch) throws Exception {
ChannelPipeline pipeline = ch.pipeline();
pipeline.addLast("frameDecoder", new LengthFieldBasedFrameDecoder(Integer.MAX_VALUE, 0, 4, 0, 4));
pipeline.addLast("frameEncoder", new LengthFieldPrepender(4));
pipeline.addLast(workerGroup,new TcpServerHandler());
}
});
// b.childOption(ChannelOption.SO_KEEPALIVE,true);
// b.option(ChannelOption.SO_BACKLOG, 10000);
b.bind(IP, PORT).sync();
logger.info("TCP服务器已启动");
}

protected static void shutdown() {
workerGroup.shutdownGracefully();
bossGroup.shutdownGracefully();
}

public static void main(String[] args) throws Exception {
logger.info("开始启动TCP服务器...");
TcpServer.run();
// TcpServer.shutdown();
}
}



TcpServerHandler.java

package com.lin.netty4.tcp;

import io.netty.buffer.ByteBuf;
import io.netty.channel.ChannelFutureListener;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.ChannelInboundHandlerAdapter;
import io.netty.util.ReferenceCountUtil;

import org.apache.log4j.Logger;



public class TcpServerHandler extends ChannelInboundHandlerAdapter{
private static final Logger logger = Logger.getLogger(TcpServerHandler.class);

@Override
public void channelRead(ChannelHandlerContext ctx, Object msg)
throws Exception {
if (msg instanceof ByteBuf) {
ByteBuf buf = (ByteBuf) msg;
byte[] dst = new byte[buf.capacity()];
buf.readBytes(dst);
logger.info("SERVER接收到消息:" + new String(dst));

byte[] dest = (new String(dst)+". yes, server is accepted you ,nice !").getBytes();
ByteBuf destBuf = ctx.alloc().buffer(dest.length);
destBuf.writeBytes(dest);
ctx.channel().writeAndFlush(destBuf).addListener(ChannelFutureListener.CLOSE);

ReferenceCountUtil.release(msg);
} else {
logger.warn("error object !");
}
}
@Override
public void exceptionCaught(ChannelHandlerContext ctx,
Throwable cause) throws Exception {
logger.warn("Unexpected exception from downstream.", cause);
ctx.close();
}
}


TcpClient.java

package com.lin.netty4.tcp;

import io.netty.bootstrap.Bootstrap;
import io.netty.buffer.ByteBuf;
import io.netty.buffer.ByteBufAllocator;
import io.netty.channel.Channel;
import io.netty.channel.ChannelInitializer;
import io.netty.channel.ChannelPipeline;
import io.netty.channel.EventLoopGroup;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.nio.NioSocketChannel;
import io.netty.handler.codec.LengthFieldBasedFrameDecoder;
import io.netty.handler.codec.LengthFieldPrepender;

import org.apache.log4j.Logger;


public class TcpClient {
private static final Logger logger = Logger.getLogger(TcpClient.class);
public static String HOST = "127.0.0.1";
public static int PORT = 9999;

public static Bootstrap bootstrap = getBootstrap();
/**
* 初始化Bootstrap
* @return
*/
public static final Bootstrap getBootstrap(){
EventLoopGroup group = new NioEventLoopGroup();
Bootstrap b = new Bootstrap();
b.group(group).channel(NioSocketChannel.class);
b.handler(new ChannelInitializer<Channel>() {
@Override
protected void initChannel(Channel ch) throws Exception {
ChannelPipeline pipeline = ch.pipeline();
pipeline.addLast("frameDecoder", new LengthFieldBasedFrameDecoder(Integer.MAX_VALUE, 0, 4, 0, 4));
pipeline.addLast("frameEncoder", new LengthFieldPrepender(4));
pipeline.addLast("handler", new TcpClientHandler());
}
});
// b.option(ChannelOption.SO_KEEPALIVE, true);
return b;
}

public static final Channel getChannel(String host,int port){
Channel channel = null;
try {
channel = bootstrap.connect(host, port).sync().channel();
} catch (Exception e) {
logger.error(String.format("连接Server(IP[%s],PORT[%s])失败", host,port),e);
return null;
}
return channel;
}

public static void sendMsg(Channel channel,Object msg) throws Exception {
if(channel!=null){
channel.writeAndFlush(msg).sync();
}else{
logger.warn("消息发送失败,连接尚未建立!");
}
}

public static void main(String[] args) throws Exception {
try {
long t0 = System.nanoTime();
byte[] value = null;
Channel channel = null;
for (int i = 0; i < 50000; i++) {
channel = getChannel(HOST, PORT);
value = (i+",你好").getBytes();
ByteBufAllocator alloc = channel.alloc();
ByteBuf buf = alloc.buffer(value.length);
buf.writeBytes(value);
TcpClient.sendMsg(channel,buf);
}
long t1 = System.nanoTime();
System.out.println((t1-t0)/1000000.0);
Thread.sleep(5000);
System.exit(0);
} catch (Exception e) {
// TODO Auto-generated catch block
e.printStackTrace();
}

}
}


TcpClientHandler.java

package com.lin.netty4.tcp;

import org.apache.log4j.Logger;

import io.netty.buffer.ByteBuf;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.ChannelInboundHandlerAdapter;
import io.netty.util.ReferenceCountUtil;


public class TcpClientHandler extends ChannelInboundHandlerAdapter {
private static final Logger logger = Logger.getLogger(TcpClientHandler.class);

@Override
public void channelRead(ChannelHandlerContext ctx, Object msg)
throws Exception {
if(msg instanceof ByteBuf){
ByteBuf buf = (ByteBuf)msg;
byte[] dst = new byte[buf.capacity()];
buf.readBytes(dst);
logger.info("client接收到服务器返回的消息:"+new String(dst));
ReferenceCountUtil.release(msg);
}else{
logger.warn("error object");
}

}


}


log4j.properties

#控制台
log4j.rootLogger=DEBUG, CONSOLE,DAILY_ROLLING_FILE

log4j.appender.CONSOLE=org.apache.log4j.ConsoleAppender
log4j.appender.CONSOLE.Threshold=WARN
log4j.appender.CONSOLE.Encoding=UTF-8
log4j.appender.CONSOLE.layout=org.apache.log4j.PatternLayout
log4j.appender.CONSOLE.layout.ConversionPattern=[%t] %d [%p] - %l:%m%n

#文件方式存储
log4j.appender.DAILY_ROLLING_FILE=org.apache.log4j.DailyRollingFileAppender
log4j.appender.DAILY_ROLLING_FILE.Encoding=UTF-8
log4j.appender.DAILY_ROLLING_FILE.Threshold=WARN
log4j.appender.DAILY_ROLLING_FILE.File=E\:\\log4j\\nettyDemo.log
log4j.appender.DAILY_ROLLING_FILE.DatePattern='.'yyyy-MM-dd
log4j.appender.DAILY_ROLLING_FILE.layout=org.apache.log4j.PatternLayout
log4j.appender.DAILY_ROLLING_FILE.layout.ConversionPattern=[%t]%d [%p] - %l:%m%n


log4j.logger.org.apache.commons = WARN
log4j.logger.io.netty = WARN



 

37秒(  37870.31592 ),Eclipse未做优化,PC配置4核3G内存

举报

相关推荐

0 条评论