本文探讨了如何在 Objective-C 中实现类似 Go 语言的 defer 语句的功能。defer 语句允许在函数返回前执行一段代码，通常用于资源清理。文章分析了使用 Autoreleased 对象、Dispatch Finalizers 和 C++ 析构函数的可能性，并提供了一种基于 @finally 块的宏实现方案，并提供了详细的代码示例和使用说明，展示了如何在 Objective-C 中模拟 defer 的行为。
Go 语言中的 defer 语句提供了一种优雅的方式来确保在函数退出时执行某些操作，例如关闭文件、释放锁等。虽然 Objective-C 本身没有直接对应的语法，但我们可以利用 Objective-C 的特性来模拟类似的行为。
基于 @finally 块的宏实现
一种可行的方法是利用 Objective-C 的 @try、@catch 和 @finally 块。@finally 块保证在 @try 块执行完毕后（无论是否发生异常）都会被执行。我们可以定义一些宏来简化 defer 语句的使用。
#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__，用于存储需要在函数结束时执行的 block。@try 块开始。
- END_SCOPE：@finally 块开始。遍历 _defered_actions__ 数组，逆序执行每个 block。释放 _defered_actions__ 数组。
- DEFER(_code__)：将一个 block 添加到 _defered_actions__ 数组中。这个 block 包含了需要在函数结束时执行的代码。
- DEFER_COPY(_code__)：与 DEFER 类似，但它会先复制 block，这在需要在 block 中捕获局部变量时非常有用。
使用示例
下面是一个使用这些宏的示例：
@interface XXObject : NSObject {
}
-(int)factorial:(int)x;
@end
@implementation XXObject
-(int)factorial:(int)x { SCOPE
printf("begin foo:%d\n", x);
DEFER( printf("end foo:%d\n", 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];
DEFER({
printf("releasing %p.\n", x);
[x release];
});
int i;
for (i = 2; i >= -1; -- i) {
// use DEFER_COPY to retain the local variable 'i' and 'fact'
int fact = [x factorial:i];
DEFER_COPY( printf("%d! == %d\n", i, fact) );
}
END_SCOPE }
int main () {
NSAutoreleasePool* pool = [[NSAutoreleasePool alloc] init];
@try {
do_stuff();
} @catch(NSException* e) {
// note that the @finally statements might not be called in 64-bit if we
// left the exception uncaught.
NSLog(@"%@", e);
}
[pool drain];
return 0;
}在这个示例中，do_stuff 函数使用 DEFER 来确保在函数结束时释放 x 对象。factorial 函数也使用了 DEFER 来在函数开始和结束时打印消息。DEFER_COPY 宏用于捕获循环中的局部变量 i 和 fact。
注意事项
- 这种方法依赖于 @try、@catch 和 @finally 块。虽然 @finally 块通常保证执行，但在某些情况下（例如程序崩溃）可能不会执行。
- 使用 DEFER_COPY 宏时，需要注意循环引用的问题。如果 block 捕获了对象的所有者，可能会导致内存泄漏。
- 宏的使用可能会降低代码的可读性。建议谨慎使用，并添加适当的注释。
- 在64位系统中，未捕获的异常可能会导致 @finally 块不被调用。
总结
虽然 Objective-C 没有直接的 defer 语句，但我们可以利用 @finally 块和宏来模拟类似的行为。这种方法可以帮助我们编写更简洁、更易于维护的代码，并确保在函数退出时执行必要的清理操作。在使用这种方法时，需要注意潜在的风险，并采取适当的措施来避免问题。
