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];
}
}