/*
**防止button快速點擊Method
*/
@interfaceUIButton (touch)
/**設置點擊時間間隔,默認0.5S*/
@property(nonatomic,assign)NSTimeIntervaltimeInterval;
/**
*用于設置單個按鈕不需要被hook,忽略YES
*/
@property(nonatomic,assign)BOOLisIgnore;
@end
#define defaultInterval.5//默認時間間隔
@interfaceUIButton()
/**bool類型YES不允許點擊NO允許點擊設置是否執行點UI方法*/
@property(nonatomic,assign)BOOLisIgnoreEvent;
@end
@implementationUIButton (touch)
+ (void)load{
staticdispatch_once_tonceToken;
dispatch_once(&onceToken, ^{
SELselA =@selector(sendAction:to:forEvent:);
SELselB =@selector(mySendAction:to:forEvent:);
MethodmethodA =class_getInstanceMethod(self,selA);
MethodmethodB =class_getInstanceMethod(self, selB);
//將methodB的實現添加到系統方法中也就是說將methodA方法指針添加成方法methodB的返回值表示是否添加成功
BOOLisAdd =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);
}
});
}
- (NSTimeInterval)timeInterval{
return[objc_getAssociatedObject(self,_cmd)doubleValue];
}
- (void)setTimeInterval:(NSTimeInterval)timeInterval{
objc_setAssociatedObject(self,@selector(timeInterval),@(timeInterval),OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}
//當我們按鈕點擊事件sendAction時將會執行mySendAction
- (void)mySendAction:(SEL)action to:(id)target forEvent:(UIEvent*)event{
if(self.isIgnore) {
//不需要被hook
[selfmySendAction:actionto:targetforEvent:event];
return;
}
if([NSStringFromClass(self.class)isEqualToString:@"UIButton"]) {
self.timeInterval=self.timeInterval==0?defaultInterval:self.timeInterval;
if(self.isIgnoreEvent){
return;
}elseif(self.timeInterval>0){
[selfperformSelector:@selector(resetState)withObject:nilafterDelay:self.timeInterval];
}
}
//此處methodA和methodB方法IMP互換了，實際上執行sendAction；所以不會死循環
self.isIgnoreEvent=YES;
[selfmySendAction:actionto:targetforEvent:event];
}
//runtime動態綁定屬性
- (void)setIsIgnoreEvent:(BOOL)isIgnoreEvent{
//注意BOOL類型需要用OBJC_ASSOCIATION_RETAIN_NONATOMIC不要用錯，否則set方法會賦值出錯
objc_setAssociatedObject(self,@selector(isIgnoreEvent),@(isIgnoreEvent),OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}
- (BOOL)isIgnoreEvent{
//_cmd == @select(isIgnore);和set方法里一致
return[objc_getAssociatedObject(self,_cmd)boolValue];
}
- (void)setIsIgnore:(BOOL)isIgnore{
//注意BOOL類型需要用OBJC_ASSOCIATION_RETAIN_NONATOMIC不要用錯，否則set方法會賦值出錯
objc_setAssociatedObject(self,@selector(isIgnore),@(isIgnore),OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}
- (BOOL)isIgnore{
//_cmd == @select(isIgnore);和set方法里一致
return[objc_getAssociatedObject(self,_cmd)boolValue];
}
- (void)resetState{
[selfsetIsIgnoreEvent:NO];
}
@end
