如何进行分组、排序和排序：Kotlin 集合

Question

我正在尝试根据 Kotlin 风格以简洁易读的方式对每个组进行分组、排序和排序。

我有一个数据类。

enum class LoadingStatus { PROCESSING, SCANNING, FAIL, FINISHED }

data class FileModel(val loadingStatus: LoadingStatus, val fileName: String, val modifiedDate: Long)

我想对FileModel的集合进行排序：

项目应按以下顺序从上到下排序：

PROCESSING → SCANNING → FINISHED → OTHER

子排序顺序：

PROCESSING → 按时间修改（AS）

SCANNING → 按时间修改 (DE)

FINISHED → 按时间修改 (DE)

OTHER → 按名称 (AS)

private fun setupFilesOrder(incomeList: MutableList<FileModel>): MutableList<FileModel> 
= ...

Answer 1

这个怎么样？

fun setupFilesOrder(incomeList: MutableList<FileModel>): MutableList<FileModel> {
    val grouped = incomeList.groupBy { it.loadingStatus }
    return listOf(
        grouped[LoadingStatus.PROCESSING]?.sortedBy { it.modifiedDate }.orEmpty(),
        grouped[LoadingStatus.SCANNING]?.sortedByDescending { it.modifiedDate }.orEmpty(),
        grouped[LoadingStatus.FINISHED]?.sortedByDescending { it.modifiedDate }.orEmpty(),
        grouped[LoadingStatus.FAIL]?.sortedBy { it.fileName }.orEmpty()
    ).flatten().toMutableList()
}

如果你想要更多的表达，甚至可以这样做:)

fun setupFilesOrder(incomeList: MutableList<FileModel>): MutableList<FileModel> {
    val grouped = incomeList.groupBy { it.loadingStatus }
    infix fun LoadingStatus.sort(block: List<FileModel>.() -> List<FileModel>) =
        grouped[this]?.block().orEmpty()
    return listOf(
        LoadingStatus.PROCESSING sort { sortedBy { it.modifiedDate } },
        LoadingStatus.SCANNING sort { sortedByDescending { it.modifiedDate } },
        LoadingStatus.FINISHED sort { sortedByDescending { it.modifiedDate } },
        LoadingStatus.FAIL sort { sortedBy { it.fileName } }
    ).flatten().toMutableList()
}