Kotlin开发五 Kotlin中集合常用函数式API

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/qiyei2009/article/details/83244682

在kotlin中有一些常用的对集合进行操作的函数式API，掌握他们可以大大提高编程效率，这些API与java8中的
Stream API类似。下面主要讲解以下几大类

1 filter 和 map

filter主要用作过滤器(对集合元素)，它的返回值也是一个集合，定义如下：

public inline fun <T> Iterable<T>.filter(predicate: (T) -> Boolean): List<T> {
    return filterTo(ArrayList<T>(), predicate)
}

可以看到，返回的也是一个集合，并且类型与原类型相同，用法如下：

class User(val id:Int,var name:String,var age:Int,val sex:String){

    init {
        //println(toString())
    }

    override fun toString(): String {
        return "User(id=$id, name='$name', age=$age, sex='$sex')"
    }
}
fun main(args: Array<String>) {

    val list = arrayListOf<User>(User(1,"张三",14,"男"),User(2,"李四",16,"女"),
            User(3,"王五",65,"男"),User(4,"赵六",24,"男")
            ,User(5,"向七",89,"女"),User(6,"王八",6,"女")
            ,User(7,"韩九",27,"男"),User(8,"陈10",35,"女"))

    //测试filter
    val filterList = list.filter { it.age > 30 }
    println(filterList)
}

输出：

[User(id=3, name='王五', age=65, sex='男'), User(id=5, name='向七', age=89, sex='女'), User(id=8, name='陈10', age=35, sex='女')]

map主要对集合的元素做映射，可以映射成不同的类型，例如Int到String，定义如下

public inline fun <T, R> Iterable<T>.map(transform: (T) -> R): List<R> {
    return mapTo(ArrayList<R>(collectionSizeOrDefault(10)), transform)
}

可以看到，返回值是映射成的新类型的集合，用法如下

    //测试map
    val mapList = list.map { it.name }
    println(mapList)

输出

[张三, 李四, 王五, 赵六, 向七, 王八, 韩九, 陈10]

2 all any count find

以上4个都是对集合的所有元素检查是否符合某个条件(或者它的变种，是否存在符合的元素等)

all检查集合中是否所有元素都符合某个条件

public inline fun <T> Iterable<T>.all(predicate: (T) -> Boolean): Boolean {
    if (this is Collection && isEmpty()) return true
    for (element in this) if (!predicate(element)) return false
    return true
}

可以看到返回值是Boolean类型，都符合返回为true,有一个不符合为true，用法如下：

    //测试all
    val all = list.all { it.age > 10 }
    println(all)

输出：

false

any检测集合中是否至少存在一个元素满足条件，如果是就返回true

public inline fun <T> Iterable<T>.any(predicate: (T) -> Boolean): Boolean {
    if (this is Collection && isEmpty()) return false
    for (element in this) if (predicate(element)) return true
    return false
}

用法如下：

    //测试any
    val any = list.any { it.age > 10 }
    println(any)

输出

true

count 检查集合中有多少个元素满足该条件，返回值是Int

public inline fun <T> Iterable<T>.count(predicate: (T) -> Boolean): Int {
    if (this is Collection && isEmpty()) return 0
    var count = 0
    for (element in this) if (predicate(element)) count++
    return count
}

用法如下：

    //测试count
    val count = list.count { it.age > 25 }
    println(count)

输出

4

find找到集合中第一个满足条件的元素，返回值是集合元素类型，可能为null

@kotlin.internal.InlineOnly
public inline fun <T> Iterable<T>.find(predicate: (T) -> Boolean): T? {
    return firstOrNull(predicate)
}

用法如下：

    //测试find
    val find = list.find { it.age > 25 }
    println(find)

输出

User(id=3, name='王五', age=65, sex='男')

3 groupBy

groupBy主要用作将集合按照给定的条件进行分组，即将集合转为map

public inline fun <T, K> Iterable<T>.groupBy(keySelector: (T) -> K): Map<K, List<T>> {
    return groupByTo(LinkedHashMap<K, MutableList<T>>(), keySelector)
}

可以看到，返回的是一个map，key类型就是lambda的返回值类型，value是集合元素的List，用法如下：

    //测试groupBy
    val groupBy = list.groupBy { it.sex}
    println(groupBy)

输出

{男=[User(id=1, name='张三', age=14, sex='男'), User(id=3, name='王五', age=65, sex='男'), User(id=4, name='赵六', age=24, sex='男'), User(id=7, name='韩九', age=27, sex='男')], 女=[User(id=2, name='李四', age=16, sex='女'), User(id=5, name='向七', age=89, sex='女'), User(id=6, name='王八', age=6, sex='女'), User(id=8, name='陈10', age=35, sex='女')]}

4 flatMap

flatMap先对集合中的某个元素映射成集合，再将所有映射成的集合做一个元素平铺，即做一个list

public inline fun <T, R> Iterable<T>.flatMap(transform: (T) -> Iterable<R>): List<R> {
    return flatMapTo(ArrayList<R>(), transform)
}

返回集合元素映射的集合的list
用法如下：

fun main(args: Array<String>) {

    val list = arrayListOf<User>(User(1,"张三",14,"男"),User(2,"李四",16,"女"),
            User(3,"王五",65,"男"),User(4,"赵六",24,"男")
            ,User(5,"向七",89,"女"),User(6,"王八",6,"女")
            ,User(7,"韩九",27,"男"),User(8,"陈10",35,"女"))

	....
    var citys = arrayListOf<City>()
    citys.add(City("CD", arrayListOf(User(1,"张三",14,"男"),User(2,"李四",16,"女"),
            User(3,"王五",65,"男"))))
    citys.add(City("CD", arrayListOf(User(5,"向七",89,"女"),User(6,"王八",6,"女"))))
    //测试flatMap
    val flatmap = citys.flatMap {
        it.people
    }
    println(flatmap)
}

class City(val name:String,val people:List<User>){

    override fun toString(): String {
        return "City(name='$name', people=$people)"
    }
}

输出：

[User(id=1, name='张三', age=14, sex='男'), User(id=2, name='李四', age=16, sex='女'), User(id=3, name='王五', age=65, sex='男'), User(id=5, name='向七', age=89, sex='女'), User(id=6, name='王八', age=6, sex='女')]

以上就是Kotlin中集合常用的一些API，熟练使用这些可以非常方便的提高我们的开发效率