com.firefly.net.tcp.codec.common.decode.CharParser Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of firefly-nettool Show documentation

There is a newer version: 5.0.0-dev6

package com.firefly.net.tcp.codec.common.decode;

import com.firefly.net.tcp.codec.AbstractByteBufferMessageHandler;

import java.nio.CharBuffer;
import java.nio.charset.Charset;
import java.nio.charset.CharsetDecoder;
import java.nio.charset.CoderResult;

public class CharParser extends AbstractByteBufferMessageHandler {

    private CharsetDecoder decoder;

    public CharParser() {
        this("UTF-8");
    }

    public CharParser(String charset) {
        decoder = Charset.forName(charset).newDecoder();
    }

    @Override
    protected void parse() {
        CharBuffer charBuff = allocate();
        while (buffer.hasRemaining()) {
            CoderResult r = decoder.decode(buffer, charBuff, false);
            charBuff.flip();
            if (r.isUnderflow()) {
                if (buffer.hasRemaining()) {
                    buffer = buffer.slice();
                }
                if (charBuff.hasRemaining()) {
                    action.call(charBuff.toString());
                }
                break;
            } else if (r.isOverflow()) {
                action.call(charBuff.toString());
                charBuff = allocate();
            }
        }
    }

    private CharBuffer allocate() {
        int expectedLength = (int) (buffer.remaining() * decoder.averageCharsPerByte()) + 1;
        return CharBuffer.allocate(expectedLength);
    }

}