基于多个过滤器过滤数据集的最佳方法

时间:2021-07-15 19:06:53

标签: android kotlin mvvm kotlin-coroutines

我有一个数据集,其中包含用户的 GradeEntity 对象,其中包含对其关联的 SubjectEntity 实例的引用。我现在需要允许用户根据两个选项过滤该数据集:

  1. 主题名称
  2. 日期范围

这是我要使用的过滤过程的逻辑。 假设初始数据集包含两个成绩实体:

  1. 成绩(题目:数学,日期:7 月 11 日)
  2. 成绩(标题:数学,日期:7 月 14 日)

现在假设用户根据“数学”对成绩进行排序,但尚未选择日期范围。显示了所有相应的成绩,但现在用户想要在过滤后的数据集上应用日期范围。他/她选择从 7 月 1 日到 7 月 12 日的范围。所以,现在用户只能看到一级实体。现在假设他们想要将日期范围过滤器从 7 月 1 日更改为 7 月 15 日。我希望他们现在能够看到两个成绩。这意味着我不能像在当前的 ViewModel 代码中那样重新过滤过滤后的数据集。当用户首先根据日期、主题进行过滤,然后选择另一个主题时,也会出现同样的问题。这种过滤过程的最佳方法是什么?

现在我使用的是共享的 MutableLiveData<List<GradeEntity>> 变量,但这并不适用于我上面提到的所有情况。

ViewModel.kt:

class GradesViewModel(private val database: AppDatabase) : ViewModel() {
    private val _grades: MutableLiveData<List<GradeEntity>> = MutableLiveData(emptyList())
    val grades: LiveData<List<GradeEntity>> = _grades
    private val _filteredGrades: MutableLiveData<List<GradeEntity>> = MutableLiveData(emptyList())
    val filteredGrades: LiveData<List<GradeEntity>> = _filteredGrades

    init {
        getGradesFromDB()
    }

    private fun getGradesFromDB() {
        viewModelScope.launch(Dispatchers.IO) {
            val grades = database.getGradeDao().getAllGrades()
            if (grades.isNotEmpty() && grades != _grades.value) _grades.postValue(grades)
        }
    }

    fun getSubjectTitlesRelatedToGrades(): MutableSet<String> {
        val subjectTitles: MutableSet<String> = mutableSetOf("All Subjects")
        for (gradeEntity in grades.value!!) {
            subjectTitles.add(gradeEntity.subject.title)
        }
        return subjectTitles
    }

    fun filterGradesBasedOnSubject(subjectTitle: String) {
        viewModelScope.launch {
            filterGradesBasedOnSubjectAsync(subjectTitle)
        }
    }

    fun filterGradesBasedOnDate(startDate: Date, endDate: Date) {
        viewModelScope.launch {
            filterGradesBasedOnDateAsync(startDate, endDate)
        }
    }

    private suspend fun filterGradesBasedOnSubjectAsync(subjectTitle: String) {
        val filteredData = viewModelScope.async(Dispatchers.IO) {
            if (filteredGrades.value!!.isEmpty()) {
                grades.value?.stream()
                    ?.filter { gradeEntity -> gradeEntity.subject.title == subjectTitle }?.toList()
                    ?: emptyList()
            } else filteredGrades.value?.stream()
                ?.filter { gradeEntity -> gradeEntity.subject.title == subjectTitle }?.toList()
                ?: emptyList()
        }
        _filteredGrades.postValue(filteredData.await())
    }

    private suspend fun filterGradesBasedOnDateAsync(startDate: Date, endDate: Date) {
        val filteredData = viewModelScope.async(Dispatchers.IO) {
            val formatter = SimpleDateFormat("MMMM dd yyyy", Locale.getDefault())
            return@async if (filteredGrades.value!!.isEmpty()) {
                grades.value?.stream()
                    ?.filter { gradeEntity ->
                        formatter.parse(gradeEntity.formattedDateTime).after(startDate) &&
                                formatter.parse(gradeEntity.formattedDateTime).before(endDate)
                    }?.toList()
                    ?: emptyList()
            } else {
                filteredGrades.value?.stream()
                    ?.filter { gradeEntity ->
                        formatter.parse(gradeEntity.formattedDateTime).after(startDate) &&
                                formatter.parse(gradeEntity.formattedDateTime).before(endDate)
                    }?.toList()
                    ?: emptyList()
            }
        }
        _filteredGrades.postValue(filteredData.await())
    }
}

2 个答案:

答案 0 :(得分:2)

我认为只公开一个 LiveData 并将未过滤结果的完整列表保密会更清晰。那么 Fragment/Activity 只需要观察单个 LiveData 一次,并且可以单独修改要应用的过滤器。当过滤器更改时,通过从头开始重新过滤所有值的列表,让 ViewModel 自动更新 LiveData 中的值。

让您的支持数据也自动更新也很好。为此,您应该添加一个返回 Flow 的 repo/DAO 函数。基本上是一个 dao.getAllGrades() 的副本,它不是一个挂起函数并返回一个 Flow<List<GradeEntity>>。我们称之为getAllGradesFlow()

要更改过滤器,我们可以公开用于设置当前过滤器的属性,并将它们设置为触发后备数据的重新过滤器,以便自动刷新 LiveData。为了减少函数/属性的数量,我们可以使用 ClosedRange<Date> 作为日期过滤器的属性类型。也可以使用 in 运算符轻松检查日期是否在两个日期之间。在调用方使用 .. 运算符很容易指定。

我还想提一下,您不必指定 Dispatchers.IO 来调用 DAO 中的挂起函数。按照约定挂起函数不会阻塞,因此无需指定任何调度程序来调用它们。

class GradesViewModel(private val database: AppDatabase) : ViewModel() {
    private var allGrades: List<GradeEntity> = emptyList()
    private val _grades: MutableLiveData<List<GradeEntity>> = MutableLiveData(emptyList())
    val grades: LiveData<List<GradeEntity>> = _grades
    var datesFilter: ClosedRange<Date>? by Delegates.observable(null) { _, _, _ -> onFilterChange() }
    var subjectTitleFilter: String? by Delegates.observable(null) { _, _, _ -> onFilterChange() }

    init {
        getGradesFromDB()
    }

    private fun getGradesFromDB() {
        database.getGradeDao().getAllGradesFlow()
            .onEach { 
                allGrades = it 
                publishFilteredGrades()
            }.launchIn(viewModelScope)
    }

    private fun onFilterChange() = viewModelScope.launch {
        publishFilteredGrades()
    }

    // Not related to your question but I simplified this.
    fun getSubjectTitlesRelatedToGrades(): Set<String> {
        return setOf("All Subjects") + allGrades.map { it.subject.title }
    }

    // Specifying a suspend function with dispatcher here because filtering 
    // a long list might be CPU heavy.
    private suspend fun publishFilteredGrades() = withContext(Dispatchers.Default) {
        var filteredGrades = allGrades
        subjectTitleFilter?.let { subjectTitle ->
            filteredGrades = filteredGrades.filter { it.subject.title == subjectTitle }
        }
        datesFilter?.let { dates ->
            val formatter = SimpleDateFormat("MMMM dd yyyy", Locale.getDefault())
            filteredGrades = filteredGrades.filter { 
                formatter.parse(it.formattedDateTime) in dates 
            }
        }
        _grades.postValue(filteredGrades)
    }
}

从片段中,您只需要观察单个 LiveData 即可更新您的列表视图。随意更改两个过滤器属性,将它们设置为空以清除过滤器。

答案 1 :(得分:0)

您可以在 SwitchMap 上设置 MutableLiveData 运算符。每当 SwitchMap 被调用的 MutableLiveData 的值发生变化时,它都会再次触发代码。

例如,假设您只想根据日期或主题进行过滤,您可以使用以下内容:

private val _isDateSelected: MutableLiveData<Boolean> = MutableLiveData(false)

val currentResult = _isDateSelected.switchMap { dateIsSelected ->

   if(dateIsSelected) {

   // here the code that filters by date and returns a LiveData appears
  } else {

  // here the code that filters by subject and returns a LiveData appears 
  }
}

fun setSubjectSelectedFilter() {
  _isDateSelected = false
}

fun setDateSelectedFilter() {
  _isDateSelected = true
}

然后,您可以在您的视图中观察这个 currentResult LiveData,以通过 SubjectDate 获得过滤结果。如果您有任何疑问,请随时在评论中提问:)