iOS开发／Category与Extentsion的区别／以及Runtime常见的使用

一、在项目中新建简单一个Robot类，该类只有name、uid属性，以及一个简单- (void)printRobotData 的方法：

Robot.h文件中代码：

image.png

//
//  Robot.h
//  RuntimeDemo
//
//  Created by GrabinWong on 2017/12/12.
//  Copyright © 2017年 GrabinWong. All rights reserved.
//

#import 

@interface Robot : NSObject

@property (nonatomic, copy) NSString *name;

@property (nonatomic, copy) NSString *uid;

- (void)printRobotData;

@end

Robot.m文件中代码：

image.png

//
//  Robot.m
//  RuntimeDemo
//
//  Created by GrabinWong on 2017/12/12.
//  Copyright © 2017年 GrabinWong. All rights reserved.
//

#import "Robot.h"

@implementation Robot

- (void)printRobotData
{
    NSLog(@"This robit named %@,and it's id is %@.",self.name,self.uid);
}

@end

二、接下来就是为Robot类创建Category和Extentsion来对比它们之间的区别了。

image.png

image.png

创建完可以发现，Category有.h和.m两个文件，而Extentsion只有.h文件。

image.png

A、尝试在Extentsion中添加属性：master、price和方法- (void)sayHello;

image.png

//
//  Robot+RobotExtentsion.h
//  RuntimeDemo
//
//  Created by GrabinWong on 2017/12/12.
//  Copyright © 2017年 GrabinWong. All rights reserved.
//

#import "Robot.h"

@interface Robot ()

@property (nonatomic, copy) NSString *master;

@property (nonatomic, copy) NSString *price;

- (void)sayHello;

@end

实例化属性和方法，运行会发现，添加属性是没问题的，但添加方法因为Extentsion只有.h文件。所以方法只有声明，没有方法实现，所以崩了：

image.png

解决方法是，在原有的类Robot中添加方法实现：

image.png

总结一下：类的Extentsion(延展)只有.h文件，它可以为类添加属性，不需要去实现setter,getter方法，可以去为类添加方法，但方法的实现是得在原有类的.m文件中实现。

B、尝试在Category中添加属性：user和方法- (void)printUser;

Robot+RobotCategory.h文件中代码：

image.png

Robot+RobotCategory.m文件中代码：

image.png

运行崩了，原因是属性user没有setter方法，上面的图其实也有给了警告。

image.png

解决方法是，在Robot+RobotCategory.m中用runtime实现属性的setter、getter方法实现，这也是runtime其中的用法，可以为category添加属性：

image.png

//
//  Robot+RobotCategory.m
//  RuntimeDemo
//
//  Created by GrabinWong on 2017/12/12.
//  Copyright © 2017年 GrabinWong. All rights reserved.
//

#import "Robot+RobotCategory.h"
#import 

static const char *kUserKey = "kUserKey";

@implementation Robot (RobotCategory)

- (void)setUser:(NSString *)user
{
    objc_setAssociatedObject(self, kUserKey, user, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}

- (NSString *)user
{
    return objc_getAssociatedObject(self, kUserKey);
}

- (void)printUser
{
    NSLog(@"Robot's user is %@",self.user);
}

@end

总结一下：类的Category(类别)有.h和.m文件，它可以为类添加方法，可以为类添加属性，但得用runtime为属性添加setter、getter方法。

Runtime的基本使用：

1.可以为类添加属性，

2.方法交换。

方法交换 method_exchangeImplementations

在分类中添加一个方法，用于测试方法交换

image.png

//
//  Robot+RobotCategory.m
//  RuntimeDemo
//
//  Created by GrabinWong on 2017/12/12.
//  Copyright © 2017年 GrabinWong. All rights reserved.
//

#import "Robot+RobotCategory.h"
#import 

static const char *kUserKey = "kUserKey";

@implementation Robot (RobotCategory)

- (void)setUser:(NSString *)user
{
  objc_setAssociatedObject(self, kUserKey, user, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}

- (NSString *)user
{
  return objc_getAssociatedObject(self, kUserKey);
}

- (void)printUser
{
  NSLog(@"Robot's user is %@",self.user);
}

- (void)printHelloWorld
{
  NSLog(@"这是用来测试runtime方法交换的");
  NSLog(@"HelloWorld");
}

+ (void)load
{
  Method printRobotDataMethod = class_getInstanceMethod([self class], @selector(printUser));
  Method sayHelloMethod = class_getClassMethod([self class], @selector(printHelloWorld));
  method_exchangeImplementations(printRobotDataMethod, sayHelloMethod);
}

@end

class_getInstanceMethod 得到类的实例方法

class_getClassMethod 得到类方法

再用method_exchangeImplementations 进行方法交换。

可以看到结果，调用[curRobot printUser] 得到的是 printHelloWorld 方法执行的内容。

image.png

注意：把runtime方法交换的操作放在+(void)load里面是为了避免多次调用

+ (void)load
{
    Method printRobotDataMethod = class_getInstanceMethod([self class], @selector(printUser));
    Method sayHelloMethod = class_getClassMethod([self class], @selector(printHelloWorld));
    method_exchangeImplementations(printRobotDataMethod, sayHelloMethod);
}