美文网首页
iOS-防止UIButton重复点击

iOS-防止UIButton重复点击

作者: th先生 | 来源:发表于2017-10-13 13:36 被阅读0次

    通过添加UIButton类别,即可轻松搞定。
    代码如下:

    .h

    #import <UIKit/UIKit.h>
    
    @interface UIButton (CS_FixMultiClick)
    
    @property (nonatomic, assign) NSTimeInterval cs_acceptEventInterval; // 重复点击的间隔
    
    @property (nonatomic, assign) NSTimeInterval cs_acceptEventTime;
    
    @end
    

    .m

    #import "UIButton+CS_FixMultiClick.h"
    #import <objc/runtime.h>
    
    @implementation UIButton (CS_FixMultiClick)
    // 因category不能添加属性,只能通过关联对象的方式。
    static const char *UIControl_acceptEventInterval = "UIControl_acceptEventInterval";
    
    - (NSTimeInterval)cs_acceptEventInterval {
        return  [objc_getAssociatedObject(self, UIControl_acceptEventInterval) doubleValue];
    }
    
    - (void)setCs_acceptEventInterval:(NSTimeInterval)cs_acceptEventInterval {
        objc_setAssociatedObject(self, UIControl_acceptEventInterval, @(cs_acceptEventInterval), OBJC_ASSOCIATION_RETAIN_NONATOMIC);
    }
    
    static const char *UIControl_acceptEventTime = "UIControl_acceptEventTime";
    
    - (NSTimeInterval)cs_acceptEventTime {
        return  [objc_getAssociatedObject(self, UIControl_acceptEventTime) doubleValue];
    }
    
    - (void)setCs_acceptEventTime:(NSTimeInterval)cs_acceptEventTime {
        objc_setAssociatedObject(self, UIControl_acceptEventTime, @(cs_acceptEventTime), OBJC_ASSOCIATION_RETAIN_NONATOMIC);
    }
    
    
    // 在load时执行hook
    + (void)load {
        
    //    Method before   = class_getInstanceMethod(self, @selector(sendAction:to:forEvent:));
    //    Method after    = class_getInstanceMethod(self, @selector(cs_sendAction:to:forEvent:));
    //    method_exchangeImplementations(before, after);
        
        static dispatch_once_t onceToken;
        dispatch_once(&onceToken, ^{
            Class class = [self class];
            //分别获取
            SEL beforeSelector = @selector(sendAction:to:forEvent:);
            SEL afterSelector = @selector(cs_sendAction:to:forEvent:);
            
            Method beforeMethod = class_getInstanceMethod(class, beforeSelector);
            Method afterMethod = class_getInstanceMethod(class, afterSelector);
            //先尝试给原来的方法添加实现,如果原来的方法不存在就可以添加成功。返回为YES,否则
            //返回为NO。
            //UIButton 真的没有sendAction方法的实现,这是继承了UIControl的而已,UIControl才真正的实现了。
            BOOL didAddMethod =
            class_addMethod(class,
                            beforeSelector,
                            method_getImplementation(afterMethod),
                            method_getTypeEncoding(afterMethod));
    
            if (didAddMethod) {
                // 如果之前不存在,但是添加成功了,此时添加成功的是cs_sendAction方法的实现
                // 这里只需要方法替换
                class_replaceMethod(class,
                                    afterSelector,
                                    method_getImplementation(beforeMethod),
                                    method_getTypeEncoding(beforeMethod));
            } else {
                //本来如果存在就进行交换
                method_exchangeImplementations(afterMethod, beforeMethod);
            }
        });
        
    }
    
    - (void)cs_sendAction:(SEL)action to:(id)target forEvent:(UIEvent *)event {
        
        if ([NSDate date].timeIntervalSince1970 - self.cs_acceptEventTime < self.cs_acceptEventInterval) {
            return;
        }
        
        if (self.cs_acceptEventInterval > 0) {
            self.cs_acceptEventTime = [NSDate date].timeIntervalSince1970;
        }
        
        [self cs_sendAction:action to:target forEvent:event];
    }
    
    @end
    

    用法

    btn.cs_acceptEventInterval = 3;
    

    搞定了~

    相关文章

      网友评论

          本文标题:iOS-防止UIButton重复点击

          本文链接:https://www.haomeiwen.com/subject/xuznyxtx.html