使用关联对象（AssociatedObject）为UIButton添加Block响应

工程师宇航技术 2022年11月13日

0 收藏 813 点赞 4,635 浏览 1196 个字

在开发中，要给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

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