一生只爱一个：iOS倒计时按钮

写在前面

封装，是因为下次懒得再写！一生只爱一个。

分析

倒计时，肯定是要用到计时器了。我们会想到这个NSTimer和dispatch_source_set_timer。二者的区别点这里。

关于定时器

1、首先声明一个本地对象：

dispatch_source_t verifyTimer;//验证码定时器

2、创建一个队列，用来装载这个定时器

dispatch_queue_t queue = dispatch_queue_create(DISPATCH_QUEUE_PRIORITY_DEFAULT, 0);

3、创建这个定时器对象

verifyTimer = dispatch_source_create(DISPATCH_SOURCE_TYPE_TIMER, 0, 0, queue);

4、设置定时器从现在起执行，执行的频率为每秒执行一次。

dispatch_source_set_timer(verifyTimer, dispatch_walltime(NULL, 0), 1.0*NSEC_PER_SEC, 0);//每秒执行

5、监听每次执行

__weak typeof(self)weakSelf = self;
dispatch_source_set_event_handler(verifyTimer, ^{
        if (timeout <=  0) {
            
        //倒计时结束
        dispatch_async(dispatch_get_main_queue(), ^{
             //执行一些UI上的代码

            //并释放定时器
            [weakSelf stopCountDown];
        });
}else{
       //倒计时中    
       dispatch_async(dispatch_get_main_queue(), ^{
               if (_countDownChanging) {
                    //设置按钮文字，外部返回title
                    [weakSelf setTitle:_countDownChanging(weakSelf,timeout) forState:UIControlStateNormal];
                    [weakSelf setTitle:_countDownChanging(weakSelf,timeout) forState:UIControlStateDisabled];
                }else{
                    //设置按钮文字，外部不放回，设置默认的
                    NSString *title = [NSString stringWithFormat:@"%zd秒",timeout];
                    [weakSelf setTitle:title forState:UIControlStateNormal];
                    [weakSelf setTitle:title forState:UIControlStateDisabled];
                }
                
       });
            //每次执行减1秒
            timeout--;
        }
});

6、开启定时器

dispatch_resume(verifyTimer);

7、结束定时器

dispatch_source_cancel(verifyTimer);

以上是关于整个定时器的创建、开启和结束。

全部代码

.h文件

#import <UIKit/UIKit.h>
@class WxxCountDownButton;

typedef void (^TouchedCountDownButtonHandler)(WxxCountDownButton  *countDownButton,NSInteger tag);
typedef NSString* (^CountDownChanging)(WxxCountDownButton  *countDownButton,NSUInteger second);
typedef NSString* (^CountDownFinished)(WxxCountDownButton  *countDownButton,NSUInteger second);

@interface WxxCountDownButton : UIButton

///带圆角边框的按钮
-(instancetype)initHasBorderWithNormalColor:(UIColor*)normalColor
                                changeColor:(UIColor*)changeColor;
///监听按钮事件
-(void)countButtonHandler:(TouchedCountDownButtonHandler)touchedCountDownButtonHandler;
///开始倒计时
- (void)startCountDownWithSecond:(NSUInteger)totalSecond;
///停止倒计时
- (void)stopCountDown;
///倒计时中
- (void)countDownChanging:(CountDownChanging)countDownChanging;
///倒计时结束(需要在切换页面的时候调用，以便释放定时对象)
- (void)countDownFinished:(CountDownFinished)countDownFinished;

@end

.m文件

#import "WxxCountDownButton.h"

@interface WxxCountDownButton(){
    NSInteger _second;
    NSUInteger _totalSecond;
    
    UIColor *_normalColor;
    UIColor *_changeColor;
    
    CountDownChanging _countDownChanging;
    CountDownFinished _countDownFinished;
    TouchedCountDownButtonHandler _touchedCountDownButtonHandler;
    
    dispatch_source_t verifyTimer;//定时器
    BOOL isHasBorder;//是否带边框
}

@end

@implementation WxxCountDownButton

-(instancetype)initHasBorderWithNormalColor:(UIColor*)normalColor
                                changeColor:(UIColor*)changeColor{
    self = [super init];
    if (self) {
        _normalColor = normalColor;
        _changeColor = changeColor;
        
        isHasBorder = YES;
        //边框
        self.layer.borderWidth = 1;
        self.layer.masksToBounds = YES;
        self.layer.cornerRadius = 4;
        self.layer.borderColor = normalColor.CGColor;
        //字体颜色
        [self setTitleColor:normalColor forState:UIControlStateNormal];
        [self setTitleColor:changeColor forState:UIControlStateDisabled];
    }
    return self;
}

-(void)countButtonHandler:(TouchedCountDownButtonHandler)touchedCountDownButtonHandler{
    _touchedCountDownButtonHandler = [touchedCountDownButtonHandler copy];
    [self addTarget:self action:@selector(touched:) forControlEvents:UIControlEventTouchUpInside];
}

- (void)touched:(WxxCountDownButton *)sender{
    if (_touchedCountDownButtonHandler) {
        _touchedCountDownButtonHandler(sender,sender.tag);
    }
}

#pragma -mark count down method
- (void)startCountDownWithSecond:(NSUInteger)totalSecond
{
    self.enabled = NO;
    _totalSecond = totalSecond;
    _second = totalSecond;
    
    __block NSInteger timeout = totalSecond;
    dispatch_queue_t queue = dispatch_queue_create(DISPATCH_QUEUE_PRIORITY_DEFAULT, 0);
    verifyTimer = dispatch_source_create(DISPATCH_SOURCE_TYPE_TIMER, 0, 0, queue);
    
    dispatch_source_set_timer(verifyTimer, dispatch_walltime(NULL, 0), 1.0*NSEC_PER_SEC, 0);//每秒执行
    __weak typeof(self)weakSelf = self;
    dispatch_source_set_event_handler(verifyTimer, ^{
        if (timeout <=  0) {
            
            //倒计时结束
            dispatch_async(dispatch_get_main_queue(), ^{
//                [weakSelf setBackgroundColor:_normalColor];
                if (isHasBorder) {
                    self.layer.borderColor = _normalColor.CGColor;
                }
                [weakSelf stopCountDown];
            });
        }else{
            //倒计时中
            
            dispatch_async(dispatch_get_main_queue(), ^{
//               [weakSelf  setBackgroundColor:_changeColor];
                if (isHasBorder) {
                    self.layer.borderColor = _changeColor.CGColor;
                }
                if (_countDownChanging) {
                    [weakSelf setTitle:_countDownChanging(weakSelf,timeout) forState:UIControlStateNormal];
                    [weakSelf setTitle:_countDownChanging(weakSelf,timeout) forState:UIControlStateDisabled];
                }else{
                    NSString *title = [NSString stringWithFormat:@"%zd秒",timeout];
                    [weakSelf setTitle:title forState:UIControlStateNormal];
                    [weakSelf setTitle:title forState:UIControlStateDisabled];
                }
            });
            timeout--;
        }
    });
    dispatch_resume(verifyTimer);
}


- (void)stopCountDown{
    //停止计时线程
    if (verifyTimer) {
        dispatch_source_cancel(verifyTimer);
        verifyTimer = nil;
    }
    //按钮可用
    self.enabled = YES;
    //设置正常颜色
    if (isHasBorder) {
        self.layer.borderColor = _normalColor.CGColor;
    }
    _second = _totalSecond;
    if (_countDownFinished) {
        [self setTitle:_countDownFinished(self,_totalSecond)forState:UIControlStateNormal];
        [self setTitle:_countDownFinished(self,_totalSecond)forState:UIControlStateDisabled];
    }else{
        [self setTitle:@"重新获取" forState:UIControlStateNormal];
        [self setTitle:@"重新获取" forState:UIControlStateDisabled];
    }
}

- (void)countDownChanging:(CountDownChanging)countDownChanging{
    _countDownChanging = [countDownChanging copy];
}
- (void)countDownFinished:(CountDownFinished)countDownFinished{
    _countDownFinished = [countDownFinished copy];
}

@end

具体的按钮样式，可以自己根据需要去做修改。本demo的按钮样式是：带圆角边框，文字颜色同边框，无背景颜色

具体的使用方式如下：

 button = [[WxxCountDownButton alloc]initHasBorderWithNormalColor:[UIColor blueColor] changeColor:[UIColor lightGrayColor]];
[self.view addSubview:button];
button.frame = CGRectMake(120, 150, 80, 30);
button.titleLabel.font = [UIFont systemFontOfSize:12];
[button setTitle:@"发送验证码" forState:UIControlStateNormal];

[button countButtonHandler:^(WxxCountDownButton *countDownButton, NSInteger tag) {
    [countDownButton startCountDownWithSecond:59];
    [countDownButton countDownChanging:^NSString *(WxxCountDownButton *countDownButton, NSUInteger second) {
        NSString *title = [NSString stringWithFormat:@"(%lu)正在发送",(unsigned long)second];
        return title;
    }];
    [countDownButton countDownFinished:^NSString *(WxxCountDownButton *countDownButton, NSUInteger second) {
        return @"发送验证码";
    }];
}];

切换页面的时候，记得停止和释放：

-(void)viewWillDisappear:(BOOL)animated{
    [button stopCountDown];
}

小结

  • NSTimer的话，如果是主线程的RunLoop，需要设置NSRunLoopCommonModes模式，以防止UI等操作对计时器的影响。如果是开启新的线程的话，虽然不受UI影响，但是需要手动开启RunLoop。
  • 如果是dispatch_source_set_timer，计时准确，写起来也是挺方便的。
  • 计时过程和计时结束对外暴露方法，可以由用户自定义显示文字，做相关UI修改。
  • 自定义UIButton对外暴露的方法，用block来写，外部调用可以写到一起，简洁！
最后编辑于 ：
©著作权归作者所有,转载或内容合作请联系作者
  • 人面猴
    序言：七十年代末，一起剥皮案震惊了整个滨河市，随后出现的几起案子，更是在滨河造成了极大的恐慌，老刑警刘岩，带你破解...
    沈念sama阅读 218,204评论 6赞 506
  • 死咒
    序言：滨河连续发生了三起死亡事件，死亡现场离奇诡异，居然都是意外死亡，警方通过查阅死者的电脑和手机，发现死者居然都...
    沈念sama阅读 93,091评论 3赞 395
  • 救了他两次的神仙让他今天三更去死
    文/潘晓璐 我一进店门，熙熙楼的掌柜王于贵愁眉苦脸地迎上来，“玉大人，你说我怎么就摊上这事。” “怎么了？”我有些...
    开封第一讲书人阅读 164,548评论 0赞 354
  • 道士缉凶录：失踪的卖姜人
    文/不坏的土叔 我叫张陵，是天一观的道长。 经常有香客问我，道长，这世上最难降的妖魔是什么？ 我笑而不...
    开封第一讲书人阅读 58,657评论 1赞 293
  • 港岛之恋（遗憾婚礼）
    正文 为了忘掉前任，我火速办了婚礼，结果婚礼上，老公的妹妹穿的比我还像新娘。我一直安慰自己，他们只是感情好，可当我...
    茶点故事阅读 67,689评论 6赞 392
  • 恶毒庶女顶嫁案：这布局不是一般人想出来的
    文/花漫 我一把揭开白布。 她就那样静静地躺着，像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上，一...
    开封第一讲书人阅读 51,554评论 1赞 305
  • 城市分裂传说
    那天，我揣着相机与录音，去河边找鬼。 笑死，一个胖子当着我的面吹牛，可吹牛的内容都是我干的。 我是一名探鬼主播，决...
    沈念sama阅读 40,302评论 3赞 418
  • 双鸳鸯连环套：你想象不到人心有多黑
    文/苍兰香墨 我猛地睁开眼，长吁一口气：“原来是场噩梦啊……” “哼！你这毒妇竟也来了？” 一声冷哼从身侧响起，我...
    开封第一讲书人阅读 39,216评论 0赞 276
  • 万荣杀人案实录
    序言：老挝万荣一对情侣失踪，失踪者是张志新（化名）和其女友刘颖，没想到半个月后，有当地人在树林里发现了一具尸体，经...
    沈念sama阅读 45,661评论 1赞 314
  • 护林员之死
    正文 独居荒郊野岭守林人离奇死亡，尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,851评论 3赞 336
  • 白月光启示录
    正文 我和宋清朗相恋三年，在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,977评论 1赞 348
  • 活死人
    序言：一个原本活蹦乱跳的男人离奇死亡，死状恐怖，灵堂内的尸体忽然破棺而出，到底是诈尸还是另有隐情，我是刑警宁泽，带...
    沈念sama阅读 35,697评论 5赞 347
  • 日本核电站爆炸内幕
    正文 年R本政府宣布，位于F岛的核电站，受9级特大地震影响，放射性物质发生泄漏。R本人自食恶果不足惜，却给世界环境...
    茶点故事阅读 41,306评论 3赞 330
  • 男人毒药：我在死后第九天来索命
    文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹，春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,898评论 0赞 22
  • 一桩弑父案，背后竟有这般阴谋
    文/苍兰香墨 我抬头看了看天上的太阳。三九已至，却和暖如春，着一层夹袄步出监牢的瞬间，已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 33,019评论 1赞 270
  • 情欲美人皮
    我被黑心中介骗来泰国打工， 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留，地道东北人。 一个月前我还...
    沈念sama阅读 48,138评论 3赞 370
  • 代替公主和亲
    正文 我出身青楼，却偏偏与公主长得像，于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子，可洞房花烛夜当晚...
    茶点故事阅读 44,927评论 2赞 355