有時(shí)候，我們需要在button點(diǎn)擊后，取出點(diǎn)擊方法執(zhí)行完后的結(jié)果，然而原生的點(diǎn)擊方法沒有提供回調(diào)，那么就需要我們自己進(jìn)行封裝了。下面分兩種情況進(jìn)行了封裝，一種是繼承，一種是分類。具體看Demo。

Jietu20180206-101510.jpg

繼承

#import <UIKit/UIKit.h>
@interface BlockButton : UIButton
@property(nonatomic,copy) void(^block)(UIButton*);
-(void)addTapBlock:(void(^)(UIButton *btn))block;
@end


#import "BlockButton.h"
@implementation BlockButton
-(void)addTapBlock:(void(^)(UIButton*))block{
    self.block= block;
    [self addTarget:self action:@selector(click:) forControlEvents:UIControlEventTouchUpInside];
}
-(void)click:(UIButton*)btn{
    if(self.block) {
        self.block(btn);
    }
}
@end

調(diào)用

#import "BlockButton.h"http://引入繼承
BlockButton *button0 = [[BlockButton alloc] initWithFrame:CGRectMake(100, 100, 100, 100)];
button0.backgroundColor = [UIColor redColor];
[button0 setTitle:@"測(cè)試1" forState:0];
[self.view addSubview:button0];
[button0 addTapBlock:^(UIButton *btn) {
     //在這里調(diào)用所需要執(zhí)行的方法
     [self buttonClick];
}];

-(void)buttonClick{
    NSLog(@"點(diǎn)擊方法");
}

分類

#import <UIKit/UIKit.h>
@interface UIButton (Block)
@property(nonatomic ,copy)void(^block)(UIButton*);
-(void)addTapBlock:(void(^)(UIButton*btn))block;
@end


#import "UIButton+Block.h"
#import<objc/runtime.h>
@implementation UIButton (Block)
-(void)setBlock:(void(^)(UIButton*))block
{
    objc_setAssociatedObject(self,@selector(block), block,OBJC_ASSOCIATION_COPY_NONATOMIC);
    [self addTarget:self action:@selector(click:)forControlEvents:UIControlEventTouchUpInside];
}
-(void(^)(UIButton*))block
{
    return objc_getAssociatedObject(self,@selector(block));
}
-(void)addTapBlock:(void(^)(UIButton*))block
{
    self.block= block;
    [self addTarget:self action:@selector(click:)forControlEvents:UIControlEventTouchUpInside];
}

-(void)click:(UIButton*)btn
{
    if(self.block) {
        self.block(btn);
    }
}
@end

調(diào)用

#import "UIButton+Block.h"http://引入分類
UIButton *button1 = [[UIButton alloc] initWithFrame:CGRectMake(100, 300, 100, 100)];
button1.backgroundColor = [UIColor blueColor];
button1.tag = 1001;
[button1 setTitle:@"測(cè)試2" forState:0];
[self.view addSubview:button1];
    
__block UIButton *blockSelf = button1;
[button1 addTapBlock:^(UIButton *btn) {
     //在這里調(diào)用所需要執(zhí)行的方法
     [self button:blockSelf callBack:^(NSInteger selectIndex, NSString *title) {
         NSLog(@"tag:%ld    title:%@",(long)selectIndex,title);
     }];
}];

-(void)button:(UIButton *)button callBack:(void (^)(NSInteger selectIndex ,NSString *title))callBack{
    callBack(button.tag,button.currentTitle);
}

這兩種方法都可以實(shí)現(xiàn)傳參和回調(diào)，結(jié)合具體情況使用吧。