获取短信验证码-> 倒计时 -> 未收到重新获取,这是个在有手机登陆的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];
}
需要代码戳这里
网友评论