iOS “获取验证码”按钮的倒计时功能

iOS 的倒计时有多种实现细节，Cocoa Touch 为我们提供了 NSTimer 类和 GCD 的dispatch_source_set_timer方法去更加方便的使用计时器。我们也可以很容易的的各种 UI 控件上添加倒计时功能，你只需

iOS 的倒计时有多种实现细节，Cocoa Touch 为我们提供了 NSTimer 类和 GCD 的dispatch_source_set_timer 方法去更加方便的使用计时器。我们也可以很容易的的各种 UI 控件上添加倒计时功能，你只需要定时刷新一次界面，给控件文本属性重新赋新值即可，但在实际项目中，可能并没有你想的这么简单美好。

我们不妨设想一下这样的情景：

主界面上有一个注册按钮，你点击按钮 push 到下一级页面，这个页面让你输入手机号并有一个获取验证码的按钮。你填完号码，再点击“获取验证码”按钮，然后按钮上的文字开始了 60 秒的倒计时。20 秒之后你 pop 回上一级页面，那么现在的页面应该被销毁了，10 秒后再次 push 到这个注册页面，那么倒计时按钮上的文字应该是【获取验证码】还是 【30 秒后重试】？

合理交互应该是：按钮正在从 30 秒开始，继续倒计时，直到秒数为 0 然后按钮文字再次变为【获取验证码】，并且倒计时过程中，按钮不可点击。否则倒计时限制将不再有意义。

有 2 个细节需要注意：

• 界面销毁以后，计时器还在继续计时
• 重新创建界面后，如果该按钮的计时器存在，则应当立即继续倒计时。

我们要做的就是 获取该按钮对应对的计时器 。

有人会说，何必如此麻烦，直接将这个页面或者这个按钮写成单例不就得了？是的，单例可以轻松解决这个问题，但是这种设计模式切不可滥用，假如你的 App 有20 个页面需要获取验证码按钮，那岂不是得生成 20 个单例的 View controller ？要知道，你并不是经常需要这些页面。如果把按钮设计成单例，那更不可取，一但你修改了一个按钮，其他地方的按钮必受牵连，引发不可估计的后果。

我的一个方案是， 生成一个全局的计时器管理类，它负责为每一个需要倒计时功能的按钮分配一个计时器，按钮和计时器通过一个 key 相互绑定。按钮被 dealloc 之后，计时器任然存在，直至计时结束。按钮重新生成时，计时器管理类会根据 key 决定该按钮是否需要继续倒计时 。

Do it!

首先需要思考，这个计时器管理类应该是是什么样子？它的具体功能又是什么？我给它命名为 WLButtonCountdownManager ，它是一个全局类，可用单例设计（1 个单例类比 20 个单例页面划算得多）。它负责分配计时器并将其与按钮绑定，所以它需要有一个容器属性来存储计时器，并且还要知道，容器里是否已经有计时器在跑了。那么 WLButtonCountdownManager 的头文件大概类似于这样：

NS_ASSUME_NONNULL_BEGIN
@interface WLButtonCountdownManager : NSObject

/**
 *  获取单例
 *
 *  @return 该类的唯一实例
 */
+ (instancetype)defaultManager;

/**
 *  开始倒计时，如果倒计时管理器里具有相同的key，则直接开始回调。
 *
 *  @param aKey         任务key，用于标示唯一性
 *  @param timeInterval 倒计时总时间，受操作系统后台时间限制，倒计时时间规定不得大于 120 秒.
 *  @param countingDown 倒计时时，会多次回调，提供当前秒数
 *  @param finished     倒计时结束时调用，提供当前秒数，值恒为 0
 */
- (void)scheduledCountDownWithKey:(NSString *)aKey
                     timeInterval:(NSTimeInterval)timeInterval
                     countingDown:(nullable void (^)(NSTimeInterval leftTimeInterval))countingDown
                         finished:(nullable void (^)(__unused NSTimeInterval finalTimeInterval))finished;

/**
 *  查询倒计时任务是否存在
 *
 *  @param akey 任务key
 *  @param task 任务
 *  @return YES - 存在， NO - 不存在
 */
- (BOOL)coundownTaskExistWithKey:(NSString *)akey task:(NSOperation * _Nullable * _Nullable)task;


@end
NS_ASSUME_NONNULL_END

[*注]这里不得不提一下，对于单例的写法，各位看官仁者见仁智者见智。一个严谨的单例至少需要满足以下条件：

1. 全局唯一性 
   ①不可通过 alloc 再次分配资源 
   ②线程安全
2. 不可继承

很多人忽略了第一条的第一点和第二条。

About timer

关于计时器，我使用子线程每秒睡眠一次进行模拟，计时操作精度要求并不高，且线程池也利于管理。

第一步，子类化一个 NSOperation 的类，名为 WLCountdownTask :

@interface WLCountdownTask : NSOperation
/**
 *  计时中回调
 */
@property (copy, nonatomic) void (^countingDownBlcok)(NSTimeInterval timeInterval);
/**
 *  计时结束后回调
 */
@property (copy, nonatomic) void (^finishedBlcok)(NSTimeInterval timeInterval);
/**
 *  计时剩余时间
 */
@property (assign, nonatomic) NSTimeInterval leftTimeInterval;
/**
 *  后台任务标识，确保程序进入后台依然能够计时
 */
@property (assign, nonatomic) UIBackgroundTaskIdentifier taskIdentifier;
@end


- (void)main {
    self.taskIdentifier = [[UIApplication sharedApplication] beginBackgroundTaskWithExpirationHandler:nil];

    while (--_leftTimeInterval > 0) {
        dispatch_async(dispatch_get_main_queue(), ^{
            if (_countingDownBlcok) _countingDownBlcok(_leftTimeInterval);
        });

        [NSThread sleepForTimeInterval:1];
    }

    dispatch_async(dispatch_get_main_queue(), ^{
        if (_finishedBlcok) {
            _finishedBlcok(0);
        }
    });

    if (self.taskIdentifier != UIBackgroundTaskInvalid) {
        [[UIApplication sharedApplication] endBackgroundTask:self.taskIdentifier];
        self.taskIdentifier = UIBackgroundTaskInvalid;
    }
}

@end

下一步，WLButtonCountdownManager 拥有一个线程池（也叫并发操作队列，规定队列中最多只允许存在 20 个并发线程），每分配一个计时器（即创建一个子线程）就将其放入池子中，计时器跑完以后会自动从池子里销毁。

在创建计时任务之前，Manager 从池子里检索是否有相同 key 的计时任务，如果任务存在，直接回调计时操作。否则，新建一个标识为 key 的任务。

- (void)scheduledCountDownWithKey:(NSString *)aKey
                     timeInterval:(NSTimeInterval)timeInterval
                     countingDown:(void (^)(NSTimeInterval))countingDown
                         finished:(void (^)(NSTimeInterval))finished
{
    if (timeInterval > 120) {
        NSCAssert(NO, @"受操作系统后台时间限制，倒计时时间规定不得大于 120 秒.");
    }

    if (_pool.operations.count >= 20)  // 最多 20 个并发线程
        return;

    WLCountdownTask *task = nil;
    if ([self coundownTaskExistWithKey:aKey task:&task]) {
        task.countingDownBlcok = countingDown;
        task.finishedBlcok     = finished;
        if (countingDown) {
            countingDown(task.leftTimeInterval);
        }
    } else {
        task                   = [[WLCountdownTask alloc] init];
        task.name              = aKey;
        task.leftTimeInterval  = timeInterval;
        task.countingDownBlcok = countingDown;
        task.finishedBlcok     = finished;
        [_pool addOperation:task];
    }
}


- (BOOL)coundownTaskExistWithKey:(NSString *)akey
                            task:(NSOperation *__autoreleasing  _Nullable *)task
{
    __block BOOL taskExist = NO;
    [_pool.operations enumerateObjectsUsingBlock:^(__kindof NSOperation * _Nonnull obj, NSUInteger idx, BOOL * _Nonnull stop) {
        if ([obj.name isEqualToString:akey]) {
            if (task) *task = obj;
            taskExist = YES;
            *stop     = YES;
        }
    }];

    return taskExist;
}

Last

至此，解决方案到此就结束了