Merge pull request #102 from soil-kt/purge

Add `purgeAll` API
soil-kt · Sep 29, 2024 · 17ddc16 · 17ddc16
2 parents 5bda7b2 + 09a6b99
commit 17ddc16
Show file tree

Hide file tree

Showing 7 changed files with 97 additions and 37 deletions.
diff --git a/soil-query-compose/src/commonMain/kotlin/soil/query/compose/tooling/SwrPreviewClient.kt b/soil-query-compose/src/commonMain/kotlin/soil/query/compose/tooling/SwrPreviewClient.kt
@@ -34,6 +34,7 @@ class SwrPreviewClient(
     override val errorRelay: Flow<ErrorRecord> = flow { }
 ) : SwrClient, SwrClientPlus, QueryClient by query, MutationClient by mutation, SubscriptionClient by subscription {
     override fun gc(level: MemoryPressureLevel) = Unit
+    override fun purgeAll() = Unit
     override fun perform(sideEffects: QueryEffect): Job = Job()
     override fun onMount(id: String) = Unit
     override fun onUnmount(id: String) = Unit

diff --git a/soil-query-core/src/commonMain/kotlin/soil/query/SwrCache.kt b/soil-query-core/src/commonMain/kotlin/soil/query/SwrCache.kt
@@ -95,6 +95,24 @@ open class SwrCache(private val policy: SwrCachePolicy) : SwrClient, QueryMutabl
         }
     }
 
+    override fun purgeAll() {
+        purgeAllQueries()
+        purgeAllMutations()
+    }
+
+    private fun purgeAllQueries() {
+        val queryStoreCopy = queryStore.toMap()
+        queryStore.clear()
+        queryCache.clear()
+        queryStoreCopy.values.forEach { it.close() }
+    }
+
+    private fun purgeAllMutations() {
+        val mutationStoreCopy = mutationStore.toMap()
+        mutationStore.clear()
+        mutationStoreCopy.values.forEach { it.close() }
+    }
+
     override fun perform(sideEffects: QueryEffect): Job {
         return coroutineScope.launch {
             with(this@SwrCache) { sideEffects() }
@@ -577,6 +595,7 @@ open class SwrCache(private val policy: SwrCachePolicy) : SwrClient, QueryMutabl
         return model?.let(predicate) ?: false
     }
 
+
     internal class ManagedMutation<T>(
         val scope: CoroutineScope,
         val id: UniqueId,

diff --git a/soil-query-core/src/commonMain/kotlin/soil/query/SwrCachePlus.kt b/soil-query-core/src/commonMain/kotlin/soil/query/SwrCachePlus.kt
@@ -48,6 +48,9 @@ class SwrCachePlus(private val policy: SwrCachePlusPolicy) : SwrCache(policy), S
     )
     private val batchScheduler: BatchScheduler = policy.batchSchedulerFactory.create(coroutineScope)
 
+
+    // ----- SwrClientPlus ----- //
+
     override val defaultSubscriptionOptions: SubscriptionOptions = policy.subscriptionOptions
 
     override fun gc(level: MemoryPressureLevel) {
@@ -58,6 +61,18 @@ class SwrCachePlus(private val policy: SwrCachePlusPolicy) : SwrCache(policy), S
         }
     }
 
+    override fun purgeAll() {
+        super.purgeAll()
+        purgeAllSubscriptions()
+    }
+
+    private fun purgeAllSubscriptions() {
+        val subscriptionStoreCopy = subscriptionStore.toMap()
+        subscriptionStore.clear()
+        subscriptionCache.clear()
+        subscriptionStoreCopy.values.forEach { it.close() }
+    }
+
     @Suppress("UNCHECKED_CAST")
     override fun <T> getSubscription(
         key: SubscriptionKey<T>,

diff --git a/soil-query-core/src/commonMain/kotlin/soil/query/SwrClient.kt b/soil-query-core/src/commonMain/kotlin/soil/query/SwrClient.kt
@@ -28,6 +28,15 @@ interface SwrClient : MutationClient, QueryClient {
      */
     fun gc(level: MemoryPressureLevel = MemoryPressureLevel.Low)
 
+    /**
+     * Removes all queries and mutations from the in-memory stored data.
+     *
+     * **Note:**
+     * If there are any active queries or mutations, they will be stopped as well.
+     * This method should only be used for full resets, such as during sign-out.
+     */
+    fun purgeAll()
+
     /**
      * Executes side effects for queries.
      */

diff --git a/soil-query-core/src/commonMain/kotlin/soil/query/SwrClientPlus.kt b/soil-query-core/src/commonMain/kotlin/soil/query/SwrClientPlus.kt
@@ -6,4 +6,14 @@ package soil.query
 /**
  * An enhanced version of [SwrClient] that integrates [SubscriptionClient] into SwrClient.
  */
-interface SwrClientPlus : SwrClient, SubscriptionClient
+interface SwrClientPlus : SwrClient, SubscriptionClient {
+
+    /**
+     * Removes all queries, mutations and subscriptions from the in-memory stored data.
+     *
+     * **Note:**
+     * If there are any active queries or mutations or subscriptions, they will be stopped as well.
+     * This method should only be used for full resets, such as during sign-out.
+     */
+    override fun purgeAll()
+}
diff --git a/soil-query-core/src/commonMain/kotlin/soil/query/core/Actor.kt b/soil-query-core/src/commonMain/kotlin/soil/query/core/Actor.kt
@@ -31,37 +31,39 @@ interface Actor {
     fun launchIn(scope: CoroutineScope): Job
 }
 
-internal typealias ActorSequenceNumber = Int
+internal typealias ActorSequenceNumber = String
 
 internal class ActorBlockRunner(
+    private val id: String = uuid(),
     private val scope: CoroutineScope,
     private val options: ActorOptions,
     private val onTimeout: (ActorSequenceNumber) -> Unit,
     private val block: suspend () -> Unit
 ) : Actor {
 
-    var seq: ActorSequenceNumber = 0
-        private set
+    val seq: ActorSequenceNumber
+        get() = "$id#$versionCounter"
 
-    private var launchedCount: Int = 0
+    private var versionCounter: Int = 0
+    private var activeCounter: Int = 0
     private var hasActiveScope: Boolean = false
     private var runningJob: Job? = null
     private var cancellationJob: Job? = null
 
     override fun launchIn(scope: CoroutineScope): Job {
-        seq++
+        versionCounter++
         return scope.launch(start = CoroutineStart.UNDISPATCHED) {
             cancellationJob?.cancelAndJoin()
             cancellationJob = null
             suspendCancellableCoroutine { continuation ->
-                launchedCount++
-                if (!hasActiveScope && launchedCount > 0) {
+                activeCounter++
+                if (!hasActiveScope && activeCounter > 0) {
                     hasActiveScope = true
                     start()
                 }
                 continuation.invokeOnCancellation {
-                    launchedCount--
-                    if (hasActiveScope && launchedCount <= 0) {
+                    activeCounter--
+                    if (hasActiveScope && activeCounter <= 0) {
                         hasActiveScope = false
                         stop()
                     }

diff --git a/soil-query-core/src/commonTest/kotlin/soil/query/core/ActorTest.kt b/soil-query-core/src/commonTest/kotlin/soil/query/core/ActorTest.kt
@@ -16,6 +16,7 @@ import kotlinx.coroutines.yield
 import soil.testing.UnitTest
 import kotlin.test.Test
 import kotlin.test.assertEquals
+import kotlin.test.assertNull
 import kotlin.time.Duration
 import kotlin.time.Duration.Companion.seconds
 
@@ -29,31 +30,32 @@ class ActorTest : UnitTest() {
         val blockHandler = TestBlockHandler()
         val timeoutHandler = TestTimeoutHandler()
         val actor = ActorBlockRunner(
+            id = "test",
             scope = backgroundScope,
             options = TestActorOptions(),
             onTimeout = timeoutHandler::onTimeout,
             block = blockHandler::handle
         )
 
-        assertEquals(0, actor.seq)
+        assertEquals("test#0", actor.seq)
         assertEquals(0, blockHandler.count)
-        assertEquals(-1, timeoutHandler.seq)
+        assertNull(timeoutHandler.seq)
 
         val scope1 = CoroutineScope(uiDispatcher)
         val job1 = actor.launchIn(scope1)
         yield()
 
-        assertEquals(1, actor.seq)
+        assertEquals("test#1", actor.seq)
         assertEquals(1, blockHandler.count)
-        assertEquals(-1, timeoutHandler.seq)
+        assertNull(timeoutHandler.seq)
 
         job1.cancel()
         // Wait for the actor to be canceled. (keepAliveTime = 5.seconds)
         advanceTimeBy(6.seconds)
 
-        assertEquals(1, actor.seq)
+        assertEquals("test#1", actor.seq)
         assertEquals(1, blockHandler.count)
-        assertEquals(1, timeoutHandler.seq)
+        assertEquals("test#1", timeoutHandler.seq)
     }
 
     @Test
@@ -63,46 +65,47 @@ class ActorTest : UnitTest() {
         val blockHandler = TestBlockHandler()
         val timeoutHandler = TestTimeoutHandler()
         val actor = ActorBlockRunner(
+            id = "test",
             scope = backgroundScope,
             options = TestActorOptions(),
             onTimeout = timeoutHandler::onTimeout,
             block = blockHandler::handle
         )
 
-        assertEquals(0, actor.seq)
+        assertEquals("test#0", actor.seq)
         assertEquals(0, blockHandler.count)
-        assertEquals(-1, timeoutHandler.seq)
+        assertNull(timeoutHandler.seq)
 
         val scope1 = CoroutineScope(uiDispatcher)
         val job1 = actor.launchIn(scope1)
         yield()
 
-        assertEquals(1, actor.seq)
+        assertEquals("test#1", actor.seq)
         assertEquals(1, blockHandler.count)
-        assertEquals(-1, timeoutHandler.seq)
+        assertNull(timeoutHandler.seq)
 
         val scope2 = CoroutineScope(uiDispatcher)
         val job2 = actor.launchIn(scope2)
         yield()
 
-        assertEquals(2, actor.seq)
+        assertEquals("test#2", actor.seq)
         assertEquals(1, blockHandler.count)
-        assertEquals(-1, timeoutHandler.seq)
+        assertNull(timeoutHandler.seq)
 
         job1.cancel()
         yield()
 
-        assertEquals(2, actor.seq)
+        assertEquals("test#2", actor.seq)
         assertEquals(1, blockHandler.count)
-        assertEquals(-1, timeoutHandler.seq)
+        assertNull(timeoutHandler.seq)
 
         job2.cancel()
 
         // Wait for the actor to be canceled. (keepAliveTime = 5.seconds)
         advanceTimeBy(6.seconds)
-        assertEquals(2, actor.seq)
+        assertEquals("test#2", actor.seq)
         assertEquals(1, blockHandler.count)
-        assertEquals(2, timeoutHandler.seq)
+        assertEquals("test#2", timeoutHandler.seq)
     }
 
     @Test
@@ -113,31 +116,32 @@ class ActorTest : UnitTest() {
         val blockHandler = TestBlockHandler()
         val timeoutHandler = TestTimeoutHandler()
         val actor = ActorBlockRunner(
+            id = "test",
             scope = actorScope,
             options = TestActorOptions(),
             onTimeout = timeoutHandler::onTimeout,
             block = blockHandler::handle
         )
 
-        assertEquals(0, actor.seq)
+        assertEquals("test#0", actor.seq)
         assertEquals(0, blockHandler.count)
-        assertEquals(-1, timeoutHandler.seq)
+        assertNull(timeoutHandler.seq)
 
         val scope1 = CoroutineScope(uiDispatcher)
         val job1 = actor.launchIn(scope1)
         yield()
 
-        assertEquals(1, actor.seq)
+        assertEquals("test#1", actor.seq)
         assertEquals(1, blockHandler.count)
-        assertEquals(-1, timeoutHandler.seq)
+        assertNull(timeoutHandler.seq)
 
         job1.cancel()
 
         // Wait for the actor to be canceled. (keepAliveTime = 5.seconds)
         advanceTimeBy(6.seconds)
-        assertEquals(1, actor.seq)
+        assertEquals("test#1", actor.seq)
         assertEquals(1, blockHandler.count)
-        assertEquals(1, timeoutHandler.seq)
+        assertEquals("test#1", timeoutHandler.seq)
 
         actorScope.cancel()
         advanceUntilIdle()
@@ -146,17 +150,17 @@ class ActorTest : UnitTest() {
         val job2 = actor.launchIn(scope2)
         yield()
 
-        assertEquals(2, actor.seq)
+        assertEquals("test#2", actor.seq)
         // Unchanged (already canceled)
         assertEquals(1, blockHandler.count)
-        assertEquals(1, timeoutHandler.seq)
+        assertEquals("test#1", timeoutHandler.seq)
 
         job2.cancel()
 
         advanceTimeBy(6.seconds)
-        assertEquals(2, actor.seq)
+        assertEquals("test#2", actor.seq)
         assertEquals(1, blockHandler.count)
-        assertEquals(1, timeoutHandler.seq)
+        assertEquals("test#1", timeoutHandler.seq)
     }
 
     private class TestBlockHandler {
@@ -173,7 +177,7 @@ class ActorTest : UnitTest() {
     }
 
     private class TestTimeoutHandler {
-        var seq: ActorSequenceNumber = -1
+        var seq: ActorSequenceNumber? = null
             private set
 
         fun onTimeout(seq: ActorSequenceNumber) {