From 1efda6d70f0bfde57ee8a33e256e313953e86f36 Mon Sep 17 00:00:00 2001
From: Cesar Munoz <56847527+LikeTheSalad@users.noreply.github.com>
Date: Wed, 3 Jan 2024 11:16:57 +0100
Subject: [PATCH] Passing the initialized OpenTelemetry instance to the
 initialization listeners

---
 .../io/opentelemetry/android/OpenTelemetryRumBuilder.java  | 7 ++++---
 .../instrumentation/startup/InitializationListener.kt      | 5 ++++-
 2 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/instrumentation/src/main/java/io/opentelemetry/android/OpenTelemetryRumBuilder.java b/instrumentation/src/main/java/io/opentelemetry/android/OpenTelemetryRumBuilder.java
index 07c114295..009d799fe 100644
--- a/instrumentation/src/main/java/io/opentelemetry/android/OpenTelemetryRumBuilder.java
+++ b/instrumentation/src/main/java/io/opentelemetry/android/OpenTelemetryRumBuilder.java
@@ -21,6 +21,7 @@
 import io.opentelemetry.android.instrumentation.startup.SdkInitializationEvents;
 import io.opentelemetry.android.internal.features.persistence.DiskManager;
 import io.opentelemetry.android.internal.features.persistence.SimpleTemporaryFileProvider;
+import io.opentelemetry.api.OpenTelemetry;
 import io.opentelemetry.api.baggage.propagation.W3CBaggagePropagator;
 import io.opentelemetry.api.trace.propagation.W3CTraceContextPropagator;
 import io.opentelemetry.context.propagation.ContextPropagators;
@@ -266,7 +267,7 @@ public OpenTelemetryRum build() {
                 new SdkPreconfiguredRumBuilder(application, sdk, sessionId);
         instrumentationInstallers.forEach(delegate::addInstrumentation);
 
-        notifyInitializationEnd();
+        notifyInitializationEnd(sdk);
         return delegate.build();
     }
 
@@ -276,9 +277,9 @@ private void notifyInitializationStart() {
         }
     }
 
-    private void notifyInitializationEnd() {
+    private void notifyInitializationEnd(OpenTelemetry openTelemetry) {
         for (InitializationListener listener : config.getInitializationListeners()) {
-            listener.onEnd();
+            listener.onEnd(openTelemetry);
         }
     }
 
diff --git a/instrumentation/src/main/java/io/opentelemetry/android/instrumentation/startup/InitializationListener.kt b/instrumentation/src/main/java/io/opentelemetry/android/instrumentation/startup/InitializationListener.kt
index 85970c90f..413d3284d 100644
--- a/instrumentation/src/main/java/io/opentelemetry/android/instrumentation/startup/InitializationListener.kt
+++ b/instrumentation/src/main/java/io/opentelemetry/android/instrumentation/startup/InitializationListener.kt
@@ -5,6 +5,8 @@
 
 package io.opentelemetry.android.instrumentation.startup
 
+import io.opentelemetry.api.OpenTelemetry
+
 /**
  * Provides callbacks to know the sate of the initialization.
  */
@@ -16,6 +18,7 @@ interface InitializationListener {
 
     /**
      * Called when the RUM initialization ends.
+     * @param openTelemetry - The initialized OpenTelemetry instance.
      */
-    fun onEnd()
+    fun onEnd(openTelemetry: OpenTelemetry)
 }