com.hedera.hapi.node.base.codec.TokenBalanceProtoCodec Maven / Gradle / Ivy
The newest version!
package com.hedera.hapi.node.base.codec;
import com.hedera.pbj.runtime.*;
import com.hedera.pbj.runtime.io.*;
import com.hedera.pbj.runtime.io.buffer.*;
import com.hedera.pbj.runtime.io.stream.EOFException;
import java.io.IOException;
import java.nio.*;
import java.nio.charset.*;
import java.util.*;
import edu.umd.cs.findbugs.annotations.NonNull;
import com.hedera.hapi.node.base.TokenBalance;
import com.hedera.hapi.node.base.*;
import com.hedera.hapi.node.base.schema.*;
import static com.hedera.hapi.node.base.schema.TokenBalanceSchema.*;
import static com.hedera.pbj.runtime.ProtoWriterTools.*;
import static com.hedera.pbj.runtime.ProtoParserTools.*;
import static com.hedera.pbj.runtime.ProtoConstants.*;
/**
* Protobuf Codec for TokenBalance model object. Generated based on protobuf schema.
*/
public final class TokenBalanceProtoCodec implements Codec {
/**
* Parses a TokenBalance object from ProtoBuf bytes in a {@link ReadableSequentialData}. Throws if in strict mode ONLY.
*
* @param input The data input to parse data from, it is assumed to be in a state ready to read with position at start
* of data to read and limit set at the end of data to read. The data inputs limit will be changed by this
* method. If there are no bytes remaining in the data input,
* then the method also returns immediately.
* @param strictMode when {@code true}, the parser errors out on unknown fields; otherwise they'll be simply skipped.
* @param maxDepth a ParseException will be thrown if the depth of nested messages exceeds the maxDepth value.
* @return Parsed TokenBalance model object or null if data input was null or empty
* @throws ParseException If parsing fails
*/
public @NonNull TokenBalance parse(
@NonNull final ReadableSequentialData input,
final boolean strictMode,
final int maxDepth) throws ParseException {
if (maxDepth < 0) {
throw new ParseException("Reached maximum allowed depth of nested messages");
}
try {
// -- TEMP STATE FIELDS --------------------------------------
TokenID temp_tokenId = null;
long temp_balance = 0;
int temp_decimals = 0;
// -- PARSE LOOP ---------------------------------------------
// Continue to parse bytes out of the input stream until we get to the end.
while (input.hasRemaining()) {
// Note: ReadableStreamingData.hasRemaining() won't flip to false
// until the end of stream is actually hit with a read operation.
// So we catch this exception here and **only** here, because an EOFException
// anywhere else suggests that we're processing malformed data and so
// we must re-throw the exception then.
final int tag;
try {
// Read the "tag" byte which gives us the field number for the next field to read
// and the wire type (way it is encoded on the wire).
tag = input.readVarInt(false);
} catch (EOFException e) {
// There's no more fields. Stop the parsing loop.
break;
}
// The field is the top 5 bits of the byte. Read this off
final int field = tag >>> TAG_FIELD_OFFSET;
// Ask the Schema to inform us what field this represents.
final var f = getField(field);
// Given the wire type and the field type, parse the field
switch (tag) {
case 10 /* type=2 [MESSAGE] field=1 [tokenId] */ -> {
final var messageLength = input.readVarInt(false);
final TokenID value;
if (messageLength == 0) {
value = TokenID.DEFAULT;
} else {
if (messageLength > 2097152) {
throw new ParseException("tokenId size " + messageLength + " is greater than max " + 2097152);
}
final var limitBefore = input.limit();
// Make sure that we have enough bytes in the message
// to read the subObject.
// If the buffer is truncated on the boundary of a subObject,
// we will not throw.
final var startPos = input.position();
try {
if ((startPos + messageLength) > limitBefore) {
throw new BufferUnderflowException();
}
input.limit(startPos + messageLength);
value = TokenID.PROTOBUF.parse(input, strictMode, maxDepth - 1);
// Make sure we read the full number of bytes. for the types
if ((startPos + messageLength) != input.position()) {
throw new BufferOverflowException();
}
} finally {
input.limit(limitBefore);
}
}
temp_tokenId = value;
}
case 16 /* type=0 [UINT64] field=2 [balance] */ -> {
final var value = readUint64(input);
temp_balance = value;
}
case 24 /* type=0 [UINT32] field=3 [decimals] */ -> {
final var value = readUint32(input);
temp_decimals = value;
}
default -> {
// The wire type is the bottom 3 bits of the byte. Read that off
final int wireType = tag & TAG_WIRE_TYPE_MASK;
// handle error cases here, so we do not do if statements in normal loop
// Validate the field number is valid (must be > 0)
if (field == 0) {
throw new IOException("Bad protobuf encoding. We read a field value of "
+ field);
}
// Validate the wire type is valid (must be >=0 && <= 5).
// Otherwise we cannot parse this.
// Note: it is always >= 0 at this point (see code above where it is defined).
if (wireType > 5) {
throw new IOException("Cannot understand wire_type of " + wireType);
}
// It may be that the parser subclass doesn't know about this field
if (f == null) {
if (strictMode) {
// Since we are parsing is strict mode, this is an exceptional condition.
throw new UnknownFieldException(field);
} else {
// We just need to read off the bytes for this field to skip it
// and move on to the next one.
skipField(input, ProtoConstants.get(wireType), 2097152);
}
} else {
throw new IOException("Bad tag [" + tag + "], field [" + field
+ "] wireType [" + wireType + "]");
}
}
}
}
return new TokenBalance(temp_tokenId, temp_balance, temp_decimals);
} catch (final Exception anyException) {
if (anyException instanceof ParseException parseException) {
throw parseException;
}
throw new ParseException(anyException);
}
}
/**
* Write out a TokenBalance model to output stream in protobuf format.
*
* @param data The input model data to write
* @param out The output stream to write to
* @throws IOException If there is a problem writing
*/
public void write(@NonNull TokenBalance data, @NonNull final WritableSequentialData out) throws IOException {
// [1] - tokenId
writeMessage(out, TOKEN_ID, data.tokenId(), com.hedera.hapi.node.base.TokenID.PROTOBUF::write, com.hedera.hapi.node.base.TokenID.PROTOBUF::measureRecord);
// [2] - balance
writeLong(out, BALANCE, data.balance(), true);
// [3] - decimals
writeInteger(out, DECIMALS, data.decimals(), true);
}
/**
* Reads from this data input the length of the data within the input. The implementation may
* read all the data, or just some special serialized data, as needed to find out the length of
* the data.
*
* @param input The input to use
* @return The length of the data item in the input
* @throws ParseException If parsing fails
*/
public int measure(@NonNull final ReadableSequentialData input) throws ParseException {
final var start = input.position();
parse(input);
final var end = input.position();
return (int)(end - start);
}
/**
* Compute number of bytes that would be written when calling {@code write()} method.
*
* @param data The input model data to measure write bytes for
* @return The length in bytes that would be written
*/
public int measureRecord(TokenBalance data) {
int size = 0;
// [1] - tokenId
size += sizeOfMessage(TOKEN_ID, data.tokenId(), com.hedera.hapi.node.base.TokenID.PROTOBUF::measureRecord);
// [2] - balance
size += sizeOfLong(BALANCE, data.balance(), true);
// [3] - decimals
size += sizeOfInteger(DECIMALS, data.decimals(), true);
return size;
}
/**
* Compares the given item with the bytes in the input, and returns false if it determines that
* the bytes in the input could not be equal to the given item. Sometimes we need to compare an
* item in memory with serialized bytes and don't want to incur the cost of deserializing the
* entire object, when we could have determined the bytes do not represent the same object very
* cheaply and quickly.
*
* @param item The item to compare. Cannot be null.
* @param input The input with the bytes to compare
* @return true if the bytes represent the item, false otherwise.
* @throws ParseException If parsing fails
*/
public boolean fastEquals(@NonNull TokenBalance item, @NonNull final ReadableSequentialData input) throws ParseException {
return item.equals(parse(input));
}
}