40 lines
1.2 KiB
Kotlin
40 lines
1.2 KiB
Kotlin
package org.pixeldroid.app
|
|
|
|
import androidx.lifecycle.ViewModel
|
|
import androidx.lifecycle.viewModelScope
|
|
import dagger.hilt.android.lifecycle.HiltViewModel
|
|
import kotlinx.coroutines.Dispatchers
|
|
import kotlinx.coroutines.flow.MutableStateFlow
|
|
import kotlinx.coroutines.flow.asStateFlow
|
|
import kotlinx.coroutines.flow.flowOn
|
|
import kotlinx.coroutines.flow.update
|
|
import kotlinx.coroutines.launch
|
|
import org.pixeldroid.app.utils.db.AppDatabase
|
|
import org.pixeldroid.app.utils.db.entities.UserDatabaseEntity
|
|
import javax.inject.Inject
|
|
|
|
@HiltViewModel
|
|
class MainActivityViewModel @Inject constructor(
|
|
private val db: AppDatabase
|
|
): ViewModel() {
|
|
|
|
// Mutable state flow that will be used internally in the ViewModel, empty list is given as initial value.
|
|
private val _users = MutableStateFlow(emptyList<UserDatabaseEntity>())
|
|
|
|
// Immutable state flow exposed to UI
|
|
val users = _users.asStateFlow()
|
|
|
|
|
|
init {
|
|
getUsers()
|
|
}
|
|
|
|
private fun getUsers() {
|
|
viewModelScope.launch {
|
|
db.userDao().getAllFlow().flowOn(Dispatchers.IO)
|
|
.collect { users: List<UserDatabaseEntity> ->
|
|
_users.update { users }
|
|
}
|
|
}
|
|
}
|
|
} |