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是动态创建的，且事件处理逻辑较少（比如就几行代码 ），我觉得还是可以使用。我目前是用在自定义导航栏按钮上。