要實現(xiàn)的效果如下:
btn.addTarget(for: .touchUpInside) { (sender) in
print("button click懒叛!")
}
很多時候我們的的button點擊事件就一行代碼,而這一行代碼還需要另寫一個方法耽梅。然后用selector給button薛窥。這樣:
當(dāng)代碼量增大,按鈕的點擊事件和按鈕的定義就分散在代碼的不同區(qū)域眼姐,給后期的代碼維護造成不便诅迷。為了解決這個問題,我給UIButton擴展了一個用尾隨閉包給按鈕添加事件的方法众旗。然后代碼變成這樣:
顯然要代碼集中了罢杉,也好好看了!
實現(xiàn):
想法:給UIButton擴展一個方法帶閉包參數(shù)的方法贡歧,在這個方法中去調(diào)用系統(tǒng)的addTraget方法滩租,并把閉包保存起來,在addTraget中Selector對應(yīng)的方法去執(zhí)行閉包艘款。
實現(xiàn)步驟:
1持际、在extension中給UIButton擴展一個帶尾隨閉包添加事件的方法:
func addTarget(for controlEvents: UIControlEvents,action:@escaping (UIButton)->())
{
}
在該方法中,要做兩件事:
一哗咆、是給按鈕添加響應(yīng)事件蜘欲,這個沒啥可說的,直接調(diào)用系統(tǒng)API晌柬。
二姥份、是把閉包保存起來郭脂,以便于在響應(yīng)事件方法中去執(zhí)行閉包。怎么保存呢澈歉,只能通過一個屬性保存展鸡。extension擴展屬性,用到runtime的關(guān)聯(lián)屬性方法埃难。直接寫一個閉包屬性去保存事件閉包:
寫是沒問題莹弊,但一運行就會錯誤,通過實驗涡尘,發(fā)現(xiàn)關(guān)聯(lián)對象不支持基本數(shù)據(jù)類型忍弛,和閉包類型(具體原因我也沒弄清楚,等搞清楚再修改過此處)考抄。
鑒于這種情況细疚,定義了一個結(jié)構(gòu)體,在結(jié)構(gòu)體中定義了一個閉包屬性川梅。關(guān)聯(lián)一個該結(jié)構(gòu)體類型的對象疯兼。問題便迎刃而解。
下面是全部擴展代碼:
extension UIButton{
struct AssociatedClosureClass {
var eventClosure: (UIButton)->()
}
private struct AssociatedKeys {
static var eventClosureObj:AssociatedClosureClass?
}
private var eventClosureObj: AssociatedClosureClass{
set{
objc_setAssociatedObject(self, &AssociatedKeys.eventClosureObj, newValue, .OBJC_ASSOCIATION_RETAIN_NONATOMIC)
}
get{
return (objc_getAssociatedObject(self, &AssociatedKeys.eventClosureObj) as? AssociatedClosureClass)!
}
}
func addTarget(for controlEvents: UIControlEvents,action:@escaping (UIButton)->()) {
let eventObj = AssociatedClosureClass(eventClosure: action)
eventClosureObj = eventObj
addTarget(self, action: #selector(eventExcuate(_:)), for: controlEvents)
}
@objc private func eventExcuate(_ sender: UIButton){
eventClosureObj.eventClosure(sender)
}
}
同理贫途,我們可以寫監(jiān)聽通知吧彪,timer執(zhí)行,segement切換的方法擴展潮饱。
如果你有更好的想法来氧,或者能解答文中問題,希望看到你的留言香拉。