org.noear.solon.socketd.client.smartsocket.AioProtocol Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of solon.socketd.client.smartsocket Show documentation

There is a newer version: 2.6.0-M1

Show newest version

package org.noear.solon.socketd.client.smartsocket;

import org.noear.solon.core.message.Message;
import org.noear.solon.socketd.ProtocolManager;
import org.noear.solon.socketd.client.smartsocket.decoder.FixedLengthFrameDecoder;
import org.smartboot.socket.Protocol;
import org.smartboot.socket.transport.AioSession;

import java.nio.ByteBuffer;


/**
 * /date/xxx/sss\n...
 *
 * */
public class AioProtocol implements Protocol {
    public static final AioProtocol instance = new AioProtocol();

    @Override
    public Message decode(ByteBuffer buffer, AioSession session) {
        FixedLengthFrameDecoder decoder = session.getAttachment();

        if(decoder == null){
            if (buffer.remaining() < Integer.BYTES) {
                return null;
            }else{
                buffer.mark();
                decoder = new FixedLengthFrameDecoder(buffer.getInt());
                buffer.reset();
                session.setAttachment(decoder);
            }
        }

        if(decoder.read(buffer) == false){
            return null;
        }else{
            session.setAttachment(null);
            buffer = decoder.getBuffer();
            buffer.flip();
        }

        return ProtocolManager.decode(buffer);
    }
}