fix: clearing new clients was being executed on main thread (#2116)

wireapp · Oct 6, 2023 · 882fbf9 · 882fbf9
1 parent 9200a35
commit 882fbf9
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/...tence/src/commonMain/kotlin/com/wire/kalium/persistence/dao/newclient/NewClientDAOImpl.kt b/...tence/src/commonMain/kotlin/com/wire/kalium/persistence/dao/newclient/NewClientDAOImpl.kt
@@ -24,6 +24,7 @@ import com.wire.kalium.persistence.dao.client.InsertClientParam
 import com.wire.kalium.persistence.util.mapToList
 import kotlinx.coroutines.flow.Flow
 import kotlinx.coroutines.flow.flowOn
+import kotlinx.coroutines.withContext
 import kotlinx.datetime.Instant
 import kotlin.coroutines.CoroutineContext
 
@@ -49,12 +50,14 @@ internal class NewClientDAOImpl(
 ) : NewClientDAO {
 
     override suspend fun insertNewClient(client: InsertClientParam) = with(client) {
-        newClientsQueries.insertNewClient(
-            id = id,
-            device_type = deviceType,
-            registration_date = registrationDate,
-            model = model
-        )
+        withContext(queriesContext) {
+            newClientsQueries.insertNewClient(
+                id = id,
+                device_type = deviceType,
+                registration_date = registrationDate,
+                model = model
+            )
+        }
     }
 
     override suspend fun observeNewClients(): Flow<List<NewClientEntity>> =
@@ -63,7 +66,7 @@ internal class NewClientDAOImpl(
             .flowOn(queriesContext)
             .mapToList()
 
-    override suspend fun clearNewClients() {
+    override suspend fun clearNewClients() = withContext(queriesContext) {
         newClientsQueries.clearNewClients()
     }
 }