Swap original StrictContextStorage with simple wrapper to avoid missing reference issue on native-image

Author: mcculls

dd-java-agent/agent-otel/otel-bootstrap/build.gradle

@@ -56,6 +56,7 @@ tasks.named('compileJava') {
 shadowJar {
   dependencies deps.excludeShared
 
+  exclude 'io/opentelemetry/context/StrictContextStorage*'
   exclude 'io/opentelemetry/context/internal/shaded/**'
 
   relocate 'io.opentelemetry', 'datadog.trace.bootstrap.otel'

dd-java-agent/agent-otel/otel-bootstrap/src/main/java/datadog/trace/bootstrap/otel/context/StrictContextStorage.java

@@ -0,0 +1,34 @@
+package datadog.trace.bootstrap.otel.context;
+
+import io.opentelemetry.context.Context;
+import io.opentelemetry.context.ContextStorage;
+import io.opentelemetry.context.Scope;
+
+/**
+ * Replaces original class with a simple wrapper to avoid missing reference issue on native-image.
+ *
+ * <p>The original class is only used for testing purposes when a particular property is set, but
+ * native-image follows the reference in {@code LazyStorage} and attempts to load everything it
+ * touches, including some types we are not embedding. This simple replacement fixes this issue.
+ */
+final class StrictContextStorage implements ContextStorage {
+  private final ContextStorage delegate;
+
+  static StrictContextStorage create(ContextStorage delegate) {
+    return new StrictContextStorage(delegate);
+  }
+
+  public StrictContextStorage(ContextStorage delegate) {
+    this.delegate = delegate;
+  }
+
+  @Override
+  public Scope attach(Context context) {
+    return delegate.attach(context);
+  }
+
+  @Override
+  public Context current() {
+    return delegate.current();
+  }
+}