fix: keep provider state during storage outages
This commit is contained in:
+4
-1
@@ -7,6 +7,8 @@ import androidx.compose.foundation.layout.fillMaxSize
|
|||||||
import androidx.compose.foundation.layout.fillMaxWidth
|
import androidx.compose.foundation.layout.fillMaxWidth
|
||||||
import androidx.compose.foundation.layout.padding
|
import androidx.compose.foundation.layout.padding
|
||||||
import androidx.compose.foundation.layout.widthIn
|
import androidx.compose.foundation.layout.widthIn
|
||||||
|
import androidx.compose.foundation.rememberScrollState
|
||||||
|
import androidx.compose.foundation.verticalScroll
|
||||||
import androidx.compose.material3.ExperimentalMaterial3Api
|
import androidx.compose.material3.ExperimentalMaterial3Api
|
||||||
import androidx.compose.material3.MaterialTheme
|
import androidx.compose.material3.MaterialTheme
|
||||||
import androidx.compose.material3.Scaffold
|
import androidx.compose.material3.Scaffold
|
||||||
@@ -45,8 +47,9 @@ fun AuthenticatedPlaceholderScreen(
|
|||||||
) {
|
) {
|
||||||
Column(
|
Column(
|
||||||
modifier = Modifier
|
modifier = Modifier
|
||||||
.fillMaxWidth()
|
|
||||||
.widthIn(max = 520.dp)
|
.widthIn(max = 520.dp)
|
||||||
|
.fillMaxWidth()
|
||||||
|
.verticalScroll(rememberScrollState())
|
||||||
.padding(24.dp),
|
.padding(24.dp),
|
||||||
horizontalAlignment = Alignment.CenterHorizontally,
|
horizontalAlignment = Alignment.CenterHorizontally,
|
||||||
verticalArrangement = Arrangement.spacedBy(14.dp),
|
verticalArrangement = Arrangement.spacedBy(14.dp),
|
||||||
|
|||||||
@@ -118,8 +118,8 @@ fun LoginScreen(
|
|||||||
) {
|
) {
|
||||||
Column(
|
Column(
|
||||||
modifier = Modifier
|
modifier = Modifier
|
||||||
.fillMaxWidth()
|
|
||||||
.widthIn(max = 520.dp)
|
.widthIn(max = 520.dp)
|
||||||
|
.fillMaxWidth()
|
||||||
.verticalScroll(rememberScrollState())
|
.verticalScroll(rememberScrollState())
|
||||||
.padding(horizontal = 24.dp, vertical = 16.dp),
|
.padding(horizontal = 24.dp, vertical = 16.dp),
|
||||||
verticalArrangement = Arrangement.spacedBy(16.dp),
|
verticalArrangement = Arrangement.spacedBy(16.dp),
|
||||||
|
|||||||
+1
-1
@@ -65,8 +65,8 @@ fun ProviderSelectionScreen(
|
|||||||
) {
|
) {
|
||||||
Column(
|
Column(
|
||||||
modifier = Modifier
|
modifier = Modifier
|
||||||
.fillMaxWidth()
|
|
||||||
.widthIn(max = 640.dp)
|
.widthIn(max = 640.dp)
|
||||||
|
.fillMaxWidth()
|
||||||
.verticalScroll(rememberScrollState())
|
.verticalScroll(rememberScrollState())
|
||||||
.padding(horizontal = 20.dp, vertical = 12.dp),
|
.padding(horizontal = 20.dp, vertical = 12.dp),
|
||||||
verticalArrangement = Arrangement.spacedBy(14.dp),
|
verticalArrangement = Arrangement.spacedBy(14.dp),
|
||||||
|
|||||||
+12
-28
@@ -4,7 +4,6 @@ import androidx.datastore.core.DataStore
|
|||||||
import androidx.datastore.preferences.core.Preferences
|
import androidx.datastore.preferences.core.Preferences
|
||||||
import androidx.datastore.preferences.core.booleanPreferencesKey
|
import androidx.datastore.preferences.core.booleanPreferencesKey
|
||||||
import androidx.datastore.preferences.core.edit
|
import androidx.datastore.preferences.core.edit
|
||||||
import androidx.datastore.preferences.core.emptyPreferences
|
|
||||||
import androidx.datastore.preferences.core.stringPreferencesKey
|
import androidx.datastore.preferences.core.stringPreferencesKey
|
||||||
import androidx.datastore.preferences.core.stringSetPreferencesKey
|
import androidx.datastore.preferences.core.stringSetPreferencesKey
|
||||||
import com.flagship.abox.manager.domain.DomainResult
|
import com.flagship.abox.manager.domain.DomainResult
|
||||||
@@ -18,35 +17,26 @@ import java.io.IOException
|
|||||||
import kotlinx.coroutines.CancellationException
|
import kotlinx.coroutines.CancellationException
|
||||||
import kotlinx.coroutines.delay
|
import kotlinx.coroutines.delay
|
||||||
import kotlinx.coroutines.flow.Flow
|
import kotlinx.coroutines.flow.Flow
|
||||||
import kotlinx.coroutines.flow.catch
|
|
||||||
import kotlinx.coroutines.flow.MutableSharedFlow
|
|
||||||
import kotlinx.coroutines.flow.distinctUntilChanged
|
import kotlinx.coroutines.flow.distinctUntilChanged
|
||||||
import kotlinx.coroutines.flow.first
|
import kotlinx.coroutines.flow.first
|
||||||
import kotlinx.coroutines.flow.flatMapLatest
|
|
||||||
import kotlinx.coroutines.flow.flowOf
|
|
||||||
import kotlinx.coroutines.flow.map
|
import kotlinx.coroutines.flow.map
|
||||||
import kotlinx.coroutines.flow.onStart
|
|
||||||
import kotlinx.coroutines.flow.retryWhen
|
import kotlinx.coroutines.flow.retryWhen
|
||||||
|
|
||||||
@OptIn(kotlinx.coroutines.ExperimentalCoroutinesApi::class)
|
|
||||||
class ProviderProfileStoreImpl(
|
class ProviderProfileStoreImpl(
|
||||||
private val dataStore: DataStore<Preferences>,
|
private val dataStore: DataStore<Preferences>,
|
||||||
) : ProviderProfileStore {
|
) : ProviderProfileStore {
|
||||||
private val refresh = MutableSharedFlow<Unit>(extraBufferCapacity = 1)
|
|
||||||
private val safeData: Flow<Preferences>
|
private val safeData: Flow<Preferences>
|
||||||
get() = refresh
|
get() = dataStore.data
|
||||||
.onStart { emit(Unit) }
|
.retryWhen { cause, attempt ->
|
||||||
.flatMapLatest { readDataOnce() }
|
if (cause !is IOException) return@retryWhen false
|
||||||
|
val delayMillis = (READ_RETRY_DELAY_MILLIS shl
|
||||||
|
attempt.coerceAtMost(MAX_BACKOFF_SHIFT).toInt())
|
||||||
|
.coerceAtMost(MAX_READ_RETRY_DELAY_MILLIS)
|
||||||
|
delay(delayMillis)
|
||||||
|
true
|
||||||
|
}
|
||||||
.distinctUntilChanged()
|
.distinctUntilChanged()
|
||||||
|
|
||||||
private fun readDataOnce(): Flow<Preferences> = dataStore.data
|
|
||||||
.retryWhen { cause, attempt ->
|
|
||||||
val retry = cause is IOException && attempt < MAX_READ_RETRIES
|
|
||||||
if (retry) delay(READ_RETRY_DELAY_MILLIS)
|
|
||||||
retry
|
|
||||||
}
|
|
||||||
.catchIOExceptionAsEmpty()
|
|
||||||
|
|
||||||
override fun observeProfiles(): Flow<List<ProviderProfile>> = safeData.map(::profilesFrom)
|
override fun observeProfiles(): Flow<List<ProviderProfile>> = safeData.map(::profilesFrom)
|
||||||
|
|
||||||
override fun observeActiveProfile(): Flow<ProviderProfile?> = safeData.map(::activeProfileFrom)
|
override fun observeActiveProfile(): Flow<ProviderProfile?> = safeData.map(::activeProfileFrom)
|
||||||
@@ -67,7 +57,6 @@ class ProviderProfileStoreImpl(
|
|||||||
preferences[profileTypeKey(profile.id)] = profile.type.name
|
preferences[profileTypeKey(profile.id)] = profile.type.name
|
||||||
preferences[profileNameKey(profile.id)] = profile.displayName
|
preferences[profileNameKey(profile.id)] = profile.displayName
|
||||||
}
|
}
|
||||||
refresh.tryEmit(Unit)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
override suspend fun setActiveProfile(profileId: ProviderProfileId?): DomainResult<Unit> = operation {
|
override suspend fun setActiveProfile(profileId: ProviderProfileId?): DomainResult<Unit> = operation {
|
||||||
@@ -75,7 +64,6 @@ class ProviderProfileStoreImpl(
|
|||||||
if (profileId == null) preferences.remove(ACTIVE_PROFILE_ID)
|
if (profileId == null) preferences.remove(ACTIVE_PROFILE_ID)
|
||||||
else preferences[ACTIVE_PROFILE_ID] = profileId.value
|
else preferences[ACTIVE_PROFILE_ID] = profileId.value
|
||||||
}
|
}
|
||||||
refresh.tryEmit(Unit)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
override suspend fun saveLoginPreferences(
|
override suspend fun saveLoginPreferences(
|
||||||
@@ -89,7 +77,6 @@ class ProviderProfileStoreImpl(
|
|||||||
if (username == null) values.remove(usernameKey(profileId))
|
if (username == null) values.remove(usernameKey(profileId))
|
||||||
else values[usernameKey(profileId)] = username
|
else values[usernameKey(profileId)] = username
|
||||||
}
|
}
|
||||||
refresh.tryEmit(Unit)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
private fun profilesFrom(preferences: Preferences): List<ProviderProfile> =
|
private fun profilesFrom(preferences: Preferences): List<ProviderProfile> =
|
||||||
@@ -131,8 +118,9 @@ class ProviderProfileStoreImpl(
|
|||||||
}
|
}
|
||||||
|
|
||||||
companion object {
|
companion object {
|
||||||
private const val MAX_READ_RETRIES = 2L
|
private const val READ_RETRY_DELAY_MILLIS = 100L
|
||||||
private const val READ_RETRY_DELAY_MILLIS = 50L
|
private const val MAX_READ_RETRY_DELAY_MILLIS = 5_000L
|
||||||
|
private const val MAX_BACKOFF_SHIFT = 6L
|
||||||
private val PROFILE_IDS = stringSetPreferencesKey("profile_ids")
|
private val PROFILE_IDS = stringSetPreferencesKey("profile_ids")
|
||||||
private val ACTIVE_PROFILE_ID = stringPreferencesKey("active_profile_id")
|
private val ACTIVE_PROFILE_ID = stringPreferencesKey("active_profile_id")
|
||||||
|
|
||||||
@@ -152,7 +140,3 @@ class ProviderProfileStoreImpl(
|
|||||||
booleanPreferencesKey("profile_${id.value}_auto_login")
|
booleanPreferencesKey("profile_${id.value}_auto_login")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private fun Flow<Preferences>.catchIOExceptionAsEmpty(): Flow<Preferences> = catch { error ->
|
|
||||||
if (error is IOException) emit(emptyPreferences()) else throw error
|
|
||||||
}
|
|
||||||
|
|||||||
+10
-54
@@ -3,58 +3,32 @@ package com.flagship.abox.manager.session
|
|||||||
import androidx.datastore.core.DataStore
|
import androidx.datastore.core.DataStore
|
||||||
import androidx.datastore.preferences.core.Preferences
|
import androidx.datastore.preferences.core.Preferences
|
||||||
import androidx.datastore.preferences.core.emptyPreferences
|
import androidx.datastore.preferences.core.emptyPreferences
|
||||||
import com.flagship.abox.manager.domain.ProviderProfile
|
|
||||||
import com.flagship.abox.manager.domain.ProviderProfileId
|
import com.flagship.abox.manager.domain.ProviderProfileId
|
||||||
import com.flagship.abox.manager.domain.ProviderType
|
|
||||||
import com.flagship.abox.manager.session.profile.ProviderProfileStoreImpl
|
import com.flagship.abox.manager.session.profile.ProviderProfileStoreImpl
|
||||||
import java.io.IOException
|
import java.io.IOException
|
||||||
import kotlinx.coroutines.flow.Flow
|
import kotlinx.coroutines.flow.Flow
|
||||||
import kotlinx.coroutines.flow.first
|
|
||||||
import kotlinx.coroutines.flow.flow
|
import kotlinx.coroutines.flow.flow
|
||||||
import kotlinx.coroutines.flow.take
|
|
||||||
import kotlinx.coroutines.flow.toList
|
|
||||||
import kotlinx.coroutines.launch
|
import kotlinx.coroutines.launch
|
||||||
import kotlinx.coroutines.test.advanceTimeBy
|
import kotlinx.coroutines.test.advanceTimeBy
|
||||||
import kotlinx.coroutines.test.advanceUntilIdle
|
|
||||||
import kotlinx.coroutines.test.runTest
|
import kotlinx.coroutines.test.runTest
|
||||||
import org.junit.Assert.assertEquals
|
import kotlinx.coroutines.withTimeout
|
||||||
|
import org.junit.Assert.assertFalse
|
||||||
import org.junit.Test
|
import org.junit.Test
|
||||||
|
|
||||||
class ProviderProfileStoreFailureTest {
|
class ProviderProfileStoreFailureTest {
|
||||||
@Test
|
|
||||||
fun ioFailureFallsBackToEmptyPreferences() = runTest {
|
|
||||||
val store = ProviderProfileStoreImpl(FailingReadDataStore())
|
|
||||||
|
|
||||||
assertEquals(emptyList<Any>(), store.observeProfiles().first())
|
|
||||||
assertEquals(null, store.observeActiveProfile().first())
|
|
||||||
assertEquals(emptyList<Any>(), store.profiles())
|
|
||||||
assertEquals(null, store.activeProfile())
|
|
||||||
assertEquals(false, store.loginPreferences(ProviderProfileId("profile")).autoLoginEnabled)
|
|
||||||
}
|
|
||||||
|
|
||||||
@OptIn(kotlinx.coroutines.ExperimentalCoroutinesApi::class)
|
@OptIn(kotlinx.coroutines.ExperimentalCoroutinesApi::class)
|
||||||
@Test
|
@Test
|
||||||
fun observerResubscribesAfterSuccessfulWrite() = runTest {
|
fun persistentIoFailureDoesNotEmitFalseEmptyState() = runTest {
|
||||||
val store = ProviderProfileStoreImpl(RecoveringReadDataStore())
|
val store = ProviderProfileStoreImpl(FailingReadDataStore())
|
||||||
val emissions = mutableListOf<ProviderProfile?>()
|
var completed = false
|
||||||
val collector = backgroundScope.launch {
|
|
||||||
store.observeActiveProfile().take(2).toList(emissions)
|
backgroundScope.launch {
|
||||||
|
runCatching { withTimeout(250) { store.activeProfile() } }
|
||||||
|
completed = true
|
||||||
}
|
}
|
||||||
advanceTimeBy(200)
|
advanceTimeBy(200)
|
||||||
advanceUntilIdle()
|
|
||||||
assertEquals(listOf<ProviderProfile?>(null), emissions)
|
|
||||||
|
|
||||||
val profile = ProviderProfile(
|
assertFalse(completed)
|
||||||
ProviderProfileId("recovered"),
|
|
||||||
ProviderType.LOCAL_EXPERIENCE,
|
|
||||||
"Recovered",
|
|
||||||
)
|
|
||||||
store.saveProfile(profile)
|
|
||||||
store.setActiveProfile(profile.id)
|
|
||||||
advanceUntilIdle()
|
|
||||||
collector.join()
|
|
||||||
|
|
||||||
assertEquals(listOf(null, profile), emissions)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
private class FailingReadDataStore : DataStore<Preferences> {
|
private class FailingReadDataStore : DataStore<Preferences> {
|
||||||
@@ -66,22 +40,4 @@ class ProviderProfileStoreFailureTest {
|
|||||||
transform: suspend (t: Preferences) -> Preferences,
|
transform: suspend (t: Preferences) -> Preferences,
|
||||||
): Preferences = transform(emptyPreferences())
|
): Preferences = transform(emptyPreferences())
|
||||||
}
|
}
|
||||||
|
|
||||||
private class RecoveringReadDataStore : DataStore<Preferences> {
|
|
||||||
private var attempts = 0
|
|
||||||
private var current: Preferences = emptyPreferences()
|
|
||||||
|
|
||||||
override val data: Flow<Preferences>
|
|
||||||
get() = flow {
|
|
||||||
if (attempts++ < 3) throw IOException("temporarily unavailable")
|
|
||||||
emit(current)
|
|
||||||
}
|
|
||||||
|
|
||||||
override suspend fun updateData(
|
|
||||||
transform: suspend (t: Preferences) -> Preferences,
|
|
||||||
): Preferences {
|
|
||||||
current = transform(current)
|
|
||||||
return current
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user