UIButton+Block

UIButton的一个Category，使用block处理UIControlEvent事件，如常用的TouchUpInside等。
代码非原创，也是从网上看到的，用到了实际项目中，目前还没发现什么问题。

UIButton+Block.h：

#import <UIKit/UIKit.h>
#import <objc/runtime.h>

typedef void (^ActionBlock)();

@interface UIButton (Block)

/**
 *  UIButton添加UIControlEvents事件的block
 *
 *  @param event 事件
 *  @param action block代码
 */
- (void) handleControlEvent:(UIControlEvents)event withBlock:(ActionBlock)action;

@end

UIButton+Block.m:

#import "UIButton+Block.h"

@implementation UIButton (Block)

static char eventKey;

/**
 *  UIButton添加UIControlEvents事件的block
 *
 *  @param controlEvent 事件
 *  @param action block代码
 */
- (void) handleControlEvent:(UIControlEvents)event withBlock:(void (^)())action {
    objc_setAssociatedObject(self, &eventKey, action, OBJC_ASSOCIATION_COPY_NONATOMIC);
    [self addTarget:self action:@selector(callActionBlock:) forControlEvents:event];
}

- (void)callActionBlock:(id)sender {
    ActionBlock block = (ActionBlock)objc_getAssociatedObject(self, &eventKey);
    if (block) {
        block();
    }
}

@end

具体使用：

- (void)viewDidLoad {
    [super viewDidLoad];
    // Do any additional setup after loading the view, typically from a nib.
    
    [self.button handleControlEvent:UIControlEventTouchUpInside withBlock:^{
       
        NSLog(@"button touched!");
    }];
}

附上demo，这里是链接地址。

另外，毕竟是用到了objc_setAssociatedObject动态关联，虽然不知道有没有什么性能影响，但是个人感觉最好不要滥用，比如Button的点击事件里代码量较大时最好还是用原生的处理方法。
如果UIButton是动态创建的，且事件处理逻辑较少（比如就几行代码 ），我觉得还是可以使用。我目前是用在自定义导航栏按钮上。