iOS UIButton 传递参数的解决办法

需求

原生的UIButton的点击事件唯一的参数就是UIButton本身，我们通常使用UIButton自带的tag来使用不同的参数，在简单的业务场景下，通过tag都是可以满足需求的，但是在某些业务复杂的情况下，tag显得有些无力了，毕竟通过tag来传递点击事件传递参数只是一种间接的方式，没有将数据源关联到控件上。例如 table 视图有多个section，cell上有多个btn，那么btn的点击事件要如何获取到对应的数据呢？事情变得有些复杂。

解决方案

• 间接获取：tag

// 设置tag
btn.tag = indexPath.row;
[btn addTarget:self action:@selector(btnAction:) forControlEvents:UIControlEventTouchUpInside];

// 通过tag获取数据
-(void)btnAction:(UIButton *)btn{
    NSLog(@"%@",self.data[btn.tag]);
}

• 间接获取：父视图

这种方式需要将数据源绑定到父视图上，当点击btn时，通过父视图来获取数据
业务应用：一个cell上有多个按钮，共用cell的数据时。

• 直接获取：自定义子类

继承 UIButton 新增属性作为参数。

@interface MyButton : UIButton
@property (strong ,nonatomic) NSDictionary *paramDic; // 用来传递参数
@end

// 直接赋值
btn.paramDic = @{@"name":@"LOLITA",@"age":@"24"};

-(void)btnAction:(MyButton *)btn{
    NSLog(@"%@",btn.paramDic);
}

• 直接获取：分类，新增属性

如果你不想通过自定义btn的方式来传递参数，你可以通过分类来为你的UIButton来新增一个属性。

@interface UIButton (PassValue)
@property (strong ,nonatomic) NSDictionary *paramDic;
@end

// 实现setter、getter方法
-(NSDictionary *)paramDic{
    return objc_getAssociatedObject(self, _cmd);
}
-(void)setParamDic:(NSDictionary *)paramDic{
    objc_setAssociatedObject(self, @selector(paramDic), paramDic, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}

使用 UIButton 的新属性

UIButton *btn = [[UIButton alloc] initWithFrame:CGRectMake(0, 0, 200, 100)];
btn.paramDic = @{@"name":@"LOLITA",@"age":@"24"};

-(void)btnAction:(MyButton *)btn{
    NSLog(@"%@",btn.paramDic); 
}

• 直接获取： 动态运行时绑定

在上一个办法中，我们已经使用了动态运行时实现了btn的setter和getter方法来为分类新增属性了，你可以直接使用运行时绑定数据。

// 绑定数据源
objc_setAssociatedObject(btn, @"myBtn", dataDic, OBJC_ASSOCIATION_RETAIN_NONATOMIC);

-(void)myBtnClick:(UIButton *)sender{
    NSDictionary *dic = objc_getAssociatedObject(sender, @"myBtn");
    NSLog(@"%@",dic);
}

但是这中方式的问题在于，每个 UIButton 都需要一个关联 key，这个 key 可以是字符串、方法等，所以在复杂的情况下，还是推荐上一种办法。

• 方式变更

target-action 是 OC 中经典的事件传递，但是正像之前所说的那样，传递参数非常的麻烦，我们可以尝试改变这种方式。这里演示一下通过匿名函数 block 来传递参数。

@interface UIButton (LLTool)
- (void)addEventBlock:(void(^)(UIButton *sender))block forControlEvents:(UIControlEvents)controlEvents;
@end

#import <objc/runtime.h>
@interface UIButton ()
@property (strong, nonatomic) NSDictionary *event_blocks;//block事件缓存
@end
@implementation UIButton (LLTool)
/// 添加处理事件和回调
- (void)addEventBlock:(void(^)(UIButton *sender))block forControlEvents:(UIControlEvents)controlEvents{
    NSAssert(block, @"不行，block必须实现！");
    SEL sel = NULL;
    switch (controlEvents) {
        case UIControlEventTouchDown:
            sel = @selector(UIControlEventTouchDown);
            break;
        case UIControlEventTouchDownRepeat:
            sel = @selector(UIControlEventTouchDownRepeat);
            break;
        case UIControlEventTouchDragInside:
            sel = @selector(UIControlEventTouchDragInside);
            break;
        case UIControlEventTouchDragOutside:
            sel = @selector(UIControlEventTouchDragOutside);
            break;
        case UIControlEventTouchDragEnter:
            sel = @selector(UIControlEventTouchDragEnter);
            break;
        case UIControlEventTouchDragExit:
            sel = @selector(UIControlEventTouchDragExit);
            break;
        case UIControlEventTouchUpInside:
            sel = @selector(UIControlEventTouchUpInside);
            break;
        case UIControlEventTouchUpOutside:
            sel = @selector(UIControlEventTouchUpOutside);
            break;
        case UIControlEventTouchCancel:
            sel = @selector(UIControlEventTouchCancel);
            break;
        default:
            break;
    }
    /// 将所有的 block 存储起来
    [self.event_blocks setValue:block forKey:NSStringFromSelector(sel)];
    [self addTarget:self action:sel forControlEvents:controlEvents];
}
- (void)UIControlEventTouchDown{[self block:_cmd];}
- (void)UIControlEventTouchDownRepeat{[self block:_cmd];}
- (void)UIControlEventTouchDragInside{[self block:_cmd];}
- (void)UIControlEventTouchDragOutside{[self block:_cmd];}
- (void)UIControlEventTouchDragEnter{[self block:_cmd];}
- (void)UIControlEventTouchDragExit{[self block:_cmd];}
- (void)UIControlEventTouchUpInside{[self block:_cmd];}
- (void)UIControlEventTouchUpOutside{[self block:_cmd];}
- (void)UIControlEventTouchCancel{[self block:_cmd];}
- (void)block:(SEL)cmd{
    NSMutableDictionary *dic = objc_getAssociatedObject(self, @selector(event_blocks));
    void (^blcok)(UIButton *) = [dic objectForKey:NSStringFromSelector(cmd)];
    if(blcok){
        blcok(self);
    }
}
/// 运行时绑定属性
- (NSDictionary *)event_blocks{
    NSMutableDictionary *dic = objc_getAssociatedObject(self, _cmd);
    if(!dic){
        objc_setAssociatedObject(self, _cmd, [NSMutableDictionary dictionary], OBJC_ASSOCIATION_RETAIN_NONATOMIC);
        dic = objc_getAssociatedObject(self, _cmd);
    }
    return dic;
}
@end

那么使用的时候就可以将事件通过 block 回调回来，注意循环引用问题即可。

[cell.btn addEventBlock:^(UIButton *sender) {
	// do something
} forControlEvents:UIControlEventTouchUpInside];

总结

大多数情况下，UIButton 都可以通过 tag 标识位完成数据获取，在一些复杂的情况下，就有些捉襟见肘了，解决办法多样，各有各的的优缺点，自定义UIButton需要你替换旧的btn，分类方式则显得自然很多。改变事件传递的方式也不妨是一个优良的解决方式，它让你的代码变得高内聚的特点。
因此，个人比较推荐使用分类和回调的形式来解决 UIButton 的参数传递。