block自定义UIButton

ViewController

- (void)viewDidLoad
{
    [super viewDidLoad];
	
    MyButton *button = [MyButton buttonWithType:UIButtonTypeSystem];
    button.frame = CGRectMake(20, 50, 280, 50);
    button.backgroundColor = [UIColor redColor];
    [self.view addSubview:button];
    
    //实现button的block回调
    [button setButtonShouldBlock:^(MyButton *sender) {
        NSLog(@"我是Block: should方法\n");
    }];
    
    [button setButtonWillBlock:^(MyButton *sender) {
        NSLog(@"我是Block: Will方法\n");
    }];
    
    [button setButtonDidBlock:^(MyButton *sender) {
        NSLog(@"我是Blcok: Did方法\n");
    }];
    
}

@interface MyButton : UIButton

typedef void (^ButtonWillBlock)(MyButton *sender);
typedef void (^ButtonDidBlock)(MyButton *sender);
typedef void (^ButtonShouldBlock)(MyButton *sender);

@property (nonatomic, copy) ButtonWillBlock willBlock;
@property (nonatomic, copy) ButtonDidBlock didBlock;
@property (nonatomic, copy) ButtonShouldBlock shouldBlock;

//接受block的方法
-(void)setButtonShouldBlock: (ButtonShouldBlock) block;
-(void)setButtonWillBlock: (ButtonWillBlock) block;
-(void)setButtonDidBlock:(ButtonDidBlock) block;

//接受block块
-(void)setButtonShouldBlock: (ButtonShouldBlock) block
{
    self.shouldBlock = block;
    [self addTarget:self action:@selector(buttonClick) forControlEvents:UIControlEventTouchUpInside];
}

-(void)setButtonWillBlock: (ButtonWillBlock) block
{
    self.willBlock = block;
}

-(void)setButtonDidBlock:(ButtonDidBlock) block
{
    self.didBlock = block;
}

- (void)buttonClick
{
    if (_shouldBlock)
    {
        _shouldBlock(self);
        NSLog(@"调用了");
        self.backgroundColor = [UIColor grayColor];
    }
}