org.apache.rocketmq.shaded.io.opentelemetry.sdk.trace.AttributesMap Maven / Gradle / Ivy
/*
* Copyright The OpenTelemetry Authors
* SPDX-License-Identifier: Apache-2.0
*/
package org.apache.rocketmq.shaded.io.opentelemetry.sdk.trace;
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.common.AttributesBuilder;
import org.apache.rocketmq.shaded.io.opentelemetry.sdk.internal.AttributeUtil;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;
import java.util.function.BiConsumer;
import javax.annotation.Nullable;
/**
* A map with a fixed capacity that drops attributes when the map gets full, and which truncates
* string and array string attribute values to the {@link #lengthLimit}.
*/
final class AttributesMap extends HashMap, Object> implements Attributes {
private static final long serialVersionUID = -5072696312123632376L;
private final long capacity;
private final int lengthLimit;
private int totalAddedValues = 0;
AttributesMap(long capacity, int lengthLimit) {
this.capacity = capacity;
this.lengthLimit = lengthLimit;
}
void put(AttributeKey key, T value) {
totalAddedValues++;
if (size() >= capacity && !containsKey(key)) {
return;
}
super.put(key, AttributeUtil.applyAttributeLengthLimit(value, lengthLimit));
}
int getTotalAddedValues() {
return totalAddedValues;
}
@SuppressWarnings("unchecked")
@Override
@Nullable
public T get(AttributeKey key) {
return (T) super.get(key);
}
@Override
public Map, Object> asMap() {
// Because Attributes is marked Immutable, IDEs may recognize this as redundant usage. However,
// this class is private and is actually mutable, so we need to wrap with unmodifiableMap
// anyways. We implement the immutable Attributes for this class to support the
// Attributes.builder().putAll usage - it is tricky but an implementation detail of this private
// class.
return Collections.unmodifiableMap(this);
}
@Override
public AttributesBuilder toBuilder() {
return Attributes.builder().putAll(this);
}
@Override
public void forEach(BiConsumer, ? super Object> action) {
// https://github.com/open-telemetry/opentelemetry-java/issues/4161
// Help out android desugaring by having an explicit call to HashMap.forEach, when forEach is
// just called through Attributes.forEach desugaring is unable to correctly handle it.
super.forEach(action);
}
@Override
public String toString() {
return "AttributesMap{"
+ "data="
+ super.toString()
+ ", capacity="
+ capacity
+ ", totalAddedValues="
+ totalAddedValues
+ '}';
}
Attributes immutableCopy() {
return Attributes.builder().putAll(this).build();
}
}