block自定义UIButton

旧城等待， 2023-10-17 06:25 20阅读 0赞

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