美文网首页iOS
runtime简单封装获取短信验证码Button

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

作者: weiweilong | 来源:发表于2017-06-08 10:42 被阅读53次

    获取短信验证码-> 倒计时 -> 未收到重新获取,这是个在有手机登陆的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;
    }
    

    记得导入头文件<b><i>#import <objc/runtime.h></i></b>

    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,但是注意<b>设置背景颜色的时候是设置我们添加的属性.color 而不是.backgroundColor。</b>然后再需要开始倒计时的时候调用 <b><i>startTime</i></b>就行了。

        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];
    }
    

    需要代码戳这里

    相关文章

      网友评论

        本文标题:runtime简单封装获取短信验证码Button

        本文链接:https://www.haomeiwen.com/subject/rwszfxtx.html