Creating export to disk implementations

LikeTheSalad · LikeTheSalad · commit 1ca3cb210970 · 2025-08-24T14:06:30.000+02:00
diff --git a/disk-buffering/src/main/java/io/opentelemetry/contrib/disk/buffering/exporters/LogRecordToDiskExporter.java b/disk-buffering/src/main/java/io/opentelemetry/contrib/disk/buffering/exporters/LogRecordToDiskExporter.java
@@ -0,0 +1,78 @@
+/*
+ * Copyright The OpenTelemetry Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+package io.opentelemetry.contrib.disk.buffering.exporters;
+
+import com.google.errorprone.annotations.CanIgnoreReturnValue;
+import io.opentelemetry.contrib.disk.buffering.SignalType;
+import io.opentelemetry.contrib.disk.buffering.exporters.callback.ExporterCallback;
+import io.opentelemetry.contrib.disk.buffering.internal.exporters.SignalStorageExporter;
+import io.opentelemetry.contrib.disk.buffering.storage.SignalStorage;
+import io.opentelemetry.sdk.common.CompletableResultCode;
+import io.opentelemetry.sdk.logs.data.LogRecordData;
+import io.opentelemetry.sdk.logs.export.LogRecordExporter;
+import java.time.Duration;
+import java.util.Collection;
+
+/** Exporter that stores logs into disk. */
+public final class LogRecordToDiskExporter implements LogRecordExporter {
+  private final SignalStorageExporter<LogRecordData> storageExporter;
+  private final ExporterCallback callback;
+  private static final SignalType TYPE = SignalType.LOG;
+
+  private LogRecordToDiskExporter(
+      SignalStorageExporter<LogRecordData> storageExporter, ExporterCallback callback) {
+    this.storageExporter = storageExporter;
+    this.callback = callback;
+  }
+
+  public static Builder builder(SignalStorage.LogRecord storage) {
+    return new Builder(storage);
+  }
+
+  @Override
+  public CompletableResultCode export(Collection<LogRecordData> logs) {
+    return storageExporter.exportToStorage(logs);
+  }
+
+  @Override
+  public CompletableResultCode flush() {
+    return CompletableResultCode.ofSuccess();
+  }
+
+  @Override
+  public CompletableResultCode shutdown() {
+    callback.onShutdown(TYPE);
+    return CompletableResultCode.ofSuccess();
+  }
+
+  public static final class Builder {
+    private final SignalStorage.LogRecord storage;
+    private ExporterCallback callback = ExporterCallback.noop();
+    private Duration writeTimeout = Duration.ofSeconds(10);
+
+    @CanIgnoreReturnValue
+    public Builder setExporterCallback(ExporterCallback value) {
+      callback = value;
+      return this;
+    }
+
+    @CanIgnoreReturnValue
+    public Builder setWriteTimeout(Duration value) {
+      writeTimeout = value;
+      return this;
+    }
+
+    public LogRecordToDiskExporter build() {
+      SignalStorageExporter<LogRecordData> storageExporter =
+          new SignalStorageExporter<>(storage, callback, writeTimeout, TYPE);
+      return new LogRecordToDiskExporter(storageExporter, callback);
+    }
+
+    private Builder(SignalStorage.LogRecord storage) {
+      this.storage = storage;
+    }
+  }
+}
diff --git a/disk-buffering/src/main/java/io/opentelemetry/contrib/disk/buffering/exporters/MetricToDiskExporter.java b/disk-buffering/src/main/java/io/opentelemetry/contrib/disk/buffering/exporters/MetricToDiskExporter.java
@@ -0,0 +1,98 @@
+/*
+ * Copyright The OpenTelemetry Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+package io.opentelemetry.contrib.disk.buffering.exporters;
+
+import com.google.errorprone.annotations.CanIgnoreReturnValue;
+import io.opentelemetry.contrib.disk.buffering.SignalType;
+import io.opentelemetry.contrib.disk.buffering.exporters.callback.ExporterCallback;
+import io.opentelemetry.contrib.disk.buffering.internal.exporters.SignalStorageExporter;
+import io.opentelemetry.contrib.disk.buffering.storage.SignalStorage;
+import io.opentelemetry.sdk.common.CompletableResultCode;
+import io.opentelemetry.sdk.metrics.InstrumentType;
+import io.opentelemetry.sdk.metrics.data.AggregationTemporality;
+import io.opentelemetry.sdk.metrics.data.MetricData;
+import io.opentelemetry.sdk.metrics.export.AggregationTemporalitySelector;
+import io.opentelemetry.sdk.metrics.export.MetricExporter;
+import java.time.Duration;
+import java.util.Collection;
+
+/** Exporter that stores metrics into disk. */
+public final class MetricToDiskExporter implements MetricExporter {
+  private final SignalStorageExporter<MetricData> storageExporter;
+  private final AggregationTemporalitySelector aggregationTemporalitySelector;
+  private final ExporterCallback callback;
+  private static final SignalType TYPE = SignalType.METRIC;
+
+  private MetricToDiskExporter(
+      SignalStorageExporter<MetricData> storageExporter,
+      AggregationTemporalitySelector aggregationTemporalitySelector,
+      ExporterCallback callback) {
+    this.storageExporter = storageExporter;
+    this.aggregationTemporalitySelector = aggregationTemporalitySelector;
+    this.callback = callback;
+  }
+
+  public static Builder builder(SignalStorage.Metric storage) {
+    return new Builder(storage);
+  }
+
+  @Override
+  public CompletableResultCode export(Collection<MetricData> metrics) {
+    return storageExporter.exportToStorage(metrics);
+  }
+
+  @Override
+  public CompletableResultCode flush() {
+    return CompletableResultCode.ofSuccess();
+  }
+
+  @Override
+  public CompletableResultCode shutdown() {
+    callback.onShutdown(TYPE);
+    return CompletableResultCode.ofSuccess();
+  }
+
+  @Override
+  public AggregationTemporality getAggregationTemporality(InstrumentType instrumentType) {
+    return aggregationTemporalitySelector.getAggregationTemporality(instrumentType);
+  }
+
+  public static final class Builder {
+    private final SignalStorage.Metric storage;
+    private AggregationTemporalitySelector aggregationTemporalitySelector =
+        AggregationTemporalitySelector.alwaysCumulative();
+    private ExporterCallback callback = ExporterCallback.noop();
+    private Duration writeTimeout = Duration.ofSeconds(10);
+
+    @CanIgnoreReturnValue
+    public Builder setExporterCallback(ExporterCallback value) {
+      callback = value;
+      return this;
+    }
+
+    @CanIgnoreReturnValue
+    public Builder setWriteTimeout(Duration value) {
+      writeTimeout = value;
+      return this;
+    }
+
+    @CanIgnoreReturnValue
+    public Builder setAggregationTemporalitySelector(AggregationTemporalitySelector value) {
+      aggregationTemporalitySelector = value;
+      return this;
+    }
+
+    public MetricToDiskExporter build() {
+      SignalStorageExporter<MetricData> storageExporter =
+          new SignalStorageExporter<>(storage, callback, writeTimeout, TYPE);
+      return new MetricToDiskExporter(storageExporter, aggregationTemporalitySelector, callback);
+    }
+
+    private Builder(SignalStorage.Metric storage) {
+      this.storage = storage;
+    }
+  }
+}
diff --git a/disk-buffering/src/main/java/io/opentelemetry/contrib/disk/buffering/exporters/SpanToDiskExporter.java b/disk-buffering/src/main/java/io/opentelemetry/contrib/disk/buffering/exporters/SpanToDiskExporter.java
@@ -7,6 +7,8 @@
 
 import com.google.errorprone.annotations.CanIgnoreReturnValue;
 import io.opentelemetry.contrib.disk.buffering.SignalType;
+import io.opentelemetry.contrib.disk.buffering.exporters.callback.ExporterCallback;
+import io.opentelemetry.contrib.disk.buffering.internal.exporters.SignalStorageExporter;
 import io.opentelemetry.contrib.disk.buffering.storage.SignalStorage;
 import io.opentelemetry.sdk.common.CompletableResultCode;
 import io.opentelemetry.sdk.trace.data.SpanData;
diff --git a/disk-buffering/src/main/java/io/opentelemetry/contrib/disk/buffering/exporters/callback/ExporterCallback.java b/disk-buffering/src/main/java/io/opentelemetry/contrib/disk/buffering/exporters/callback/ExporterCallback.java
@@ -3,7 +3,7 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 
-package io.opentelemetry.contrib.disk.buffering.exporters;
+package io.opentelemetry.contrib.disk.buffering.exporters.callback;
 
 import io.opentelemetry.contrib.disk.buffering.SignalType;
 import javax.annotation.Nullable;
diff --git a/disk-buffering/src/main/java/io/opentelemetry/contrib/disk/buffering/exporters/callback/NoopExporterCallback.java b/disk-buffering/src/main/java/io/opentelemetry/contrib/disk/buffering/exporters/callback/NoopExporterCallback.java
@@ -3,7 +3,7 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 
-package io.opentelemetry.contrib.disk.buffering.exporters;
+package io.opentelemetry.contrib.disk.buffering.exporters.callback;
 
 import io.opentelemetry.contrib.disk.buffering.SignalType;
 import javax.annotation.Nullable;
diff --git a/disk-buffering/src/main/java/io/opentelemetry/contrib/disk/buffering/internal/exporters/SignalStorageExporter.java b/disk-buffering/src/main/java/io/opentelemetry/contrib/disk/buffering/internal/exporters/SignalStorageExporter.java
@@ -3,9 +3,10 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 
-package io.opentelemetry.contrib.disk.buffering.exporters;
+package io.opentelemetry.contrib.disk.buffering.internal.exporters;
 
 import io.opentelemetry.contrib.disk.buffering.SignalType;
+import io.opentelemetry.contrib.disk.buffering.exporters.callback.ExporterCallback;
 import io.opentelemetry.contrib.disk.buffering.storage.SignalStorage;
 import io.opentelemetry.contrib.disk.buffering.storage.result.WriteResult;
 import io.opentelemetry.sdk.common.CompletableResultCode;
@@ -17,7 +18,7 @@
 import java.util.concurrent.TimeoutException;
 
 /** Internal utility for common export to disk operations across all exporters. */
-final class SignalStorageExporter<T> {
+public final class SignalStorageExporter<T> {
   private final SignalStorage<T> storage;
   private final ExporterCallback callback;
   private final Duration writeTimeout;
diff --git a/disk-buffering/src/test/java/io/opentelemetry/contrib/disk/buffering/exporters/SignalStorageExporterTest.java b/disk-buffering/src/test/java/io/opentelemetry/contrib/disk/buffering/exporters/SignalStorageExporterTest.java
@@ -14,6 +14,7 @@
 import static org.mockito.Mockito.when;
 
 import io.opentelemetry.contrib.disk.buffering.SignalType;
+import io.opentelemetry.contrib.disk.buffering.exporters.callback.ExporterCallback;
 import io.opentelemetry.contrib.disk.buffering.storage.SignalStorage;
 import io.opentelemetry.contrib.disk.buffering.storage.result.WriteResult;
 import io.opentelemetry.sdk.common.CompletableResultCode;
@@ -40,8 +41,8 @@ class SignalStorageExporterTest {
   void verifyExportToStorage_success() {
     SignalStorage.Span storage = new TestSpanStorage();
     SignalType signalType = SignalType.SPAN;
-    SignalStorageExporter<SpanData> storageExporter =
-        new SignalStorageExporter<>(storage, callback, Duration.ofSeconds(1), signalType);
+    io.opentelemetry.contrib.disk.buffering.internal.exporters.SignalStorageExporter<SpanData> storageExporter =
+        new io.opentelemetry.contrib.disk.buffering.internal.exporters.SignalStorageExporter<>(storage, callback, Duration.ofSeconds(1), signalType);
     SpanData item1 = mock();
     SpanData item2 = mock();
     SpanData item3 = mock();
@@ -72,8 +73,8 @@ void verifyExportToStorage_success() {
   void verifyExportToStorage_failure() {
     SignalStorage.Span storage = mock();
     SignalType signalType = SignalType.SPAN;
-    SignalStorageExporter<SpanData> storageExporter =
-        new SignalStorageExporter<>(storage, callback, Duration.ofSeconds(1), signalType);
+    io.opentelemetry.contrib.disk.buffering.internal.exporters.SignalStorageExporter<SpanData> storageExporter =
+        new io.opentelemetry.contrib.disk.buffering.internal.exporters.SignalStorageExporter<>(storage, callback, Duration.ofSeconds(1), signalType);
     SpanData item1 = mock();
 
     // Without exception
