大部分业务代码，都在处理数据！所以，高效很重要！！

一、序

在日常编写业务代码的时候，大部分代码其实就是在处理数据、展示数据，这些操作占用了我们大量的编码时间。

比较常见的，从某个集合中找到满足某个限制条件的数据。最简单的方法就是利用循环来遍历数据集合，找到我们需要的数据。

fun findFive() {
    val list = arrayListOf(1, 2, 3, 4, 5, 6, 7)
    val findList = mutableListOf<Int>()
    for (i in list) {
        if (i > 5) {
            findList.add(i)
        }
    }
    // findList
}

大部分数据集合的操作，我们都可以通过循环来解决，针对一些更复杂的操作，例如从两个集合中找出交集集合，可以通过两层嵌套循环解决。

除了用粗暴的循环解决问题之外，我们还可以利用 Kotlin 对集合封装的一些高阶函数，让我们对集合的处理，更得心应手。

上面例子中的需求，用 Kotlin 集合的高阶函数，一个方法就可以解决。

val findList = list.filter { it > 5 }

集合的高阶函数，还有很多更巧妙的使用方式。

本文就从业务场景出发，讲解 Kotlin 中与集合相关的一些高阶函数的使用和场景。

二、集合的高阶函数

Kotlin 原生就支持 Lambda 表达式，高阶函数中都可以基于 Lambda 来实现的。

假如我们需要实现一个公司内部的员工管理系统，那我们首先定义一个员工类(EmployeModel)。

data class EmployeModel(var id: Long, // 员工 ID
                        var name: String,// 姓名
                        var age: Int,// 年龄
                        var department: String,// 部门
                        var salary: Int // 薪水
) {
}

// var list = arrayListOf(EmployeModel(...),...)

接下来我们就围绕这个员工类，实现各种必要的需求。

2.1 数据筛选

需求：从给定的员工列表中，筛选出月薪大于 2w 的所有员工。

正如前文中举例的样子，对于数据筛选，可以使用 .filter() 方法，此方法接受一个 Boolean 类型的条件，它会返回所有满足此条件的数据集合，最终其实返回的依然是一个 List。

var findList = list.filter {it.salary > 20000}

.filter() 方法内不是其实也是用 for 循环来实现的，没有什么高深的。而除了 filter() 之外，Kotlin 还提供了带上一些特殊条件的过滤器。

例如：

• filterIndexed，带 Index 的过滤器。

• filterNot，过滤所有不满足条件的数据。

这里就不展开了。

2.2 数据排序

需求：将给定的员工列表，根据员工的薪酬进行排序。

通过 sortedBy() 方法，指定排序依据，它会根据指定的数值，进行自然顺序排序。

var findList = list.sortedBy{it.salary}

sortedBy() 方法为顺序排序，如果想要逆序排序，可以使用 sortedByDescending() 方法。

需要注意的是，除了 sortedBy() 之外，配套的还有 sortBy() 和 sortByDescending() 方法，这两套方法的区别在于操作的集合以及返回值是什么。

在 Kotlin 中，所有的集合可以分为可变集合和不可变集合，sortBy/sortByDescending 方法，只能作用在可变集合上，它返回的是一个 Unit 类型的对象，会在集合上进行自排序。而 sortBy() 方法会返回一个排序好的新集合。

2.3 集合切片

需求：将给定的员工列表，只显示前 100 位员工。

如果想获取某个集合中前 N 个员工，可以使用 take() 方法。

var findList = list.take(100)

和 take() 取前 N 个员工类似，还可以使用 takeLast() 方法获取末尾 N 个员工。

更灵活的切片，例如取第 10~20 的员工，就需要用到 slice() 方法，如果想要截取连续的一段数据，可以使用 IntRange 类来指定。

var findList = list.slice(IntRange(10,20))

slice() 方法还支持直接指定集合的下标，来获取不连续的数据，例如选取排名第 1、10、25 位员工。

var findList = list.slice(arrayListOf(1, 10, 15))

2.4 集合变换

需求：要调薪了，准备将一批员工涨薪 20%。那就需要批量上报一批员工信息，和服务端商量，只需要上报这一批调薪的员工 ID 即可。

原本我们有一个员工类的集合 List<EmployeModel> list，但是这个集合里都是一个个员工的对象。在上报之前，我们需要将他们转换成 List<Int>。

这里就可以用到 map() 方法，它可以对集合中的数据做一次变换。

var findList = list.map { it.id }

map() 原本是不带 index 的，如果对 index 有需求，可以使用 mapIndexed() 方法。

map() 代表了一种一对一的变换关系。如果想做到一对多，可以使用 flatMap() 方法，flatMap() 可以将每个元素变换为一个新的集合，再将其平铺成一个集合。

2.5 数据分组

需求：将所有的员工，用年龄以 20 岁为界限进行分组。

这种两分的分组，可以使用 partition() 方法，它需要制定一个分组的条件，以 Boolean 类型区分，最终返回一个 Pair 对象，Pair.first 就是满足条件的集合，Pair.second 则是不满足条件的集合。

var findList = list.partition { it.age > 20 }
findList.first.forEach{
    Log.i("cxmyDev","${it.name} 大于 20");
}
findList.second.forEach{
    Log.i("cxmyDev","${it.name} 小于或等于 20");
}

此时需求又变了，要根据部门来分组，公司的部门可不止两个。针对这种情况，可以通过 groupBy() 方法进行分组。

var findList = list.groupBy { it.department }

groupBy() 方法会返回一个 Map<K,List<T>> 的 Map 对象，其中 Key 就是我们分组的条件，value 就是分组后的集合。

2.6 其他操作

前面介绍的这些针对 Kotlin 集合的高阶函数，基本上满足我对集合的大部分操作。当然，Kotlin 还有一些更丰富的高阶函数，有兴趣可以直接查看 _Collections.kt 源码。

高阶函数之所以强大，更多的是因为支持链式调用，我们可以通过一个很清晰代码结构，就实现我们对集合的复杂需求。

我拍脑袋举个例子，取员工集合里，A 部门的员工，根据员工 ID 排序后，取前 100 号员工的 ID 进行上报。

var findList = list.filter {
    "A".equals(it.department)
}.sortedBy {
    it.id
}.map {
    it.id
}.take(100)

这样一个复杂的集合操作，在 Kotlin 的高阶函数中，只需要一组链式调用即可实现。

另外需要注意的是，高阶函数中很多方法，都不是线程安全的，所以在使用的时候，需要考虑多线程场景下的同步问题。

三、小结时刻

本文介绍了一些比较常用的集合操作函数，随着 Kotlin 版本的升级，还有会更多的高阶函数添加进来，更多细节可以阅读源码或者参考 kotlin.collections 文档。

本文对你有帮助吗？留言、点赞、转发是最大的支持，谢谢！

---

「联机圆桌」????推荐我的知识星球，一年 50 个优质问题，上桌联机学习。

公众号后台回复成长『成长』，将会得到我准备的学习资料，也能回复『加群』，一起学习进步；你还能回复『提问』，向我发起提问。

推荐阅读：