org.apache.rocketmq.shaded.io.opentelemetry.exporter.internal.otlp.KeyValueMarshaler Maven / Gradle / Ivy
/*
* Copyright The OpenTelemetry Authors
* SPDX-License-Identifier: Apache-2.0
*/
package org.apache.rocketmq.shaded.io.opentelemetry.exporter.internal.otlp;
import org.apache.rocketmq.shaded.io.opentelemetry.api.common.AttributeKey;
import org.apache.rocketmq.shaded.io.opentelemetry.api.common.Attributes;
import org.apache.rocketmq.shaded.io.opentelemetry.api.internal.InternalAttributeKeyImpl;
import org.apache.rocketmq.shaded.io.opentelemetry.exporter.internal.marshal.CodedOutputStream;
import org.apache.rocketmq.shaded.io.opentelemetry.exporter.internal.marshal.Marshaler;
import org.apache.rocketmq.shaded.io.opentelemetry.exporter.internal.marshal.MarshalerUtil;
import org.apache.rocketmq.shaded.io.opentelemetry.exporter.internal.marshal.MarshalerWithSize;
import org.apache.rocketmq.shaded.io.opentelemetry.exporter.internal.marshal.Serializer;
import org.apache.rocketmq.shaded.io.opentelemetry.proto.common.v1.internal.AnyValue;
import org.apache.rocketmq.shaded.io.opentelemetry.proto.common.v1.internal.ArrayValue;
import org.apache.rocketmq.shaded.io.opentelemetry.proto.common.v1.internal.KeyValue;
import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.util.List;
import java.util.function.BiConsumer;
/**
* A Marshaler of {@link Attributes}.
*
* This class is internal and is hence not for public use. Its APIs are unstable and can change
* at any time.
*/
public final class KeyValueMarshaler extends MarshalerWithSize {
private static final byte[] EMPTY_BYTES = new byte[0];
private static final KeyValueMarshaler[] EMPTY_REPEATED = new KeyValueMarshaler[0];
/** Returns Marshalers for the given Attributes. */
public static KeyValueMarshaler[] createRepeated(Attributes attributes) {
if (attributes.isEmpty()) {
return EMPTY_REPEATED;
}
KeyValueMarshaler[] attributeMarshalers = new KeyValueMarshaler[attributes.size()];
attributes.forEach(
new BiConsumer, Object>() {
int index = 0;
@Override
public void accept(AttributeKey attributeKey, Object o) {
attributeMarshalers[index++] = KeyValueMarshaler.create(attributeKey, o);
}
});
return attributeMarshalers;
}
private final byte[] keyUtf8;
private final Marshaler value;
@SuppressWarnings("unchecked")
private static KeyValueMarshaler create(AttributeKey attributeKey, Object value) {
byte[] keyUtf8;
if (attributeKey.getKey().isEmpty()) {
keyUtf8 = EMPTY_BYTES;
} else if (attributeKey instanceof InternalAttributeKeyImpl) {
keyUtf8 = ((InternalAttributeKeyImpl) attributeKey).getKeyUtf8();
} else {
keyUtf8 = attributeKey.getKey().getBytes(StandardCharsets.UTF_8);
}
switch (attributeKey.getType()) {
case STRING:
return new KeyValueMarshaler(
keyUtf8, new StringAnyValueMarshaler(MarshalerUtil.toBytes((String) value)));
case LONG:
return new KeyValueMarshaler(keyUtf8, new Int64AnyValueMarshaler((long) value));
case BOOLEAN:
return new KeyValueMarshaler(keyUtf8, new BoolAnyValueMarshaler((boolean) value));
case DOUBLE:
return new KeyValueMarshaler(keyUtf8, new AnyDoubleFieldMarshaler((double) value));
case STRING_ARRAY:
return new KeyValueMarshaler(
keyUtf8,
new ArrayAnyValueMarshaler(ArrayValueMarshaler.createString((List) value)));
case LONG_ARRAY:
return new KeyValueMarshaler(
keyUtf8,
new ArrayAnyValueMarshaler(ArrayValueMarshaler.createInt64((List) value)));
case BOOLEAN_ARRAY:
return new KeyValueMarshaler(
keyUtf8,
new ArrayAnyValueMarshaler(ArrayValueMarshaler.createBool((List) value)));
case DOUBLE_ARRAY:
return new KeyValueMarshaler(
keyUtf8,
new ArrayAnyValueMarshaler(ArrayValueMarshaler.createDouble((List) value)));
}
// Error prone ensures the switch statement is complete, otherwise only can happen with
// unaligned versions which are not supported.
throw new IllegalArgumentException("Unsupported attribute type.");
}
private KeyValueMarshaler(byte[] keyUtf8, Marshaler value) {
super(calculateSize(keyUtf8, value));
this.keyUtf8 = keyUtf8;
this.value = value;
}
@Override
public void writeTo(Serializer output) throws IOException {
output.serializeString(KeyValue.KEY, keyUtf8);
output.serializeMessage(KeyValue.VALUE, value);
}
private static int calculateSize(byte[] keyUtf8, Marshaler value) {
int size = 0;
size += MarshalerUtil.sizeBytes(KeyValue.KEY, keyUtf8);
size += MarshalerUtil.sizeMessage(KeyValue.VALUE, value);
return size;
}
private static class BoolAnyValueMarshaler extends MarshalerWithSize {
private final boolean value;
BoolAnyValueMarshaler(boolean value) {
super(calculateSize(value));
this.value = value;
}
@Override
public void writeTo(Serializer output) throws IOException {
// Do not call serialize* method because we always have to write the message tag even if the
// value is empty since it's a oneof.
output.writeBool(AnyValue.BOOL_VALUE, value);
}
private static int calculateSize(boolean value) {
return AnyValue.BOOL_VALUE.getTagSize() + CodedOutputStream.computeBoolSizeNoTag(value);
}
}
private static class Int64AnyValueMarshaler extends MarshalerWithSize {
private final long value;
Int64AnyValueMarshaler(long value) {
super(calculateSize(value));
this.value = value;
}
@Override
public void writeTo(Serializer output) throws IOException {
// Do not call serialize* method because we always have to write the message tag even if the
// value is empty since it's a oneof.
output.writeInt64(AnyValue.INT_VALUE, value);
}
private static int calculateSize(long value) {
return AnyValue.INT_VALUE.getTagSize() + CodedOutputStream.computeInt64SizeNoTag(value);
}
}
private static class AnyDoubleFieldMarshaler extends MarshalerWithSize {
private final double value;
AnyDoubleFieldMarshaler(double value) {
super(calculateSize(value));
this.value = value;
}
@Override
public void writeTo(Serializer output) throws IOException {
// Do not call serialize* method because we always have to write the message tag even if the
// value is empty since it's a oneof.
output.writeDouble(AnyValue.DOUBLE_VALUE, value);
}
private static int calculateSize(double value) {
return AnyValue.DOUBLE_VALUE.getTagSize() + CodedOutputStream.computeDoubleSizeNoTag(value);
}
}
private static class ArrayAnyValueMarshaler extends MarshalerWithSize {
private final Marshaler value;
private ArrayAnyValueMarshaler(Marshaler value) {
super(calculateSize(value));
this.value = value;
}
@Override
public void writeTo(Serializer output) throws IOException {
output.serializeMessage(AnyValue.ARRAY_VALUE, value);
}
private static int calculateSize(Marshaler value) {
return MarshalerUtil.sizeMessage(AnyValue.ARRAY_VALUE, value);
}
}
private static class ArrayValueMarshaler extends MarshalerWithSize {
static ArrayValueMarshaler createString(List values) {
int len = values.size();
Marshaler[] marshalers = new StringAnyValueMarshaler[len];
for (int i = 0; i < len; i++) {
marshalers[i] = new StringAnyValueMarshaler(values.get(i).getBytes(StandardCharsets.UTF_8));
}
return new ArrayValueMarshaler(marshalers);
}
static ArrayValueMarshaler createBool(List values) {
int len = values.size();
Marshaler[] marshalers = new BoolAnyValueMarshaler[len];
for (int i = 0; i < len; i++) {
marshalers[i] = new BoolAnyValueMarshaler(values.get(i));
}
return new ArrayValueMarshaler(marshalers);
}
static ArrayValueMarshaler createInt64(List values) {
int len = values.size();
Marshaler[] marshalers = new Int64AnyValueMarshaler[len];
for (int i = 0; i < len; i++) {
marshalers[i] = new Int64AnyValueMarshaler(values.get(i));
}
return new ArrayValueMarshaler(marshalers);
}
static ArrayValueMarshaler createDouble(List values) {
int len = values.size();
Marshaler[] marshalers = new AnyDoubleFieldMarshaler[len];
for (int i = 0; i < len; i++) {
marshalers[i] = new AnyDoubleFieldMarshaler(values.get(i));
}
return new ArrayValueMarshaler(marshalers);
}
private final Marshaler[] values;
private ArrayValueMarshaler(Marshaler[] values) {
super(calculateSize(values));
this.values = values;
}
@Override
public void writeTo(Serializer output) throws IOException {
output.serializeRepeatedMessage(ArrayValue.VALUES, values);
}
private static int calculateSize(Marshaler[] values) {
return MarshalerUtil.sizeRepeatedMessage(ArrayValue.VALUES, values);
}
}
}