iOS 获取验证码倒计时方法

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/walkerwqp/article/details/82592708

#import <UIKit/UIKit.h>

@interface UIButton (YasinTimerCategory)

- (void)startCountDownTime:(int)time withCountDownBlock:(void(^)(void))countDownBlock;

@end

#import "UIButton+YasinTimerCategory.h"

static NSString *yasinTempText;

@implementation UIButton (YasinTimerCategory)

- (void)startCountDownTime:(int)time withCountDownBlock:(void(^)(void))countDownBlock{
    
    [self initButtonData];
    
    [self startTime:time];
    
    if (countDownBlock) {
        countDownBlock();
    }
}

- (void)initButtonData{
    
    yasinTempText = [NSString stringWithFormat:@"%@",self.titleLabel.text];
    
}

- (void)startTime:(int)time{
    
    __block int timeout = time;
    
    dispatch_queue_t queue = dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_DEFAULT, 0);
    
    dispatch_source_t _timer = dispatch_source_create(DISPATCH_SOURCE_TYPE_TIMER, 0, 0,queue);
    
    dispatch_source_set_timer(_timer,dispatch_walltime(NULL, 0),1.0 * NSEC_PER_SEC, 0);
    
    dispatch_source_set_event_handler(_timer, ^{
        
        //倒计时结束
        if(timeout <= 0){
            
            dispatch_source_cancel(_timer);
            
            dispatch_async(dispatch_get_main_queue(), ^{
                
                [self setTitle:yasinTempText forState:UIControlStateNormal];
                self.userInteractionEnabled = YES;
                
            });
            
        }else{
            
            dispatch_async(dispatch_get_main_queue(), ^{
                
                NSString *text = [NSString stringWithFormat:@"%02d秒重新获取",timeout];
                [self setTitle:text forState:UIControlStateNormal];
                self.userInteractionEnabled = NO;
                
            });
            
            timeout --;
            
        }
        
    });
    
    dispatch_resume(_timer);
    
}

@end

调用方法

[self.verificationBtn startCountDownTime:60 withCountDownBlock:^{
            
            NSLog(@"开始倒计时");
            
            //获取验证码方法  .倒计时60s
            
        }];