org.apache.rocketmq.shaded.io.opentelemetry.exporter.internal.marshal.MarshalerUtil Maven / Gradle / Ivy
/*
* Copyright The OpenTelemetry Authors
* SPDX-License-Identifier: Apache-2.0
*/
package org.apache.rocketmq.shaded.io.opentelemetry.exporter.internal.marshal;
import org.apache.rocketmq.shaded.io.opentelemetry.api.trace.SpanId;
import org.apache.rocketmq.shaded.io.opentelemetry.api.trace.TraceId;
import org.apache.rocketmq.shaded.io.opentelemetry.sdk.common.InstrumentationScopeInfo;
import org.apache.rocketmq.shaded.io.opentelemetry.sdk.resources.Resource;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.UncheckedIOException;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.Collection;
import java.util.IdentityHashMap;
import java.util.List;
import java.util.Map;
import java.util.function.Function;
import javax.annotation.Nullable;
/**
* Marshaler utilities.
*
* This class is internal and is hence not for public use. Its APIs are unstable and can change
* at any time.
*/
public final class MarshalerUtil {
private static final int TRACE_ID_VALUE_SIZE =
CodedOutputStream.computeLengthDelimitedFieldSize(TraceId.getLength() / 2);
private static final int SPAN_ID_VALUE_SIZE =
CodedOutputStream.computeLengthDelimitedFieldSize(SpanId.getLength() / 2);
private static final boolean JSON_AVAILABLE;
static {
boolean jsonAvailable = false;
try {
Class.forName("com.fasterxml.jackson.core.JsonFactory");
jsonAvailable = true;
} catch (ClassNotFoundException e) {
// Not available
}
JSON_AVAILABLE = jsonAvailable;
}
private static final byte[] EMPTY_BYTES = new byte[0];
/** Groups SDK items by resource and instrumentation scope. */
public static
Map>> groupByResourceAndScope(
Collection dataList,
Function getResource,
Function getInstrumentationScope,
Function createMarshaler) {
// expectedMaxSize of 8 means initial map capacity of 16 to match HashMap
IdentityHashMap>> result =
new IdentityHashMap<>(8);
for (T data : dataList) {
Map> scopeInfoListMap =
result.computeIfAbsent(getResource.apply(data), unused -> new IdentityHashMap<>(8));
List marshalerList =
scopeInfoListMap.computeIfAbsent(
getInstrumentationScope.apply(data), unused -> new ArrayList<>());
marshalerList.add(createMarshaler.apply(data));
}
return result;
}
/** Preserialize into JSON format. */
public static String preserializeJsonFields(Marshaler marshaler) {
if (!MarshalerUtil.JSON_AVAILABLE) {
return "";
}
ByteArrayOutputStream jsonBos = new ByteArrayOutputStream();
try {
marshaler.writeJsonTo(jsonBos);
} catch (IOException e) {
throw new UncheckedIOException(
"Serialization error, this is likely a bug in OpenTelemetry.", e);
}
// We effectively cache `writeTo`, however Jackson would not allow us to only write out
// fields
// which is what writeTo does. So we need to write to an object but skip the object start
// /
// end.
byte[] jsonBytes = jsonBos.toByteArray();
return new String(jsonBytes, 1, jsonBytes.length - 2, StandardCharsets.UTF_8);
}
/** Returns the size of a repeated fixed64 field. */
public static int sizeRepeatedFixed64(ProtoFieldInfo field, List values) {
return sizeRepeatedFixed64(field, values.size());
}
private static int sizeRepeatedFixed64(ProtoFieldInfo field, int numValues) {
if (numValues == 0) {
return 0;
}
int dataSize = WireFormat.FIXED64_SIZE * numValues;
int size = 0;
size += field.getTagSize();
size += CodedOutputStream.computeLengthDelimitedFieldSize(dataSize);
return size;
}
/**
* Returns the size of a repeated uint64 field.
*
* Packed repeated fields contain the tag, an integer representing the incoming payload size,
* and an actual payload of repeated varints.
*/
public static int sizeRepeatedUInt64(ProtoFieldInfo field, long[] values) {
if (values.length == 0) {
return 0;
}
int payloadSize = 0;
for (long v : values) {
payloadSize += CodedOutputStream.computeUInt64SizeNoTag(v);
}
// tag size + payload indicator size + actual payload size
return field.getTagSize() + CodedOutputStream.computeUInt32SizeNoTag(payloadSize) + payloadSize;
}
/** Returns the size of a repeated double field. */
public static int sizeRepeatedDouble(ProtoFieldInfo field, List values) {
// Same as fixed64.
return sizeRepeatedFixed64(field, values.size());
}
/** Returns the size of a repeated message field. */
public static int sizeRepeatedMessage(
ProtoFieldInfo field, T[] repeatedMessage) {
int size = 0;
int fieldTagSize = field.getTagSize();
for (Marshaler message : repeatedMessage) {
int fieldSize = message.getBinarySerializedSize();
size += fieldTagSize + CodedOutputStream.computeUInt32SizeNoTag(fieldSize) + fieldSize;
}
return size;
}
/** Returns the size of a repeated message field. */
public static int sizeRepeatedMessage(
ProtoFieldInfo field, List repeatedMessage) {
int size = 0;
int fieldTagSize = field.getTagSize();
for (Marshaler message : repeatedMessage) {
int fieldSize = message.getBinarySerializedSize();
size += fieldTagSize + CodedOutputStream.computeUInt32SizeNoTag(fieldSize) + fieldSize;
}
return size;
}
/** Returns the size of a message field. */
public static int sizeMessage(ProtoFieldInfo field, Marshaler message) {
int fieldSize = message.getBinarySerializedSize();
return field.getTagSize() + CodedOutputStream.computeUInt32SizeNoTag(fieldSize) + fieldSize;
}
/** Returns the size of a bool field. */
public static int sizeBool(ProtoFieldInfo field, boolean value) {
if (!value) {
return 0;
}
return field.getTagSize() + CodedOutputStream.computeBoolSizeNoTag(value);
}
/** Returns the size of a int64 field. */
public static int sizeInt64(ProtoFieldInfo field, long message) {
if (message == 0) {
return 0;
}
return field.getTagSize() + CodedOutputStream.computeInt64SizeNoTag(message);
}
/** Returns the size of a uint32 field. */
public static int sizeUInt32(ProtoFieldInfo field, int message) {
if (message == 0) {
return 0;
}
return field.getTagSize() + CodedOutputStream.computeUInt32SizeNoTag(message);
}
/** Returns the size of a sint32 field. */
public static int sizeSInt32(ProtoFieldInfo field, int message) {
if (message == 0) {
return 0;
}
return field.getTagSize() + CodedOutputStream.computeSInt32SizeNoTag(message);
}
/** Returns the size of a int32 field. */
public static int sizeInt32(ProtoFieldInfo field, int message) {
if (message == 0) {
return 0;
}
return field.getTagSize() + CodedOutputStream.computeInt32SizeNoTag(message);
}
/** Returns the size of a double field. */
public static int sizeDouble(ProtoFieldInfo field, double value) {
if (value == 0D) {
return 0;
}
return sizeDoubleOptional(field, value);
}
/** Returns the size of a double field. */
public static int sizeDoubleOptional(ProtoFieldInfo field, double value) {
return field.getTagSize() + CodedOutputStream.computeDoubleSizeNoTag(value);
}
/** Returns the size of a fixed64 field. */
public static int sizeFixed64(ProtoFieldInfo field, long value) {
if (value == 0L) {
return 0;
}
return sizeFixed64Optional(field, value);
}
/** Returns the size of a fixed64 field. */
public static int sizeFixed64Optional(ProtoFieldInfo field, long value) {
return field.getTagSize() + CodedOutputStream.computeFixed64SizeNoTag(value);
}
/** Returns the size of a fixed32 field. */
public static int sizeFixed32(ProtoFieldInfo field, int message) {
if (message == 0L) {
return 0;
}
return field.getTagSize() + CodedOutputStream.computeFixed32SizeNoTag(message);
}
/** Returns the size of a bytes field. */
public static int sizeBytes(ProtoFieldInfo field, byte[] message) {
if (message.length == 0) {
return 0;
}
return field.getTagSize() + CodedOutputStream.computeByteArraySizeNoTag(message);
}
/** Returns the size of a enum field. */
// Assumes OTLP always defines the first item in an enum with number 0, which it does and will.
public static int sizeEnum(ProtoFieldInfo field, ProtoEnumInfo enumValue) {
int number = enumValue.getEnumNumber();
if (number == 0) {
return 0;
}
return field.getTagSize() + CodedOutputStream.computeEnumSizeNoTag(number);
}
/** Returns the size of a trace_id field. */
public static int sizeTraceId(ProtoFieldInfo field, @Nullable String traceId) {
if (traceId == null) {
return 0;
}
return field.getTagSize() + TRACE_ID_VALUE_SIZE;
}
/** Returns the size of a span_id field. */
public static int sizeSpanId(ProtoFieldInfo field, @Nullable String spanId) {
if (spanId == null) {
return 0;
}
return field.getTagSize() + SPAN_ID_VALUE_SIZE;
}
/** Converts the string to utf8 bytes for encoding. */
public static byte[] toBytes(@Nullable String value) {
if (value == null || value.isEmpty()) {
return EMPTY_BYTES;
}
return value.getBytes(StandardCharsets.UTF_8);
}
private MarshalerUtil() {}
}