Clean up forked collectAsStateWithLifecycle
Use the the one from androidx.lifecycle.compose instead. Bug: 236346018 Test: Unit test Change-Id: I90217ed9eb6ee0e97db87edf4e69e8255fcc8197
This commit is contained in:
@@ -1,189 +0,0 @@
|
||||
/*
|
||||
* Copyright 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.settingslib.spa.framework.compose
|
||||
|
||||
import android.annotation.SuppressLint
|
||||
import androidx.compose.runtime.Composable
|
||||
import androidx.compose.runtime.State
|
||||
import androidx.compose.runtime.produceState
|
||||
import androidx.compose.ui.platform.LocalLifecycleOwner
|
||||
import androidx.lifecycle.Lifecycle
|
||||
import androidx.lifecycle.LifecycleOwner
|
||||
import androidx.lifecycle.repeatOnLifecycle
|
||||
import kotlinx.coroutines.flow.Flow
|
||||
import kotlinx.coroutines.flow.StateFlow
|
||||
import kotlinx.coroutines.withContext
|
||||
import kotlin.coroutines.CoroutineContext
|
||||
import kotlin.coroutines.EmptyCoroutineContext
|
||||
|
||||
/**
|
||||
* *************************************************************************************************
|
||||
* This file was forked from AndroidX:
|
||||
* lifecycle/lifecycle-runtime-compose/src/main/java/androidx/lifecycle/compose/FlowExt.kt
|
||||
* TODO: Replace with AndroidX when it's usable.
|
||||
*/
|
||||
|
||||
/**
|
||||
* Collects values from this [StateFlow] and represents its latest value via [State] in a
|
||||
* lifecycle-aware manner.
|
||||
*
|
||||
* The [StateFlow.value] is used as an initial value. Every time there would be new value posted
|
||||
* into the [StateFlow] the returned [State] will be updated causing recomposition of every
|
||||
* [State.value] usage whenever the [lifecycleOwner]'s lifecycle is at least [minActiveState].
|
||||
*
|
||||
* This [StateFlow] is collected every time the [lifecycleOwner]'s lifecycle reaches the
|
||||
* [minActiveState] Lifecycle state. The collection stops when the [lifecycleOwner]'s lifecycle
|
||||
* falls below [minActiveState].
|
||||
*
|
||||
* @sample androidx.lifecycle.compose.samples.StateFlowCollectAsStateWithLifecycle
|
||||
*
|
||||
* Warning: [Lifecycle.State.INITIALIZED] is not allowed in this API. Passing it as a
|
||||
* parameter will throw an [IllegalArgumentException].
|
||||
*
|
||||
* @param lifecycleOwner [LifecycleOwner] whose `lifecycle` is used to restart collecting `this`
|
||||
* flow.
|
||||
* @param minActiveState [Lifecycle.State] in which the upstream flow gets collected. The
|
||||
* collection will stop if the lifecycle falls below that state, and will restart if it's in that
|
||||
* state again.
|
||||
* @param context [CoroutineContext] to use for collecting.
|
||||
*/
|
||||
@SuppressLint("StateFlowValueCalledInComposition")
|
||||
@Composable
|
||||
fun <T> StateFlow<T>.collectAsStateWithLifecycle(
|
||||
lifecycleOwner: LifecycleOwner = LocalLifecycleOwner.current,
|
||||
minActiveState: Lifecycle.State = Lifecycle.State.STARTED,
|
||||
context: CoroutineContext = EmptyCoroutineContext
|
||||
): State<T> = collectAsStateWithLifecycle(
|
||||
initialValue = this.value,
|
||||
lifecycle = lifecycleOwner.lifecycle,
|
||||
minActiveState = minActiveState,
|
||||
context = context
|
||||
)
|
||||
|
||||
/**
|
||||
* Collects values from this [StateFlow] and represents its latest value via [State] in a
|
||||
* lifecycle-aware manner.
|
||||
*
|
||||
* The [StateFlow.value] is used as an initial value. Every time there would be new value posted
|
||||
* into the [StateFlow] the returned [State] will be updated causing recomposition of every
|
||||
* [State.value] usage whenever the [lifecycle] is at least [minActiveState].
|
||||
*
|
||||
* This [StateFlow] is collected every time [lifecycle] reaches the [minActiveState] Lifecycle
|
||||
* state. The collection stops when [lifecycle] falls below [minActiveState].
|
||||
*
|
||||
* @sample androidx.lifecycle.compose.samples.StateFlowCollectAsStateWithLifecycle
|
||||
*
|
||||
* Warning: [Lifecycle.State.INITIALIZED] is not allowed in this API. Passing it as a
|
||||
* parameter will throw an [IllegalArgumentException].
|
||||
*
|
||||
* @param lifecycle [Lifecycle] used to restart collecting `this` flow.
|
||||
* @param minActiveState [Lifecycle.State] in which the upstream flow gets collected. The
|
||||
* collection will stop if the lifecycle falls below that state, and will restart if it's in that
|
||||
* state again.
|
||||
* @param context [CoroutineContext] to use for collecting.
|
||||
*/
|
||||
@SuppressLint("StateFlowValueCalledInComposition")
|
||||
@Composable
|
||||
fun <T> StateFlow<T>.collectAsStateWithLifecycle(
|
||||
lifecycle: Lifecycle,
|
||||
minActiveState: Lifecycle.State = Lifecycle.State.STARTED,
|
||||
context: CoroutineContext = EmptyCoroutineContext
|
||||
): State<T> = collectAsStateWithLifecycle(
|
||||
initialValue = this.value,
|
||||
lifecycle = lifecycle,
|
||||
minActiveState = minActiveState,
|
||||
context = context
|
||||
)
|
||||
|
||||
/**
|
||||
* Collects values from this [Flow] and represents its latest value via [State] in a
|
||||
* lifecycle-aware manner.
|
||||
*
|
||||
* Every time there would be new value posted into the [Flow] the returned [State] will be updated
|
||||
* causing recomposition of every [State.value] usage whenever the [lifecycleOwner]'s lifecycle is
|
||||
* at least [minActiveState].
|
||||
*
|
||||
* This [Flow] is collected every time the [lifecycleOwner]'s lifecycle reaches the [minActiveState]
|
||||
* Lifecycle state. The collection stops when the [lifecycleOwner]'s lifecycle falls below
|
||||
* [minActiveState].
|
||||
*
|
||||
* @sample androidx.lifecycle.compose.samples.FlowCollectAsStateWithLifecycle
|
||||
*
|
||||
* Warning: [Lifecycle.State.INITIALIZED] is not allowed in this API. Passing it as a
|
||||
* parameter will throw an [IllegalArgumentException].
|
||||
*
|
||||
* @param initialValue The initial value given to the returned [State.value].
|
||||
* @param lifecycleOwner [LifecycleOwner] whose `lifecycle` is used to restart collecting `this`
|
||||
* flow.
|
||||
* @param minActiveState [Lifecycle.State] in which the upstream flow gets collected. The
|
||||
* collection will stop if the lifecycle falls below that state, and will restart if it's in that
|
||||
* state again.
|
||||
* @param context [CoroutineContext] to use for collecting.
|
||||
*/
|
||||
@Composable
|
||||
fun <T> Flow<T>.collectAsStateWithLifecycle(
|
||||
initialValue: T,
|
||||
lifecycleOwner: LifecycleOwner = LocalLifecycleOwner.current,
|
||||
minActiveState: Lifecycle.State = Lifecycle.State.STARTED,
|
||||
context: CoroutineContext = EmptyCoroutineContext
|
||||
): State<T> = collectAsStateWithLifecycle(
|
||||
initialValue = initialValue,
|
||||
lifecycle = lifecycleOwner.lifecycle,
|
||||
minActiveState = minActiveState,
|
||||
context = context
|
||||
)
|
||||
|
||||
/**
|
||||
* Collects values from this [Flow] and represents its latest value via [State] in a
|
||||
* lifecycle-aware manner.
|
||||
*
|
||||
* Every time there would be new value posted into the [Flow] the returned [State] will be updated
|
||||
* causing recomposition of every [State.value] usage whenever the [lifecycle] is at
|
||||
* least [minActiveState].
|
||||
*
|
||||
* This [Flow] is collected every time [lifecycle] reaches the [minActiveState] Lifecycle
|
||||
* state. The collection stops when [lifecycle] falls below [minActiveState].
|
||||
*
|
||||
* @sample androidx.lifecycle.compose.samples.FlowCollectAsStateWithLifecycle
|
||||
*
|
||||
* Warning: [Lifecycle.State.INITIALIZED] is not allowed in this API. Passing it as a
|
||||
* parameter will throw an [IllegalArgumentException].
|
||||
*
|
||||
* @param initialValue The initial value given to the returned [State.value].
|
||||
* @param lifecycle [Lifecycle] used to restart collecting `this` flow.
|
||||
* @param minActiveState [Lifecycle.State] in which the upstream flow gets collected. The
|
||||
* collection will stop if the lifecycle falls below that state, and will restart if it's in that
|
||||
* state again.
|
||||
* @param context [CoroutineContext] to use for collecting.
|
||||
*/
|
||||
@Composable
|
||||
fun <T> Flow<T>.collectAsStateWithLifecycle(
|
||||
initialValue: T,
|
||||
lifecycle: Lifecycle,
|
||||
minActiveState: Lifecycle.State = Lifecycle.State.STARTED,
|
||||
context: CoroutineContext = EmptyCoroutineContext
|
||||
): State<T> {
|
||||
return produceState(initialValue, this, lifecycle, minActiveState, context) {
|
||||
lifecycle.repeatOnLifecycle(minActiveState) {
|
||||
if (context == EmptyCoroutineContext) {
|
||||
this@collectAsStateWithLifecycle.collect { this@produceState.value = it }
|
||||
} else withContext(context) {
|
||||
this@collectAsStateWithLifecycle.collect { this@produceState.value = it }
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user