Swift - defer關鍵字(推遲執行)

在一些語言中，有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()  // 輸出結果: 跳出該方法的執行 這里是即將要推遲(但一定會)執行的代碼