iOS|iOS - runtime实现“防止button被重复点击”的背后

好吧,今天中午刚吃完饭,屁股还没坐稳呢,CTO就跑过来问我,小姑凉啊、扫码购这块有bug啊,如果我狂点、、、、、、去结算,会给我生成尼玛一堆订单,而且支付宝无限重复唤起啊。。。我内心一开始是拒绝的,这尼玛CTO估计是闲的,后来想想,这种闲的没事干的用户估计还挺多,然后联想到这将近5、60个界面啊,我该咋办。后来网上一找,果然一大推方案啊,其中最一劳永逸的方案即是利用我们伟大的runtime机制去实现一劳永逸的做法。我内心一阵狂喜,按照方案做了一遍,果然,大功告成,啥?你们要我贴方案啊!其实网上一大堆的,不过既然你们想要,那我就贴出来吧。多个入口,省点时间,本着造福跟我一样萌萌的程序媛,我就牺牲一下自己把。
step1:创建个UIControl的分类
step2:利用runtime动态的给分类绑定属性(不会?不着急,一会就给出)
step3:关键:利用method_exchangeImplementations方法交换函数的实现(只能交换一次哦)。
代码如下:

#import "UIControl+HQStopMultiTap.h"@implementation UIControl (HQStopMultiTap) - (NSTimeInterval)timeInterval { return[objc_getAssociatedObject(self,_cmd)doubleValue]; } - (void)setTimeInterval:(NSTimeInterval)timeInterval { objc_setAssociatedObject(self,@selector(timeInterval),@(timeInterval),OBJC_ASSOCIATION_RETAIN_NONATOMIC); }//runtime动态绑定属性 - (void)setIsIgnoreEvent:(BOOL)isIgnoreEvent{objc_setAssociatedObject(self,@selector(isIgnoreEvent),@(isIgnoreEvent),OBJC_ASSOCIATION_RETAIN_NONATOMIC); }- (BOOL)isIgnoreEvent{return[objc_getAssociatedObject(self,_cmd)boolValue]; } - (void)resetState{[self setIsIgnoreEvent:NO]; } + (void)load{static dispatch_once_t onceToken; dispatch_once(&onceToken, ^{SEL selA =@selector(sendAction:to:forEvent:); SEL selB =@selector(mySendAction:to:forEvent:); Method methodA =class_getInstanceMethod(self, selA); Method methodB =class_getInstanceMethod(self, selB); //将methodB的实现添加到系统方法中也就是说将methodA方法指针添加成方法methodB的。返回值表示是否添加成功BOOL isAdd =class_addMethod(self, selA,method_getImplementation(methodB),method_getTypeEncoding(methodB)); //添加成功了说明本类中不存在methodB所以此时必须将方法b的实现指针换成方法A的,否则b方法将没有实现。if(isAdd) {class_replaceMethod(self, selB,method_getImplementation(methodA),method_getTypeEncoding(methodA)); }else{//添加失败了说明本类中有methodB的实现,此时只需要将methodA和methodB的IMP互换一下即可。 method_exchangeImplementations(methodA, methodB); } }); } - (void)mySendAction:(SEL)action to:(id)target forEvent:(UIEvent*)event {if([NSStringFromClass(self.class)isEqualToString:@"UIButton"]) {self.timeInterval =self.timeInterval==0 ? 1.5:self.timeInterval; if(self.isIgnoreEvent){return; }else if(self.timeInterval>0){[self performSelector:@selector(resetState) withObject:nil afterDelay:self.timeInterval]; } } //此处methodA和methodB方法IMP互换了,实际上执行sendAction;所以不会死循环 self.isIgnoreEvent=YES; [self mySendAction:action to:target forEvent:event]; } @end

分类代码就贴到这里了。可是,小姐姐是这种只想修复bug的人么?当然得知道他背后的故事,对不对。
首先这个思想绝对不是我想出来的,我只是看完别人贴的代码以后,做了点自己的思考,欢迎各位小哥哥们来喷,不过,请手下留情。
首先runtime动态绑定属性也没什么好分析的。主要也就是 objc_setAssociatedObject/objc_getAssociatedObject这两个方法。
那这两个属性
timeInterval 代表多长时间内不能被连续点击
isIgnoreEvent代表某个button是否不需要这种机制,可以自己设置。
其实主要的思想还是交换方法的实现。
对于一个给定的事件,UIControl会调用sendAction:to:forEvent:方法,那我们就可以写一个自己的方法,让用户点击button的时候其实执行的是我们自己的方法,在我们自己的方法里面再去调用自己的方法(因为方法调换了等同于去调用了系统的方法),但是确保用户下次点击的时候不会再进行调换,那尼玛就瞎了,又换回去了,搞毛啊。所以,用
static dispatch_once_t onceToken;
dispatch_once(&onceToken, ^{

【iOS|iOS - runtime实现“防止button被重复点击”的背后】}
保证只会交换一次。OK?
哎,最近小姐姐我也在研究runtime,好高深的样子,欢迎各位小哥哥来指教。对了,还有什么问题也可以给我留言,我尽量正经的回答你。。。

    推荐阅读