UIButton+Block分類代替addTarget實現快捷監聽按鈕點選
阿新 • • 發佈:2019-02-09
平常我們寫button的點選一般都是用addTarget
[_button addTarget:self action:@selector(didTapButton:) forControlEvents:UIControlEventTouchUpInside];
這種寫法還需要單獨去寫點選執行的方法;
以下程式碼是把addTarget寫入分類,用block傳值,這樣當監聽點選時直接在block傳入點選事件即可
UIButton+Block.h
#import <UIKit/UIKit.h>
typedef void(^ButtonBlock)(UIButton* btn);
@interface UIButton (Block)
- (void)addAction:(ButtonBlock)block;
- (void)addAction:(ButtonBlock)block forControlEvents:(UIControlEvents)controlEvents;
@end
UIButton+Block.m
#import "UIButton+Block.h"
#import <objc/runtime.h>
@implementation UIButton (Block)
static char ActionTag;
- (void)addAction:(ButtonBlock)block
{
objc_setAssociatedObject(self , &ActionTag, block, OBJC_ASSOCIATION_COPY_NONATOMIC);
[self addTarget:self action:@selector(action:) forControlEvents:UIControlEventTouchUpInside];
}
- (void)addAction:(ButtonBlock)block forControlEvents:(UIControlEvents)controlEvents
{
objc_setAssociatedObject(self, &ActionTag, block, OBJC_ASSOCIATION_COPY_NONATOMIC);
[self addTarget:self action:@selector(action:) forControlEvents:controlEvents];
}
- (void)action:(id)sender
{
ButtonBlock blockAction = (ButtonBlock)objc_getAssociatedObject(self, &ActionTag);
if (blockAction)
{
blockAction(self);
}
}
@end
呼叫示例:
__weak typeof (self)weakSelf = self;
[footer.button addAction:^(UIButton *btn) {
WTAddressControlViewController *vc = [[WTAddressControlViewController alloc] initWithViewControllerType:WTAddressControlTypeAdd];
vc.popBlock = ^(){
[weakSelf loadData];
};
[weakSelf.navigationController pushViewController:vc animated:YES];
}];
需要注意迴圈引用