[iOS ]Runtime防止按钮重复点击

前言

最近老是被QA提bug，原因是按钮连续连击跳转两次页面。原先觉得不是什么大事，所以直接用UIButtom的enable属性来解决，但是这种方法治标不治本。刚好最近在学习Runtime，所以直接应用一下吧。

首先，新建一个UIButtom的分类

image.png

在UIButton+Delay.m文件中增加属性

// 重复点击间隔
@property (nonatomic, assign) NSTimeInterval acceptEventInterval;
// 上一次点击时间戳
@property (nonatomic, assign) NSTimeInterval acceptEventTime;

重写load方法，利用Runtime进行方法交换

 + (void)load {
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        Class class = [self class];

        SEL originalSelector = @selector(sendAction:to:forEvent:);
        SEL swizzledSelector = @selector(mySendAction:to:forEvent:);

        Method originalMethod = class_getInstanceMethod(class, originalSelector);
        Method swizzledMethod = class_getInstanceMethod(class, swizzledSelector);

        BOOL didAddMethod = class_addMethod(class,
                                            originalSelector,
                                            method_getImplementation(swizzledMethod),
                                            method_getTypeEncoding(swizzledMethod));

        if (didAddMethod) {
            class_replaceMethod(class,
                                swizzledSelector,
                                method_getImplementation(originalMethod),
                                method_getTypeEncoding(originalMethod));
        } else {
            method_exchangeImplementations(originalMethod, swizzledMethod);
        }
    });
}

实现要交换的方法判断是否短时间内连续点击

- (void)mySendAction:(SEL)action to:(id)target forEvent:(UIEvent *)event {
    
    // 如果想要设置统一的间隔时间，可以在此处加上以下几句
    if (self.acceptEventInterval <= 0) {
        // 如果没有自定义时间间隔，则默认为 0.4 秒
        self.acceptEventInterval = 0.4;
    }
    
    // 是否小于设定的时间间隔
    BOOL needSendAction = (NSDate.date.timeIntervalSince1970 - self.acceptEventTime >= self.acceptEventInterval);
    
    // 更新上一次点击时间戳
    if (self.acceptEventInterval > 0) {
        self.acceptEventTime = NSDate.date.timeIntervalSince1970;
    }
    
    // 两次点击的时间间隔小于设定的时间间隔时，才执行响应事件
    if (needSendAction) {
        [self mySendAction:action to:target forEvent:event];
    }else{
        NSLog(@"连续点击，不响应");
    }
}

利用Runtime动态添加属性

- (NSTimeInterval )acceptEventInterval{
    return [objc_getAssociatedObject(self, "UIControl_acceptEventInterval") doubleValue];
}

- (void)setAcceptEventInterval:(NSTimeInterval)acceptEventInterval{
    objc_setAssociatedObject(self, "UIControl_acceptEventInterval", @(acceptEventInterval), OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}

- (NSTimeInterval )acceptEventTime{
    return [objc_getAssociatedObject(self, "UIControl_acceptEventTime") doubleValue];
}

- (void)setAcceptEventTime:(NSTimeInterval)acceptEventTime{
    objc_setAssociatedObject(self, "UIControl_acceptEventTime", @(acceptEventTime), OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}

补充：最近被QA提了一个bug，调用系统摄像头拍照的时候点击拍摄按钮没有反应，经排查，是因为添加了UIButton+Delay.h类。解决方法如下：

- (void)mySendAction:(SEL)action to:(id)target forEvent:(UIEvent *)event {
    
    // 如果想要设置统一的间隔时间，可以在此处加上以下几句
    if (self.acceptEventInterval <= 0) {
        // 如果没有自定义时间间隔，则默认为 0.4 秒
        self.acceptEventInterval = 0.4;
    }
    
    // 是否小于设定的时间间隔
    BOOL needSendAction = (NSDate.date.timeIntervalSince1970 - self.acceptEventTime >= self.acceptEventInterval);
    
    // 更新上一次点击时间戳
    if (self.acceptEventInterval > 0) {
        self.acceptEventTime = NSDate.date.timeIntervalSince1970;
    }
    NSString *method = NSStringFromSelector(action);
    if ([method isEqualToString:@"_handleShutterButtonPressed:"]) {
        //点击相机拍照按钮事件则不判断是连续点击.
        [self mySendAction:action to:target forEvent:event];
        return;
    }
    // 两次点击的时间间隔小于设定的时间间隔时，才执行响应事件
    if (needSendAction) {
        [self mySendAction:action to:target forEvent:event];
    }else{
        NSLog(@"连续点击，不响应");
    }
}

继续补充：最近又发现一个bug，当同一个按钮设置了不同的UIControlEvent事件是，会被判断成连续点击，例如

 [_recordButton addTarget:self action:@selector(recordBtnDown:) forControlEvents:UIControlEventTouchDown];
    [_recordButton addTarget:self action:@selector(recordBtnUp:) forControlEvents:UIControlEventTouchUpInside];

上面的代码一个是按下后触发，一个是抬起后触发，事件间隔几乎同时。因此会变成连续点击，解决方法如下
在UIButton+Delay.h文件中增加属性

// 上一次点击的方法
@property (nonatomic, assign) NSString *lastActionName;

在UIButton+Delay.m文件中关联属性

- (NSString * )lastActionName{
    return objc_getAssociatedObject(self, "UIControl_lastActionName");
}
- (void)setLastActionName:(NSString*)lastActionName{
    objc_setAssociatedObject(self, "UIControl_lastActionName", lastActionName, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}

修改方法

- (void)mySendAction:(SEL)action to:(id)target forEvent:(UIEvent *)event {
    
    // 如果想要设置统一的间隔时间，可以在此处加上以下几句
    if (self.acceptEventInterval <= 0) {
        // 如果没有自定义时间间隔，则默认为 0.4 秒
        self.acceptEventInterval = 0.4;
    }
    
    // 是否小于设定的时间间隔
    BOOL needSendAction = (NSDate.date.timeIntervalSince1970 - self.acceptEventTime >= self.acceptEventInterval);
    // 更新上一次点击时间戳
    if (self.acceptEventInterval > 0) {
        self.acceptEventTime = NSDate.date.timeIntervalSince1970;
    }
    NSString *method = NSStringFromSelector(action);
    if ([method isEqualToString:@"_handleShutterButtonPressed:"]) {
        //点击相机拍照按钮事件则不判断是连续点击.
        [self mySendAction:action to:target forEvent:event];
        return;
    }
    NSLog(@"---```%@",NSStringFromSelector(action));
    // 两次点击的时间间隔小于设定的时间间隔时，才执行响应事件
    if (needSendAction) {
        [self mySendAction:action to:target forEvent:event];
    }else{
        if ([self.lastActionName isEqualToString:NSStringFromSelector(action)]) {
            NSLog(@"连续点击，不响应");
        }else{
            [self mySendAction:action to:target forEvent:event];
        }
    }
    self.lastActionName = NSStringFromSelector(action);
}