本文探讨了在Objective-C中实现类似go语言defer语句的延迟执行机制。通过巧妙结合Objective-C的@try/@finally异常处理块和Block特性，我们设计了一套宏，能够在函数或特定作用域结束时自动执行清理代码，有效简化资源管理和错误处理逻辑，确保关键操作的可靠完成，即使在异常情况下也能正常执行。
Go语言defer机制简介
Go语言的defer语句是一种强大的机制，用于确保某个函数调用（或表达式）在当前函数返回之前执行。它常用于资源清理，例如关闭文件、释放锁或取消网络连接。defer语句将函数调用推入一个栈中，当外层函数执行完毕时，这些被延迟的调用会以后进先出（LIFO）的顺序执行。这种模式极大地简化了资源管理，使得清理代码与资源分配代码紧密相邻，提高了代码的可读性和健壮性，尤其是在存在多条返回路径或异常发生时。
Objective-C中实现defer的挑战
Objective-C本身没有内置的defer或类似的延迟执行机制。开发者通常依赖于以下几种方式进行资源清理：
- dealloc方法：对象生命周期结束时调用，但其执行时机与函数或作用域的返回不直接关联，且仅限于对象级别的清理。
- @finally块：@try/@catch/@finally结构中的@finally块能够确保在@try或@catch块执行完毕后（无论是否发生异常）被执行。这是实现函数级延迟执行的有效途径。
- c++析构函数：在Objective-C++ (.mm文件)中，可以利用C++局部对象的析构函数在作用域结束时执行清理代码。例如，Boost库的Scope Exit就提供了这种能力。然而，这要求将文件从.m改为.mm，可能引入C++的复杂性，并且仅限于作用域级别。
- __attribute__((cleanup))：gnu C扩展，允许为局部变量指定一个清理函数，当变量超出作用域时自动调用。虽然强大，但它不是标准的Objective-C特性，且语法相对不直观。
为了在Objective-C中模拟Go语言的defer行为，我们需要一个能够捕获任意代码块，并在函数或特定作用域结束时以LIFO顺序执行这些代码的机制，同时要兼顾异常安全性。
基于@try/@finally的延迟执行实现
利用Objective-C的Block特性和@try/@finally结构，我们可以构建一个相对通用的defer机制。核心思想是在函数开始时初始化一个可变数组来存储待执行的Block，然后在函数结束的@finally块中，以相反的顺序遍历并执行这些Block。
立即学习“go语言免费学习笔记（深入）”；
以下是实现这一机制的宏定义：
#define SCOPE {id _defered_actions__=[[NSMutableArray alloc]init];@try{ #define END_SCOPE }@finally{for(void(^action)()in[_defered_actions__ reverseObjectEnumerator])action();[_defered_actions__ release];}} #define DEFER_copy(_code__) {id _blk__=[^{_code__;}copy];[_defered_actions__ addObject:_blk__];[_blk__ release];} #define DEFER(_code__) ([_defered_actions__ addObject:(^{_code__;})])
宏解释：
- SCOPE: 标记延迟执行作用域的开始。它声明并初始化一个名为_defered_actions__的NSMutableArray实例，用于存储所有待执行的Block，并开始一个@try块。
- END_SCOPE: 标记延迟执行作用域的结束。它关闭@try块，并开启一个@finally块。在@finally块中，它通过reverseObjectEnumerator遍历_defered_actions__数组中的所有Block，并依次执行它们，从而实现LIFO的执行顺序。最后，它释放了_defered_actions__数组（在非ARC环境下）。
- DEFER(_code__): 用于添加一个延迟执行的Block。它将_code__封装成一个Block并直接添加到_defered_actions__数组中。此宏创建的Block默认会按引用捕获其外部变量。
- DEFER_COPY(_code__): 同样用于添加一个延迟执行的Block，但它会复制（copy）这个Block。这在Block需要捕获的局部变量在Block执行前可能改变或超出作用域时非常有用，确保Block内部使用的是变量的“快照”值，或者确保Block能够持有其捕获的对象。
示例使用：
#import <Foundation/Foundation.h> // 宏定义如上所示 #define SCOPE {id _defered_actions__=[[NSMutableArray alloc]init];@try{ #define END_SCOPE }@finally{for(void(^action)()in[_defered_actions__ reverseObjectEnumerator])action();[_defered_actions__ release];}} #define DEFER_COPY(_code__) {id _blk__=[^{_code__;}copy];[_defered_actions__ addObject:_blk__];[_blk__ release];} #define DEFER(_code__) ([_defered_actions__ addObject:(^{_code__;})]) @interface XXObject : NSObject { } -(int)factorial:(int)x; @end @implementation XXObject -(int)factorial:(int)x { SCOPE // 开始延迟执行作用域 printf("begin foo:%dn", x); DEFER( printf("end foo:%dn", x) ); // 延迟打印函数结束信息 if (x > 0) return x * [self factorial:x-1]; else if (x == 0) return 1; else { @throw [NSException exceptionWithName:@"NegativeFactorialException" reason:@"Cannot call factorial on negative numbers" userInfo:nil]; return 0; } END_SCOPE } // 结束延迟执行作用域 -(void)dealloc { printf("%p has been released.n", self); [super dealloc]; } @end void do_stuff() { SCOPE // 开始延迟执行作用域 __block XXObject* x = [[XXObject alloc] init]; // 使用__block修饰，以便在Block中修改 DEFER({ printf("releasing %p.n", x); [x release]; // 延迟释放对象 }); int i; for (i = 2; i >= -1; -- i) { // 使用 DEFER_COPY 来保留局部变量 'i' 和 'fact' 的当前值 int fact = [x factorial:i]; DEFER_COPY( printf("%d! == %dn", i, fact) ); // 延迟打印阶乘结果 } END_SCOPE } // 结束延迟执行作用域 int main () { NSAutoreleasePool* pool = [[NSAutoreleasePool alloc] init]; @try { do_stuff(); } @catch(NSException* e) { // 注意：在64位环境下，如果异常未被捕获，@finally语句可能不会被调用。 NSLog(@"%@", e); } [pool drain]; return 0; }
示例输出分析：
begin foo:2 begin foo:1 begin foo:0 end foo:0 // factorial:0 的 defer 打印 end foo:1 // factorial:1 的 defer 打印 end foo:2 // factorial:2 的 defer 打印 begin foo:1 begin foo:0 end foo:0 // factorial:0 的 defer 打印 end foo:1 // factorial:1 的 defer 打印 begin foo:0 end foo:0 // factorial:0 的 defer 打印 begin foo:-1 end foo:-1 // factorial:-1 的 defer 打印 (此处会抛出异常，但 defer 依然执行) 0! == 1 // do_stuff 循环中 i=0 的 defer_copy 打印 1! == 1 // do_stuff 循环中 i=1 的 defer_copy 打印 2! == 2 // do_stuff 循环中 i=2 的 defer_copy 打印 releasing 0x100116500. // do_stuff 中 x 对象的 defer 释放打印 0x100116500 has been released. // x 对象的 dealloc 打印 2011-02-05 23:06:21.192 a.out[51141:903] Cannot call factorial on negative numbers
从输出中可以看出，defer语句的执行顺序是LIFO，并且即使在factorial:方法抛出异常后，其内部的DEFER语句依然能够正常执行，这验证了@finally块的异常安全性。do_stuff中的DEFER_COPY也成功捕获了i和fact的当前值，并在循环结束后按LIFO顺序打印。
使用注意事项与最佳实践
-
变量捕获与DEFER_COPY:
- DEFER宏创建的Block默认按引用捕获外部变量。如果被捕获的局部变量在Block执行前可能发生变化，或者是一个在Block执行时可能已释放的对象，这可能导致意外行为。
- DEFER_COPY宏通过copy操作确保Block持有其捕获的变量的“快照”或对象引用。对于基本类型，它捕获值；对于对象，它通常会增加引用计数（在MRC下）或强引用（在ARC下）。
- 当Block需要修改捕获的局部变量时，应使用__block修饰符。
- 注意避免Block的循环引用，尤其是在Block捕获self时。可以使用__weak或__unsafe_unretained来打破循环。
-
异常安全:
- @finally块的特性保证了其内部的代码无论@try块是否正常完成或抛出异常，都会被执行。这使得defer机制在处理异常情况下的资源清理非常可靠。
- 然而，原始答案中提到一个重要警告：在64位环境下，如果异常未被捕获（即@catch块没有处理），那么@finally语句可能不会被调用。这通常意味着程序将崩溃，所以确保异常被适当地捕获和处理是关键。
-
性能考量:
- 每次调用DEFER或DEFER_COPY都会创建一个Block对象并将其添加到NSMutableArray中。这会带来一定的内存分配和CPU开销。
- 在END_SCOPE时，需要遍历数组并执行所有Block。对于大量defer语句或性能敏感的循环，需要评估其影响。但在大多数常规资源清理场景下，这些开销通常可以忽略不计。
-
内存管理（MRC vs. ARC）:
- 提供的宏示例是基于MRC（手动引用计数）环境编写的，其中包含了release调用。
- 在ARC（自动引用计数）环境下，[_defered_actions__ release];和[_blk__ release];应该被移除，因为ARC会自动管理内存。NSMutableArray和Block的生命周期将由ARC自动处理。
-
可读性与维护:
- 使用宏可以简化代码，使其看起来更像Go的defer。然而，过度使用宏可能会降低代码的可读性和调试的便利性。
- 建议在团队中统一使用规范，并确保所有开发者都理解这些宏的工作原理。
-
C++集成:
- 如果项目允许使用Objective-C++ (.mm文件)，那么使用C++的RAII（Resource Acquisition Is Initialization）模式，结合局部对象的析构函数是实现作用域级别延迟执行的更“原生”和健壮的方式。例如，Boost库的Scope Exit或自定义的ScopeGuard类都是很好的选择。
总结
通过巧妙地结合Objective-C的@try/@finally结构和Block特性，我们可以有效地模拟Go语言中defer语句的延迟执行行为。这种模式在需要确保资源在函数或特定作用域结束时（包括异常情况下）被正确清理的场景中非常有用。虽然引入了宏和一些额外的内存开销，但它为Objective-C开发者提供了一种简洁、可靠的资源管理方式，有助于编写更健壮、更易于维护的代码。在实际项目中，应根据具体需求和团队规范，权衡其优缺点并选择最适合的实现方案。
© 版权声明
文章版权归作者所有，未经允许请勿转载。
THE END
