交換方法Method Swizzling[swift]

1. dispatch_once替代方案

OC中用來保證代碼塊只執(zhí)行一次的dispatch_once在swfit中已經(jīng)被廢棄了,取而代之的是使用static let,let本身就帶有線程安全性質(zhì)的.

例如單例的實現(xiàn).

final public class MySingleton {
    static let shared = MySingleton()
    private init() {}
}

但如果我們不想定義常量,需要某個代碼塊執(zhí)行一次呢?

private lazy var takeOnceTime: Void = {
    // 代碼塊...
}()

_ = takeOnceTime

定義一個懶加載的變量,防止在初始化的時候被執(zhí)行.后面加一個void,為了在_ = takeOnceTime賦值時不耗性能,返回一個Void類型.

lazy var改為static let也可以,為了使用方便,我們用一個類方法封裝下

class ClassName {
    private static let takeOnceTime: Void = {
        // 代碼塊...
    }()
    static func takeOnceTimeFunc() {
        ClassName.takeOnceTime
    }
}

// 使用
ClassName.takeOnceTimeFunc()

這樣就可以做到和dispatch_once一樣的效果了.

2. 被廢棄的+load()和+initialize()

我們都知道OC中兩個方法+load()+initialize().

+load(): app啟動的時候會加載所有的類,此時就會調(diào)用每個類的load方法.
+initialize(): 第一次初始化這個類的時候會被調(diào)用.

然而在目前的swift版本中這兩個方法都不可用了,那現(xiàn)在我們要在這個階段搞事情該怎么做? 例如method swizzling.

JORDAN SMITH大神給出了一種很巧解決方案.UIApplication有一個next屬性,它會在applicationDidFinishLaunching之前被調(diào)用,這個時候通過runtime獲取到所有類的列表,然后向所有遵循SelfAware協(xié)議的類發(fā)送消息.

extension UIApplication {
    private static let runOnce: Void = {
        NothingToSeeHere.harmlessFunction()
    }()
    override open var next: UIResponder? {
        // Called before applicationDidFinishLaunching
        UIApplication.runOnce
        return super.next
    }
}

protocol SelfAware: class {
    static func awake()
}
class NothingToSeeHere {
    static func harmlessFunction() {
        let typeCount = Int(objc_getClassList(nil, 0))
        let types = UnsafeMutablePointer<AnyClass>.allocate(capacity: typeCount)
        let autoreleasingTypes = AutoreleasingUnsafeMutablePointer<AnyClass>(types)
        objc_getClassList(autoreleasingTypes, Int32(typeCount))
        for index in 0 ..< typeCount {
            (types[index] as? SelfAware.Type)?.awake()
        }
        types.deallocate()
    }
}

之后任何遵守SelfAware協(xié)議實現(xiàn)的+awake()方法在這個階段都會被調(diào)用.

3. 交換方法 Method Swizzling

黑魔法Method Swizzling在swift中實現(xiàn)的兩個困難點

  • swizzling 應(yīng)該保證只會執(zhí)行一次.
  • swizzling 應(yīng)該在加載所有類的時候調(diào)用.

分別在上面給出了解決方案.

下面給出了兩個示例供參考:

protocol SelfAware: class {
    static func awake()
    static func swizzlingForClass(_ forClass: AnyClass, originalSelector: Selector, swizzledSelector: Selector)
}

extension SelfAware {
    
    static func swizzlingForClass(_ forClass: AnyClass, originalSelector: Selector, swizzledSelector: Selector) {
        let originalMethod = class_getInstanceMethod(forClass, originalSelector)
        let swizzledMethod = class_getInstanceMethod(forClass, swizzledSelector)
        guard (originalMethod != nil && swizzledMethod != nil) else {
            return
        }
        if class_addMethod(forClass, originalSelector, method_getImplementation(swizzledMethod!), method_getTypeEncoding(swizzledMethod!)) {
            class_replaceMethod(forClass, swizzledSelector, method_getImplementation(originalMethod!), method_getTypeEncoding(originalMethod!))
        } else {
            method_exchangeImplementations(originalMethod!, swizzledMethod!)
        }
    }
}

class NothingToSeeHere {
    static func harmlessFunction() {
        let typeCount = Int(objc_getClassList(nil, 0))
        let types = UnsafeMutablePointer<AnyClass>.allocate(capacity: typeCount)
        let autoreleasingTypes = AutoreleasingUnsafeMutablePointer<AnyClass>(types)
        objc_getClassList(autoreleasingTypes, Int32(typeCount))
        for index in 0 ..< typeCount {
            (types[index] as? SelfAware.Type)?.awake()
        }
        types.deallocate()
    }
}
extension UIApplication {
    private static let runOnce: Void = {
        NothingToSeeHere.harmlessFunction()
    }()
    override open var next: UIResponder? {
        UIApplication.runOnce
        return super.next
    }
}

SelfAwareextension中為swizzlingForClass做了默認(rèn)實現(xiàn),相當(dāng)于一層封裝.

1. 給按鈕添加點擊計數(shù)
extension UIButton: SelfAware {
    static func awake() {
        UIButton.takeOnceTime
    }
    private static let takeOnceTime: Void = {
        let originalSelector = #selector(sendAction)
        let swizzledSelector = #selector(xxx_sendAction(action:to:forEvent:))
        
        swizzlingForClass(UIButton.self, originalSelector: originalSelector, swizzledSelector: swizzledSelector)
    }()
    
    @objc public func xxx_sendAction(action: Selector, to: AnyObject!, forEvent: UIEvent!) {
        struct xxx_buttonTapCounter {
            static var count: Int = 0
        }
        xxx_buttonTapCounter.count += 1
        print(xxx_buttonTapCounter.count)
        xxx_sendAction(action: action, to: to, forEvent: forEvent)
    }
}
2. 替換控制器的viewWillAppear方法
extension UIViewController: SelfAware {
    static func awake() {
        swizzleMethod
    }
    private static let swizzleMethod: Void = {
        let originalSelector = #selector(viewWillAppear(_:))
        let swizzledSelector = #selector(swizzled_viewWillAppear(_:))
        
        swizzlingForClass(UIViewController.self, originalSelector: originalSelector, swizzledSelector: swizzledSelector)
    }()
    
    @objc func swizzled_viewWillAppear(_ animated: Bool) {
        swizzled_viewWillAppear(animated)
        print("swizzled_viewWillAppear")
    }
}
本文收錄于 SwiftTips

如有疑問,歡迎留言 :-D

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末罚屋,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子柱徙,更是在濱河造成了極大的恐慌,老刑警劉巖盛嘿,帶你破解...
    沈念sama閱讀 218,755評論 6 507
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件削锰,死亡現(xiàn)場離奇詭異妹卿,居然都是意外死亡,警方通過查閱死者的電腦和手機坑夯,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,305評論 3 395
  • 文/潘曉璐 我一進(jìn)店門岖寞,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人柜蜈,你說我怎么就攤上這事仗谆。” “怎么了淑履?”我有些...
    開封第一講書人閱讀 165,138評論 0 355
  • 文/不壞的土叔 我叫張陵隶垮,是天一觀的道長。 經(jīng)常有香客問我秘噪,道長狸吞,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 58,791評論 1 295
  • 正文 為了忘掉前任指煎,我火速辦了婚禮蹋偏,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘至壤。我一直安慰自己威始,他們只是感情好,可當(dāng)我...
    茶點故事閱讀 67,794評論 6 392
  • 文/花漫 我一把揭開白布像街。 她就那樣靜靜地躺著黎棠,像睡著了一般。 火紅的嫁衣襯著肌膚如雪镰绎。 梳的紋絲不亂的頭發(fā)上脓斩,一...
    開封第一講書人閱讀 51,631評論 1 305
  • 那天,我揣著相機與錄音畴栖,去河邊找鬼随静。 笑死,一個胖子當(dāng)著我的面吹牛吗讶,可吹牛的內(nèi)容都是我干的燎猛。 我是一名探鬼主播叼丑,決...
    沈念sama閱讀 40,362評論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼扛门!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起纵寝,我...
    開封第一講書人閱讀 39,264評論 0 276
  • 序言:老撾萬榮一對情侶失蹤论寨,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后爽茴,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體葬凳,經(jīng)...
    沈念sama閱讀 45,724評論 1 315
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 37,900評論 3 336
  • 正文 我和宋清朗相戀三年室奏,在試婚紗的時候發(fā)現(xiàn)自己被綠了火焰。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 40,040評論 1 350
  • 序言:一個原本活蹦亂跳的男人離奇死亡胧沫,死狀恐怖昌简,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情绒怨,我是刑警寧澤纯赎,帶...
    沈念sama閱讀 35,742評論 5 346
  • 正文 年R本政府宣布,位于F島的核電站南蹂,受9級特大地震影響犬金,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜六剥,卻給世界環(huán)境...
    茶點故事閱讀 41,364評論 3 330
  • 文/蒙蒙 一晚顷、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧疗疟,春花似錦该默、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,944評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至锅锨,卻和暖如春叽赊,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背必搞。 一陣腳步聲響...
    開封第一講書人閱讀 33,060評論 1 270
  • 我被黑心中介騙來泰國打工必指, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人恕洲。 一個月前我還...
    沈念sama閱讀 48,247評論 3 371
  • 正文 我出身青樓塔橡,卻偏偏與公主長得像梅割,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子葛家,可洞房花燭夜當(dāng)晚...
    茶點故事閱讀 44,979評論 2 355

推薦閱讀更多精彩內(nèi)容

  • 今天看到一條新聞癞谒,一個蕭山姑娘跟老板談工資底燎,結(jié)果老板甩出一大段雞湯文跟她談理想。 看到這些弹砚,我都想笑双仍,不知道這位老...
    青鸞微光閱讀 14,369評論 0 0
  • 相比于人類的丑陋不堪,大自然的美可歌可贊桌吃!天天晨練的大廟山朱沃,雪后的一草一木都透露出蓬勃生機!
    青島寒江雪閱讀 118評論 0 1
  • 子墨8點40左右在媽媽的陪同下入園茅诱,入園情緒平穩(wěn)逗物,進(jìn)園就朝教室內(nèi)走去,媽媽引導(dǎo)子墨換鞋瑟俭,子墨不愿意換室內(nèi)鞋敬察,理由是...
    純?nèi)粌和?/span>閱讀 492評論 0 0