高性能TcpServer(Java) - Netty

女生的网名这么多〃 提交于 2020-11-26 13:21:37

源码下载 -> 提取码  QQ:505645074

 


 

Netty 是一个高性能、异步事件驱动的 NIO 框架,它提供了对 TCPUDP 和文件传输的支持,作为一个异步 NIO 框架,Netty 的所有 IO 操作都是异步非阻塞的,通过 Future-Listener 机制,用户可以方便的主动获取或者通过通知机制获得 IO 操作结果。

作为当前最流行的 NIO 框架,Netty 在互联网领域、大数据分布式计算领域、游戏行业、通信行业等获得了广泛的应用,一些业界著名的开源组件也基于 Netty 的 NIO 框架构建。

协议解析代码:

package nettyserver;

 

import io.netty.buffer.ByteBuf;

import io.netty.buffer.Unpooled;

import io.netty.channel.ChannelHandlerContext;

import io.netty.channel.ChannelInboundHandlerAdapter;

import io.netty.handler.timeout.IdleStateEvent;

import io.netty.util.ReferenceCountUtil;

import java.util.HashMap;

import java.util.Map;

 

class MyInHandler extends ChannelInboundHandlerAdapter {

 

    final byte HEAD1 = 0x48;// H

    final byte HEAD2 = 0x54;// T

    final byte HEAD3 = 0x45;// E

    final byte HEAD4 = 0x4D;// M

    final byte HEAD5 = 0x50;// P

    final byte HEAD6 = 0x3D;// =

    public static final String SPLIT1 = "#";

 

    public static Map<String, io.netty.buffer.ByteBuf> dicSn2Buffer = new HashMap<>();

 

    // HTEMP=0026#1533022506#Meter-001#01##

 

    @Override

    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause)

    {

        // 当出现异常就关闭连接

        //cause.printStackTrace();

        RemoveClient(ctx, "exceptionCaught");

    }

 

    @Override

    public void channelActive(io.netty.channel.ChannelHandlerContext ctx) throws Exception

    {

        CTxtHelp.AppendLog("客户端已经连接,sn:" + GetSN(ctx));

    }

 

    @Override

    public void channelInactive(io.netty.channel.ChannelHandlerContext ctx) throws Exception

    {

        RemoveClient(ctx, "channelInactive");

    }

 

    @Override

    public void userEventTriggered(ChannelHandlerContext ctx, Object evt) throws Exception

    {

        if (evt instanceof IdleStateEvent)

        {  

            String type = "未知";

            IdleStateEvent event = (IdleStateEvent) evt;  

            if (null != event.state())

            {

                switch (event.state())

                {

                    case READER_IDLE: type = "read idle"; break;

                    case WRITER_IDLE: type = "write idle"; break;

                    case ALL_IDLE: type = "all idle"; break;

                    default: break;

                }

            }

            RemoveClient(ctx, "超时退出(" + type + ")");

        }

        else

        {

            super.userEventTriggered(ctx, evt);

        }

    }

 

    @Override

    public void channelRead(io.netty.channel.ChannelHandlerContext ctx, Object msg)

    {

        if (!ctx.channel().isActive()) return;

        

        String sn = GetSN(ctx);

        

        if(!dicSn2Buffer.containsKey(sn)) dicSn2Buffer.put(sn, Unpooled.buffer(1024));

        io.netty.buffer.ByteBuf oldBuffer = dicSn2Buffer.get(sn);  

        

        if (null != msg)

        {

            io.netty.buffer.ByteBuf recvBuffer = (ByteBuf) msg;

            int size = recvBuffer.readableBytes();

            if (size > 0)

            {

                //recvBuffer.markReaderIndex();

                oldBuffer.writeBytes(recvBuffer);

                ReferenceCountUtil.release(recvBuffer);

                //recvBuffer.resetReaderIndex();

                //byte[] recv = new byte[size];

                //recvBuffer.readBytes(recv);

                //CTxtHelp.AppendLog("接收数据:" + CDataHelper.ArrayByteToString(recv));

            }

        }

 

        byte head1 = 0; byte head2 = 0; byte head3 = 0; byte head4 = 0; byte head5 = 0; byte head6 = 0;

 

        while (oldBuffer.isReadable())

        {

            oldBuffer.markReaderIndex();

            head1 = oldBuffer.readByte(); if (!oldBuffer.isReadable()) { oldBuffer.resetReaderIndex(); return ; } // 掉包处理

            if (HEAD1 == head1)// 垃圾包处理

            {

                head2 = oldBuffer.readByte(); if (!oldBuffer.isReadable()) { oldBuffer.resetReaderIndex(); return ; }

                head3 = oldBuffer.readByte(); if (!oldBuffer.isReadable()) { oldBuffer.resetReaderIndex(); return ; }

                head4 = oldBuffer.readByte(); if (!oldBuffer.isReadable()) { oldBuffer.resetReaderIndex(); return ; }

                head5 = oldBuffer.readByte(); if (!oldBuffer.isReadable()) { oldBuffer.resetReaderIndex(); return ; }

                head6 = oldBuffer.readByte(); if (!oldBuffer.isReadable()) { oldBuffer.resetReaderIndex(); return ; }

                if (HEAD2 == head2 && HEAD3 == head3 && HEAD4 == head4 && HEAD5 == head5 && HEAD6 == head6)

                {

                    break;

                }

                else

                {

                    CTxtHelp.AppendLog("Error,Unable to parse the data2");

                }

            }

            else

            {

                CTxtHelp.AppendLog("Error,Unable to parse the data1");

            }

        }

 

        int lengthsize = 4;

        if (GetWaitRecvRemain(oldBuffer, lengthsize)) { oldBuffer.resetReaderIndex(); return ; }

        byte[] arrlen = new byte[lengthsize]; oldBuffer.readBytes(arrlen);

        int lengthdata = CDataHelper.String2Int(CDataHelper.ArrayByteToString(arrlen));// if (-1 == len) return;

        if (GetWaitRecvRemain(oldBuffer, lengthdata)) { oldBuffer.resetReaderIndex(); return ; }

        byte[] source = new byte[lengthdata]; oldBuffer.readBytes(source);  

 

        String data = CDataHelper.ArrayByteToString(source);

        CTxtHelp.AppendLog("解析后:" + data);

 

        if (null == data || 0 == data.length() || data.length() - 1 != data.lastIndexOf(SPLIT1)) return;

        data = data.substring(1, data.length() - 1);

        String[] items = data.split(SPLIT1, -1);

        if (null == items || 4 != items.length) return;

        String uid = items[0];

        String taskid = items[1];

        int cmd = CDataHelper.String2Int(items[2]);

        String content = items[3];

        String back = "";

        try

        {

            switch (cmd)

            {

                case 1: back = "ok"; break;

                case 2: break;

                case 3: break;

            }

        }

        catch (Exception ex)

        {

            CTxtHelp.AppendLog("error@" + ex.getMessage());

        }

        

        if (!"".equals(back))

        {

            byte[] sendBuff = GetProtocol(uid, cmd, back);

            ByteBuf out = ctx.alloc().directBuffer(sendBuff.length);

            out.writeBytes(sendBuff);

            ctx.channel().writeAndFlush(out);

        }

        

        if (!oldBuffer.isReadable())

        {

            oldBuffer.clear();

        }

        else

        {

            channelRead(ctx, null);

        }

    }

 

    boolean GetWaitRecvRemain( io.netty.buffer.ByteBuf buf, int len) {

        return buf.readerIndex() + len > buf.writerIndex();

    }

    

    String GetSN(io.netty.channel.ChannelHandlerContext ctx)

    {

        return ctx.channel().remoteAddress().toString().replace("/", "");

    }

 

    void RemoveClient(io.netty.channel.ChannelHandlerContext ctx, String errmsg)

    {

        String sn = GetSN(ctx);

        CTxtHelp.AppendLog("客户端退出,errmsg" + errmsg);

        if(dicSn2Buffer.containsKey(sn)) {

            io.netty.buffer.ByteBuf byteBuf = dicSn2Buffer.get(sn);

            ReferenceCountUtil.release(byteBuf);

            dicSn2Buffer.remove(sn);

        }

        ctx.close();

    }

 

    byte[] GetProtocol(String uid, int cmd, String msg)

    {

        String content = SPLIT1 + uid + SPLIT1 +

                        System.currentTimeMillis() + SPLIT1 +

                        CDataHelper.lpad(cmd, 2) + SPLIT1 +

                        msg + SPLIT1;

        StringBuilder data = new StringBuilder();

        data.append("HTEMP=");

        data.append(CDataHelper.lpad(CDataHelper.GetStringLength(content), 4));

        data.append(content);

        return CDataHelper.StringToByte(data.toString());

    }

 

    String lpad(int number, int length) {

        String f = "%0" + length + "d";

        return String.format(f, number);

    }

}

程序结构:

运行结果:

1. 包解析(正常包+掉包+粘包+垃圾包)

2. 超时清理链路

3. 并发测试

 

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!