实现按钮点击倒计时的三种方法

在登陆页面经常会需要实现一种功能，点击“获取验证码“按钮后，按钮被禁用，并且开始显示倒计时。

Demo地址 : https://github.com/TWLemontree/ThreeCountdown

1. 通过 NSThread 的 performSelectorInBackground、performSelectorOnMainThread、sleep 来实现

#import "CountdownOneViewController.h"

@interface CountdownOneViewController ()

/** 显示倒计时效果的按钮 */
@property (nonatomic, strong) UIButton *btn;

@end

@implementation CountdownOneViewController {
    int _count; // 倒计时数字
}

#pragma mark - 生命周期
- (void)viewDidLoad {
    [super viewDidLoad];

    [self.view addSubview:self.btn];
    self.view.backgroundColor = [UIColor whiteColor];
}

#pragma mark - 私有方法
- (void)btnAction
{
    [self performSelectorInBackground:@selector(thread) withObject:nil];
}

- (void)thread
{
    for (int i = 59; i >= 0; i--) {
        _count = i;
        [self performSelectorOnMainThread:@selector(mainThread) withObject:nil waitUntilDone:YES];
        sleep(1);
    }
}

- (void)mainThread
{
    if (_count > 0) {
        [self.btn setUserInteractionEnabled:NO];
        [self.btn setTitle:[NSString stringWithFormat:@"(%d)重发验证码",_count] forState:UIControlStateNormal];
    } else {
        [self.btn setUserInteractionEnabled:YES];
        [self.btn setTitle:[NSString stringWithFormat:@"重发验证码"] forState:UIControlStateNormal];
    }
}

#pragma mark - 懒加载
- (UIButton *)btn
{
    if (!_btn) {
        _btn = [UIButton buttonWithType:UIButtonTypeSystem];
        _btn.frame = CGRectMake(10, 200, 300, 50);
        [_btn setTitle:@"获取验证码" forState:UIControlStateNormal];
        [_btn addTarget:self action:@selector(btnAction) forControlEvents:UIControlEventTouchUpInside];
    }
    return _btn;
}

@end

2. 通过 NSTimer 来实现

#import "CountdownTwoViewController.h"

@interface CountdownTwoViewController ()

/** 显示倒计时效果的按钮 */
@property (nonatomic, strong) UIButton *btn;

@end

@implementation CountdownTwoViewController {
    int _count; // 倒计时数字
    NSTimer *_timer; // 定时器
}

#pragma mark - 生命周期
- (void)viewDidLoad {
    [super viewDidLoad];

    [self.view addSubview:self.btn];
    self.view.backgroundColor = [UIColor whiteColor];
}

#pragma mark - 私有方法
- (void)btnAction
{
    _count = 59;
    [self handleTimer];
    _timer = [NSTimer scheduledTimerWithTimeInterval:(1.0) target:self selector:@selector(handleTimer) userInfo:nil repeats:YES];
}

- (void)handleTimer
{
    if (_count > 0) {
        [self.btn setUserInteractionEnabled:NO];
        [self.btn setTitle:[NSString stringWithFormat:@"(%d)重发验证码",_count] forState:UIControlStateNormal];
    } else {
        [self.btn setUserInteractionEnabled:YES];
        [self.btn setTitle:[NSString stringWithFormat:@"重发验证码"] forState:UIControlStateNormal];

        [_timer invalidate];
    }
    _count -= 1;
}

#pragma mark - 懒加载
- (UIButton *)btn
{
    if (!_btn) {
        _btn = [UIButton buttonWithType:UIButtonTypeSystem];
        _btn.frame = CGRectMake(10, 200, 300, 50);
        [_btn setTitle:@"获取验证码" forState:UIControlStateNormal];
        [_btn addTarget:self action:@selector(btnAction) forControlEvents:UIControlEventTouchUpInside];
    }
    return _btn;
}

@end

3. 通过 GCD 中的 dispatch_source 来实现

#import "CountdownThreeViewController.h"

@interface CountdownThreeViewController ()

/** 显示倒计时效果的按钮 */
@property (nonatomic, strong) UIButton *btn;

@end

@implementation CountdownThreeViewController

#pragma mark - 生命周期
- (void)viewDidLoad {
    [super viewDidLoad];

    [self.view addSubview:self.btn];
    self.view.backgroundColor = [UIColor whiteColor];
}

#pragma mark - 私有方法
- (void)btnAction
{
    __block NSInteger second = 60;
    //全局队列    默认优先级
    dispatch_queue_t quene = dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_DEFAULT, 0);
    //定时器模式  事件源
    dispatch_source_t timer = dispatch_source_create(DISPATCH_SOURCE_TYPE_TIMER, 0, 0, quene);
    //NSEC_PER_SEC是秒，＊1是每秒
    dispatch_source_set_timer(timer, dispatch_walltime(NULL, 0), NSEC_PER_SEC * 1, 0);
    //设置响应dispatch源事件的block，在dispatch源指定的队列上运行
    dispatch_source_set_event_handler(timer, ^{
        //回调主线程，在主线程中操作UI
        dispatch_async(dispatch_get_main_queue(), ^{
            if (second >= 0) {
                [self.btn setTitle:[NSString stringWithFormat:@"(%ld)重发验证码",second] forState:UIControlStateNormal];
                second--;
            }
            else
            {
                //这句话必须写否则会出问题
                dispatch_source_cancel(timer);
                [self.btn setTitle:@"获取验证码" forState:UIControlStateNormal];

            }
        });
    });
    //启动源
    dispatch_resume(timer);
}



#pragma mark - 懒加载
- (UIButton *)btn
{
    if (!_btn) {
        _btn = [UIButton buttonWithType:UIButtonTypeSystem];
        _btn.frame = CGRectMake(10, 200, 300, 50);
        [_btn setTitle:@"获取验证码" forState:UIControlStateNormal];
        [_btn addTarget:self action:@selector(btnAction) forControlEvents:UIControlEventTouchUpInside];
    }
    return _btn;
}

@end

参考文章

1. iOS开发－三种倒计时的写法