org.redkale.convert.ext.ByteBufferSimpledCoder Maven / Gradle / Ivy

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package org.redkale.convert.ext;

import java.nio.ByteBuffer;
import org.redkale.convert.Reader;
import org.redkale.convert.SimpledCoder;
import org.redkale.convert.Writer;

/**
 * ByteBuffer 的SimpledCoder实现
 *
 * 
 * 详情见: https://redkale.org
 *
 * @author zhangjx
 * @param  Reader输入的子类型
 * @param  Writer输出的子类型
 */
public final class ByteBufferSimpledCoder extends SimpledCoder {

    public static final ByteBufferSimpledCoder instance = new ByteBufferSimpledCoder();

    @Override
    public void convertTo(W out, ByteBuffer value) {
        if (value == null) {
            out.writeNull();
            return;
        }
        if (out.writeArrayB(value.remaining(), ByteSimpledCoder.instance, value) < 0) {
            boolean flag = false;
            for (byte v : value.array()) {
                if (flag) out.writeArrayMark();
                out.writeByte(v);
                flag = true;
            }
        }
        out.writeArrayE();
    }

    @Override
    public ByteBuffer convertFrom(R in) {
        int len = in.readArrayB(null, ByteSimpledCoder.instance);
        int contentLength = -1;
        if (len == Reader.SIGN_NULL) return null;
        if (len == Reader.SIGN_NOLENBUTBYTES) {
            contentLength = in.readMemberContentLength(null, ByteSimpledCoder.instance);
            len = Reader.SIGN_NOLENGTH;
        }
        if (len == Reader.SIGN_NOLENGTH) {
            int size = 0;
            byte[] data = new byte[8];
            int startPosition = in.position();
            while (in.hasNext(startPosition, contentLength)) {
                if (size >= data.length) {
                    byte[] newdata = new byte[data.length + 4];
                    System.arraycopy(data, 0, newdata, 0, size);
                    data = newdata;
                }
                data[size++] = in.readByte();
            }
            in.readArrayE();
            return ByteBuffer.wrap(data, 0, size);
        } else {
            byte[] values = new byte[len];
            for (int i = 0; i < values.length; i++) {
                values[i] = in.readByte();
            }
            in.readArrayE();
            return ByteBuffer.wrap(values);
        }
    }

}