最近开始接手一些iOS工程且都是OC代码,虽然大学时学过C语言,但早已忘得一干二净。工程里面的 @weakify 和 @strongify 引起了我的好奇:他们到底是干啥的?
打破循环引用的标准方法
假设我们有一个 model 属性的 controller,当模型中的数据发生变化时更新 label。为此,我们建立了一个模型:
- (void)setUpModel
{
Model *model = [Model new];
// 该block会在model的数据变化调用
model.dataChanged = ^(NSString *title) {
self.label.text = title;
};
self.model = model;
}
几行简单的代码便创建了一个引用循环。我们的 controller 引用了 model,而 model 又引用了一个 controller 的 block。好在我们可以通过引入带有__weak和__strong存储类型修饰符的局部变量轻松打破这个引用循环:
Model *model = [Model new];
__weak typeof(self) weakSelf = self;
model.dataChanged = ^(NSString *title) {
__strong typeof(self) strongSelf = weakSelf;
strongSelf.label.text = title;
};
self.model = model;
Objective-C 工程里到处都可以看到这种 weak/strong 群魔乱舞的代码。它是一种正确的处理方式,但却很容易出错。当引入你新功能让你的 block 变得越来越大时,最终会有人在其中使用 self。我们不会注意到它何时发生,编译器只在最简单的情况下提供帮助。这时候 weakify 和 strongify 宏便派上了用场。
@weakify 和 @strongify
@weakify 和 @strongify 宏的原始实现比较复杂,因为它们接受多个参数。为了使分析更简单,我将自己的版本说明,每个版本只接受一个参数:
#define weakify(var) __weak typeof(var) AHKWeak_##var = var;
#define strongify(var) \
_Pragma("clang diagnostic push") \
_Pragma("clang diagnostic ignored "-Wshadow"") \
__strong typeof(var) var = AHKWeak_##var; \
_Pragma("clang diagnostic pop")
有了这些宏,我们的示例变成了一下形式:
Model *model = [Model new];
weakify(self);
model.dataChanged = ^(NSString *title) {
strongify(self);
self.label.text = title;
};
self.model = model;
最后将编译成:
Model *model = [Model new];
__weak typeof(self) AHKWeak_self = self;
model.dataChanged = ^(NSString *title) {
__strong typeof(self) self = AHKWeak_self;
self.label.text = title;
};
self.model = model;
在 block 中self被一个同名的局部变量所掩盖。self可以在块内安全地使用,因为它引用了该局部变量,同时该变量被强持有,但又仅在块结束执行之前有效。
如果我忘记使用了strongify会怎么样?其实:weakify创建一个新的局部变量,如果没有使用它,我们会收到warning:
Unused variable 'AHKWeak_self'
注意:如果在多个块中使用了 strongify(self),则该 warning 也无济于事。我们可以通过为每个 block 引入一个新scope 来解决,例如:
{
weakify(self);
model.dataChanged = ^(NSString *title) {
strongify(self);
self.label.text = title;
};
}
{
weakify(self);
model.syncFinished = ^(BOOL success) {
strongify(self);
[self update];
};
}
看起来并不优美,但在某些情况下还是很有用。如果您忘记使用weakify,但有strongify,编译器会显示错误:
使用未声明的标识符“AHKWeak_self”
最后
在某些情况下创建新 block 时,我们仍需要考虑保留 self(或其他一些对象)引用的可能性,这是任何宏也无法帮忙做到的,需要实际分析。但当使用了weakify 和strongify 宏,以后对 block 的任何更改都会比标准的weak/strong 方式更安全。