Kotlin 常见内联扩展函数的使用

2,499 阅读4分钟

Kotlin一个强大之处就在于它的扩展函数,巧妙的运用这些扩展函数可以让你写出的代码更加优雅,阅读起来更加流畅,下面总结了在开发中经常用到的一些内联扩展函数。经常有小伙伴搞不懂with,run,apply等等这些函数该怎么用,在哪里用,我的建议是先记住每个函数的功能(无非就是它需要什么参数?返回值是什么?)记住这两点再根据实际开发中的场景慢慢的就能熟练运用了。其实这些函数极其类似,不同的函数可以完成同样的功能,通过下面的实例也能看出。而在我以往的开发经验中这些函数主要的使用场景有两个,一是非空判断,二是对象的初始化或者本身及方法的频繁调用。

1. with

定义:fun <T, R> with(receiver: T, block: T.() -> R): R

功能:将对象作为函数的参数,在函数内可以通过 this指代该对象。返回值为函数的最后一行或return表达式。

实例: 1.在自定义view中当我们初始化画笔时很多时候我们会写下边的代码

    var paint = Paint()
    paint.color = Color.BLACK
    paint.strokeWidth = 1.0f
    paint.textSize = 18.0f
    paint.isAntiAlias = true

如果使用with,那么就可以写成这样

var paint = Paint()
with(paint) {
        color = Color.BLACK
        strokeWidth = 1.0f
        textSize = 18.0f
        isAntiAlias = true
    }

省去了paint.后书写起来感觉会更加自然 2.在声明一些集合的场景比如:

    var list= mutableListOf<String>()
    list.add("1")
    list.add("2")
    list.add("3")

使用with可以写成

var list = with(mutableListOf<String>()) {
        add("1")
        add("2")
        add("3")
        this
    }

开发中还有很多场景可以使用with功能,理解了with的功能也就能够灵活运用了。

2. takeIf和takeUnless

takeif

定义:fun T.takeIf(predicate: (T) -> Boolean): T?

功能:传递一个函数参数,如果函数结果为true,返回T对象,否则返回null。

实例:使用File文件时通常会判断file是否存在,比如

   var file = File("filePath")
    if (file.exists()) {
        //do something
    } else {
        return false
    }

使用takeif后

var file = File("filePath").takeIf { it.exists() }?:return false
//do something

takeUnless

定义:fun T.takeUnless(predicate: (T) -> Boolean): T?

功能:与takeIf相反,参数函数返回false时返回T对象,否则返回null,这里不再举例。

3. run

定义:

(1)fun run(block: () -> R): R (2)fun <T, R> T.run(block: T.() -> R): R

功能:调用run函数返回值为函数体最后一行,或return表达式。

实例: 返回最后一行

kotlin.run {
    println("11")
    println("22")
}

结果:

I/System.out: 11

I/System.out: 22

返回return表达式,return后面的代码不再执行(注意写法@run)

kotlin.run {
        return@run println("11")
        println("22")
    }

结果: I/System.out: 11

4. repeat

定义:fun repeat(times: Int, action: (Int) -> Unit)

功能:重复执行action函数times次,times从0开始

实例:与for循环功能类似,例如

repeat(5){ println("count:$it") }

等价于

for (i in 0..4) { println("count:$i") }

或者

(0..4).forEach{println("count:$it")}

5. let

定义:fun <T, R> T.let(block: (T) -> R): R

功能:调用对象(T)的let函数,则该对象为函数的参数。在函数内可以通过 it 指代该对象。返回值为函数的最后一行或指定return表达式。

实例:有点类似于run(),let在使用中可用于空安全验证,变量?.let{} 例如

val data = ……
data?.let {
…… // 假如data不为null,代码会执行到此处
}

6. apply

定义:fun T.apply(block: T.() -> Unit): T

功能:调用对象的apply函数,在函数范围内,可以任意调用该对象的任意方法,并返回该对象。

实例:

var list = mutableListOf<Int>().apply {
    add(1)
    add(2)
    add(3)
}

注意他和run函数的区别,run返回的是最后一行,apply返回的是对象本身,由apply函数的定义我们可以看出apply适用于那些对象初始化需要给其属性赋值的情况。 还是用画笔的例子

原始的

    var paint = Paint()
    paint.textSize = 14.0f
    paint.color = Color.WHITE
    paint.isAntiAlias = false

使用apply后

    var paint = Paint().apply {
        textSize = 14.0f
        color = Color.WHITE
        isAntiAlias = false
    }

此外由于apply函数返回的是其对象本身,那么可以配合?.完成多级的非空判断操作,或者用于建造者模式的Builder中

7. also

定义:fun T.also(block: (T) -> Unit): T

功能:调用对象的also函数,在函数块内可以通过 it 指代该对象,返回值为该对象本身。(注意其和let函数的区别,let返回的是最后一行,also返回的是对象本身)

实例:需要返回对象本身(this)的情况下,例如建造者模式。