iOS method swizz

1: 防止按钮在一定时间内重复响应默认1秒

//
//  UIButton+AvoidDoubleClick.h
//  51WaywardShop
//
//  Created by jisa on 2018/9/30.
//  Copyright © 2018年 Anyike. All rights reserved.
//

#import <UIKit/UIKit.h>

NS_ASSUME_NONNULL_BEGIN

@interface UIButton (AvoidDoubleClick)

/**
 避免多次点击重复响应。设置时间间隔（默认0.5S）
 */
@property (class, nonatomic, assign) NSTimeInterval intervalTime;

/**
 上一次点击的时间
 */
@property (nonatomic, assign) NSTimeInterval clickTime;
@end

NS_ASSUME_NONNULL_END

//
//  UIButton+AvoidDoubleClick.m
//  51WaywardShop
//
//  Created by jisa on 2018/9/30.
//  Copyright © 2018年 Anyike. All rights reserved.
//  

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

@implementation UIButton (AvoidDoubleClick)

+ (void)load {
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        UIButton.intervalTime = 1;
        SEL systemSEL = @selector(sendAction:to:forEvent:);
        Method systemMethod = class_getInstanceMethod([self class], systemSEL);
        
        SEL customSEL = @selector(FF_sendAction:target:event:);
        Method customMethod = class_getInstanceMethod(self, customSEL);
        /// 防止子类没有重载父类的方法，找不到消息接收的对象。故使用方法交换时，通常配合，方法的添加和替换使用。
        BOOL addSuccess = class_addMethod(self, systemSEL, method_getImplementation(customMethod), method_getTypeEncoding(customMethod));
        if (addSuccess) {
            class_replaceMethod(self, customSEL, method_getImplementation(systemMethod), method_getTypeEncoding(systemMethod));
        }else {
            method_exchangeImplementations(systemMethod, customMethod);
        }
    });
}


#pragma mark -- method
- (void)FF_sendAction:(SEL)action target:(id)target event:(UIEvent *)event {

    if (self.clickTime == 0) {
        self.clickTime = [NSDate date].timeIntervalSince1970;
        [self FF_sendAction:action target:target event:event];
    }
    if ([NSDate date].timeIntervalSince1970 - self.clickTime < UIButton.intervalTime) {
        return;
    }
    if (UIButton.intervalTime > 0) {
        self.clickTime = [NSDate date].timeIntervalSince1970;
    }
    [self FF_sendAction:action target:target event:event];
}


#pragma mark -- setter getter
+ (void)setIntervalTime:(NSTimeInterval)intervalTime {
    objc_setAssociatedObject(self, _cmd, @(intervalTime), OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}

+ (NSTimeInterval)intervalTime {
    NSNumber *needNumber = objc_getAssociatedObject(self, @selector(setIntervalTime:));
    return [needNumber integerValue];
}

- (void)setClickTime:(NSTimeInterval)clickTime {
    objc_setAssociatedObject(self, _cmd, @(clickTime), OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}

- (NSTimeInterval)clickTime {
    NSNumber *num = objc_getAssociatedObject(self, @selector(setClickTime:));
    return [num integerValue];
}
@end

 

2: 防止数组下标越界。

    NSArray是一个抽象类。实际上使用的是“__NSArray0(空数组), __NSSingleObjectArrayI(只有一个元素的数组), __NSArrayI（超过两个元素的数组）, __NSArrayM(可变数组)”。通过‘语法糖’获取元素时会走‘objectAtIndexedSubscript:’方法。

//
//  NSArray+Crash.h
//  NetWork
//
//  Created by jisa on 2018/11/17.
//  Copyright © 2018 jisa. All rights reserved.
//  .h文件

#import <Foundation/Foundation.h>

NS_ASSUME_NONNULL_BEGIN

@interface NSArray (Crash)

@end

NS_ASSUME_NONNULL_END

//
//  NSArray+Crash.m
//  NetWork
//
//  Created by jisa on 2018/11/17.
//  Copyright © 2018 jisa. All rights reserved.
//  如果是一个空数组，即时'__NSArray0'时可以不用单独处理,会走'__NSSingleObjectArrayI'的方法。

#import "NSArray+Crash.h"
#import <objc/runtime.h>

@implementation NSArray (Crash)
+ (void)load {
    
    /* 数组 元素1个 */
    SEL oriSingleSEL = @selector(objectAtIndex:);
    Method oriSingleMethod = class_getInstanceMethod(NSClassFromString(@"__NSSingleObjectArrayI"), oriSingleSEL);
    SEL autoSingleSEL = @selector(FF_SingleObjectAtIndex:);
    Method autoSingleMethod = class_getInstanceMethod(NSClassFromString(@"__NSSingleObjectArrayI"), autoSingleSEL);
    
    if (class_addMethod(self, oriSingleSEL, method_getImplementation(autoSingleMethod), method_getTypeEncoding(autoSingleMethod))) {
        class_replaceMethod(self, autoSingleSEL, method_getImplementation(oriSingleMethod), method_getTypeEncoding(oriSingleMethod));
    }else {
        method_exchangeImplementations(oriSingleMethod, autoSingleMethod);
    }
    
    /* 数组 语法糖 元素1个 */
    SEL oriSingleTangSEL = @selector(objectAtIndexedSubscript:);
    Method oriSingleTangMethod = class_getInstanceMethod(NSClassFromString(@"__NSSingleObjectArrayI"), oriSingleTangSEL);
    SEL autoSingleTangSEL = @selector(FF_SingleTangObjectAtIndex:);
    Method autoSingleTangMethod = class_getInstanceMethod(NSClassFromString(@"__NSSingleObjectArrayI"), autoSingleTangSEL);
    if (class_addMethod(self, oriSingleTangSEL, method_getImplementation(autoSingleTangMethod), method_getTypeEncoding(autoSingleTangMethod))) {
        class_replaceMethod(self, autoSingleTangSEL, method_getImplementation(oriSingleTangMethod), method_getTypeEncoding(oriSingleTangMethod));
    }else {
        method_exchangeImplementations(oriSingleTangMethod, autoSingleTangMethod);
    }

    
    /* 数组 元素超过2 */
    SEL oriSEL = @selector(objectAtIndex:);
    Method oriMethod = class_getInstanceMethod(NSClassFromString(@"__NSArrayI"), oriSEL);
    SEL autoSEL = @selector(FF_ObjectAtIndex:);
    Method autoMethod = class_getInstanceMethod(NSClassFromString(@"__NSArrayI"), autoSEL);

    if (class_addMethod(self, oriSEL, method_getImplementation(autoMethod), method_getTypeEncoding(oriMethod))) {
        class_replaceMethod(self, autoSEL, method_getImplementation(oriMethod), method_getTypeEncoding(oriMethod));
    }else {
        method_exchangeImplementations(oriMethod, autoMethod);
    }
    
    /* 数组 语法糖 元素超过2 */
    SEL oriTangSEL = @selector(objectAtIndexedSubscript:);
    Method oriTangMethod = class_getInstanceMethod(NSClassFromString(@"__NSArrayI"), oriTangSEL);
    SEL autoTangSEL = @selector(FF_ObjectAtIndexSubscript:);
    Method autoTangMethod = class_getInstanceMethod(NSClassFromString(@"__NSArrayI"), autoTangSEL);
    if (class_addMethod(self, oriTangSEL, method_getImplementation(autoTangMethod), method_getTypeEncoding(autoTangMethod))) {
        class_replaceMethod(self, autoTangSEL, method_getImplementation(oriTangMethod), method_getTypeEncoding(oriTangMethod));
    }else {
        method_exchangeImplementations(oriTangMethod, autoTangMethod);
    }

    /*   可变数组    */
    SEL oriMSEL = @selector(objectAtIndex:);
    Method oriMMethod = class_getInstanceMethod(NSClassFromString(@"__NSArrayM"), oriMSEL);
    SEL autoMSEL = @selector(FF_MutableObjectAtIndex:);
    Method autoMMethod = class_getInstanceMethod(NSClassFromString(@"__NSArrayM"), autoMSEL);
    
    if (class_addMethod(self, oriMSEL, method_getImplementation(autoMMethod), method_getTypeEncoding(autoMMethod))) {
        class_replaceMethod(self, autoMSEL, method_getImplementation(oriMMethod), method_getTypeEncoding(oriMMethod));
    }else {
        method_exchangeImplementations(oriMMethod, autoMMethod);
    }
    
    /*  可变数组语法糖   */
    SEL oriMTangSEL = @selector(objectAtIndexedSubscript:);
    Method oriMTangMethod = class_getInstanceMethod(NSClassFromString(@"__NSArrayM"), oriMTangSEL);
    SEL autoMTangSEL = @selector(FF_MutableObjectAtIndexSubscript:);
    Method autoMTangMethod = class_getInstanceMethod(NSClassFromString(@"__NSArrayM"), autoMTangSEL);
    if (class_addMethod(self, oriMTangSEL, method_getImplementation(autoMTangMethod), method_getTypeEncoding(autoMTangMethod))) {
        class_replaceMethod(self, autoMTangSEL, method_getImplementation(oriMTangMethod), method_getTypeEncoding(oriMTangMethod));
    }else {
        method_exchangeImplementations(oriMTangMethod, autoMTangMethod);
    }

    
}

#pragma mark -- method
- (id)FF_SingleObjectAtIndex:(NSInteger)index {
    if (self.count - 1 < index || self.count == 0) {
        @try {
            return [self FF_SingleObjectAtIndex:index];
        } @catch (NSException *exception) {
            NSLog(@"%@", exception.reason);
            return nil;
        } @finally {
            NSLog(@"下标越界");
        }
    }else {
        return [self FF_SingleObjectAtIndex:index];
    }
}


- (id)FF_SingleTangObjectAtIndex:(NSInteger)index {
    if (self.count - 1 < index || self.count == 0) {
        @try {
            return [self FF_SingleTangObjectAtIndex:index];
        } @catch (NSException *exception) {
            NSLog(@"%@", exception.reason);
            return nil;
        } @finally {
            NSLog(@"下标越界");
        }
    }else {
        return [self FF_SingleTangObjectAtIndex:index];
    }
}


- (id)FF_ObjectAtIndex:(NSInteger)index {
    if (self.count - 1 < index || self.count == 0) {
        @try {
            return [self FF_ObjectAtIndex:index];
        } @catch (NSException *exception) {
            NSLog(@"%@", exception.reason);
            return nil;
        } @finally {
            NSLog(@"下标越界");
        }
    }else {
        return [self FF_ObjectAtIndex:index];
    }
}

- (id)FF_ObjectAtIndexSubscript:(NSInteger)index {
    if (self.count - 1 < index || self.count == 0) {
        @try {
            return [self FF_ObjectAtIndexSubscript:index];
        } @catch (NSException *exception) {
            NSLog(@"%@", exception.reason);
            return nil;
        } @finally {
            NSLog(@"下标越界");
        }
    }else {
        return [self FF_ObjectAtIndexSubscript:index];
    }
}

- (id)FF_MutableObjectAtIndex:(NSInteger)index {
    if (self.count - 1 < index || self.count == 0 ) {
        @try {
            return [self FF_MutableObjectAtIndex:index];
        } @catch (NSException *exception) {
            NSLog(@"%@", exception.reason);
            return nil;
        } @finally {
            NSLog(@"下标越界");
        }
    }else {
        return [self FF_MutableObjectAtIndex:index];
    }
}

- (id)FF_MutableObjectAtIndexSubscript:(NSInteger)index {
    if (self.count - 1 < index || self.count == 0 ) {
        @try {
            return [self FF_MutableObjectAtIndexSubscript:index];
        } @catch (NSException *exception) {
            NSLog(@"%@", exception.reason);
            return nil;
        } @finally {
            NSLog(@"下标越界");
        }
    }else {
        return [self FF_MutableObjectAtIndexSubscript:index];
    }
}

@end