Compose benchmark app crud

engine/benchmarks/.gitignore

@@ -0,0 +1 @@
+synthea

engine/benchmarks/app/src/main/java/com/google/android/fhir/engine/benchmarks/app/BenchmarkResult.kt

@@ -0,0 +1,32 @@
+/*
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *       http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.google.android.fhir.engine.benchmarks.app
+
+import kotlin.time.Duration
+
+internal sealed interface BenchmarkResult {
+  data object Nil : BenchmarkResult
+}
+
+internal data class BenchmarkDuration(val size: Int, val duration: Duration) : BenchmarkResult {
+  val averageDuration: Duration
+    get() = duration.div(size)
+}
+
+internal operator fun BenchmarkDuration.plus(other: BenchmarkDuration): BenchmarkResult {
+  return BenchmarkDuration(this.size + other.size, this.duration + other.duration)
+}

engine/benchmarks/app/src/main/java/com/google/android/fhir/engine/benchmarks/app/CrudApiViewModel.kt

@@ -18,25 +18,138 @@ package com.google.android.fhir.engine.benchmarks.app
 
 import androidx.lifecycle.ViewModel
 import androidx.lifecycle.viewModelScope
+import androidx.tracing.trace
 import com.google.android.fhir.FhirEngine
 import com.google.android.fhir.engine.benchmarks.app.data.ResourcesDataProvider
-import java.time.LocalDateTime
+import kotlin.time.measureTime
+import kotlin.time.measureTimedValue
 import kotlinx.coroutines.flow.MutableStateFlow
-import kotlinx.coroutines.flow.StateFlow
 import kotlinx.coroutines.flow.asStateFlow
+import kotlinx.coroutines.flow.update
 import kotlinx.coroutines.launch
+import kotlinx.coroutines.runBlocking
+import org.hl7.fhir.r4.model.Resource
+import org.hl7.fhir.r4.model.ResourceType
 
 @Suppress("unused")
-class CrudApiViewModel(
+internal class CrudApiViewModel(
   private val resourcesDataProvider: ResourcesDataProvider,
   private val fhirEngine: FhirEngine,
 ) : ViewModel() {
-  private val _detailMutableStateFlow = MutableStateFlow("")
-  val detailStateFlow: StateFlow<String> = _detailMutableStateFlow.asStateFlow()
+
+  private val _crudMutableStateFlow = MutableStateFlow(CRUDUiState())
+  val crudStateFlow = _crudMutableStateFlow.asStateFlow()
 
   init {
-    viewModelScope.launch(benchmarkingViewModelWorkDispatcher) {
-      _detailMutableStateFlow.value = "CRUD: ${LocalDateTime.now()}"
+    viewModelScope.launch(benchmarkingViewModelWorkDispatcher) { traceCRUD() }
+  }
+
+  private suspend fun traceCRUD() {
+    // Create
+    fhirEngine.clearDatabase()
+    val savedResourceTypeIdPairs: MutableList<List<Pair<ResourceType, String>>> = mutableListOf()
+
+    resourcesDataProvider.provideResources { resources ->
+      val (logicalIds, duration) = traceCreateResources(resources)
+      val result = BenchmarkDuration(logicalIds.size, duration)
+      _crudMutableStateFlow.update {
+        val newUpdatedResult =
+          when (it.create) {
+            is BenchmarkResult.Nil -> result
+            is BenchmarkDuration -> it.create + result
+          }
+        it.copy(create = newUpdatedResult)
+      }
+
+      savedResourceTypeIdPairs += resources.zip(logicalIds) { r, l -> Pair(r.resourceType, l) }
+    }
+
+    // Get
+    val dbResources =
+      savedResourceTypeIdPairs.mapIndexed { index, list ->
+        val (resourceType, logicalId) = list.shuffled().random()
+        val (resource, duration) = traceGetResource(resourceType, logicalId)
+
+        val result = BenchmarkDuration(1, duration)
+        _crudMutableStateFlow.update {
+          val newUpdatedResult =
+            when (it.read) {
+              is BenchmarkResult.Nil -> result
+              is BenchmarkDuration -> it.read + result
+            }
+          it.copy(read = newUpdatedResult)
+        }
+        resource
+      }
+
+    // Update
+    val updateDbResources =
+      dbResources.shuffled().mapIndexed { index, resource ->
+        val duration = traceUpdateResources(listOf(resource))
+
+        val result = BenchmarkDuration(1, duration)
+        _crudMutableStateFlow.update {
+          val newUpdatedResult =
+            when (it.update) {
+              is BenchmarkResult.Nil -> result
+              is BenchmarkDuration -> it.update + result
+            }
+          it.copy(update = newUpdatedResult)
+        }
+        resource
+      }
+
+    // Delete
+    updateDbResources.shuffled().forEachIndexed { index, resource ->
+      val logicalId = resource.idElement?.idPart.orEmpty()
+      val duration = runBlocking { traceDeleteResources(resource.resourceType, logicalId) }
+      val result = BenchmarkDuration(1, duration)
+      _crudMutableStateFlow.update {
+        val newUpdatedResult =
+          when (it.delete) {
+            is BenchmarkResult.Nil -> result
+            is BenchmarkDuration -> it.delete + result
+          }
+        it.copy(delete = newUpdatedResult)
+      }
+    }
+  }
+
+  /**
+   * measureTime wraps to get time elapsed for display in UI while trace wraps for use in
+   * macrobenchmarking of the application as a TraceMetric
+   */
+  private fun traceCreateResources(resources: List<Resource>) = measureTimedValue {
+    trace(TRACE_CREATE_SECTION_NAME) {
+      runBlocking { fhirEngine.create(*resources.toTypedArray()) }
+    }
+  }
+
+  private fun traceUpdateResources(resources: List<Resource>) = measureTime {
+    trace(TRACE_UPDATE_SECTION_NAME) {
+      runBlocking { fhirEngine.update(*resources.toTypedArray()) }
     }
   }
+
+  private fun traceGetResource(resourceType: ResourceType, resourceId: String) = measureTimedValue {
+    trace(TRACE_GET_SECTION_NAME) { runBlocking { fhirEngine.get(resourceType, resourceId) } }
+  }
+
+  private fun traceDeleteResources(resourceType: ResourceType, resourceId: String) = measureTime {
+    trace(TRACE_DELETE_SECTION_NAME) { runBlocking { fhirEngine.delete(resourceType, resourceId) } }
+  }
+
+  companion object {
+    const val TRACE_CREATE_SECTION_NAME = "Create API"
+    const val TRACE_UPDATE_SECTION_NAME = "Update API"
+    const val TRACE_GET_SECTION_NAME = "Get API"
+    const val TRACE_DELETE_SECTION_NAME = "Delete API"
+  }
 }
+
+internal data class CRUDUiState(
+  val create: BenchmarkResult = BenchmarkResult.Nil,
+  val read: BenchmarkResult = BenchmarkResult.Nil,
+  val update: BenchmarkResult = BenchmarkResult.Nil,
+  val delete: BenchmarkResult = BenchmarkResult.Nil,
+)

engine/benchmarks/app/src/main/java/com/google/android/fhir/engine/benchmarks/app/MainApplication.kt

@@ -19,6 +19,7 @@ package com.google.android.fhir.engine.benchmarks.app
 import android.app.Application
 import android.content.Context
 import android.util.Log
+import androidx.tracing.Trace
 import com.google.android.fhir.DatabaseErrorStrategy.RECREATE_AT_OPEN
 import com.google.android.fhir.FhirEngine
 import com.google.android.fhir.FhirEngineConfiguration
@@ -31,6 +32,8 @@ class MainApplication : Application() {
 
   override fun onCreate() {
     super.onCreate()
+    Trace.forceEnableAppTracing()
+
     FhirEngineProvider.init(
       FhirEngineConfiguration(
         enableEncryptionIfSupported = true,

engine/benchmarks/app/src/main/java/com/google/android/fhir/engine/benchmarks/app/MainViewModel.kt

@@ -18,4 +18,4 @@ package com.google.android.fhir.engine.benchmarks.app
 
 import androidx.lifecycle.ViewModel
 
-class MainViewModel() : ViewModel()
+internal class MainViewModel() : ViewModel()

engine/benchmarks/app/src/main/java/com/google/android/fhir/engine/benchmarks/app/SearchApiViewModel.kt

@@ -27,7 +27,7 @@ import kotlinx.coroutines.flow.asStateFlow
 import kotlinx.coroutines.launch
 
 @Suppress("unused")
-class SearchApiViewModel(
+internal class SearchApiViewModel(
   private val resourcesDataProvider: ResourcesDataProvider,
   private val fhirEngine: FhirEngine,
 ) : ViewModel() {

engine/benchmarks/app/src/main/java/com/google/android/fhir/engine/benchmarks/app/SyncApiViewModel.kt

@@ -27,7 +27,7 @@ import kotlinx.coroutines.flow.asStateFlow
 import kotlinx.coroutines.launch
 
 @Suppress("unused")
-class SyncApiViewModel(
+internal class SyncApiViewModel(
   private val resourcesDataProvider: ResourcesDataProvider,
   private val fhirEngine: FhirEngine,
 ) : ViewModel() {

engine/benchmarks/app/src/main/java/com/google/android/fhir/engine/benchmarks/app/data/ResourcesDataProvider.kt

@@ -27,7 +27,7 @@ class ResourcesDataProvider(
   private val assetManager: AssetManager,
   private val fhirR4JsonParser: IParser,
 ) {
-  suspend fun provideResources(onResourcesProvided: (Iterable<Resource>) -> Unit) {
+  suspend fun provideResources(onResourcesProvided: (List<Resource>) -> Unit) {
     try {
       readFileResources()
         .chunked(READ_CHUNK_SIZE)

engine/benchmarks/app/src/main/java/com/google/android/fhir/engine/benchmarks/app/ui/CrudDetail.kt

@@ -18,27 +18,66 @@ package com.google.android.fhir.engine.benchmarks.app.ui
 
 import androidx.compose.foundation.layout.Arrangement
 import androidx.compose.foundation.layout.Column
-import androidx.compose.foundation.layout.fillMaxSize
+import androidx.compose.foundation.layout.Spacer
 import androidx.compose.foundation.layout.padding
+import androidx.compose.foundation.layout.size
+import androidx.compose.material3.MaterialTheme
 import androidx.compose.material3.Text
 import androidx.compose.runtime.Composable
+import androidx.compose.runtime.getValue
+import androidx.compose.runtime.remember
 import androidx.compose.ui.Modifier
-import androidx.compose.ui.text.style.TextAlign
+import androidx.compose.ui.tooling.preview.Preview
 import androidx.compose.ui.unit.dp
-import androidx.compose.ui.unit.sp
 import androidx.lifecycle.compose.collectAsStateWithLifecycle
+import com.google.android.fhir.engine.benchmarks.app.BenchmarkDuration
+import com.google.android.fhir.engine.benchmarks.app.BenchmarkResult
 import com.google.android.fhir.engine.benchmarks.app.CrudApiViewModel
+import kotlin.time.Duration.Companion.milliseconds
 
 @Composable
-fun CrudDetail(
+internal fun CrudDetail(
   viewModel: CrudApiViewModel,
   navigateToHome: () -> Unit,
 ) {
-  val detailState = viewModel.detailStateFlow.collectAsStateWithLifecycle()
+  val crudUiState = viewModel.crudStateFlow.collectAsStateWithLifecycle()
+  val crudUiStateValue by remember { crudUiState }
 
   DetailScaffold("CRUD", navigateToHome) {
-    Column(Modifier.fillMaxSize().padding(8.dp), verticalArrangement = Arrangement.Center) {
-      Text(detailState.value, fontSize = 20.sp, textAlign = TextAlign.Center)
+    Column(verticalArrangement = Arrangement.spacedBy(8.dp)) {
+      CrudBenchmarkResultView("FhirEngine#Create", crudUiStateValue.create)
+      CrudBenchmarkResultView("FhirEngine#Get", crudUiStateValue.read)
+      CrudBenchmarkResultView("FhirEngine#Update", crudUiStateValue.update)
+      CrudBenchmarkResultView("FhirEngine#Delete", crudUiStateValue.delete)
     }
   }
 }
+
+@Composable
+internal fun CrudBenchmarkResultView(headline: String, result: BenchmarkResult) {
+  Column(Modifier.padding(8.dp)) {
+    Text(headline, style = MaterialTheme.typography.headlineMedium)
+    Spacer(Modifier.size(8.dp))
+    when (result) {
+      is BenchmarkDuration -> {
+        Text("Takes ~${result.duration} for ${result.size} resources")
+        Text("Averages: ~${result.averageDuration}")
+      }
+      is BenchmarkResult.Nil -> {
+        Text("Waiting for results\u2026")
+      }
+    }
+  }
+}
+
+@Preview(showBackground = true)
+@Composable
+internal fun PreviewCrudBenchmarkResultView() {
+  CrudBenchmarkResultView("FhirEngine#Create", BenchmarkDuration(20, 1008.milliseconds))
+}
+
+@Preview(showBackground = true)
+@Composable
+internal fun PreviewCrudNilBenchmarkResultView() {
+  CrudBenchmarkResultView("FhirEngine#Create", BenchmarkResult.Nil)
+}

engine/benchmarks/app/src/main/java/com/google/android/fhir/engine/benchmarks/app/ui/DetailScaffold.kt

@@ -33,6 +33,8 @@ import androidx.compose.material3.TopAppBar
 import androidx.compose.runtime.Composable
 import androidx.compose.ui.Modifier
 import androidx.compose.ui.graphics.Color
+import androidx.compose.ui.semantics.semantics
+import androidx.compose.ui.semantics.testTagsAsResourceId
 import androidx.compose.ui.tooling.preview.Preview
 import androidx.compose.ui.unit.dp
 
@@ -57,7 +59,7 @@ fun DetailScaffold(
         },
       )
     },
-    modifier = Modifier.fillMaxSize(),
+    modifier = Modifier.fillMaxSize().semantics { testTagsAsResourceId = true },
   ) { innerPadding ->
     Card(
       shape = RoundedCornerShape(10.dp),

engine/benchmarks/app/src/main/java/com/google/android/fhir/engine/benchmarks/app/ui/Home.kt

@@ -40,8 +40,11 @@ import androidx.compose.runtime.Composable
 import androidx.compose.ui.Alignment
 import androidx.compose.ui.Modifier
 import androidx.compose.ui.graphics.Color
+import androidx.compose.ui.platform.testTag
 import androidx.compose.ui.res.painterResource
 import androidx.compose.ui.res.stringResource
+import androidx.compose.ui.semantics.semantics
+import androidx.compose.ui.semantics.testTagsAsResourceId
 import androidx.compose.ui.tooling.preview.Preview
 import androidx.compose.ui.unit.dp
 import androidx.compose.ui.unit.sp
@@ -57,7 +60,7 @@ fun Home(
 ) {
   Scaffold(
     topBar = { TopAppBar(title = { Text(stringResource(R.string.app_name)) }) },
-    modifier = Modifier.fillMaxSize(),
+    modifier = Modifier.fillMaxSize().semantics { testTagsAsResourceId = true },
   ) { innerPadding ->
     Box(
       Modifier.padding(innerPadding).padding(horizontal = 16.dp),
@@ -66,18 +69,21 @@ fun Home(
         ActionCard(
           title = "CRUD",
           imageId = R.drawable.ic_outline_database_24,
+          modifier = Modifier.testTag("crudBenchmarkSection"),
           onActionClick = navigateToCrudScreen,
         )
 
         ActionCard(
           title = "Search API",
           imageId = R.drawable.ic_home_search,
+          modifier = Modifier.testTag("searchBenchmarkSection"),
           onActionClick = navigateToSearchScreen,
         )
 
         ActionCard(
           title = "Sync API",
           imageId = R.drawable.ic_home_sync,
+          modifier = Modifier.testTag("syncBenchmarkSection"),
           onActionClick = navigateToSyncScreen,
         )
       }

engine/benchmarks/app/src/main/java/com/google/android/fhir/engine/benchmarks/app/ui/SearchApiDetail.kt

@@ -30,7 +30,7 @@ import androidx.lifecycle.compose.collectAsStateWithLifecycle
 import com.google.android.fhir.engine.benchmarks.app.SearchApiViewModel
 
 @Composable
-fun SearchApiDetail(viewModel: SearchApiViewModel, navigateToHome: () -> Unit) {
+internal fun SearchApiDetail(viewModel: SearchApiViewModel, navigateToHome: () -> Unit) {
   val detailState = viewModel.detailStateFlow.collectAsStateWithLifecycle()
 
   DetailScaffold("Search API", navigateToHome) {

engine/benchmarks/app/src/main/java/com/google/android/fhir/engine/benchmarks/app/ui/SyncApiDetail.kt

@@ -30,7 +30,7 @@ import androidx.lifecycle.compose.collectAsStateWithLifecycle
 import com.google.android.fhir.engine.benchmarks.app.SyncApiViewModel
 
 @Composable
-fun SyncApiDetail(viewModel: SyncApiViewModel, navigateToHome: () -> Unit) {
+internal fun SyncApiDetail(viewModel: SyncApiViewModel, navigateToHome: () -> Unit) {
   val detailState = viewModel.detailStateFlow.collectAsStateWithLifecycle()
 
   DetailScaffold("Sync API", navigateToHome) {