Add compression configuration to ZipkinSpanExporterBuilder

open-telemetry · Oct 5, 2022 · 55c1d0e · 55c1d0e
1 parent d5ec773
commit 55c1d0e
Show file tree

Hide file tree

Showing 3 changed files with 76 additions and 4 deletions.
diff --git a/docs/apidiffs/current_vs_latest/opentelemetry-exporter-zipkin.txt b/docs/apidiffs/current_vs_latest/opentelemetry-exporter-zipkin.txt
@@ -1,2 +1,4 @@
 Comparing source compatibility of  against 
-No changes.
+***  MODIFIED CLASS: PUBLIC FINAL io.opentelemetry.exporter.zipkin.ZipkinSpanExporterBuilder  (not serializable)
+	===  CLASS FILE FORMAT VERSION: 52.0 <- 52.0
+	+++  NEW METHOD: PUBLIC(+) io.opentelemetry.exporter.zipkin.ZipkinSpanExporterBuilder setCompression(java.lang.String)
diff --git a/...ters/zipkin/src/main/java/io/opentelemetry/exporter/zipkin/ZipkinSpanExporterBuilder.java b/...ters/zipkin/src/main/java/io/opentelemetry/exporter/zipkin/ZipkinSpanExporterBuilder.java
@@ -26,11 +26,12 @@ public final class ZipkinSpanExporterBuilder {
   private Supplier<InetAddress> localIpAddressSupplier = LocalInetAddressSupplier.getInstance();
   @Nullable private Sender sender;
   private String endpoint = ZipkinSpanExporter.DEFAULT_ENDPOINT;
+  private boolean compressionEnabled = false;
   private long readTimeoutMillis = TimeUnit.SECONDS.toMillis(10);
   private MeterProvider meterProvider = MeterProvider.noop();
 
   /**
-   * Sets the Zipkin sender. Implements the client side of the span transport. A {@link
+   * Sets the Zipkin sender. Implements the client side of the span transport. An {@link
    * OkHttpSender} is a good default.
    *
    * <p>The {@link Sender#close()} method will be called when the exporter is shut down.
@@ -75,7 +76,7 @@ public ZipkinSpanExporterBuilder setLocalIpAddressSupplier(Supplier<InetAddress>
   }
 
   /**
-   * Sets the zipkin endpoint. This will use the endpoint to assign a {@link OkHttpSender} instance
+   * Sets the zipkin endpoint. This will use the endpoint to assign an {@link OkHttpSender} instance
    * to this builder.
    *
    * @param endpoint The Zipkin endpoint URL, ex. "http://zipkinhost:9411/api/v2/spans".
@@ -88,6 +89,26 @@ public ZipkinSpanExporterBuilder setEndpoint(String endpoint) {
     return this;
   }
 
+  /**
+   * Sets the method used to compress payloads. If unset, compression is enabled. Currently
+   * supported compression methods include "gzip" and "none".
+   *
+   * <p>The compression method is ignored when a custom Zipkin sender is set via {@link
+   * #setSender(Sender)}.
+   *
+   * @param compressionMethod The compression method, ex. "gzip".
+   * @return this.
+   * @see OkHttpSender
+   */
+  public ZipkinSpanExporterBuilder setCompression(String compressionMethod) {
+    requireNonNull(compressionMethod, "compressionMethod");
+    checkArgument(
+        compressionMethod.equals("gzip") || compressionMethod.equals("none"),
+        "Unsupported compression method. Supported compression methods include: gzip, none.");
+    this.compressionEnabled = compressionMethod.equals("gzip");
+    return this;
+  }
+
   /**
    * Sets the maximum time to wait for the export of a batch of spans. If unset, defaults to 10s.
    *
@@ -135,7 +156,11 @@ public ZipkinSpanExporter build() {
     Sender sender = this.sender;
     if (sender == null) {
       sender =
-          OkHttpSender.newBuilder().endpoint(endpoint).readTimeout((int) readTimeoutMillis).build();
+          OkHttpSender.newBuilder()
+              .endpoint(endpoint)
+              .compressionEnabled(compressionEnabled)
+              .readTimeout((int) readTimeoutMillis)
+              .build();
     }
     OtelToZipkinSpanTransformer transformer =
         OtelToZipkinSpanTransformer.create(localIpAddressSupplier);

diff --git a/exporters/zipkin/src/test/java/io/opentelemetry/exporter/zipkin/ZipkinSpanExporterTest.java b/exporters/zipkin/src/test/java/io/opentelemetry/exporter/zipkin/ZipkinSpanExporterTest.java
@@ -136,6 +136,10 @@ void invalidConfig() {
         .isInstanceOf(NullPointerException.class)
         .hasMessage("endpoint");
 
+    assertThatThrownBy(() -> ZipkinSpanExporter.builder().setCompression(null))
+        .isInstanceOf(NullPointerException.class)
+        .hasMessage("compressionMethod");
+
     assertThatThrownBy(() -> ZipkinSpanExporter.builder().setSender(null))
         .isInstanceOf(NullPointerException.class)
         .hasMessage("sender");
@@ -144,4 +148,45 @@ void invalidConfig() {
         .isInstanceOf(NullPointerException.class)
         .hasMessage("encoder");
   }
+
+  @Test
+  void compressionDefault() {
+    ZipkinSpanExporter exporter = ZipkinSpanExporter.builder().build();
+    try {
+      assertThat(exporter).extracting("sender.compressionEnabled").isEqualTo(false);
+    } finally {
+      exporter.shutdown();
+    }
+  }
+
+  @Test
+  void compressionNone() {
+    ZipkinSpanExporter exporter = ZipkinSpanExporter.builder().setCompression("none").build();
+    try {
+      assertThat(exporter).extracting("sender.compressionEnabled").isEqualTo(false);
+    } finally {
+      exporter.shutdown();
+    }
+  }
+
+  @Test
+  void compressionGzip() {
+    ZipkinSpanExporter exporter = ZipkinSpanExporter.builder().setCompression("gzip").build();
+    try {
+      assertThat(exporter).extracting("sender.compressionEnabled").isEqualTo(true);
+    } finally {
+      exporter.shutdown();
+    }
+  }
+
+  @Test
+  void compressionEnabledAndDisabled() {
+    ZipkinSpanExporter exporter =
+        ZipkinSpanExporter.builder().setCompression("gzip").setCompression("none").build();
+    try {
+      assertThat(exporter).extracting("sender.compressionEnabled").isEqualTo(false);
+    } finally {
+      exporter.shutdown();
+    }
+  }
 }