分类（category）

应用场景

在我们开发中经常会有这样的问题：

• 多个人同时维护同一个类 ，彼此之间交叉修改同一个文件，导致冲突以及文件臃肿。
• 想给已经封装好的类（或者是系统类，第三方库的类）增加方法，而又不想使用继承后的子类。
• 希望拆分一个过于庞大的类。

分类（category）可以很好的解决这些问题。

介绍

分类是OC的一种语法，他可以在一个新的文件中为一个即有的类添加新的方法（理论上不能添加新的属性）

我们为NSObject添加一个分类名为haha的分类,并在这个分类中为NSObject添加一个laugh方法：

.h
@interface NSObject (haha)

-(void)laugh;

@end

.m
#import "NSObject+haha.h"

@implementation NSObject (haha)

-(void)laugh
{
    NSLog(@"hahahahahaha");
}

@end

在程序入口处，我们只需要引用这个分类，就可以直接调用分类里的方法了：

#import "NSObject+haha.h"

NSObject * obj = [NSObject new];
[obj laugh];

源码分析

我们在runtime.h中可以找到category的运行时结构体：

struct objc_category {
    char * _Nonnull category_name                            OBJC2_UNAVAILABLE;
    char * _Nonnull class_name                               OBJC2_UNAVAILABLE;
    struct objc_method_list * _Nullable instance_methods     OBJC2_UNAVAILABLE;
    struct objc_method_list * _Nullable class_methods        OBJC2_UNAVAILABLE;
    struct objc_protocol_list * _Nullable protocols          OBJC2_UNAVAILABLE;
}    

我们可以看到，在运行时，分类的结构体里有分类名，原类名，实例方法列表，类方法列表，协议列表。但是并没有变量列表。

这就是分类中是无法添加变量的原因。

如果非要为分类添加属性，我们可以用动态添加属性（Associated Object）技术间接实现

我们在haha分类中为NSObject添加一个name属性：

.h
@interface NSObject (haha)

@property(nonatomic,copy)NSString *name;

-(void)laugh;

@end

这时在.m文件中系统给我们警告，让我们实现name方法和setName方法，因为，在分类中定义属性，系统并不会为我们生成get，set方法，也不会生成实例变量。

我们需要利用动态添加属性（Associated Object）手动编写get和set方法：

#import "NSObject+haha.h"
#import <objc/runtime.h>

static NSString *nameKey = @"nameKey";

@implementation NSObject (haha)

-(void)laugh
{
    NSLog(@"hahahahahaha");
}

-(NSString *)name
{
    return objc_getAssociatedObject(self, &nameKey);
}

-(void)setName:(NSString *)name
{
    objc_setAssociatedObject(self, &nameKey, name, OBJC_ASSOCIATION_COPY_NONATOMIC);
}

通过这种方式，我们看起来就为分类添加了一个name属性,但是任然不能访问_name实例变量，因为它并不存在。

小细节

• 分类中可以使用原类的.h中的属性，方法。
• 如果分类中有原类的同名方法，那么，会覆盖原类方法。同名方法优先级：分类>原类>父类
• 如果多个分类有同名方法，那么执行最后加入编译的分类的方法。

类扩展（Class Extension）

类扩展介绍

类扩展其实我们平时使用非常多，只是可能很多同学不知道，那个就是类扩展而已，我们平时创建控制器时IDE就会帮我们在.m文件里自动生成一个类扩展，供我们写一些私有的方法和属性：

@interface ViewController ()

@end

类扩展最常见的使用场景就是在.m中定义私有的属性和方法。

类扩展的几个特点

• 与分类不同的是类扩展是可以添加属性的，并且会自动生成get方法，set方法，实例变量。

• 类扩展可以定义在.m 中，其中的属性方法为私有，也可以定义在.h中，它的属性和方法为共有。

• 类扩展的中方法的实现，必须在原类的.m文件中。