iOS开发 | 让你的UIButton自带block

背景

升级到Xcode9后，项目抽风了，只要遇到RAC一定崩溃。

试了各种方法，都不行，不得已，我决定将RAC从项目中移除，RAC在我项目中只是辅助，主要用来简化代码，所以用得也不是很多。

移除RAC后必然要找到东西来替代RAC，我请教了一下iOS高端妓术裙的师傅，师傅们向我推荐了BlocksKit，这东西简直就是RAC的孪生兄弟，太好用了。我分分钟就完成了RAC到BlocksKit的替换，然而令我万万没想到的是运行到BlocksKit代码的时候还是崩了。。。

遇到的问题

既然不让我用三方，看来就只有自己动手改了，先来看一段类似代码：

+ (void)showAlertWithType:(NSInteger)type buttonClickedBlock:(void (^)(void))buttonClickedBlock {
    
    UIWindow *delegateWindow = [[[UIApplication sharedApplication] delegate] window];
    
    // 灰色view
    UIView *bgView = [[UIView alloc] initWithFrame:CGRectMake(90, 90, 200, 200)];
    [delegateWindow addSubview:bgView];
    bgView.backgroundColor = [UIColor grayColor];
    
    // 按钮（点击按钮让灰色view变成红色）
    UIButton *button = [UIButton buttonWithType:UIButtonTypeSystem];
    [bgView addSubview:button];
    button.frame = CGRectMake(30, 30, 90, 40);
    [button setTitle:@"点击变红" forState:UIControlStateNormal];
    
    // 怎么实现点击按钮让灰色view变红并且执行block？（不使用RAC）
    //button addTarget:<#(nullable id)#> action:<#(nonnull SEL)#> forControlEvents:<#(UIControlEvents)#>
}

这是一个类方法，点击按钮后会回调传过来的block，用RAC或BlocksKit来实现的话就太简单了：

[[button rac_signalForControlEvents:UIControlEventTouchUpInside] subscribeNext:^(id x) {
    if (buttonClickedBlock) {
        buttonClickedBlock();
    }
}];

但如果不让你用RAC或BlocksKit呢？

解决问题

要解决这个问题最完美的方法无疑是用自己的方式实现一个类似于RAC或BlocksKit那样的功能。
正当我为此发愁的时候，我师傅上午来报（很遗憾他不写简书）给我分享了一个他写的自定义button，看了我师傅的button后我得到了启发，然后在这基础上写了一个category：

让你的UIButton自带block

#import "UIButton+CQBlock.h"
#import <objc/runtime.h>

typedef void(^CQ_ButtonEventsBlock)(void);

@interface UIButton ()

/** 事件回调的block */
@property (nonatomic, copy) CQ_ButtonEventsBlock cq_buttonEventsBlock;

@end

@implementation UIButton (CQBlock)

//------- 添加属性 -------//

static void *cq_buttonEventsBlockKey = &cq_buttonEventsBlockKey;

- (CQ_ButtonEventsBlock)cq_buttonEventsBlock {
    return objc_getAssociatedObject(self, &cq_buttonEventsBlockKey);
}

- (void)setCq_buttonEventsBlock:(CQ_ButtonEventsBlock)cq_buttonEventsBlock {
    objc_setAssociatedObject(self, &cq_buttonEventsBlockKey, cq_buttonEventsBlock, OBJC_ASSOCIATION_COPY);
}

/**
 给按钮绑定事件回调block
 
 @param block 回调的block
 @param controlEvents 回调block的事件
 */
- (void)cq_addEventHandler:(void (^)(void))block forControlEvents:(UIControlEvents)controlEvents {
    self.cq_buttonEventsBlock = block;
    [self addTarget:self action:@selector(cq_blcokButtonClicked) forControlEvents:controlEvents];
}

// 按钮点击
- (void)cq_blcokButtonClicked {
    if (self.cq_buttonEventsBlock) {
        self.cq_buttonEventsBlock();
    }
}

@end

使用：

[button cq_addEventHandler:^{
    NSLog(@"按钮点击");
} forControlEvents:UIControlEventTouchUpInside];

方法命名我是模仿的BlocksKit:

[button bk_addEventHandler:^(id sender) {
     NSLog(@"按钮点击");
 } forControlEvents:UIControlEventTouchUpInside];

然后所有button都带有block了，用起来和BlocksKit感觉一样。
这样就以很小的改动替换了RAC。

扩展

学会这个套路之后，你要实现类似的需求就都很easy了，比如说BlocksKit中有一个方法：

[self.view bk_whenTapped:^{
    NSLog(@"单击手势");
}];

我模仿它的方法写了一个：

[self.view cq_whenTapped:^{
    NSLog(@"单击手势");
}];

详细代码：

#import "UIView+CQBlock.h"
#import <objc/runtime.h>

typedef void(^CQ_ViewTappedBlock)(void);

@interface UIView ()

/** 单击手势事件回调的block */
@property (nonatomic, copy) CQ_ViewTappedBlock cq_viewTappedBlock;

@end

@implementation UIView (CQBlock)

//------- 添加属性 -------//

static void *cq_viewTappedBlockKey = &cq_viewTappedBlockKey;

- (CQ_ViewTappedBlock)cq_viewTappedBlock {
    return objc_getAssociatedObject(self, &cq_viewTappedBlockKey);
}

- (void)setCq_viewTappedBlock:(CQ_ViewTappedBlock)cq_viewTappedBlock {
    objc_setAssociatedObject(self, &cq_viewTappedBlockKey, cq_viewTappedBlock, OBJC_ASSOCIATION_COPY);
}

/**
 与单击手势绑定的block
 
 @param tappedBlock 单击手势事件回调的block
 */
- (void)cq_whenTapped:(void(^)(void))tappedBlock {
    self.cq_viewTappedBlock = tappedBlock;
    UITapGestureRecognizer *tapGesture = [[UITapGestureRecognizer alloc] initWithTarget:self action:@selector(viewTapped)];
    [self addGestureRecognizer:tapGesture];
}

// 单击view
- (void)viewTapped {
    if (self.cq_viewTappedBlock) {
        self.cq_viewTappedBlock();
    }
}

其实本质就是利用category和runtime给UIView添加了一个block属性，单击的时候回调这个block。

这里是demo

鸣谢：

感谢我师傅上午来报的分享。

总结

有师傅的孩子是幸福的！

2017年10月19日更新

调用block有一种简化写法：

if (self.cq_viewTappedBlock) {
    self.cq_viewTappedBlock();
}

等价于：

!self.cq_viewTappedBlock ?: self.cq_viewTappedBlock();