io.datakernel.codec.binary.BinaryUtils Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of datakernel-codec Show documentation

Tools for encoding/decoding of primitives and objects.

There is a newer version: 3.1.0

/*
 * Copyright (C) 2015-2018 SoftIndex LLC.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package io.datakernel.codec.binary;

import io.datakernel.bytebuf.ByteBuf;
import io.datakernel.bytebuf.ByteBufPool;
import io.datakernel.codec.StructuredDecoder;
import io.datakernel.codec.StructuredEncoder;
import io.datakernel.common.parse.ParseException;

public final class BinaryUtils {

	public static  T decode(StructuredDecoder decoder, byte[] bytes) throws ParseException {
		return decode(decoder, ByteBuf.wrapForReading(bytes));
	}

	public static  T decode(StructuredDecoder decoder, ByteBuf buf) throws ParseException {
		try {
			BinaryStructuredInput in = new BinaryStructuredInput(buf);
			T result = decoder.decode(in);
			if (buf.readRemaining() != 0) {
				throw new ParseException("Byte buffer was not fully consumed when decoding");
			}
			return result;
		} finally {
			buf.recycle();
		}
	}

	public static  ByteBuf encode(StructuredEncoder encoder, T item) {
		BinaryStructuredOutput out = new BinaryStructuredOutput();
		encoder.encode(out, item);
		return out.getBuf();
	}

	public static  byte[] encodeAsArray(StructuredEncoder encoder, T item) {
		return encode(encoder, item).asArray();
	}

	public static  void encodeInto(StructuredEncoder encoder, T item, ByteBuf dest) {
		ByteBuf encoded = encode(encoder, item);
		dest.write(encoded.array(), encoded.head(), encoded.readRemaining());
	}

	public static  ByteBuf encodeWithSizePrefix(StructuredEncoder encoder, T item) {
		BinaryStructuredOutput out = new BinaryStructuredOutput();
		encoder.encode(out, item);
		ByteBuf buf = ByteBufPool.allocate(out.getBuf().readRemaining() + 5);
		buf.writeVarInt(out.getBuf().readRemaining());
		buf.write(out.getBuf().array(), out.getBuf().head(), out.getBuf().readRemaining());
		out.getBuf().recycle();
		return buf;
	}
}