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

如何解决基于多个过滤器过滤数据集的最佳方法

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

主题名称
日期范围

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

成绩（题目：数学，日期：7 月 11 日）
成绩（标题：数学，日期：7 月 14 日）

现在假设用户根据“数学”对成绩进行排序，但尚未选择日期范围。显示了所有相应的成绩，但现在用户想要在过滤后的数据集上应用日期范围。他/她选择从 7 月 1 日到 7 月 12 日的范围。所以，现在用户只能看到一级实体。现在假设他们想要将日期范围过滤器从 7 月 1 日更改为 12 月 13 日。我希望他们现在能够看到两个成绩。这意味着我不能像在当前的 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())
    }
}

解决方法

我认为只公开一个 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 即可更新您的列表视图。随意更改两个过滤器属性，将它们设置为空以清除过滤器。

您可以在 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，以通过 Subject 或 Date 获得过滤结果。如果您有任何疑问，请随时在评论中提问:)