一般給按鈕添加事件是這樣的組合:
// 把點(diǎn)擊后要調(diào)用的方法名傳到@selector中
self.btn addTarget:self action:@selector(btnClick) forControlEvents:UIControlEventTouchUpInside];
// 找別的地方寫(xiě)上方法的實(shí)現(xiàn)
- (void)btnClick {
}
這樣寫(xiě)代碼多不說(shuō),工程大的話還得到處找方法。
所以我寫(xiě)了一個(gè)分類,可以用block來(lái)代替@selector(),防止代碼分散。
使用起來(lái)是這樣的:
// button
[self.btn addTarget:self action:[self selectorBlock:^(id arg) {
NSLog(@"clicked %@", arg);
}] forControlEvents:UIControlEventTouchUpInside];
下面講一下分類NSObject+BlockSEL.h中的實(shí)現(xiàn)原理
- (SEL)selectorBlock:(void (^)(id))block {
NSString *selName = [NSString stringWithFormat:@"selector_%p:", block];
SEL sel = NSSelectorFromString(selName);
class_addMethod([self class], sel, (IMP)selectorImp, "v@:@");
objc_setAssociatedObject(self, sel, block, OBJC_ASSOCIATION_COPY_NONATOMIC);
return sel;
}
static void selectorImp(id self, SEL _cmd, id arg) {
void (^block)(id arg) = objc_getAssociatedObject(self, _cmd);
if (block) block(arg);
}
用block的內(nèi)存地址創(chuàng)建一個(gè)唯一方法,并將方法實(shí)現(xiàn)綁定到selectorImp函數(shù)。將block用關(guān)聯(lián)對(duì)象保存下來(lái),在調(diào)用時(shí)執(zhí)行block。
需要注意一下,當(dāng)block中使用self時(shí)會(huì)產(chǎn)生循環(huán)引用,需要用__weak來(lái)修飾防止self無(wú)法釋放。
__weak typeof(self) weakSelf = self;
[self.btn addTarget:self action:[self selectorBlock:^(id arg) {
weakSelf.view.backgroundColor = [UIColor whiteColor];
}] forControlEvents:UIControlEventTouchUpInside];