swift中defer的小结

看看苹果官方的介绍

用 defer 语句在即将离开当前代码块时执行一系列语句。该语句让你能执行一些必要的清理工作，不管是以何种方式离开当前代码块的——无论是由于抛出错误而离开，或是由于诸如 return、break 的语句。例如，你可以用 defer 语句来确保文件描述符得以关闭，以及手动分配的内存得以释放。
defer 语句将代码的执行延迟到当前的作用域退出之前。该语句由 defer 关键字和要被延迟执行的语句组成。延迟执行的语句不能包含任何控制转移语句，例如 break、return 语句，或是抛出一个错误。延迟执行的操作会按照它们声明的顺序从后往前执行——也就是说，第一条 defer 语句中的代码最后才执行，第二条 defer 语句中的代码倒数第二个执行，以此类推。最后一条语句会第一个执行。

总结一下苹果官方的介绍

• defer语句在代码块（方法、闭包等，可以理解为大括号包装起来的代码）执行，也就是代码块中其他应该执行的代码都执行完了，才执行defer中的代码
• 一个代码块允许多个defer，多个defer执行的顺序

一些测试及误区纠正

测试案例1

func testDefer() {
    defer {
        print("方法中defer内容")
    }
    if true {
        defer {
            print("if 中defer内容")
        }
        print("if中最后的代码")
    }
    print("方法中的代码")
    if true {
        return
    }
    print("方法结束前最后一句代码")
}
testDefer()

以上代码打印结果：

if中最后的代码
if 中defer内容
方法中的代码
方法中defer内容

打印结果中，第一个if中的代码及里面的defer最先执行，方法中的defer最后执行，由此可以看出，代码块中其他能够执行的代码先执行，最后执行defer的内容；defer的作用范围不能简单的看成方法，而是代码块（可能有些同学会有这样的误区）

测试案例2

func testDefer() {
    print("开始")
    defer {
        print("defer 1 中的内容")
    }
    defer {
        print("defer 2 中的内容")
    }
    if true {
        return
    }
    defer {
        print("defer 3 中的内容")
    }
    print("方法结束前最后一句代码")
}
testDefer()

打印结果

开始
defer 2 中的内容
defer 1 中的内容

我们可以看到最后一个defer没有执行，所以defer定义的位置很重要，如果没有执行defer定义的代码，在代码块结束前不会执行defer中的内容
多个defer的执行顺序从后到前

一些实际应用场景

场景1：一些资源用完后需释放，这里给的是官方的一个案例

func processFile(filename: String) throws {
    if exists(filename) {
        let file = open(filename)
        defer {
            close(file)
        }
        while let line = try file.readline() {
            // 处理文件。
        }
        // close(file) 会在这里被调用，即作用域的最后。
    }
}

开始用到资源的时候就使用defer去释放，避免忘记释放资源

场景2：加锁解锁，借鉴了kingfisher

let lock = NSLock()
func testDefer() {
    lock.lock()
    defer {
        lock.unlock()
    }
    
    doSomething()
}
testDefer()

在加锁后立刻用defer解锁，避免忘记解锁

场景3：处理一些代码块作用域结束前的重复操作，比如请求网络数据的时候

通常的一种写法

func loadCityList(_ finish: ((Error?, [String]?) -> ())?) {
    DispatchQueue.global().async { // 模拟网络请求
        let data: AnyObject? // 模拟服务器返回的数据
        guard let dict = data as? [String: AnyObject] else {
            DispatchQueue.main.async {
                finish?(error, nil)
            }
            return
        }
        guard let code = dict["code"] as? Int, code == 200 else {
            DispatchQueue.main.async {
                finish?(error, nil)
            }
            return
        }
        guard let citys = dict["data"] as? [String]? else {
            DispatchQueue.main.async {
                finish?(error, nil)
            }
            return
        }
        DispatchQueue.main.async {
            finish?(nil, citys)
        }
    }
}

当每次有错误处理时和结果正确时都需要去做回调，而且回调可能有一堆代码，看起来代码会比较冗余，而且在一些错误处理时很容易造成忘记回调

defer怎么去写呢

func loadCityList(_ finish: ((Error?, [String]?) -> ())?) {
    DispatchQueue.global().async { // 模拟网络请求
        var error: Error? = nil
        var citys: [String]? = nil
        defer {
            DispatchQueue.main.async {
                finish?(error, citys)
            }
        }
        
        let data: AnyObject? // 模拟服务器返回的数据
        guard let dict = data as? [String: AnyObject] else {
            error = ...
            return
        }
        guard let code = dict["code"] as? Int, code == 200 else {
            error = ...
            return
        }
        guard let tempCitys = dict["data"] as? [String]? else {
            error = ...
            return
        }
        citys = tempCitys
    }
}

使用defer既解决了代码冗余，又解决了可能忘记回调的问题，还有当我们看到defer时，我们很清楚知道，无论网络请求结果如果，都会回调

总结

• 本文主要介绍了defer的定义、作用及一些用法