O-C的鸡尾酒戏法"Method Swizzling"

最新推荐文章于 2024-04-29 20:34:59 发布

weixin_33860528

最新推荐文章于 2024-04-29 20:34:59 发布

阅读量54

点赞数

原文链接：https://my.oschina.net/u/1984662/blog/291129

版权

2019独角兽企业重金招聘Python工程师标准>>>

偶然在stackoverflow上看到有人问“如何用类别覆盖一个o-c类的方法，同时在这个方法中调用其原来的实现？”。例如：

// Base Class

@interface ClassA : NSObject

- (NSString *) myMethod;

@end

@implementation ClassA

- (NSString*) myMethod {return @"A"; }

@end

//Category

@interface ClassA (CategoryB)

- (NSString *) myMethod;

@end

@implementation ClassA(CategoryB)

- (NSString*) myMethod {return @"B"; }

@end

类A中有一个方法myMethod，如果用类别B覆盖类A的myMethod方法，这是很容易做到的。但问题是，我想在类别方法myMethod中，还要调用类A的默认实现，即return@"A";一句，这该怎么办？

这是一个很有趣的问题。实际上，仅仅用O-C的类别机制我们无法做到这一点。因为类别的方法覆盖是完全覆盖，如果你在类别中重写了类方法，则类方法原来的实现就完全被替代了。

你也许想到了self关键字。比如在类别方法中这样写：

- (NSString*) myMethod {

return [[self myMethod]stringByAppendingString:@"B"];

}

这会导致一个myMethod方法的循环调用。因为在类别看来，ClassA的myMethod方法已经被类别替换掉了，[self myMethod]调用的仍然是类别的实现，而不是类的实现。

如果使用[super myMethod]呢？类别并不是继承，CategoryB并不是ClassA的子类，它仍然是ClassA类。那么super指向的仍然是ClassA的父类，即NSObject。NSObject中并没有myMethod方法，O-C会报一个错误。

解决这个问题的最终答案是“方法混合”。即mac osx 10.5以后的新的运行时API“Method Swizzling”。

Swizzling一词是“搅动、混合”之意，常用于鸡尾酒制作。所以Method Swizzling其实是方法混合或方法交换。

我们先来看一个简单的例子。这个例子来自于Andras的博客 http://codeshaker.blogspot.com/2012/01/calling-original-overridden-method-from.html：

假设有一个类Test，它有一个方法length:

-(NSUInteger)length{

return 4571;

}

我们想为Test定义一个类别，并覆盖这个方法，但同时，我们还想调用Test类的默认实现。因此我们可以这样写：

#import<objc/runtime.h>

@implementation Test(Logging)

- (NSUInteger)logLength {

NSUInteger length = [self logLength];

NSLog(@"Test Logging: %d", length);

return length;

}

+ (void)load {

method_exchangeImplementations(class_getInstanceMethod(self,@selector(length)), class_getInstanceMethod(self, @selector(logLength)));

}

@end

提示: swizzling无法在类簇上使用。

首先看load方法。在load方法中我们调用了鸡尾酒魔法“method_exchangeImpementations”函数。这个函数将实例方法（在这里，其实是类别方法）length和logLength方法的实现进行交换。也就是说，当我们调用length时，实际上是调用logLength，当我们调用logLength时，实际上是调用length。

实际上，一个方法的方法名和方法体是分开的。方法体是花括号 {} 之间的代码，即方法的 IMP，而在这之前的是方法名，即 SEL。通常情况下，SEL 是与 IMP 匹配的，但在 swizzling之后，length方法的SEL还是叫做length，但IMP却变成了logLength的IMP，logLength的IMP却变成了length的IMP，如下图所示。