Objective-C 中实现类似 Go 语言的 "defer" 语句-Golang-PHP中文网

Objective-C 中实现类似 Go 语言的 "defer" 语句

心靈之曲

发布： 2025-07-28 14:26:20

原创

341人浏览过

objective-c 中实现类似 go 语言的

本文探讨了如何在 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__;})])

登录后复制

这些宏的工作原理如下：

ViiTor实时翻译

AI实时多语言翻译专家！强大的语音识别、AR翻译功能。

116

查看详情

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。