runtime简单封装获取短信验证码Button

获取短信验证码-> 倒计时 -> 未收到重新获取，这是个在有手机登陆的APP中必有的小功能。最常用的做法就是使用UIButton加NSTimer计时器，要是没有对button子类化封装的话，在控制器中需要创建button，并使用计时器控制button状态，还需要管理倒计时，无疑这使代码看起来更复杂且增加了代码量。
而封装这个小功能的话有两种方法：

可以子类化button，给button的子类添加属性，并在button子类中定义计时器控制button状态。

或者新建button的类目，使用runtime给button的类目添加属性，并定义计时器。

这篇介绍使用runtime封装一个获取短信验证码的button。

1、首先新建一个UIButton的类目，然后再.h中定义需要的属性和方法。

@interface UIButton (Timer)
@property (nonatomic, copy) NSString *time;
@property (nonatomic, retain) UIColor *color;
- (void)startTime;
@end

我们知道在类目中定义的方法，外部调用的时候会报错的。这时候就要用到runtime了。

2、在.m中实现定义属性的get set 方法。

- (NSString *)time {
    return objc_getAssociatedObject(self, @selector(time));
}
- (void)setTime:(NSString *)time {
    objc_setAssociatedObject(self, @selector(time), time, OBJC_ASSOCIATION_COPY_NONATOMIC);
}

- (UIColor *)color {
    return objc_getAssociatedObject(self, @selector(color));
}
- (void)setColor:(UIColor *)color {
    objc_setAssociatedObject(self, @selector(color), color, OBJC_ASSOCIATION_COPY_NONATOMIC);
    self.backgroundColor = color;
}

记得导入头文件#import

3、实现开始计时方法
方法中使用计时器控制按钮状态就行了，逻辑比较简单。

- (void)startTime {
    _second = self.time.integerValue;
    
    [self setTitle:[NSString stringWithFormat:@"倒计时%ld s", (long)_second] forState:UIControlStateNormal];
    
    _timer = [NSTimer scheduledTimerWithTimeInterval:1 target:self selector:@selector(timerMethod) userInfo:nil repeats:YES];
    [[NSRunLoop mainRunLoop] addTimer:_timer forMode:NSDefaultRunLoopMode];
    self.backgroundColor = [UIColor grayColor];
    self.enabled = NO;
    
}

- (void)timerMethod {
    if(_second <= 0){
        [_timer invalidate];
        _timer = nil;
        [self setTitle:@"重新获取验证码" forState:UIControlStateNormal];
        self.enabled = YES;
        self.backgroundColor = self.color;
    }else {
        _second --;
        [self setTitle:[NSString stringWithFormat:@"倒计时%ld s", (long)_second] forState:UIControlStateNormal];
    }
    
}

4、最后如何使用
导入类目，按平常创建UIbutton的方法创建button，但是注意设置背景颜色的时候是设置我们添加的属性.color 而不是.backgroundColor。然后再需要开始倒计时的时候调用 startTime就行了。

    button = [UIButton buttonWithType:UIButtonTypeCustom];
    button.frame = CGRectMake(40, 100, KScreenWidth - 80, 45);
    [button setTitle:@"点击获取验证码" forState:UIControlStateNormal];
    [button setTitleColor:[UIColor whiteColor] forState:UIControlStateNormal];
    button.color = [UIColor colorWithRed:40.2f/255 green:180.2f/255 blue:247.2f/255 alpha:1];
    [button addTarget:self action:@selector(clickAction) forControlEvents:UIControlEventTouchUpInside];
    [self.view addSubview:button];
    
    button.time = @"3";

---

- (void)clickAction {
    
    [button startTime];
}

需要代码戳这里