为什么需要使用 @weakify 和 @strongify

最近开始接手一些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 方式更安全。

Guess you like

Origin juejin.im/post/7049932857807994888