kotlin实现defer功能

2019-01-04  本文已影响7人  mihope
class Deferrer {
    private val actions = arrayListOf<() -> Unit>()

    fun defer(f: () -> Unit) {
        actions.add(f)
    }

    fun done() {
        actions.reverse()
        for (action in actions) {
            action()
        }
    }
}

inline fun <T> withDefers(body: Deferrer.() -> T): T {
    val deferrer = Deferrer()
    val result = deferrer.body()
    deferrer.done()
    return result
}


fun main(args: Array<String>) {
    withDefers {
        println("A")
        defer { println("deferred: should happen last") }
        defer { println("deferred: should happen first") }
        println("B")
    }
}
上一篇 下一篇

猜你喜欢

热点阅读