iOS runtime 应用之给UIButton添加点击事件的 block

2018-02-09 12:48:39来源:https://www.jianshu.com/p/06f5f5da6899作者:two哥人点击

分享



在之前的一篇文章中已经介绍过, 可以通过 runtime 机制给已有的类"添加"对象属性和非对象属性)[详情]. 那么接下来, 再给系统的 UIButton 添加一个 block 属性(毕竟在 iOS 中, block 是一种 id 类型), 当点击 button 时, 直接触发 block 块中的代码, 这样就可以避免每次使用 UIButton 时都要调用系统的addTarget: action: forControlEvents: 方法.



首先给需要使用的 block 类型定义一个别名:
typedef void(^MHButtonActionCallBack)(UIButton *button);

这样当需要使用这种 bolck 类型的时候, 就不需要把注意力放在 block 的具体内容上了, 即: 不需要考虑 block 的传参和返回值类型了.


利用iOS runtime 应用之给 NSString 添加对象属性和非对象属性中介绍的原理给UIButton"添加"一个MHButtonActionCallBack类型的属性(诚然, 这种"添加"并不是严格意义上的添加, 只不过是添加了一对gettersetter 方法而已):
- (void)setMHCallBack:(MHButtonActionCallBack)callBack {
objc_setAssociatedObject(self, &_callBack, callBack, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}
- (MHButtonActionCallBack)getMHCallBack {
return (MHButtonActionCallBack)objc_getAssociatedObject(self, &_callBack);
}

这样就可以放心大胆的使用UIButtonsetMHCallBack:getMHCallBack:方法进行赋值和取值操作了.


为了便于调用, 还要把上一步"添加"的"属性", 封装到UIButton 内部, 只暴露出下面的方法:
- (void)addMHClickAction:(MHButtonActionCallBack)callBack;

至此, 当需要给UIButton 添加点击事件的时候, 就可以直接调用这个方法就可以了.


附上完整代码:
UIButton+MHExtra.h


#import <UIKit/UIKit.h>
typedef void(^MHButtonActionCallBack)(UIButton *button);
@interface UIButton (MHExtra)

/**
* @brief replace the method 'addTarget:forControlEvents:'
*/
- (void)addMHCallBackAction:(MHButtonActionCallBack)callBack forControlEvents:(UIControlEvents)controlEvents;
/**
* @brief replace the method 'addTarget:forControlEvents:UIControlEventTouchUpInside'
* the property 'alpha' being 0.5 while 'UIControlEventTouchUpInside'
*/
- (void)addMHClickAction:(MHButtonActionCallBack)callBack;
@end

UIButton+MHExtra.m


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

/**
* @brief add action callback to uibutton
*/
@interface UIButton (MHAddCallBackBlock)
- (void)setMHCallBack:(MHButtonActionCallBack)callBack;
- (MHButtonActionCallBack)getMHCallBack;
@end
@implementation UIButton (MHAddCallBackBlock)
static MHButtonActionCallBack _callBack;
- (void)setMHCallBack:(MHButtonActionCallBack)callBack {
objc_setAssociatedObject(self, &_callBack, callBack, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}
- (MHButtonActionCallBack)getMHCallBack {
return (MHButtonActionCallBack)objc_getAssociatedObject(self, &_callBack);
}
@end;

@implementation UIButton (MHExtra)
/**
* @brief replace the method 'addTarget:forControlEvents:UIControlEventTouchUpInside'
* the property 'alpha' being 0.5 while 'UIControlEventTouchUpInside'
*/
- (void)addMHClickAction:(MHButtonActionCallBack)callBack {
[self addMHCallBackAction:callBack forControlEvents:(UIControlEventTouchUpInside)];
[self addTarget:self action:@selector(mhTouchDownAction:) forControlEvents:(UIControlEventTouchDown)];
[self addTarget:self action:@selector(mhTouchUpAction:) forControlEvents:(UIControlEventTouchUpInside | UIControlEventTouchUpOutside | UIControlEventTouchCancel | UIControlEventTouchDragOutside)];
}
/**
* @brief replace the method 'addTarget:forControlEvents:'
*/
- (void)addMHCallBackAction:(MHButtonActionCallBack)callBack forControlEvents:(UIControlEvents)controlEvents{
[self setMHCallBack:callBack];
[self addTarget:self action:@selector(mhButtonAction:) forControlEvents:controlEvents];
}
- (void)mhButtonAction:(UIButton *)btn {
self.getMHCallBack(btn);
}
- (void)mhTouchDownAction:(UIButton *)btn {
btn.enabled = NO;
btn.alpha = 0.5f;
}
- (void)mhTouchUpAction:(UIButton *)btn {
btn.enabled = YES;
btn.alpha = 1.0f;
}
@end

从现在开始, 当需要给UIBuootn 添加点击事件的时候, 就不需要先调用addTarget: action: forControlEvents:, 然后在实现其中的action 方法了, 直接:


    [btn addMHClickAction:^(UIButton *button) {
NSLog(@"clicked button");
}];

就 O 了~


至此, 给UIButton 添加点击事件的 block 就完成了!
如果各位有什么好的思路或者对我的代码有什么优化, 欢迎积极留言哦~








最新文章

123

最新摄影

闪念基因

微信扫一扫

第七城市微信公众平台