bakcode/netty/netty-intheader-codec/IntegerHeaderFrameDecoder.java

package net.dnio.codec.intheader;

import org.jboss.netty.buffer.ChannelBuffer;
import org.jboss.netty.channel.Channel;
import org.jboss.netty.channel.ChannelHandlerContext;
import org.jboss.netty.handler.codec.frame.FrameDecoder;

/**
 *
 * <p>
 *     因为Netty是异步发送和接收消息， 接收端每次只能接收1024字节的数据。
 *     因此超过1024字节的消息会被分割。利用次消息头则可以获得的完整的消息
 * </p>
 *
 * <pre>
 *
 * Created by IntelliJ IDEA.
 * User: zhenqin
 * Date: 13-6-4
 * Time: 下午8:10
 * To change this template use File | Settings | File Templates.
 *
 * </pre>
 *
 * @author zhenqin
 */
public class IntegerHeaderFrameDecoder extends FrameDecoder {


    public IntegerHeaderFrameDecoder() {
    }

    public IntegerHeaderFrameDecoder(boolean unfold) {
        super(unfold);
    }

    @Override
    public Object decode(ChannelHandlerContext ctx,
                            Channel channel,
                            ChannelBuffer buf) throws Exception {

        // Make sure if the length field was received.
        if (buf.readableBytes() < 4) {
            // 消息长度小于4字节，则是不正确的消息
            return null;
        }

        // 给当前Reader 的Buffer做标记
        buf.markReaderIndex();

        // 取得消息头，得到该消息的真实长度
        int length = buf.readInt();

        // 如果此次buffer读取的消息长度小于消息头的长度，则说明消息没有完成
        if (buf.readableBytes() < length) {
            // 重置消息的Reader的位置
            buf.resetReaderIndex();
            return null;
        }

        // There's enough bytes in the buffer. Read it.
        //直到所有的消息都存在于Buffered时， 开始读取
        ChannelBuffer frame = buf.readBytes(length);

        // Successfully decoded a frame.  Return the decoded frame.
        return frame;
    }
}