zipkin2.internal.Proto3Fields Maven / Gradle / Ivy
/*
* Copyright The OpenZipkin Authors
* SPDX-License-Identifier: Apache-2.0
*/
package zipkin2.internal;
import zipkin2.Endpoint;
import static zipkin2.internal.WriteBuffer.utf8SizeInBytes;
import static zipkin2.internal.WriteBuffer.varintSizeInBytes;
/**
* Everything here assumes the field numbers are less than 16, implying a 1 byte tag.
*/
//@Immutable
final class Proto3Fields {
/**
* Define the wire types, except the deprecated ones (groups)
*
* See https://developers.google.com/protocol-buffers/docs/encoding#structure
*/
static final int
WIRETYPE_VARINT = 0,
WIRETYPE_FIXED64 = 1,
WIRETYPE_LENGTH_DELIMITED = 2,
WIRETYPE_FIXED32 = 5;
static class Field {
final int fieldNumber;
final int wireType;
/**
* "Each key in the streamed message is a varint with the value {@code (field_number << 3) |
* wire_type}"
*
*
See https://developers.google.com/protocol-buffers/docs/encoding#structure
*/
final int key;
Field(int key) {
this(key >>> 3, key & (1 << 3) - 1, key);
}
Field(int fieldNumber, int wireType, int key) {
this.fieldNumber = fieldNumber;
this.wireType = wireType;
this.key = key;
}
static int fieldNumber(int key, int byteL) {
int fieldNumber = key >>> 3;
if (fieldNumber != 0) return fieldNumber;
throw new IllegalArgumentException("Malformed: fieldNumber was zero at byte " + byteL);
}
static int wireType(int key, int byteL) {
int wireType = key & (1 << 3) - 1;
if (wireType != 0 && wireType != 1 && wireType != 2 && wireType != 5) {
throw new IllegalArgumentException(
"Malformed: invalid wireType " + wireType + " at byte " + byteL);
}
return wireType;
}
static boolean skipValue(ReadBuffer buffer, int wireType) {
int remaining = buffer.available();
switch (wireType) {
case WIRETYPE_VARINT:
for (int i = 0; i < remaining; i++) {
if (buffer.readByte() >= 0) return true;
}
return false;
case WIRETYPE_FIXED64:
return buffer.skip(8) == 8;
case WIRETYPE_LENGTH_DELIMITED:
int length = buffer.readVarint32();
return buffer.skip(length) == length;
case WIRETYPE_FIXED32:
return buffer.skip(4) == 4;
default:
throw new IllegalArgumentException(
"Malformed: invalid wireType " + wireType + " at byte " + buffer.pos());
}
}
}
/**
* Leniently skips out null, but not on empty string, allowing tag "error" -> "" to serialize
* properly.
*
*
This won't result in empty {@link zipkin2.Span#name()} or {@link Endpoint#serviceName()}
* because in both cases constructors coerce empty values to null.
*/
static abstract class LengthDelimitedField extends Field {
LengthDelimitedField(int key) {
super(key);
assert wireType == WIRETYPE_LENGTH_DELIMITED;
}
final int sizeInBytes(T value) {
if (value == null) return 0;
int sizeOfValue = sizeOfValue(value);
return sizeOfLengthDelimitedField(sizeOfValue);
}
final void write(WriteBuffer b, T value) {
if (value == null) return;
int sizeOfValue = sizeOfValue(value);
b.writeByte(key);
b.writeVarint(sizeOfValue); // length prefix
writeValue(b, value);
}
/**
* Calling this after consuming the field key to ensures there's enough space for the data. Null
* is returned when the length prefix is zero.
*/
final T readLengthPrefixAndValue(ReadBuffer b) {
int length = b.readVarint32();
if (length == 0) return null;
return readValue(b, length);
}
abstract int sizeOfValue(T value);
abstract void writeValue(WriteBuffer b, T value);
/** @param length is greater than zero */
abstract T readValue(ReadBuffer b, int length);
}
static class BytesField extends LengthDelimitedField {
BytesField(int key) {
super(key);
}
@Override int sizeOfValue(byte[] bytes) {
return bytes.length;
}
@Override void writeValue(WriteBuffer b, byte[] bytes) {
b.write(bytes);
}
@Override byte[] readValue(ReadBuffer b, int length) {
return b.readBytes(length);
}
}
static class HexField extends LengthDelimitedField {
HexField(int key) {
super(key);
}
@Override int sizeOfValue(String hex) {
if (hex == null) return 0;
return hex.length() / 2;
}
@Override void writeValue(WriteBuffer b, String hex) {
// similar logic to okio.ByteString.decodeHex
for (int i = 0, length = hex.length(); i < length; i++) {
int d1 = decodeLowerHex(hex.charAt(i++)) << 4;
int d2 = decodeLowerHex(hex.charAt(i));
b.writeByte((byte) (d1 + d2));
}
}
static int decodeLowerHex(char c) {
if (c >= '0' && c <= '9') return c - '0';
if (c >= 'a' && c <= 'f') return c - 'a' + 10;
throw new AssertionError("not lowerHex " + c); // bug
}
@Override String readValue(ReadBuffer buffer, int length) {
return buffer.readBytesAsHex(length);
}
}
static class Utf8Field extends LengthDelimitedField {
Utf8Field(int key) {
super(key);
}
@Override int sizeOfValue(String utf8) {
return utf8 != null ? utf8SizeInBytes(utf8) : 0;
}
@Override void writeValue(WriteBuffer b, String utf8) {
b.writeUtf8(utf8);
}
@Override String readValue(ReadBuffer buffer, int length) {
return buffer.readUtf8(length);
}
}
static final class Fixed64Field extends Field {
Fixed64Field(int key) {
super(key);
assert wireType == WIRETYPE_FIXED64;
}
void write(WriteBuffer b, long number) {
if (number == 0) return;
b.writeByte(key);
b.writeLongLe(number);
}
int sizeInBytes(long number) {
if (number == 0) return 0;
return 1 + 8; // tag + 8 byte number
}
long readValue(ReadBuffer buffer) {
return buffer.readLongLe();
}
}
static class VarintField extends Field {
VarintField(int key) {
super(key);
assert wireType == WIRETYPE_VARINT;
}
int sizeInBytes(int number) {
return number != 0 ? 1 + varintSizeInBytes(number) : 0; // tag + varint
}
void write(WriteBuffer b, int number) {
if (number == 0) return;
b.writeByte(key);
b.writeVarint(number);
}
int sizeInBytes(long number) {
return number != 0 ? 1 + varintSizeInBytes(number) : 0; // tag + varint
}
void write(WriteBuffer b, long number) {
if (number == 0) return;
b.writeByte(key);
b.writeVarint(number);
}
}
static final class BooleanField extends Field {
BooleanField(int key) {
super(key);
assert wireType == WIRETYPE_VARINT;
}
int sizeInBytes(boolean bool) {
return bool ? 2 : 0; // tag + varint
}
void write(WriteBuffer b, boolean bool) {
if (!bool) return;
b.writeByte(key);
b.writeByte(1);
}
boolean read(ReadBuffer b) {
byte bool = b.readByte();
if (bool < 0 || bool > 1) {
throw new IllegalArgumentException("Malformed: invalid boolean value at byte " + b.pos());
}
return bool == 1;
}
}
// added for completion as later we will skip fields we don't use
static final class Fixed32Field extends Field {
Fixed32Field(int key) {
super(key);
assert wireType == WIRETYPE_FIXED32;
}
int sizeInBytes(int number) {
if (number == 0) return 0;
return 1 + 4; // tag + 4 byte number
}
}
static int sizeOfLengthDelimitedField(int sizeInBytes) {
return 1 + varintSizeInBytes(sizeInBytes) + sizeInBytes; // tag + len + bytes
}
}