寫在前面

今年春節的時候，支付寶“咻一咻”紅包火了起來，整個春晚都在瘋狂地點擊“咻一咻”，后來發現不對勁，要是全國人民以這樣的頻率向服務器發送請求的話，那服務器得承擔多大的重壓啊，于是細細看了一下，原來并不是每次點擊“咻一咻”，應用都會發出一次搶紅包請求，iOS系統狀態欄左側會有請求的提示圈，就算一直不停地點擊“咻一咻”，那個提示也是過個幾秒才會轉一次，原來支付寶限制了我們搶紅包的頻率。

像支付寶“咻一咻”這樣的場景，在業務中還是會經常遇到的，很多時候，UIButton的快速重復點擊，會給用戶帶來很困惑的反應，甚至直接導致crash。如果僅僅是為了某個Button的重復點擊，那很簡單，直接寫個我們自己的Button就很好解決了。但是當我們的項目已經做完了，然后發現，哇，好多重復點擊的Bug啊，這時候就需要用一種優雅的方式來解決UIButton的重復點擊問題了。

Runtime

都說Runtime是Objective-C中的黑魔法，這里有關于OC Runtime的詳細介紹。而我們在用Swift的時候，很多時候也需要用到Runtime，它與在OC中的使用還是有一點點區別的。在解決UIButton重復點擊問題時主要用到了Runtime中的關聯對象(Associated Objects)和方法交叉(Method Swizzling)。

Associated Objects

在解決UIButton重復點擊問題時，首先我們需要兩個變量，一個是點擊的間隔時間，另一個是是否需要忽視點擊事件的Bool值，那么我們就需要將這兩個變量與UIButton關聯起來。

extension UIButton {
    
    private struct AssociatedKeys {
        static var clickDurationTime = "my_clickDurationTime"
        static var isIgnoreEvent = "my_isIgnoreEvent"
    }
    
    // 點擊間隔時間
    var clickDurationTime : NSTimeInterval {
        set {
            objc_setAssociatedObject(self, &AssociatedKeys.clickDurationTime, newValue as NSTimeInterval, .OBJC_ASSOCIATION_RETAIN_NONATOMIC)
        }
        
        get {
            
            if let clickDurationTime = objc_getAssociatedObject(self, &AssociatedKeys.clickDurationTime) as? NSTimeInterval {
                return clickDurationTime
            }
            
            return defaultDuration
        }
    }
    
    // 是否忽視點擊事件
    var isIgnoreEvent : Bool {
        
        set {
            objc_setAssociatedObject(self, &AssociatedKeys.isIgnoreEvent, newValue as Bool, .OBJC_ASSOCIATION_RETAIN_NONATOMIC)
        }
        
        get {
            
            if let isIgnoreEvent = objc_getAssociatedObject(self, &AssociatedKeys.isIgnoreEvent) as? Bool {
                return isIgnoreEvent
            }
            
            return false
        }
    }
}

Method Swizzling

接下來我們需要寫一個用來替換UIButton點擊事件的方法，在這個方法里來處理重復點擊的問題。UIButton繼承自UIControl，在傳遞點擊事件是會觸發這個方法：

public func sendAction(action: Selector, to target: AnyObject?, forEvent event: UIEvent?)

于是我們重寫用來交叉的方法：

// SwizzledMethod
    func my_sendAction(action: Selector, to target: AnyObject?, forEvent event: UIEvent?) {
        
        if self.isKindOfClass(UIButton) {
            
            clickDurationTime = clickDurationTime == 0 ? defaultDuration : clickDurationTime
            
            if isIgnoreEvent {
                return
            } else if clickDurationTime > 0 {
                isIgnoreEvent = true
                // 在過了我們設置的duration之后，再將isIgnoreEvent置為false
                dispatch_after(dispatch_time(DISPATCH_TIME_NOW, Int64(clickDurationTime) * Int64(NSEC_PER_SEC)), dispatch_get_main_queue()) {
                    self.isIgnoreEvent = false
                }
                
                my_sendAction(action, to: target, forEvent: event)
            }
        } else {
            my_sendAction(action, to: target, forEvent: event)
        }
    }

接下來，就是用我們自己重寫的SwizzeldMethod來替換UIButton的OriginalMethod：

// swift現不支持重寫+load方法
    override public class func initialize() {
        struct Static {
            static var token: dispatch_once_t = 0
        }
        
        if self !== UIButton.self {
            return
        }

        dispatch_once(&Static.token) {
            
            let originalSelector = #selector(UIButton.sendAction(_:to:forEvent:))
            let swizzledSelector = #selector(UIButton.my_sendAction(_:to:forEvent:))
            
            let originalMethod = class_getInstanceMethod(self, originalSelector)
            let swizzledMethod = class_getInstanceMethod(self, swizzledSelector)
            
            // 運行時為類添加我們自己寫的my_sendAction(_:to:forEvent:)
            let didAddMethod = class_addMethod(self, originalSelector, method_getImplementation(swizzledMethod), method_getTypeEncoding(swizzledMethod))
            
            if didAddMethod {
                // 如果添加成功，則交換方法
                class_replaceMethod(self, swizzledSelector, method_getImplementation(originalMethod), method_getTypeEncoding(originalMethod))
            } else {
                // 如果添加失敗，則交換方法的具體實現
                method_exchangeImplementations(originalMethod, swizzledMethod)
            }
        }
    }

這個方法，就是在運行時將UIButton的sendAction(:to:forEvent:)替換為我們自己寫的my_sendAction(:to:forEvent:)。

load vs. initialize

在 Swift 中 load 類方法永遠不會被 runtime 調用，因此在+load方法中實現交叉就變成了不可能的事。出于安全性和一致性的考慮，我們必須確保相關的方法交叉在一個 dispatch_once 中，這樣也是很安全的。還有一種方法，在 app delegate 中實現方法交叉， 不像上面通過類擴展進行方法交叉，而是簡單地在 app delegate 的 application(_:didFinishLaunchingWithOptions:) 方法調用時中執行相關代碼也是可以的?；趯︻惖男薷模@種方法應該就足夠確保這些代碼會被執行到。