为什么需要使用 @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 方式更安全。