Android/frameworks_base
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge "Add race() utility function" into tm-qpr-dev am: ca319b38f8

Browse Source 
Original change: https://googleplex-android-review.googlesource.com/c/platform/frameworks/base/+/19859998

Change-Id: I2fb9a59690e2c04aa73c803eb02ea47bb268a681
Signed-off-by: Automerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com>
This commit is contained in:
TreeHugger Robot
2022-09-09 13:55:57 +00:00
committed by Automerger Merge Worker
parent f2ba60efaf ca319b38f8
commit 40575dece1
2 changed files with 90 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
packages/SystemUI/src/com/android/systemui/util/kotlin/Suspend.kt Normal file
View File

@@ -0,0 +1,35 @@
/*
* Copyright (C) 2022 The Android Open Source Project
*
* 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.android.systemui.util.kotlin
import kotlinx.coroutines.CompletableDeferred
import kotlinx.coroutines.coroutineScope
import kotlinx.coroutines.launch
/**
* Runs the given [blocks] in parallel, returning the result of the first one to complete, and
* cancelling all others.
*/
suspend fun <R> race(vararg blocks: suspend () -> R): R = coroutineScope {
val completion = CompletableDeferred<R>()
val raceJob = launch {
for (block in blocks) {
launch { completion.complete(block()) }
}
}
completion.await().also { raceJob.cancel() }
}

55
packages/SystemUI/tests/src/com/android/systemui/util/kotlin/SuspendUtilTests.kt Normal file
View File

@@ -0,0 +1,55 @@
/*
* Copyright (C) 2022 The Android Open Source Project
*
* 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.android.systemui.util.kotlin
import android.testing.AndroidTestingRunner
import androidx.test.filters.SmallTest
import com.android.systemui.SysuiTestCase
import com.google.common.truth.Truth.assertThat
import kotlinx.coroutines.CompletableDeferred
import kotlinx.coroutines.async
import kotlinx.coroutines.awaitCancellation
import kotlinx.coroutines.runBlocking
import org.junit.Test
import org.junit.runner.RunWith
@SmallTest
@RunWith(AndroidTestingRunner::class)
class RaceSuspendTest : SysuiTestCase() {
@Test
fun raceSimple() = runBlocking {
val winner = CompletableDeferred<Int>()
val result = async {
race(
{ winner.await() },
{ awaitCancellation() },
)
}
winner.complete(1)
assertThat(result.await()).isEqualTo(1)
}
@Test
fun raceImmediate() = runBlocking {
assertThat(
race<Int>(
{ 1 },
{ 2 },
)
).isEqualTo(1)
}
}