Moving DefaultExportScheduler instance to DefaultExportScheduleHandle…

…r's constructor
open-telemetry · Jan 11, 2024 · 2abf0af · 2abf0af
1 parent c407bab
commit 2abf0af
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/...io/opentelemetry/android/features/diskbuffering/scheduler/DefaultExportScheduleHandler.kt b/...io/opentelemetry/android/features/diskbuffering/scheduler/DefaultExportScheduleHandler.kt
@@ -9,13 +9,14 @@ import io.opentelemetry.android.internal.services.ServiceManager
 import io.opentelemetry.android.internal.services.periodicwork.PeriodicWorkService
 import java.util.concurrent.atomic.AtomicBoolean
 
-class DefaultExportScheduleHandler : ExportScheduleHandler {
+class DefaultExportScheduleHandler(private val exportScheduler: DefaultExportScheduler) :
+    ExportScheduleHandler {
     private val enabled = AtomicBoolean(false)
 
     override fun enable() {
         if (!enabled.getAndSet(true)) {
             ServiceManager.get().getService(PeriodicWorkService::class.java)
-                .enqueue(DefaultExportScheduler())
+                .enqueue(exportScheduler)
         }
     }
 }
diff --git a/...pentelemetry/android/features/diskbuffering/scheduler/DefaultExportScheduleHandlerTest.kt b/...pentelemetry/android/features/diskbuffering/scheduler/DefaultExportScheduleHandlerTest.kt
@@ -23,7 +23,7 @@ class DefaultExportScheduleHandlerTest {
 
     @BeforeEach
     fun setUp() {
-        handler = DefaultExportScheduleHandler()
+        handler = DefaultExportScheduleHandler(DefaultExportScheduler())
     }
 
     @Test