wip: fix for solving class loader error in compressionutil api.

Author: vasantteja

exporters/common/src/main/java/io/opentelemetry/exporter/internal/compression/CompressorUtil.java

@@ -24,7 +24,8 @@
  */
 public final class CompressorUtil {
 
-  private static final Map<String, Compressor> compressorRegistry = buildCompressorRegistry();
+  private static final Map<String, Compressor> compressorRegistry =
+      buildCompressorRegistry(CompressorUtil.class.getClassLoader());
 
   private CompressorUtil() {}
 
@@ -36,19 +37,36 @@ private CompressorUtil() {}
    */
   @Nullable
   public static Compressor validateAndResolveCompressor(String compressionMethod) {
-    Set<String> supportedEncodings = compressorRegistry.keySet();
-    Compressor compressor = compressorRegistry.get(compressionMethod);
+    return validateAndResolveCompressor(compressionMethod, null);
+  }
+
+  /**
+   * Validate that the {@code compressionMethod} is "none" or matches a registered compressor.
+   *
+   * @param compressionMethod the compression method to validate and resolve
+   * @param classLoader the class loader to use for loading SPI implementations, or null to use the
+   *     default
+   * @return {@code null} if {@code compressionMethod} is "none" or the registered compressor
+   * @throws IllegalArgumentException if no match is found
+   */
+  @Nullable
+  public static Compressor validateAndResolveCompressor(
+      String compressionMethod, @Nullable ClassLoader classLoader) {
+    Map<String, Compressor> registry =
+        classLoader == null ? compressorRegistry : buildCompressorRegistry(classLoader);
+
+    Set<String> supportedEncodings = registry.keySet();
+    Compressor compressor = registry.get(compressionMethod);
     checkArgument(
         "none".equals(compressionMethod) || compressor != null,
         "Unsupported compressionMethod. Compression method must be \"none\" or one of: "
             + supportedEncodings.stream().collect(joining(",", "[", "]")));
     return compressor;
   }
 
-  private static Map<String, Compressor> buildCompressorRegistry() {
+  private static Map<String, Compressor> buildCompressorRegistry(ClassLoader classLoader) {
     Map<String, Compressor> compressors = new HashMap<>();
-    for (CompressorProvider spi :
-        ServiceLoader.load(CompressorProvider.class, CompressorUtil.class.getClassLoader())) {
+    for (CompressorProvider spi : ServiceLoader.load(CompressorProvider.class, classLoader)) {
       Compressor compressor = spi.getInstance();
       compressors.put(compressor.getEncoding(), compressor);
     }

exporters/otlp/all/src/main/java/io/opentelemetry/exporter/otlp/http/logs/OtlpHttpLogRecordExporterBuilder.java

@@ -42,6 +42,7 @@ public final class OtlpHttpLogRecordExporterBuilder {
 
   private final HttpExporterBuilder<Marshaler> delegate;
   private MemoryMode memoryMode;
+  private ClassLoader serviceClassLoader = OtlpHttpLogRecordExporterBuilder.class.getClassLoader();
 
   OtlpHttpLogRecordExporterBuilder(HttpExporterBuilder<Marshaler> delegate, MemoryMode memoryMode) {
     this.delegate = delegate;
@@ -117,7 +118,8 @@ public OtlpHttpLogRecordExporterBuilder setEndpoint(String endpoint) {
    */
   public OtlpHttpLogRecordExporterBuilder setCompression(String compressionMethod) {
     requireNonNull(compressionMethod, "compressionMethod");
-    Compressor compressor = CompressorUtil.validateAndResolveCompressor(compressionMethod);
+    Compressor compressor =
+        CompressorUtil.validateAndResolveCompressor(compressionMethod, serviceClassLoader);
     delegate.setCompression(compressor);
     return this;
   }
@@ -251,6 +253,7 @@ public OtlpHttpLogRecordExporterBuilder setMemoryMode(MemoryMode memoryMode) {
    */
   public OtlpHttpLogRecordExporterBuilder setServiceClassLoader(ClassLoader serviceClassLoader) {
     requireNonNull(serviceClassLoader, "serviceClassLoader");
+    this.serviceClassLoader = serviceClassLoader;
     delegate.setServiceClassLoader(serviceClassLoader);
     return this;
   }

exporters/otlp/all/src/main/java/io/opentelemetry/exporter/otlp/http/metrics/OtlpHttpMetricExporterBuilder.java

@@ -54,6 +54,7 @@ public final class OtlpHttpMetricExporterBuilder {
   private AggregationTemporalitySelector aggregationTemporalitySelector;
   private DefaultAggregationSelector defaultAggregationSelector;
   private MemoryMode memoryMode;
+  private ClassLoader serviceClassLoader = OtlpHttpMetricExporterBuilder.class.getClassLoader();
 
   OtlpHttpMetricExporterBuilder(
       HttpExporterBuilder<Marshaler> delegate,
@@ -137,7 +138,8 @@ public OtlpHttpMetricExporterBuilder setEndpoint(String endpoint) {
    */
   public OtlpHttpMetricExporterBuilder setCompression(String compressionMethod) {
     requireNonNull(compressionMethod, "compressionMethod");
-    Compressor compressor = CompressorUtil.validateAndResolveCompressor(compressionMethod);
+    Compressor compressor =
+        CompressorUtil.validateAndResolveCompressor(compressionMethod, serviceClassLoader);
     delegate.setCompression(compressor);
     return this;
   }
@@ -310,6 +312,7 @@ public OtlpHttpMetricExporterBuilder setMemoryMode(MemoryMode memoryMode) {
    */
   public OtlpHttpMetricExporterBuilder setServiceClassLoader(ClassLoader serviceClassLoader) {
     requireNonNull(serviceClassLoader, "serviceClassLoader");
+    this.serviceClassLoader = serviceClassLoader;
     delegate.setServiceClassLoader(serviceClassLoader);
     return this;
   }

exporters/otlp/all/src/main/java/io/opentelemetry/exporter/otlp/http/trace/OtlpHttpSpanExporterBuilder.java

@@ -42,6 +42,7 @@ public final class OtlpHttpSpanExporterBuilder {
 
   private final HttpExporterBuilder<Marshaler> delegate;
   private MemoryMode memoryMode;
+  private ClassLoader serviceClassLoader = OtlpHttpSpanExporterBuilder.class.getClassLoader();
 
   OtlpHttpSpanExporterBuilder(HttpExporterBuilder<Marshaler> delegate, MemoryMode memoryMode) {
     this.delegate = delegate;
@@ -117,7 +118,8 @@ public OtlpHttpSpanExporterBuilder setEndpoint(String endpoint) {
    */
   public OtlpHttpSpanExporterBuilder setCompression(String compressionMethod) {
     requireNonNull(compressionMethod, "compressionMethod");
-    Compressor compressor = CompressorUtil.validateAndResolveCompressor(compressionMethod);
+    Compressor compressor =
+        CompressorUtil.validateAndResolveCompressor(compressionMethod, serviceClassLoader);
     delegate.setCompression(compressor);
     return this;
   }
@@ -252,6 +254,7 @@ public OtlpHttpSpanExporterBuilder setMemoryMode(MemoryMode memoryMode) {
    */
   public OtlpHttpSpanExporterBuilder setServiceClassLoader(ClassLoader serviceClassLoader) {
     requireNonNull(serviceClassLoader, "serviceClassLoader");
+    this.serviceClassLoader = serviceClassLoader;
     delegate.setServiceClassLoader(serviceClassLoader);
     return this;
   }

exporters/otlp/all/src/main/java/io/opentelemetry/exporter/otlp/logs/OtlpGrpcLogRecordExporterBuilder.java

@@ -51,6 +51,7 @@ public final class OtlpGrpcLogRecordExporterBuilder {
   // Visible for testing
   final GrpcExporterBuilder<Marshaler> delegate;
   private MemoryMode memoryMode;
+  private ClassLoader serviceClassLoader = OtlpGrpcLogRecordExporterBuilder.class.getClassLoader();
 
   OtlpGrpcLogRecordExporterBuilder(GrpcExporterBuilder<Marshaler> delegate, MemoryMode memoryMode) {
     this.delegate = delegate;
@@ -150,7 +151,8 @@ public OtlpGrpcLogRecordExporterBuilder setEndpoint(String endpoint) {
    */
   public OtlpGrpcLogRecordExporterBuilder setCompression(String compressionMethod) {
     requireNonNull(compressionMethod, "compressionMethod");
-    Compressor compressor = CompressorUtil.validateAndResolveCompressor(compressionMethod);
+    Compressor compressor =
+        CompressorUtil.validateAndResolveCompressor(compressionMethod, serviceClassLoader);
     delegate.setCompression(compressor);
     return this;
   }
@@ -280,6 +282,7 @@ public OtlpGrpcLogRecordExporterBuilder setMemoryMode(MemoryMode memoryMode) {
    */
   public OtlpGrpcLogRecordExporterBuilder setServiceClassLoader(ClassLoader serviceClassLoader) {
     requireNonNull(serviceClassLoader, "serviceClassLoader");
+    this.serviceClassLoader = serviceClassLoader;
     delegate.setServiceClassLoader(serviceClassLoader);
     return this;
   }

exporters/otlp/all/src/main/java/io/opentelemetry/exporter/otlp/metrics/OtlpGrpcMetricExporterBuilder.java

@@ -62,6 +62,7 @@ public final class OtlpGrpcMetricExporterBuilder {
   private AggregationTemporalitySelector aggregationTemporalitySelector;
   private DefaultAggregationSelector defaultAggregationSelector;
   private MemoryMode memoryMode;
+  private ClassLoader serviceClassLoader = OtlpGrpcMetricExporterBuilder.class.getClassLoader();
 
   OtlpGrpcMetricExporterBuilder(
       GrpcExporterBuilder<Marshaler> delegate,
@@ -169,7 +170,8 @@ public OtlpGrpcMetricExporterBuilder setEndpoint(String endpoint) {
    */
   public OtlpGrpcMetricExporterBuilder setCompression(String compressionMethod) {
     requireNonNull(compressionMethod, "compressionMethod");
-    Compressor compressor = CompressorUtil.validateAndResolveCompressor(compressionMethod);
+    Compressor compressor =
+        CompressorUtil.validateAndResolveCompressor(compressionMethod, serviceClassLoader);
     delegate.setCompression(compressor);
     return this;
   }
@@ -338,6 +340,7 @@ public OtlpGrpcMetricExporterBuilder setMemoryMode(MemoryMode memoryMode) {
    */
   public OtlpGrpcMetricExporterBuilder setServiceClassLoader(ClassLoader serviceClassLoader) {
     requireNonNull(serviceClassLoader, "serviceClassLoader");
+    this.serviceClassLoader = serviceClassLoader;
     delegate.setServiceClassLoader(serviceClassLoader);
     return this;
   }

exporters/otlp/all/src/main/java/io/opentelemetry/exporter/otlp/trace/OtlpGrpcSpanExporterBuilder.java

@@ -47,6 +47,7 @@ public final class OtlpGrpcSpanExporterBuilder {
   // Visible for testing
   final GrpcExporterBuilder<Marshaler> delegate;
   private MemoryMode memoryMode;
+  private ClassLoader serviceClassLoader = OtlpGrpcSpanExporterBuilder.class.getClassLoader();
 
   OtlpGrpcSpanExporterBuilder(GrpcExporterBuilder<Marshaler> delegate, MemoryMode memoryMode) {
     this.delegate = delegate;
@@ -146,7 +147,8 @@ public OtlpGrpcSpanExporterBuilder setEndpoint(String endpoint) {
    */
   public OtlpGrpcSpanExporterBuilder setCompression(String compressionMethod) {
     requireNonNull(compressionMethod, "compressionMethod");
-    Compressor compressor = CompressorUtil.validateAndResolveCompressor(compressionMethod);
+    Compressor compressor =
+        CompressorUtil.validateAndResolveCompressor(compressionMethod, serviceClassLoader);
     delegate.setCompression(compressor);
     return this;
   }
@@ -277,6 +279,7 @@ public OtlpGrpcSpanExporterBuilder setMemoryMode(MemoryMode memoryMode) {
    */
   public OtlpGrpcSpanExporterBuilder setServiceClassLoader(ClassLoader serviceClassLoader) {
     requireNonNull(serviceClassLoader, "serviceClassLoader");
+    this.serviceClassLoader = serviceClassLoader;
     delegate.setServiceClassLoader(serviceClassLoader);
     return this;
   }

exporters/otlp/profiles/src/main/java/io/opentelemetry/exporter/otlp/profiles/OtlpGrpcProfilesExporterBuilder.java

@@ -40,6 +40,7 @@ public final class OtlpGrpcProfilesExporterBuilder {
   private static final String DEFAULT_ENDPOINT_URL = "http://localhost:4317";
   private static final URI DEFAULT_ENDPOINT = URI.create(DEFAULT_ENDPOINT_URL);
   private static final long DEFAULT_TIMEOUT_SECS = 10;
+  private ClassLoader serviceClassLoader = OtlpGrpcProfilesExporterBuilder.class.getClassLoader();
 
   // TODO maybe make more efficient by adding support for MEMORY_MODE
 
@@ -139,7 +140,8 @@ public OtlpGrpcProfilesExporterBuilder setEndpoint(String endpoint) {
    */
   public OtlpGrpcProfilesExporterBuilder setCompression(String compressionMethod) {
     requireNonNull(compressionMethod, "compressionMethod");
-    Compressor compressor = CompressorUtil.validateAndResolveCompressor(compressionMethod);
+    Compressor compressor =
+        CompressorUtil.validateAndResolveCompressor(compressionMethod, serviceClassLoader);
     delegate.setCompression(compressor);
     return this;
   }
@@ -207,6 +209,7 @@ public OtlpGrpcProfilesExporterBuilder setRetryPolicy(@Nullable RetryPolicy retr
   /** Set the {@link ClassLoader} used to load the sender API. */
   public OtlpGrpcProfilesExporterBuilder setServiceClassLoader(ClassLoader serviceClassLoader) {
     requireNonNull(serviceClassLoader, "serviceClassLoader");
+    this.serviceClassLoader = serviceClassLoader;
     delegate.setServiceClassLoader(serviceClassLoader);
     return this;
   }