Refine the handling of OpenTelemetry resource attributes

OpenTelemetryResourceAttributes now has convenient APIs to construct,
modify, and merge attributes from various sources, including environment
variables and user-defined attribute maps.

Signed-off-by: Dmytro Nosan <[email protected]>

Author: nosan

spring-boot-project/spring-boot-actuator-autoconfigure/src/main/java/org/springframework/boot/actuate/autoconfigure/metrics/export/otlp/OtlpMetricsPropertiesConfigAdapter.java

@@ -77,12 +77,11 @@ public AggregationTemporality aggregationTemporality() {
 
 	@Override
 	public Map<String, String> resourceAttributes() {
-		Map<String, String> attributes = new OpenTelemetryResourceAttributes(
-				this.openTelemetryProperties.getResourceAttributes())
-			.asMap();
-		attributes.computeIfAbsent("service.name", (key) -> getApplicationName());
-		attributes.computeIfAbsent("service.group", (key) -> getApplicationGroup());
-		return Collections.unmodifiableMap(attributes);
+		OpenTelemetryResourceAttributes attributes = OpenTelemetryResourceAttributes.fromEnv();
+		attributes.putAll(this.openTelemetryProperties.getResourceAttributes());
+		attributes.putIfAbsent("service.name", this::getApplicationName);
+		attributes.putIfAbsent("service.group", this::getApplicationGroup);
+		return Collections.unmodifiableMap(attributes.asMap());
 	}
 
 	private String getApplicationName() {

spring-boot-project/spring-boot-actuator-autoconfigure/src/main/java/org/springframework/boot/actuate/autoconfigure/opentelemetry/OpenTelemetryAutoConfiguration.java

@@ -16,8 +16,6 @@
 
 package org.springframework.boot.actuate.autoconfigure.opentelemetry;
 
-import java.util.Map;
-
 import io.opentelemetry.api.OpenTelemetry;
 import io.opentelemetry.context.propagation.ContextPropagators;
 import io.opentelemetry.sdk.OpenTelemetrySdk;
@@ -76,11 +74,11 @@ Resource openTelemetryResource(Environment environment, OpenTelemetryProperties
 
 	private Resource toResource(Environment environment, OpenTelemetryProperties properties) {
 		ResourceBuilder builder = Resource.builder();
-		Map<String, String> attributes = new OpenTelemetryResourceAttributes(properties.getResourceAttributes())
-			.asMap();
-		attributes.computeIfAbsent("service.name", (key) -> getApplicationName(environment));
-		attributes.computeIfAbsent("service.group", (key) -> getApplicationGroup(environment));
-		attributes.forEach(builder::put);
+		OpenTelemetryResourceAttributes attributes = OpenTelemetryResourceAttributes.fromEnv();
+		attributes.putAll(properties.getResourceAttributes());
+		attributes.putIfAbsent("service.name", () -> getApplicationName(environment));
+		attributes.putIfAbsent("service.group", () -> getApplicationGroup(environment));
+		attributes.asMap().forEach(builder::put);
 		return builder.build();
 	}
 

spring-boot-project/spring-boot-actuator-autoconfigure/src/main/java/org/springframework/boot/actuate/autoconfigure/opentelemetry/OpenTelemetryResourceAttributes.java

@@ -18,100 +18,135 @@
 
 import java.io.ByteArrayOutputStream;
 import java.nio.charset.StandardCharsets;
-import java.util.Collections;
 import java.util.LinkedHashMap;
 import java.util.Map;
 import java.util.function.Function;
+import java.util.function.Supplier;
 
+import org.springframework.util.CollectionUtils;
 import org.springframework.util.StringUtils;
+import org.springframework.util.function.SupplierUtils;
 
 /**
- * OpenTelemetryResourceAttributes retrieves information from the
- * {@code OTEL_RESOURCE_ATTRIBUTES} and {@code OTEL_SERVICE_NAME} environment variables
- * and merges it with the resource attributes provided by the user.
- * <p>
- * <b>User-provided resource attributes take precedence.</b>
+ * {@link OpenTelemetryResourceAttributes} for managing OpenTelemetry resource attributes
+ * and provides convenient API to construct, modify, and merge attributes from different
+ * sources.
  * <p>
  * <a href= "https://opentelemetry.io/docs/specs/otel/resource/sdk/">OpenTelemetry
  * Resource Specification</a>
  *
  * @author Dmytro Nosan
  * @since 3.5.0
+ * @see #fromEnv()
  */
 public final class OpenTelemetryResourceAttributes {
 
-	private final Map<String, String> resourceAttributes;
-
-	private final Function<String, String> getEnv;
+	private final Map<String, String> attributes = new LinkedHashMap<>();
 
 	/**
-	 * Creates a new instance of {@link OpenTelemetryResourceAttributes}.
-	 * @param resourceAttributes user provided resource attributes to be used
+	 * Return Resource attributes as a Map.
+	 * @return the resource attributes as key-value pairs
 	 */
-	public OpenTelemetryResourceAttributes(Map<String, String> resourceAttributes) {
-		this(resourceAttributes, null);
+	public Map<String, String> asMap() {
+		return new LinkedHashMap<>(this.attributes);
 	}
 
 	/**
-	 * Creates a new {@link OpenTelemetryResourceAttributes} instance.
-	 * @param resourceAttributes user provided resource attributes to be used
-	 * @param getEnv a function to retrieve environment variables by name
+	 * Adds a name-value pair to the resource attributes. Both the name and value will be
+	 * trimmed.
+	 * @param name the attribute name to add, must not be null or empty
+	 * @param value the attribute value to add, must not be null
 	 */
-	OpenTelemetryResourceAttributes(Map<String, String> resourceAttributes, Function<String, String> getEnv) {
-		this.resourceAttributes = (resourceAttributes != null) ? resourceAttributes : Collections.emptyMap();
-		this.getEnv = (getEnv != null) ? getEnv : System::getenv;
+	public void put(String name, String value) {
+		if (StringUtils.hasText(name) && value != null) {
+			this.attributes.put(name.trim(), value.trim());
+		}
 	}
 
 	/**
-	 * Returns resource attributes by combining attributes from environment variables and
-	 * user-defined resource attributes. The final resource contains all attributes from
-	 * both sources.
+	 * Merge attributes with the provided resource attributes.
+	 * <p>
+	 * If a key exists in both, the value from provided resource takes precedence, even if
+	 * it is empty.
 	 * <p>
-	 * If a key exists in both environment variables and user-defined resources, the value
-	 * from the user-defined resource takes precedence, even if it is empty.
+	 * <b>Keys that are null or empty will be ignored, and all keys will be trimmed.</b>
 	 * <p>
-	 * <b>Null keys and values are ignored.</b>
-	 * @return the resource attributes
+	 * <b>Values that are null will be ignored, and all values will be trimmed.</b>
+	 * @param resourceAttributes resource attributes
 	 */
-	public Map<String, String> asMap() {
-		Map<String, String> attributes = getResourceAttributesFromEnv();
-		this.resourceAttributes.forEach((name, value) -> {
-			if (name != null && value != null) {
-				attributes.put(name, value);
-			}
-		});
-		return attributes;
+	public void putAll(Map<String, String> resourceAttributes) {
+		if (!CollectionUtils.isEmpty(resourceAttributes)) {
+			resourceAttributes.forEach(this::put);
+		}
+	}
+
+	/**
+	 * Adds a name-value pair to the resource attributes. Both the name and supplied value
+	 * will be trimmed.
+	 * @param name the attribute name to add, must not be null or empty
+	 * @param valueSupplier the attribute value supplier
+	 */
+	public void putIfAbsent(String name, Supplier<String> valueSupplier) {
+		if (!contains(name)) {
+			put(name, SupplierUtils.resolve(valueSupplier));
+		}
+	}
+
+	@Override
+	public boolean equals(Object obj) {
+		if (obj == null || getClass() != obj.getClass()) {
+			return false;
+		}
+		OpenTelemetryResourceAttributes that = (OpenTelemetryResourceAttributes) obj;
+		return this.attributes.equals(that.attributes);
+	}
+
+	@Override
+	public int hashCode() {
+		return this.attributes.hashCode();
+	}
+
+	@Override
+	public String toString() {
+		return this.attributes.toString();
 	}
 
 	/**
-	 * Parses resource attributes from the {@link System#getenv()}. This method fetches
-	 * attributes defined in the {@code OTEL_RESOURCE_ATTRIBUTES} and
-	 * {@code OTEL_SERVICE_NAME} environment variables and provides them as key-value
-	 * pairs.
+	 * Creates an {@link OpenTelemetryResourceAttributes} instance based on environment
+	 * variables. This method fetches attributes defined in the
+	 * {@code OTEL_RESOURCE_ATTRIBUTES} and {@code OTEL_SERVICE_NAME} environment
+	 * variables.
 	 * <p>
 	 * If {@code service.name} is also provided in {@code OTEL_RESOURCE_ATTRIBUTES}, then
 	 * {@code OTEL_SERVICE_NAME} takes precedence.
-	 * @return resource attributes
+	 * @return an {@link OpenTelemetryResourceAttributes}
 	 */
-	private Map<String, String> getResourceAttributesFromEnv() {
-		Map<String, String> attributes = new LinkedHashMap<>();
-		for (String attribute : StringUtils.tokenizeToStringArray(getEnv("OTEL_RESOURCE_ATTRIBUTES"), ",")) {
+	public static OpenTelemetryResourceAttributes fromEnv() {
+		return fromEnv(System::getenv);
+	}
+
+	static OpenTelemetryResourceAttributes fromEnv(Function<String, String> getEnv) {
+		OpenTelemetryResourceAttributes attributes = new OpenTelemetryResourceAttributes();
+		for (String attribute : StringUtils.tokenizeToStringArray(getEnv.apply("OTEL_RESOURCE_ATTRIBUTES"), ",")) {
 			int index = attribute.indexOf('=');
 			if (index > 0) {
 				String key = attribute.substring(0, index);
 				String value = attribute.substring(index + 1);
-				attributes.put(key.trim(), decode(value.trim()));
+				attributes.put(key, decode(value));
 			}
 		}
-		String otelServiceName = getEnv("OTEL_SERVICE_NAME");
+		String otelServiceName = getEnv.apply("OTEL_SERVICE_NAME");
 		if (otelServiceName != null) {
 			attributes.put("service.name", otelServiceName);
 		}
 		return attributes;
 	}
 
-	private String getEnv(String name) {
-		return this.getEnv.apply(name);
+	private boolean contains(String name) {
+		if (!StringUtils.hasText(name)) {
+			return false;
+		}
+		return this.attributes.containsKey(name.trim());
 	}
 
 	/**
@@ -122,7 +157,7 @@ private String getEnv(String name) {
 	 * @param value value to decode
 	 * @return the decoded string
 	 */
-	public static String decode(String value) {
+	private static String decode(String value) {
 		if (value.indexOf('%') < 0) {
 			return value;
 		}