(#9297) add new instrumentation config otel.instrumentation.mdc.resource-attributes that will be used to configure Resource attributes to be added to logging map diagnostic context.

Author: cleverchuk

instrumentation/log4j/log4j-mdc-1.2/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/log4j/mdc/v1_2/LoggingEventInstrumentation.java

@@ -20,6 +20,7 @@
 import io.opentelemetry.javaagent.bootstrap.Java8BytecodeBridge;
 import io.opentelemetry.javaagent.extension.instrumentation.TypeInstrumentation;
 import io.opentelemetry.javaagent.extension.instrumentation.TypeTransformer;
+import io.opentelemetry.javaagent.tooling.ConfiguredResourceAttributesHolder;
 import net.bytebuddy.asm.Advice;
 import net.bytebuddy.description.type.TypeDescription;
 import net.bytebuddy.matcher.ElementMatcher;
@@ -79,6 +80,10 @@ public static void onExit(
           default:
             // do nothing
         }
+      } else if (ConfiguredResourceAttributesHolder.getAttributeValue(key) != null) {
+        if (value == null) {
+          value = ConfiguredResourceAttributesHolder.getAttributeValue(key);
+        }
       }
     }
   }

instrumentation/logback/logback-mdc-1.0/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/logback/mdc/v1_0/LoggingEventInstrumentation.java

@@ -26,6 +26,7 @@
 import io.opentelemetry.javaagent.bootstrap.Java8BytecodeBridge;
 import io.opentelemetry.javaagent.extension.instrumentation.TypeInstrumentation;
 import io.opentelemetry.javaagent.extension.instrumentation.TypeTransformer;
+import io.opentelemetry.javaagent.tooling.ConfiguredResourceAttributesHolder;
 import java.util.HashMap;
 import java.util.Map;
 import net.bytebuddy.asm.Advice;
@@ -79,6 +80,8 @@ public static void onExit(
         spanContextData.put(TRACE_ID, spanContext.getTraceId());
         spanContextData.put(SPAN_ID, spanContext.getSpanId());
         spanContextData.put(TRACE_FLAGS, spanContext.getTraceFlags().asHex());
+
+        spanContextData.putAll(ConfiguredResourceAttributesHolder.getResourceAttribute());
       }
 
       if (LogbackSingletons.addBaggage()) {

javaagent-tooling/src/main/java/io/opentelemetry/javaagent/tooling/AgentInstaller.java

@@ -41,6 +41,7 @@
 import io.opentelemetry.javaagent.tooling.muzzle.AgentTooling;
 import io.opentelemetry.javaagent.tooling.util.Trie;
 import io.opentelemetry.sdk.autoconfigure.AutoConfiguredOpenTelemetrySdk;
+import io.opentelemetry.sdk.autoconfigure.SdkAutoconfigureAccess;
 import io.opentelemetry.sdk.autoconfigure.internal.AutoConfigureUtil;
 import io.opentelemetry.sdk.autoconfigure.spi.ConfigProperties;
 import java.lang.instrument.Instrumentation;
@@ -125,6 +126,9 @@ private static void installBytebuddyAgent(
     copyNecessaryConfigToSystemProperties(sdkConfig);
 
     setBootstrapPackages(sdkConfig, extensionClassLoader);
+    ConfiguredResourceAttributesHolder.initialize(
+        SdkAutoconfigureAccess.getResourceAttribute(autoConfiguredSdk),
+        InstrumentationConfig.get());
 
     for (BeforeAgentListener agentListener :
         loadOrdered(BeforeAgentListener.class, extensionClassLoader)) {

javaagent-tooling/src/main/java/io/opentelemetry/javaagent/tooling/ConfiguredResourceAttributesHolder.java

@@ -0,0 +1,45 @@
+/*
+ * Copyright The OpenTelemetry Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+package io.opentelemetry.javaagent.tooling;
+
+import static io.opentelemetry.api.common.AttributeKey.stringKey;
+import static java.util.Collections.emptyList;
+
+import io.opentelemetry.api.common.Attributes;
+import io.opentelemetry.javaagent.bootstrap.internal.InstrumentationConfig;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import javax.annotation.Nullable;
+
+public final class ConfiguredResourceAttributesHolder {
+
+  private static final Map<String, String> resourceAttribute = new HashMap<>();
+
+  public static Map<String, String> getResourceAttribute() {
+    return resourceAttribute;
+  }
+
+  public static void initialize(Attributes resourceAttribute, InstrumentationConfig config) {
+    List<String> mdcResourceAttributes =
+        config.getList("otel.instrumentation.mdc.resource-attributes", emptyList());
+
+    for (String key : mdcResourceAttributes) {
+      String value = resourceAttribute.get(stringKey(key));
+      if (value != null) {
+        ConfiguredResourceAttributesHolder.resourceAttribute.put(
+            String.format("otel.resource.%s", key), value);
+      }
+    }
+  }
+
+  @Nullable
+  public static String getAttributeValue(String key) {
+    return resourceAttribute.get(String.format("otel.resource.%s", key));
+  }
+
+  private ConfiguredResourceAttributesHolder() {}
+}

javaagent-tooling/src/main/java/io/opentelemetry/sdk/autoconfigure/SdkAutoconfigureAccess.java

@@ -0,0 +1,16 @@
+/*
+ * Copyright The OpenTelemetry Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+package io.opentelemetry.sdk.autoconfigure;
+
+import io.opentelemetry.api.common.Attributes;
+
+public final class SdkAutoconfigureAccess {
+  public static Attributes getResourceAttribute(AutoConfiguredOpenTelemetrySdk sdk) {
+    return sdk.getResource().getAttributes();
+  }
+
+  private SdkAutoconfigureAccess() {}
+}

javaagent-tooling/src/test/java/io/opentelemetry/javaagent/tooling/ConfiguredResourceAttributesHolderTest.java

@@ -0,0 +1,50 @@
+/*
+ * Copyright The OpenTelemetry Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+package io.opentelemetry.javaagent.tooling;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertNull;
+import static org.mockito.ArgumentMatchers.anyList;
+import static org.mockito.ArgumentMatchers.eq;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
+
+import io.opentelemetry.api.common.AttributeKey;
+import io.opentelemetry.api.common.Attributes;
+import io.opentelemetry.javaagent.bootstrap.internal.InstrumentationConfig;
+import java.util.Collections;
+import org.junit.jupiter.api.Test;
+
+class ConfiguredResourceAttributesHolderTest {
+
+  @Test
+  void testGetAttributeValue() {
+    Attributes attributes = Attributes.builder().put("service.name", "test-service").build();
+
+    InstrumentationConfig config = mock(InstrumentationConfig.class);
+    when(config.getList(eq("otel.instrumentation.mdc.resource-attributes"), anyList()))
+        .thenReturn(Collections.singletonList("service.name"));
+
+    ConfiguredResourceAttributesHolder.initialize(attributes, config);
+    assertEquals(
+        "test-service", ConfiguredResourceAttributesHolder.getAttributeValue("service.name"));
+  }
+
+  @Test
+  void testGetAttributeValueWhenKeyIsNotString() {
+    Attributes attributes =
+        Attributes.builder()
+            .put(AttributeKey.stringArrayKey("items"), Collections.singletonList("test-item"))
+            .build();
+
+    InstrumentationConfig config = mock(InstrumentationConfig.class);
+    when(config.getList(eq("otel.instrumentation.mdc.resource-attributes"), anyList()))
+        .thenReturn(Collections.singletonList("items"));
+
+    ConfiguredResourceAttributesHolder.initialize(attributes, config);
+    assertNull(ConfiguredResourceAttributesHolder.getAttributeValue("items"));
+  }
+}