Mina框架断包、粘包问题解决方案
Apache Mina Server 是一个网络通信应用框架,也就是说,它主要是对基于TCP/IP、UDP/IP协议栈的通信框架(当然,也可以提供JAVA 对象的序列化服务、虚拟机管道通信服务等),Mina 可以帮助我们快速开发高性能、高扩展性的网络通信应用,Mina 提供了事件驱动、异步(Mina 的异步IO 默认使用的是JAVA NIO 作为底层支持)操作的编程模型。
在mina中,一般的应用场景用TextLine的Decode和Encode就够用了(TextLine的默认分割符虽然是\r\n,但其实分隔符是可以自己指定的,如:newTextLineDecoder(charset, decodingDelimiter);)
但默认解码器每次读取缓冲的数据是有限制的,即ReadBufferSize的大小,默认是2048个字节,当数据包比较大时将被分成多次读取,造成断包。虽然可以通过acceptor.getSessionConfig().setReadBufferSize(newsize)这种方式来增加默认容量,但毕竟不是王道(太大了浪费空间,肯定会降低数据的处理效率)。
所以,当我们接收的数据的大小不是很固定,且容易偏大的时候,默认的TextLine就不适合了。这时我们在解析之前就需要判断数据包是否完整,这样处理起来就会非常麻烦。那么Mina 中幸好提供了CumulativeProtocolDecoder
类,从名字上可以看出累积性的协议解码器,也就是说只要有数据发送过来,这个类就会去读取数据,然后累积到内部的IoBuffer 缓冲区,但是具体的拆包(把累积到缓冲区的数据解码为JAVA 对象)交由子类的doDecode()方法完成,实际上CumulativeProtocolDecoder就是在decode()反复的调用暴漏给子类实现的doDecode()方法。
具体执行过程如下所示:
A. 你的doDecode()方法返回true 时,CumulativeProtocolDecoder 的decode()方法会首先判断你是否在doDecode()方法中从内部的IoBuffer 缓冲区读取了数据,如果没有,则会抛出非法的状态异常,也就是你的doDecode()方法返回true 就表示你已经消费了本次数据(相当于聊天室中一个完整的消息已经读取完毕),进一步说,也就是此时你必须已经消费过内部的IoBuffer 缓冲区的数据(哪怕是消费了一个字节的数据)。如果验证通过,那么CumulativeProtocolDecoder会检查缓冲区内是否还有数据未读取,如果有就继续调用doDecode()方法,没有就停止对doDecode()方法的调用,直到有新的数据被缓冲。
B. 当你的doDecode()方法返回false 时,CumulativeProtocolDecoder 会停止对doDecode()方法的调用,但此时如果本次数据还有未读取完的,就将含有剩余数据的IoBuffer 缓冲区保存到IoSession 中,以便下一次数据到来时可以从IoSession 中提取合并。如果发现本次数据全都读取完毕,则清空IoBuffer 缓冲区。
简而言之,当你认为读取到的数据已经够解码了,那么就返回true,否则就返回false。这个CumulativeProtocolDecoder其实最重要的工作就是帮你完成了数据的累积,因为这个工作是很烦琐的。
一、 实现解码器
CumulativeProtocolDecoder是一个抽象类,必须继承并实现其doDecode方法,用户自定义协议的拆分就应该写在doDecode方法中,下面的MyDecoder类是一个其子类的实现:
public
class MyDecoder extends CumulativeProtocolDecoder {
public
static Logger log = Logger.getLogger(MyDecoder.class);
/**
* 包解码器组件
*/
private PacketComponent
packetComponent;
/**
* 这个方法的返回值是重点:
* 1、当内容刚好时,返回false,告知父类接收下一批内容
* 2、内容不够时需要下一批发过来的内容,此时返回false,这样父类 CumulativeProtocolDecoder
* 会将内容放进IoSession中,等下次来数据后就自动拼装再交给本类的doDecode
* 3、当内容多时,返回true,因为需要再将本批数据进行读取,父类会将剩余的数据再次推送本
* 类的doDecode
*/
public
boolean doDecode(IoSession session,IoBuffer in,
ProtocolDecoderOutput out) throws Exception {
log.info("in.remaining : "+in.remaining());
if(in.remaining() > 0){//有数据时,读取前8字节判断消息长度
byte [] sizeBytes =
new byte[8];
in.mark();//标记当前位置,以便reset
//因为我的前数据包的长度是保存在第4-8字节中,
in.get(sizeBytes,0,8);//读取4字节
//DataTypeChangeHelper是自己写的一个byte[]转int的一个工具类
int size = (int) DataTypeUtil.bytesToInt(sizeBytes,4);
log.info("size : "+size);
in.reset();
if(size > in.remaining()){//如果消息内容不够,则重置,相当于不读取size
return
false;//父类接收新数据,以拼凑成完整数据
} else{
byte[] bytes =
new byte[size];
in.get(bytes, 0, size);
//把字节转换为Java对象的工具类
PackageData pack =
packetComponent.getDataFromBuffer(IoBuffer.wrap(bytes));
out.write(pack);
if(in.remaining() > 0){//如果读取内容后还粘了包,就让父类再重读
一次,进行下一次解析
return
true;
}
}
}
return
false;//处理成功,让父类进行接收下个包
}
getter();
Setter();
}
二、 实现编解码工厂和解码器
我们还需要一个编解码工厂,用来为编解码过滤器提供编码器和解码器,解码器此处我们用不到,但是也必须提供,所以可以提供一个空的实现。
/**
*
* 编解码工厂
*
*/
public
class MyCodecFcatory implements ProtocolCodecFactory {
private ProtocolEncoder
encoder = null;
private ProtocolDecoder
decoder = null;
public MyCodecFcatory(ProtocolEncoder encoder, ProtocolDecoderdecoder) {
this.encoder = encoder;
this.decoder = decoder;
}
@Override
public ProtocolEncoder getEncoder(IoSession session)
throws Exception {
return
this.encoder;
}
@Override
public ProtocolDecoder getDecoder(IoSession session)
throws Exception {
return
this.decoder;
}
}
/**
*
* 编码器:不做任何操作,数据已是约定好的格式,按原格式编码
*
*/
public
class MyEncoder extends ProtocolEncoderAdapter {
@Override
public
void encode(IoSession session, Object message,
ProtocolEncoderOutput out) throws Exception {
// TODO Do nothing
}
}
三、 配置编解码过滤器
下面就可以配置编解码过滤器了:
<!-- 累加数据包解码器:解断丢包、粘包问题 -->
<bean
id="codec"
class="org.apache.mina.filter.codec.ProtocolCodecFilter">
<constructor-arg>
<bean
class="com.mina.codec.MyCodecFcatory">
<constructor-arg
index="0">
<bean
class="com.mina.codec.MyEncoder"></bean>
</constructor-arg>
<constructor-arg
index="1">
<bean
class="com.mina.codec.MyDecoder">
<property
name="packetComponent">
<bean
class="com. mina.component.RootComponent">
</bean>
</property>
</bean>
</constructor-arg>
</bean>
</constructor-arg>
</bean>
<bean
id="filterChainBuilder"
class="org.apache.mina.core.filterchain.DefaultIoFilterChainBuilder">
<property
name="filters">
<map>
<entry
key="codec"
value-ref="codec"/>
<entry
key="logger"
value-ref="loggerFilter"/>
<entry
key="executors"
value-ref="executors"/>
</map>
</property>
</bean>
需要注意的是:在doDecode中通过out.write(pack) 把数据输出后,官方的说明文档中说接下来会继续执行后面的过滤器,然后是IoHandle。如果你是只用了一个编解码过滤器的话,这可能完全没问题,但是如果使用了两个编解码过滤器(可能很少有人会这样做,但本人由于前期使用了另外一个自定义的编解码过滤器,后来想加上这个可累加的解码器,为了图省事就在原过滤器的前面新增加了一个编解码过滤器,后来数据流就不走我原来的编解码过滤器了,out.write()之后直接到了IoHandle里面,搞了我好久,无奈最后把两个编解码过滤器合二为一啦,其中原因我还没时间去搞个清楚,为防止大家和我犯同一个错误,特此提醒!)
Mina框架断包、粘包问题解决方案