chore: clean up

kula-app · Oct 4, 2024 · 7e7048e · 7e7048e
1 parent e91d6f4
commit 7e7048e
Showing 1 changed file with 5 additions and 11 deletions.
diff --git a/client/src/main/java/app/kula/onlaunch/client/data/local/OnLaunchDataStore.kt b/client/src/main/java/app/kula/onlaunch/client/data/local/OnLaunchDataStore.kt
@@ -6,27 +6,21 @@ import androidx.datastore.preferences.core.Preferences
 import androidx.datastore.preferences.core.edit
 import androidx.datastore.preferences.core.stringSetPreferencesKey
 import androidx.datastore.preferences.preferencesDataStore
-import kotlinx.coroutines.Dispatchers
 import kotlinx.coroutines.flow.first
-import kotlinx.coroutines.withContext
 
 private val KEY_DISMISSED_MESSAGE_IDS = stringSetPreferencesKey("isDataSharingEnabled")
 
 internal class OnLaunchDataStore(context: Context) {
 
     private val dataStore: DataStore<Preferences> = context.onLaunchDataStore
 
-    suspend fun getDismissedMessageIds(): Set<Int> = withContext(Dispatchers.IO) {
-        dataStore.data.first().let {
-            it[KEY_DISMISSED_MESSAGE_IDS]?.map(String::toInt)?.toSet() ?: emptySet()
-        }
+    suspend fun getDismissedMessageIds(): Set<Int> = dataStore.data.first().let {
+        it[KEY_DISMISSED_MESSAGE_IDS]?.map(String::toInt)?.toSet() ?: emptySet()
     }
 
-    suspend fun addDismissedMessageId(messageId: Int) = withContext(Dispatchers.IO) {
-        dataStore.edit {
-            it[KEY_DISMISSED_MESSAGE_IDS] =
-                (it[KEY_DISMISSED_MESSAGE_IDS] ?: emptySet()) + messageId.toString()
-        }
+    suspend fun addDismissedMessageId(messageId: Int) = dataStore.edit {
+        it[KEY_DISMISSED_MESSAGE_IDS] =
+            (it[KEY_DISMISSED_MESSAGE_IDS] ?: emptySet()) + messageId.toString()
     }
 }