Swift-defer关键字(推迟执行)

OStack程序员社区-中国程序员成长平台 › 门户 › 编程› 移动开发›Swift 教程

原作者: [db:作者] 来自: [db:来源] 收藏邀请

在一些语言中，有try/finally这样的控制语句，比如Java。

这种语句可以让我们在finally代码块中执行必须要执行的代码，不管之前怎样的兴风作浪。

在Swift 2.0中，Apple提供了defer关键字，让我们可以实现同样的效果。

func checkSomething() {
    
    print("CheckPoint 1")
    doSomething()
    print("CheckPoint 4")
    
}

func doSomething() {
    
    print("CheckPoint 2")
    defer {
        print("Clean up here")
    }
    print("CheckPoint 3")
    
}

checkSomething() // CheckPoint 1, CheckPoint 2, CheckPoint 3, Clean up here, CheckPoint 4

上述示例可以看到，在打印出“CheckPoint 2”之后并没有打印出“Clean up here”，而是“CheckPoint 3”，这就是defer的作用，它对进行了print("Clean up here")延迟。我们再来看一个I/O的示例：

// 伪代码
func writeSomething() {
    
    let file = OpenFile()
    
    let ioStatus = fetchIOStatus()
    guard ioStatus != "error" else {
        return
    }
    file.write()
    
    closeFile(file)
    
}

上述示例是一个I/O操作的伪代码，如果获取到的ioStatus正常，那么该方法没有问题，

如果ioStatus取到的是error，那么会被guard语句抓到执行return操作，

这样的话closeFile(file)就永远都不会执行了，一个严重的Bug就这样产生了。下面我们看看如何用defer来解决这个问题：

// 伪代码
func writeSomething() {
    
    let file = OpenFile()
    defer {
        closeFile(file)
    }
    
    let ioStatus = fetchIOStatus()
    guard ioStatus != "error" else {
        return
    }
    file.write()
    
}

我们将closeFile(file)放在defer代码块里，这样即使ioStatus为error，在执行return前会先执行defer里的代码，这样就保证了不管发生什么，最后都会将文件关闭。

需要注意的是, 虽然说defer的内容会在return之前执行, 但是如果defer定义在return之后, 那么还是不会执行defter的内容, 也就是说, defer关键字必须比return早出现。

放在return之后:

var str = "Hello, playground"

func show() {
    print("这里是即将要推迟(但一定会)执行的代码")
}

func test() {
    
    if str.characters.count >= 2 {
        print("跳出该方法的执行")
        return
    }
    
    defer {
        show()
    }
}

test()  // 输出结果: 跳出该方法的执行

放在return之前:

var str = "Hello, playground"

func show() {
    print("这里是即将要推迟(但一定会)执行的代码")
}

func test() {
    
    defer {
        show()
    }
    
    if str.characters.count >= 2 {
        print("跳出该方法的执行")
        return
    }
}

test()  // 输出结果: 跳出该方法的执行 这里是即将要推迟(但一定会)执行的代码