MINA2收包中对粘包的处理

http://scholers.iteye.com/blog/784336

  MINA2中（MINA2 RC版本，MINA2.0正式版已经发布)服务端接受数据默认有一定长度的缓冲区（可以在启动的时候设置)。那么对于大报文，怎么处理呢？比如说超过1024，甚至更多？MINA2为了节省网络流量，提高处理效率，会将大报文自动拆分（可能是存放MINA2中的缓冲区里面)：比如2048字节的报文，就会拆分成两次；那么在接受的时候，就有一个如何判断是完整报文的问题，或者说是一个拆包组包的问题。 
  MINA2中初始化服务的时候是可以设置输入和输出的缓冲区的： 
 

 acceptor.getSessionConfig().setReadBufferSize(1024);

    MINA2提供的案例是，在IoSession中设置一个类似于session，存在在当前IoSession中的全局变量，在此IoSession中有效。 
 

 private final AttributeKey TEST = new AttributeKey(getClass(), "TEST");

  大家都知道,通过 SOCKET TCP/IP传输过来的报文是不知道边界的，所以一般会约定在前端固定长度的字节加上报文长度，让SERVER来根据这个长度来确定整个报文的边界，在我前面的博文有提到。其实MINA2中有： 
  prefixedDataAvailable(4) int 
方法，来判断固定长度的报文长度，但是参数只能是1，2，4；该方法很好用。判断前四字节的整型值是否大于等于整个缓冲区的数据。可以方便的判断一次 messageReceived 过来的数据是否完整。（前提是自己设计的网络通讯协议前四字节等于发送数据的长度） ，如果你不是设定1，2，4字节来作为长度的话，那么就没辙了。 
   在你的解码操作中，MINA2的缓冲区发多少次报文，你的decode方法就会调用多少次。 
  上面设置了session之后，可以采用一个方法： 
 

 /**
  *
  * @param session
  *            会话信息
  * @return 返回session中的累积
  */
 private Context getContext(IoSession session) {
     Context ctx = (Context) session.getAttribute(CONTEXT);
     if (ctx == null) {
         ctx = new Context();
         session.setAttribute(CONTEXT, ctx);
     }
     return ctx;
 }

然后在你的decode方法中，首先从session取出数据对象，进行拼接： 
 

 Context ctx = getContext(session);

  // 先把当前buffer中的数据追加到Context的buffer当中
  ctx.append(ioBuffer);
  // 把position指向0位置，把limit指向原来的position位置
  IoBuffer buf = ctx.getBuffer();
  buf.flip();

  
接着读取每次报文的总长度： 

 // 读取消息头部分
 byte[] bLeng = new byte[packHeadLength];
 buf.get(bLeng);
 int length = -1;
 try {
   length = Integer.parseInt(new String(bLeng));
 } catch (NumberFormatException ex) {
   ex.printStackTrace();
 }
 if (length > 0) {
   ctx.setMsgLength(length);
 }

在读取到每次报文的长度之后，就接着循环判断BUF里面的字节数据是否已经全部接受完毕了，如果没有接受完毕，那么就不处理；下面是完整处理的代码： 
 

 while (buf.remaining() >= packHeadLength) {
     buf.mark();
     // 设置总长度
     if (ctx.getMsgLength() <= 0) {
         // 读取消息头部分
         byte[] bLeng = new byte[packHeadLength];
         buf.get(bLeng);
         int length = -1;
         try {
             length = Integer.parseInt(new String(bLeng));
             } catch (NumberFormatException ex) {
                         ex.printStackTrace();
             }
             if (length > 0) {
               ctx.setMsgLength(length);
                }
             }


             // 读取消息头部分
             int length = ctx.getMsgLength();
             // 检查读取的包头是否正常，不正常的话清空buffer
             if (length < 0) { // || length > maxPackLength2) {
             buf.clear();
             out.write("ERROR!");
             break;
             // 读取正常的消息包，并写入输出流中，以便IoHandler进行处理
           } else if (length > packHeadLength && buf.remaining() >= length) {
             //完整的数据读取之后，就可以开始做你自己想做的操作了
     } else {
         // 如果消息包不完整
         // 将指针重新移动消息头的起始位置
         buf.reset();
         break;
     }
      }
     if (buf.hasRemaining()) { // 如果有剩余的数据，则放入Session中
        // 将数据移到buffer的最前面
              IoBuffer temp = IoBuffer.allocate(2048).setAutoExpand(
                         true);
        temp.put(buf);
        temp.flip();
        buf.clear();
        buf.put(temp);

     } else { // 如果数据已经处理完毕，进行清空
     buf.clear();
 }

为了便于操作，最好设置一个内部类： 

 private class Context {
         private final CharsetDecoder decoder;
         private IoBuffer buf;
         private int msgLength = 0;
         private int overflowPosition = 0;

         /**
          *
          *
          */
         private Context() {
             decoder = charset.newDecoder();
             buf = IoBuffer.allocate(80).setAutoExpand(true);
         }

         /**
          *
          *
          * @return CharsetDecoder
          */
         public CharsetDecoder getDecoder() {
             return decoder;
         }

         /**
          *
          *
          * @return IoBuffer
          */
         public IoBuffer getBuffer() {
             return buf;
         }

         /**
          *
          *
          * @return overflowPosition
          */
         public int getOverflowPosition() {
             return overflowPosition;
         }

         /**
          *
          *
          * @return matchCount
          */
         public int getMsgLength() {
             return msgLength;
         }

         /**
          *
          *
          * @param matchCount
          *            报文长度
          */
         public void setMsgLength(int msgLength) {
             this.msgLength = msgLength;
         }

         /**
          *
          *
          */
         public void reset() {
             this.buf.clear();
             this.overflowPosition = 0;
             this.msgLength = 0;
             this.decoder.reset();
         }

         /**
          *
          * @param in
          *            输入流
          */
         public void append(IoBuffer in) {
             getBuffer().put(in);

         }

     }