mina自定义编解码器接收处理byte数组(同时解决数据传输中的粘包、缺包问题)

最新推荐文章于 2021-11-05 22:44:58 发布

jbgtwang

最新推荐文章于 2021-11-05 22:44:58 发布

阅读量1.8w

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.csdn.net/jbgtwang/article/details/26266309

本文介绍了一种自定义传输协议的实现方法，通过使用Apache Mina框架完成字节数组的编码与解码过程。该协议规定字节数组的前四个字节为后续数据的长度，随后跟随实际数据内容。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

我们在自定义传输协议时，通常都是采用字节数组的方式进行传送，如何正确接收和解码byte数组？

假设我们自定义了传输协议: 字节数组的前4个字节是要传输的数据长度，后面跟数据。我们用mina可以这样处理

1.自定义编码器ByteArrayEncoder.java

import org.apache.mina.core.session.IoSession;
import org.apache.mina.filter.codec.ProtocolEncoderAdapter;
import org.apache.mina.filter.codec.ProtocolEncoderOutput;

/**
* 编码器将数据直接发出去(不做处理)
*/
public class ByteArrayEncoder extends ProtocolEncoderAdapter {

    @Override
    public void encode(IoSession session, Object message,
            ProtocolEncoderOutput out) throws Exception {
        out.write(message);
        out.flush();

    }
}

2.自定义解码器(确保能读取到完整的包)

import org.apache.mina.core.buffer.IoBuffer;
import org.apache.mina.core.session.IoSession;
import org.apache.mina.filter.codec.CumulativeProtocolDecoder;
import org.apache.mina.filter.codec.ProtocolDecoderOutput;

import com.talkweb.meeting.tools.IntByteConvert;

public class ByteArrayDecoder extends CumulativeProtocolDecoder {

    @Override
    public boolean doDecode(IoSession session, IoBuffer in, ProtocolDecoderOutput out)
            throws Exception {

        if(in.remaining() > 4){//前4字节是包头
            //标记当前position的快照标记mark，以便后继的reset操作能恢复position位置
            in.mark();
            byte[] l = new byte[4];
            in.get(l);

            //包体数据长度
            int len = MyTools.bytes2int(l);//将byte转成int


            //注意上面的get操作会导致下面的remaining()值发生变化
            if(in.remaining() < len){
                //如果消息内容不够，则重置恢复position位置到操作前,进入下一轮, 接收新数据，以拼凑成完整数据
                in.reset();
                return false;
            }else{
                //消息内容足够
                in.reset();//重置恢复position位置到操作前
                int sumlen = 4+len;//总长 = 包头+包体
                byte[] packArr = new byte[sumlen];
                in.get(packArr, 0 , sumlen);

                IoBuffer buffer = IoBuffer.allocate(sumlen);
                buffer.put(packArr);
                buffer.flip();
                out.write(buffer);
                buffer.free();

                if(in.remaining() > 0){//如果读取一个完整包内容后还粘了包，就让父类再调用一次，进行下一次解析
                    return true;
                }
            }
        }
        return false;//处理成功，让父类进行接收下个包
    }

}

3.编解码工厂类

import org.apache.mina.core.session.IoSession;
import org.apache.mina.filter.codec.ProtocolCodecFactory;
import org.apache.mina.filter.codec.ProtocolDecoder;
import org.apache.mina.filter.codec.ProtocolEncoder;

/**
* @author BruceYang
*
*/
public class ByteArrayCodecFactory implements ProtocolCodecFactory {

    private ByteArrayDecoder decoder;
    private ByteArrayEncoder encoder;

    public ByteArrayCodecFactory() {
        encoder = new ByteArrayEncoder();
        decoder = new ByteArrayDecoder();
    }

    @Override
    public ProtocolDecoder getDecoder(IoSession session) throws Exception {
        return decoder;
    }

    @Override
    public ProtocolEncoder getEncoder(IoSession session) throws Exception {
        return encoder;
    }

}

4.调用编解码工厂进行编解码

NioSocketAcceptor acceptor = new NioSocketAcceptor();

acceptor.getFilterChain().addLast("mycoder", new ProtocolCodecFilter(new ByteArrayCodecFactory()));