Enable metro tracing with integration

Author: alexvanyo

android-application/build.gradle.kts

@@ -45,6 +45,7 @@ kotlin {
                 implementation(projects.injectScopes)
                 implementation(projects.processLifecycle)
                 implementation(projects.strictMode)
+                implementation(projects.tracing)
                 implementation(projects.updatable)
             }
         }

android-application/src/androidMain/kotlin/com/alexvanyo/composelife/BaseComposeLifeApplication.kt

@@ -29,6 +29,11 @@ import com.alexvanyo.composelife.scopes.ApplicationGraph
 import com.alexvanyo.composelife.scopes.ApplicationGraphArguments
 import com.alexvanyo.composelife.scopes.ApplicationGraphOwner
 import com.alexvanyo.composelife.strictmode.initStrictModeIfNeeded
+import com.alexvanyo.composelife.tracing.AbstractTraceDriver
+import com.alexvanyo.composelife.tracing.AbstractTraceDriverFactory
+import com.alexvanyo.composelife.tracing.Tracer
+import com.alexvanyo.composelife.tracing.createTraceDriver
+import com.alexvanyo.composelife.tracing.trace
 import com.alexvanyo.composelife.updatable.Updatable
 import dev.zacsweers.metro.AppScope
 import dev.zacsweers.metro.ContributesTo
@@ -49,24 +54,34 @@ internal val ApplicationGraph.composeLifeApplicationCtx: ComposeLifeApplicationC
     this as ComposeLifeApplicationCtx
 
 @Suppress("UnnecessaryAbstractClass", "AbstractClassCanBeConcreteClass")
-abstract class BaseComposeLifeApplication<G>(private val createGlobalGraph: () -> G) :
+abstract class BaseComposeLifeApplication<G>(private val createGlobalGraph: (tracer: Tracer) -> G) :
     Application(),
+    AbstractTraceDriverFactory,
     ApplicationGraphOwner {
     override lateinit var applicationGraph: ApplicationGraph
 
+    private val traceDriver by lazy { createTraceDriver(this) }
+
+    override fun create(): AbstractTraceDriver = traceDriver
+
     override fun onCreate() {
         super.onCreate()
 
         initStrictModeIfNeeded()
         Composer.setDiagnosticStackTraceMode(ComposeStackTraceMode.Auto)
 
-        val globalGraph = createGlobalGraph()
-        applicationGraph =
+        val tracer = traceDriver.tracer
+
+        val globalGraph = tracer.trace("startup", "createGlobalGraph") {
+            createGlobalGraph(tracer)
+        }
+        applicationGraph = tracer.trace("startup", "createApplicationGraph") {
             (globalGraph as ApplicationGraph.Factory).create(
                 object : ApplicationGraphArguments {
                     override val applicationContext: Context = this@BaseComposeLifeApplication
                 },
             )
+        }
         val ctx = applicationGraph.composeLifeApplicationCtx
         val processLifecycleOwner = ctx.processLifecycleOwner
         val appUpdatables = ctx.appUpdatables

app-impl/src/androidMain/kotlin/com/alexvanyo/composelife/ComposeLifeApplication.kt

@@ -17,10 +17,20 @@
 package com.alexvanyo.composelife
 
 import com.alexvanyo.composelife.scopes.GlobalScope
+import com.alexvanyo.composelife.tracing.Tracer
 import dev.zacsweers.metro.DependencyGraph
-import dev.zacsweers.metro.createGraph
+import dev.zacsweers.metro.Provides
+import dev.zacsweers.metro.createGraphFactory
 
-class ComposeLifeApplication : BaseComposeLifeApplication<GlobalGraph>({ createGraph() })
+class ComposeLifeApplication :
+    BaseComposeLifeApplication<GlobalGraph>({ tracer ->
+        createGraphFactory<GlobalGraph.Factory>().create(tracer)
+    })
 
 @DependencyGraph(GlobalScope::class)
-interface GlobalGraph
+interface GlobalGraph {
+    @DependencyGraph.Factory
+    fun interface Factory {
+        fun create(@Provides tracer: Tracer): GlobalGraph
+    }
+}

app-test-fixtures/build.gradle.kts

@@ -54,6 +54,7 @@ kotlin {
                 implementation(projects.preferencesTestFixtures)
                 implementation(projects.resourcesApp)
                 implementation(projects.strictMode)
+                implementation(projects.tracingTestFixtures)
                 implementation(projects.uiApp)
                 implementation(projects.uiCommon)
                 implementation(projects.uiMobile)

app-test-fixtures/src/androidMain/kotlin/com/alexvanyo/composelife/GlobalGraph.kt

@@ -17,18 +17,28 @@
 package com.alexvanyo.composelife
 
 import com.alexvanyo.composelife.scopes.GlobalScope
+import com.alexvanyo.composelife.tracing.TestTraceDriver
+import com.alexvanyo.composelife.tracing.Tracer
 import com.alexvanyo.composelife.ui.app.UiWithLoadedPreferencesScope
 import com.alexvanyo.composelife.ui.app.UiWithLoadedPreferencesScopeBindings
 import dev.zacsweers.metro.BindingContainer
 import dev.zacsweers.metro.ContributesTo
 import dev.zacsweers.metro.DependencyGraph
 import dev.zacsweers.metro.Provides
-import dev.zacsweers.metro.createGraph
+import dev.zacsweers.metro.createGraphFactory
 
 @DependencyGraph(GlobalScope::class)
-interface TestGlobalGraph
+interface TestGlobalGraph {
+    @DependencyGraph.Factory
+    fun interface Factory {
+        fun create(@Provides tracer: Tracer): TestGlobalGraph
+    }
+}
 
-internal val testGlobalGraph = createGraph<TestGlobalGraph>()
+private val testTraceDriver = TestTraceDriver()
+internal val testGlobalGraph = createGraphFactory<TestGlobalGraph.Factory>().create(
+    testTraceDriver.tracer,
+)
 
 @ContributesTo(UiWithLoadedPreferencesScope::class, replaces = [UiWithLoadedPreferencesScopeBindings::class])
 @BindingContainer

app/build.gradle.kts

@@ -31,6 +31,7 @@ plugins {
 }
 
 metro {
+    enableRuntimeTracing.set(true)
     unusedGraphInputsSeverity = DiagnosticSeverity.NONE
 }
 

app/dependencies/releaseRuntimeClasspath.txt

@@ -172,8 +172,9 @@ com.squareup.okio:okio-jvm:3.18.1
 com.squareup.okio:okio:3.18.1
 com.squareup.wire:wire-runtime-jvm:7.0.0-alpha05
 com.squareup.wire:wire-runtime:7.0.0-alpha05
-dev.zacsweers.metro:runtime-jvm:1.2.1
-dev.zacsweers.metro:runtime:1.2.1
+dev.zacsweers.metro:metro-trace:1.3.2
+dev.zacsweers.metro:runtime-jvm:1.3.2
+dev.zacsweers.metro:runtime:1.3.2
 io.coil-kt.coil3:coil-compose-core-android:3.5.0
 io.coil-kt.coil3:coil-compose-core:3.5.0
 io.coil-kt.coil3:coil-core-android:3.5.0

data/build.gradle.kts

@@ -108,6 +108,7 @@ kotlin {
                 implementation(projects.filesystemTestFixtures)
                 implementation(projects.injectTest)
                 implementation(projects.networkTestFixtures)
+                implementation(projects.tracingTestFixtures)
                 implementation(projects.workTestFixtures)
             }
         }

data/src/jvmTest/kotlin/com/alexvanyo/composelife/data/GlobalGraph.kt

@@ -17,10 +17,19 @@
 package com.alexvanyo.composelife.data
 
 import com.alexvanyo.composelife.scopes.GlobalScope
+import com.alexvanyo.composelife.tracing.TestTraceDriver
+import com.alexvanyo.composelife.tracing.Tracer
 import dev.zacsweers.metro.DependencyGraph
-import dev.zacsweers.metro.createGraph
+import dev.zacsweers.metro.Provides
+import dev.zacsweers.metro.createGraphFactory
 
 @DependencyGraph(GlobalScope::class)
-interface GlobalGraph
+interface GlobalGraph {
+    @DependencyGraph.Factory
+    fun interface Factory {
+        fun create(@Provides tracer: Tracer): GlobalGraph
+    }
+}
 
-internal val globalGraph = createGraph<GlobalGraph>()
+private val testTraceDriver = TestTraceDriver()
+internal val globalGraph = createGraphFactory<GlobalGraph.Factory>().create(testTraceDriver.tracer)

database/build.gradle.kts

@@ -102,6 +102,7 @@ kotlin {
                 implementation(projects.filesystemTestFixtures)
                 implementation(projects.injectTest)
                 implementation(projects.kmpAndroidRunner)
+                implementation(projects.tracingTestFixtures)
             }
         }
         val jvmTest by creating {