UIButton+Block分类代替addTarget实现快捷监听按钮点击

最新推荐文章于 2022-02-01 16:20:37 发布

手捧太极震寰宇

最新推荐文章于 2022-02-01 16:20:37 发布

阅读量1.9k

点赞数

分类专栏： iOS开发实用技巧文章标签： uibutton 分类 addTarget 点击事件 block

本文链接：https://blog.csdn.net/poplar_b/article/details/54963166

版权

iOS开发同时被 2 个专栏收录

18 篇文章 0 订阅

订阅专栏

实用技巧

1 篇文章 0 订阅

订阅专栏

平常我们写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];
    }];

需要注意循环引用