使用关联对象(AssociatedObject)为UIButton添加Block响应

时间:2021-10-13 06:49:55

在开发中,要给UIButton添加点击事件的话,通常的做法是这样的

UIButton *button = [UIButton buttonWithType:UIButtonTypeSystem];
[button addTarget:self action:@selector(touchButton:) forControlEvents:UIControlEventTouchUpInside]; - (void)touchButton:(id)sender
{ }

通过选择器来响应button点击,会使button申明和响应方法分离,我们可以通过runtime的AssociatedObject为UIButton关联一个block来进行处理

首先创建一个UIButton的分类(Category)UIButton+block,在分类中为UIButton添加addTouchHandler:方法

UIButton+block.h

@interface UIButton (block)

- (void)addTouchHandler:(TouchBlock)handler;

@end

UIButton+block.m

static void * TouchBlockKey = &TouchBlockKey;

@implementation UIButton (block)

- (void)addTouchHandler:(TouchBlock)handler
{
objc_setAssociatedObject(self, TouchBlockKey, handler, OBJC_ASSOCIATION_COPY_NONATOMIC); [self addTarget:self action:@selector(touchUpInSide:) forControlEvents:UIControlEventTouchUpInside]; } -(void)touchUpInSide:(UIButton *)btn{
TouchBlock block = objc_getAssociatedObject(self, TouchBlockKey);
if (block) {
block();
}
} @end

以上就是实现代码,接下来对代码进行解释 objc_setAssociatedObject(id object, void *key, id value, objc_AssociationPolicy policy)

此方法用于为对象添加一个关联对象,其中
object
要关联的源对象
key
用于取出关联的对象
value
要关联到对象的值
policy
关联的策略


id objc_getAssociatedObject(id object, void *key)

此方法用于从对象中通过key来获取关联对象
object 关联的源对象
key objc_setAssociatedObject中使用的key

你可以从这里下载示例代码