wip sessionmanager

Author: jrodiz

firebase-perf/src/main/java/com/google/firebase/perf/FirebasePerfEarly.java

@@ -34,29 +34,30 @@
 public class FirebasePerfEarly {
 
   public FirebasePerfEarly(
-      FirebaseApp app, @Nullable StartupTime startupTime, Executor uiExecutor) {
+      FirebaseApp app, @Nullable StartupTime startupTime, Executor uiExecutor, SessionManager sessionManager) {
     Context context = app.getApplicationContext();
 
     // Initialize ConfigResolver early for accessing device caching layer.
     ConfigResolver configResolver = ConfigResolver.getInstance();
     configResolver.setApplicationContext(context);
 
     AppStateMonitor appStateMonitor = AppStateMonitor.getInstance();
+    appStateMonitor.setSessionManager(sessionManager);
     appStateMonitor.registerActivityLifecycleCallbacks(context);
     appStateMonitor.registerForAppColdStart(new FirebasePerformanceInitializer());
 
     if (startupTime != null) {
       AppStartTrace appStartTrace = AppStartTrace.getInstance();
+      appStartTrace.setSessionManager(sessionManager);
       appStartTrace.registerActivityLifecycleCallbacks(context);
       uiExecutor.execute(new AppStartTrace.StartFromBackgroundRunnable(appStartTrace));
     }
-
     // TODO: Bring back Firebase Sessions dependency to watch for updates to sessions.
 
     // In the case of cold start, we create a session and start collecting gauges as early as
     // possible.
     // There is code in SessionManager that prevents us from resetting the session twice in case
     // of app cold start.
-    SessionManager.getInstance().initializeGaugeCollection();
+    sessionManager.initializeGaugeCollection();
   }
 }

firebase-perf/src/main/java/com/google/firebase/perf/FirebasePerfRegistrar.java

@@ -26,8 +26,12 @@
 import com.google.firebase.components.Qualified;
 import com.google.firebase.installations.FirebaseInstallationsApi;
 import com.google.firebase.perf.injection.components.DaggerFirebasePerformanceComponent;
+import com.google.firebase.perf.injection.components.DaggerSessionManagerComponent;
 import com.google.firebase.perf.injection.components.FirebasePerformanceComponent;
+import com.google.firebase.perf.injection.components.SessionManagerComponent;
 import com.google.firebase.perf.injection.modules.FirebasePerformanceModule;
+import com.google.firebase.perf.injection.modules.SessionManagerModule;
+import com.google.firebase.perf.session.SessionManager;
 import com.google.firebase.platforminfo.LibraryVersionComponent;
 import com.google.firebase.remoteconfig.RemoteConfigComponent;
 import java.util.Arrays;
@@ -59,21 +63,25 @@ public List<Component<?>> getComponents() {
             .add(Dependency.required(FirebaseInstallationsApi.class))
             .add(Dependency.requiredProvider(TransportFactory.class))
             .add(Dependency.required(FirebasePerfEarly.class))
+            .add(Dependency.required(SessionManager.class))
             .factory(FirebasePerfRegistrar::providesFirebasePerformance)
             .build(),
         Component.builder(FirebasePerfEarly.class)
             .name(EARLY_LIBRARY_NAME)
             .add(Dependency.required(FirebaseApp.class))
             .add(Dependency.optionalProvider(StartupTime.class))
             .add(Dependency.required(uiExecutor))
+            .add(Dependency.required(SessionManager.class))
             .eagerInDefaultApp()
             .factory(
                 container ->
                     new FirebasePerfEarly(
                         container.get(FirebaseApp.class),
                         container.getProvider(StartupTime.class).get(),
-                        container.get(uiExecutor)))
+                        container.get(uiExecutor),
+                        container.get(SessionManager.class)))
             .build(),
+        Component.builder(SessionManager.class).factory(container -> new SessionManager()).build(),
         /**
          * Fireperf SDK is lazily by {@link FirebasePerformanceInitializer} during {@link
          * com.google.firebase.perf.application.AppStateMonitor#onActivityResumed(Activity)}. we use
@@ -94,7 +102,8 @@ private static FirebasePerformance providesFirebasePerformance(ComponentContaine
                     container.get(FirebaseApp.class),
                     container.get(FirebaseInstallationsApi.class),
                     container.getProvider(RemoteConfigComponent.class),
-                    container.getProvider(TransportFactory.class)))
+                    container.getProvider(TransportFactory.class),
+                    container.get(SessionManager.class)))
             .build();
 
     return component.getFirebasePerformance();

firebase-perf/src/main/java/com/google/firebase/perf/FirebasePerformance.java

@@ -140,6 +140,7 @@ public static FirebasePerformance getInstance() {
   private final Provider<RemoteConfigComponent> firebaseRemoteConfigProvider;
   private final FirebaseInstallationsApi firebaseInstallationsApi;
   private final Provider<TransportFactory> transportFactoryProvider;
+  private final SessionManager sessionManager;
 
   /**
    * Constructs the FirebasePerformance class and allows injecting dependencies.
@@ -168,6 +169,7 @@ public static FirebasePerformance getInstance() {
     this.firebaseRemoteConfigProvider = firebaseRemoteConfigProvider;
     this.firebaseInstallationsApi = firebaseInstallationsApi;
     this.transportFactoryProvider = transportFactoryProvider;
+    this.sessionManager = sessionManager;
 
     if (firebaseApp == null) {
       this.mPerformanceCollectionForceEnabledState = false;
@@ -177,7 +179,7 @@ public static FirebasePerformance getInstance() {
     }
 
     TransportManager.getInstance()
-        .initialize(firebaseApp, firebaseInstallationsApi, transportFactoryProvider);
+        .initialize(firebaseApp, firebaseInstallationsApi, transportFactoryProvider, sessionManager);
 
     Context appContext = firebaseApp.getApplicationContext();
     // TODO(b/110178816): Explore moving off of main thread.
@@ -187,7 +189,7 @@ public static FirebasePerformance getInstance() {
     this.configResolver.getRemoteConfigManager().setFirebaseRemoteConfigProvider(firebaseRemoteConfigProvider);
     this.configResolver.setMetadataBundle(mMetadataBundle);
     this.configResolver.setApplicationContext(appContext);
-    sessionManager.setApplicationContext(appContext);
+    this.sessionManager.setApplicationContext(appContext);
 
     mPerformanceCollectionForceEnabledState = configResolver.getIsPerformanceCollectionEnabled();
     if (logger.isLogcatEnabled() && isPerformanceCollectionEnabled()) {
@@ -423,7 +425,7 @@ public Trace newTrace(@NonNull String traceName) {
    */
   @NonNull
   public HttpMetric newHttpMetric(@NonNull String url, @NonNull @HttpMethod String httpMethod) {
-    return new HttpMetric(url, httpMethod, TransportManager.getInstance(), new Timer());
+    return new HttpMetric(url, httpMethod, TransportManager.getInstance(), new Timer(), sessionManager);
   }
 
   /**
@@ -436,7 +438,7 @@ public HttpMetric newHttpMetric(@NonNull String url, @NonNull @HttpMethod String
    */
   @NonNull
   public HttpMetric newHttpMetric(@NonNull URL url, @NonNull @HttpMethod String httpMethod) {
-    return new HttpMetric(url, httpMethod, TransportManager.getInstance(), new Timer());
+    return new HttpMetric(url, httpMethod, TransportManager.getInstance(), new Timer(), sessionManager);
   }
 
   /**

firebase-perf/src/main/java/com/google/firebase/perf/application/AppStateMonitor.java

@@ -80,6 +80,8 @@ public class AppStateMonitor implements ActivityLifecycleCallbacks {
   private boolean isRegisteredForLifecycleCallbacks = false;
   private boolean isColdStart = true;
 
+  private SessionManager sessionManager;
+
   public static AppStateMonitor getInstance() {
     if (instance == null) {
       synchronized (AppStateMonitor.class) {
@@ -159,7 +161,7 @@ private void startFrameMonitoring(Activity activity) {
       activityToRecorderMap.put(activity, recorder);
       if (activity instanceof FragmentActivity) {
         FragmentStateMonitor fragmentStateMonitor =
-            new FragmentStateMonitor(clock, transportManager, this, recorder);
+            new FragmentStateMonitor(clock, transportManager, this, recorder, sessionManager);
         activityToFragmentStateMonitorMap.put(activity, fragmentStateMonitor);
         FragmentActivity fragmentActivity = (FragmentActivity) activity;
         fragmentActivity
@@ -198,7 +200,7 @@ public synchronized void onActivityStarted(Activity activity) {
       // Starts recording frame metrics for this activity.
       activityToRecorderMap.get(activity).start();
       // Start the Trace
-      Trace screenTrace = new Trace(getScreenTraceName(activity), transportManager, clock, this);
+      Trace screenTrace = new Trace(getScreenTraceName(activity), transportManager, clock, this, sessionManager);
       screenTrace.start();
       activityToScreenTraceMap.put(activity, screenTrace);
     }
@@ -379,7 +381,7 @@ private void sendSessionLog(String name, Timer startTime, Timer endTime) {
             .setName(name)
             .setClientStartTimeUs(startTime.getMicros())
             .setDurationUs(startTime.getDurationMicros(endTime))
-            .addPerfSessions(SessionManager.getInstance().perfSession().build());
+            .addPerfSessions(sessionManager.perfSession().build());
     // Atomically get mTsnsCount and set it to zero.
     int tsnsCount = this.tsnsCount.getAndSet(0);
     synchronized (metricToCountMap) {
@@ -463,4 +465,12 @@ Timer getResumeTime() {
   public void setIsColdStart(boolean isColdStart) {
     this.isColdStart = isColdStart;
   }
+
+  public SessionManager getSessionManager() {
+    return sessionManager;
+  }
+
+  public void setSessionManager(SessionManager sessionManager) {
+    this.sessionManager = sessionManager;
+  }
 }

firebase-perf/src/main/java/com/google/firebase/perf/application/FragmentStateMonitor.java

@@ -21,6 +21,7 @@
 import com.google.firebase.perf.logging.AndroidLogger;
 import com.google.firebase.perf.metrics.FrameMetricsCalculator.PerfFrameMetrics;
 import com.google.firebase.perf.metrics.Trace;
+import com.google.firebase.perf.session.SessionManager;
 import com.google.firebase.perf.transport.TransportManager;
 import com.google.firebase.perf.util.Clock;
 import com.google.firebase.perf.util.Constants;
@@ -33,18 +34,21 @@ public class FragmentStateMonitor extends FragmentManager.FragmentLifecycleCallb
   private final WeakHashMap<Fragment, Trace> fragmentToTraceMap = new WeakHashMap<>();
   private final Clock clock;
   private final TransportManager transportManager;
+  private final SessionManager sessionManager;
   private final AppStateMonitor appStateMonitor;
   private final FrameMetricsRecorder activityFramesRecorder;
 
   public FragmentStateMonitor(
       Clock clock,
       TransportManager transportManager,
       AppStateMonitor appStateMonitor,
-      FrameMetricsRecorder recorder) {
+      FrameMetricsRecorder recorder,
+      SessionManager sessionManager) {
     this.clock = clock;
     this.transportManager = transportManager;
     this.appStateMonitor = appStateMonitor;
     this.activityFramesRecorder = recorder;
+    this.sessionManager = sessionManager;
   }
 
   /**
@@ -63,7 +67,7 @@ public void onFragmentResumed(@NonNull FragmentManager fm, @NonNull Fragment f)
     // Start Fragment screen trace
     logger.debug("FragmentMonitor %s.onFragmentResumed", f.getClass().getSimpleName());
     Trace fragmentTrace =
-        new Trace(getFragmentScreenTraceName(f), transportManager, clock, appStateMonitor);
+        new Trace(getFragmentScreenTraceName(f), transportManager, clock, appStateMonitor, sessionManager);
     fragmentTrace.start();
 
     fragmentTrace.putAttribute(

firebase-perf/src/main/java/com/google/firebase/perf/injection/components/SessionManagerComponent.java

@@ -0,0 +1,17 @@
+package com.google.firebase.perf.injection.components;
+
+import androidx.annotation.NonNull;
+
+import com.google.firebase.perf.injection.modules.SessionManagerModule;
+import com.google.firebase.perf.session.SessionManager;
+
+import javax.inject.Singleton;
+
+import dagger.Component;
+
+@Component(modules = {SessionManagerModule.class})
+@Singleton
+public interface SessionManagerComponent {
+  @NonNull
+  SessionManager getSessionManager();
+}

firebase-perf/src/main/java/com/google/firebase/perf/injection/modules/FirebasePerformanceModule.java

@@ -34,16 +34,19 @@ public class FirebasePerformanceModule {
   private final FirebaseInstallationsApi firebaseInstallations;
   private final Provider<RemoteConfigComponent> remoteConfigComponentProvider;
   private final Provider<TransportFactory> transportFactoryProvider;
+  private final SessionManager sessionManager;
 
   public FirebasePerformanceModule(
       @NonNull FirebaseApp firebaseApp,
       @NonNull FirebaseInstallationsApi firebaseInstallations,
       @NonNull Provider<RemoteConfigComponent> remoteConfigComponentProvider,
-      @NonNull Provider<TransportFactory> transportFactoryProvider) {
+      @NonNull Provider<TransportFactory> transportFactoryProvider,
+      @NonNull SessionManager sessionManager) {
     this.firebaseApp = firebaseApp;
     this.firebaseInstallations = firebaseInstallations;
     this.remoteConfigComponentProvider = remoteConfigComponentProvider;
     this.transportFactoryProvider = transportFactoryProvider;
+    this.sessionManager = sessionManager;
   }
 
   @Provides
@@ -73,6 +76,6 @@ ConfigResolver providesConfigResolver() {
 
   @Provides
   SessionManager providesSessionManager() {
-    return SessionManager.getInstance();
+    return sessionManager;
   }
 }

firebase-perf/src/main/java/com/google/firebase/perf/injection/modules/SessionManagerModule.java

@@ -0,0 +1,22 @@
+package com.google.firebase.perf.injection.modules;
+
+import androidx.annotation.NonNull;
+
+import com.google.firebase.perf.session.SessionManager;
+
+import dagger.Module;
+import dagger.Provides;
+
+@Module
+public class SessionManagerModule {
+  private final SessionManager sessionManager;
+
+  public SessionManagerModule(@NonNull SessionManager sessionManager) {
+    this.sessionManager = sessionManager;
+  }
+
+  @Provides
+  SessionManager providesSessionManager() {
+    return sessionManager;
+  }
+}

firebase-perf/src/main/java/com/google/firebase/perf/metrics/AppStartTrace.java

@@ -92,6 +92,7 @@ public class AppStartTrace implements ActivityLifecycleCallbacks, LifecycleObser
   private final ConfigResolver configResolver;
   private final TraceMetric.Builder experimentTtid;
   private Context appContext;
+  private SessionManager sessionManager;
 
   /**
    * The first time onCreate() of any activity is called, the activity is saved as launchActivity.
@@ -413,7 +414,7 @@ public synchronized void onActivityResumed(Activity activity) {
     appStartActivity = new WeakReference<Activity>(activity);
 
     onResumeTime = clock.getTime();
-    this.startSession = SessionManager.getInstance().perfSession();
+    this.startSession = sessionManager.perfSession();
     AndroidLogger.getInstance()
         .debug(
             "onResume(): "
@@ -572,6 +573,10 @@ public static boolean isAnyAppProcessInForeground(Context appContext) {
     return false;
   }
 
+  public void setSessionManager(SessionManager sessionManager) {
+    this.sessionManager = sessionManager;
+  }
+
   /**
    * We use StartFromBackgroundRunnable to detect if app is started from background or foreground.
    * If app is started from background, we do not generate AppStart trace. This runnable is posted

firebase-perf/src/main/java/com/google/firebase/perf/metrics/HttpMetric.java

@@ -22,6 +22,7 @@
 import com.google.firebase.perf.FirebasePerformanceAttributable;
 import com.google.firebase.perf.config.ConfigResolver;
 import com.google.firebase.perf.logging.AndroidLogger;
+import com.google.firebase.perf.session.SessionManager;
 import com.google.firebase.perf.transport.TransportManager;
 import com.google.firebase.perf.util.Constants;
 import com.google.firebase.perf.util.Timer;
@@ -52,12 +53,12 @@ public class HttpMetric implements FirebasePerformanceAttributable {
    * @hide
    */
   public HttpMetric(
-      String url, @HttpMethod String httpMethod, TransportManager transportManager, Timer timer) {
+      String url, @HttpMethod String httpMethod, TransportManager transportManager, Timer timer, SessionManager sessionManager) {
     customAttributesMap = new ConcurrentHashMap<>();
     this.timer = timer;
 
     networkMetricBuilder =
-        NetworkRequestMetricBuilder.builder(transportManager).setUrl(url).setHttpMethod(httpMethod);
+        NetworkRequestMetricBuilder.builder(transportManager, sessionManager).setUrl(url).setHttpMethod(httpMethod);
     networkMetricBuilder.setManualNetworkRequestMetric();
 
     if (!ConfigResolver.getInstance().isPerformanceMonitoringEnabled()) {
@@ -72,8 +73,8 @@ public HttpMetric(
    * @hide
    */
   public HttpMetric(
-      URL url, @HttpMethod String httpMethod, TransportManager transportManager, Timer timer) {
-    this(url.toString(), httpMethod, transportManager, timer);
+      URL url, @HttpMethod String httpMethod, TransportManager transportManager, Timer timer, SessionManager sessionManager) {
+    this(url.toString(), httpMethod, transportManager, timer, sessionManager);
   }
 
   /**