Objective-C有两个扩展机制:Associative和Category。Category用来扩展类方法,Associative用于扩展属性。Associative机制的原理是把两个对象关联起来，让一个对象成为另外一个对象的一部分。它可以在不修改类的定义的前提下为其对象增加存储空间，这在我们无法访问类的源码时(例如给UILable添加一个selected的BOOL属性)是非常有用的。Associative基于关键字的,因此我们可以使用不同的关键字为任何对象添加任意多的Associative。Associative可以保证被关联的对象在对象的整个生命周期都是可用的。Associative基于runtime,是运行时里的东西,所以头文件需要引用#im port<objc/runtime.h>文件。
Associative提供了三个方法:
objc_setAssociatedObject(id object, const void *key, id value, objc_AssociationPolicy policy)
objc_getAssociatedObject(id object, const void *key)
objc_removeAssociatedObjects(id object)
详细介绍一下这几个方法,第一个用于设置关联的:
objc_setAssociatedObject(id object, const void *key, id value, objc_AssociationPolicy policy)
注意的是返回值类型为Object类型，注意一些不是Object类型的例如:BOOL是结构类型。当碰到这种情况可以考虑通过中间类型来转换，如设置BOOL类型属性的时候可以转换为NSNumber类型，获取的时候再转换成BOOL类型即可。
四个参数分别是:源对象,关键字,关联对象和关联策略
关键字是一个void类型的指针,例如 static void * myKey = (void *)@"MyKey";每一个关联的关键字必须是唯一的。关联策略是枚举类型:
enum{
OBJC_ASSOCIATION_ASSIGN;----------->@property(assign)----------->弱引用关联对象
OBJC_ASSOCIATION_COPY;------------->@property(copy,atomic)------>复制关联对象且为原子操作
OBJC_ASSOCIATION_COPY_NONATOMIC;--->@property(copy,nonatomic)--->复制关联对象且为非原子操作
OBJC_ASSOCIATION_RETAIN;----------->@property(strong,atomic)---->强引用关联对象且为原子操作
OBJC_ASSOCIATION_RETAIN_NONATOMIC;->@property(strong,nonatomic)->强引用且为非原子操作
}
第二个用于获取关联对象:
objc_getAssociatedObject(id object, const void *key)
用于获取关联对象的值。这里需要注意的是返回值类型为Object类型，注意一些不是Object类型的例如:BOOL是结构类型
第三个 objc_removeAssociatedObjects(id object) 是断开关联,需要注意的是他会断开所有关联,所以不推荐这种方式。需要断开关联的时候使用objc_setAssociatedObject函数，传入nil值即可。
下面一个示例,我给UILable添加了一个表示选中状态的selected属性,自定义了类似于UIButton的选中非选中状态下设置背景色,字体色的方法。
.h文件
#import <UIKit/UIKit.h>
#import <objc/runtime.h>
typedef enum{
ControlStateNormal,
ControlStateSelected,
}ControlState; @interface UILabel (CellLable) @property(nonatomic,assign)BOOL selected;//设置UILable的选中和非选中状态 default is NO -(void)setBackgroundColor:(UIColor *)backgroundColor forState:(ControlState)state ; -(void)setTextColor:(UIColor *)textColor forState:(ControlState)state ; @end
.m文件
#import "UILabel+CellLable.h" static UIColor * normalBackgroundColor = nil ;
static UIColor * selectedBackgorundColor = nil ;
static UIColor * normalTextColor = nil ;
static UIColor * selectedTextColor = nil ;
static UIColor * firstBackgroundColor = nil ;
static UIColor * firstTextColor = nil ; static BOOL isHaveSet = NO; //用于在未调用setSelected方法时 设置UILable的默认属性 //static char const myKey = '\a';
static void * myKey = (void *)@"MyKey"; @implementation UILabel (CellLable) @dynamic selected ;//动态绑定 声明自定义set get 方法，不自动生成 -(void)setSelected:(BOOL)selected{ #warning 注意 runtime 里边的objc_setAssociateObject 方法 第三个参数为object类型 BOOL属于结构类型 需要用NSNumber来转换
isHaveSet = YES; if (selected) { [self setBackgroundColor:nil forState:ControlStateSelected]; [self setTextColor:nil forState:ControlStateSelected]; }else if (!selected){ [self setBackgroundColor:nil forState:ControlStateNormal]; [self setTextColor:nil forState:ControlStateNormal];
} NSNumber * num = [NSNumber numberWithBool:selected]; objc_setAssociatedObject(self, &myKey, num, OBJC_ASSOCIATION_ASSIGN);
} -(BOOL)selected{ NSNumber * num = objc_getAssociatedObject(self, &myKey); return [num boolValue];
}
-(void)setBackgroundColor:(UIColor *)backgroundColor forState:(ControlState )state{ switch (state) {
case ControlStateNormal:
{
/** 当传递进来的是nil的时候,采取一种保护机制 **/
if (backgroundColor) { /** 记录上次传递进来的颜色属性 **/
normalBackgroundColor = backgroundColor ;
}
self.backgroundColor = normalBackgroundColor ;
}
break;
case ControlStateSelected:
{
if (backgroundColor) {
selectedBackgorundColor = backgroundColor ;
}
self.backgroundColor = selectedBackgorundColor ;
}
break ;
default:
break;
} if (!isHaveSet) { self.backgroundColor = normalBackgroundColor ;
}
} -(void)setTextColor:(UIColor *)textColor forState:(ControlState)state{ switch (state) {
case ControlStateNormal:
{
if (textColor) {
normalTextColor = textColor ;
}
self.textColor = normalTextColor ;
}
break;
case ControlStateSelected:
{
if (textColor) {
selectedTextColor = textColor ;
}
self.textColor = selectedTextColor ;
}
break ;
default:
break;
} if (!isHaveSet) {
self.textColor = normalTextColor ;
}
} @end
ViewController.m文件
-(void)viewDidLoad{
UILabel * lable2 =[[UILabel alloc]initWithFrame:CGRectMake(, , , )];
lable2.textAlignment = NSTextAlignmentCenter ;
lable2.layer.masksToBounds = YES ;
lable2.layer.cornerRadius = ;
lable2.selected = NO ;
[lable2 setBackgroundColor:[UIColor colorWithRed:/255.0 green:/255.0 blue:/255.0 alpha:1.0] forState:ControlStateSelected];
[lable2 setBackgroundColor:[UIColor clearColor] forState:ControlStateNormal];
[lable2 setTextColor:[UIColor whiteColor] forState:ControlStateSelected];
[lable2 setTextColor:[UIColor blackColor] forState:ControlStateNormal];
[self.view addSubview:lable2];
}