一、序
在日常编写业务代码的时候,大部分代码其实就是在处理数据、展示数据,这些操作占用了我们大量的编码时间。
比较常见的,从某个集合中找到满足某个限制条件的数据。最简单的方法就是利用循环来遍历数据集合,找到我们需要的数据。
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() 方法,这两套方法的区别在于操作的集合以及返回值是什么。
#p#分页标题#e#在 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 list,但是这个集合里都是一个个员工的对象。在上报之前,我们需要将他们转换成 List。
这里就可以用到 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
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 文档。
【本文为51CTO专栏作者“张旸”的原创稿件,转载请通过微信公众号联系作者获取授权】
戳这里,看该作者更多好文
【编辑推荐】
为什么MySQL数据库要用B+树存储索引?
想伪装成资深程序员?知道这三个数据结构就够了
数据科学难在实践,有哪些弯路可以不走?
中国移动研究院常耀斌:商用大数据平台的研发之路
人人都说大数据,那你知道它的核心价值吗?