iOS如何封装倒计时按钮HLCountDownButton

在iOS开发中，倒计时按钮是一个常见的需求，尤其是在验证码发送、短信验证等场景中。为了提高代码的复用性和可维护性，我们可以将倒计时功能封装成一个自定义的按钮控件。本文将详细介绍如何封装一个名为HLCountDownButton的倒计时按钮。

1. 需求分析

在开始编码之前，我们需要明确HLCountDownButton的功能需求：

1. 倒计时功能：按钮点击后开始倒计时，倒计时期间按钮不可点击。
2. 倒计时显示：倒计时期间，按钮显示剩余时间。
3. 倒计时结束：倒计时结束后，按钮恢复初始状态，可以再次点击。
4. 自定义倒计时时间：允许外部设置倒计时的总时间。
5. 自定义按钮样式：允许外部设置按钮的样式，如背景颜色、文字颜色等。

2. 创建HLCountDownButton类

首先，我们创建一个继承自UIButton的HLCountDownButton类。

#import <UIKit/UIKit.h>

@interface HLCountDownButton : UIButton

// 设置倒计时总时间（单位：秒）
@property (nonatomic, assign) NSInteger countDownTime;

// 开始倒计时
- (void)startCountDown;

// 停止倒计时
- (void)stopCountDown;

@end

3. 实现倒计时功能

接下来，我们在HLCountDownButton.m文件中实现倒计时功能。

#import "HLCountDownButton.h"

@interface HLCountDownButton ()

@property (nonatomic, strong) NSTimer *countDownTimer;
@property (nonatomic, assign) NSInteger remainingTime;

@end

@implementation HLCountDownButton

- (instancetype)initWithFrame:(CGRect)frame {
    self = [super initWithFrame:frame];
    if (self) {
        [self setup];
    }
    return self;
}

- (void)setup {
    self.countDownTime = 60; // 默认倒计时时间为60秒
    [self setTitle:@"获取验证码" forState:UIControlStateNormal];
    [self setTitleColor:[UIColor whiteColor] forState:UIControlStateNormal];
    self.backgroundColor = [UIColor blueColor];
    self.layer.cornerRadius = 5.0;
    [self addTarget:self action:@selector(buttonClicked) forControlEvents:UIControlEventTouchUpInside];
}

- (void)buttonClicked {
    [self startCountDown];
}

- (void)startCountDown {
    self.remainingTime = self.countDownTime;
    self.enabled = NO;
    self.countDownTimer = [NSTimer scheduledTimerWithTimeInterval:1.0 target:self selector:@selector(updateCountDown) userInfo:nil repeats:YES];
}

- (void)updateCountDown {
    if (self.remainingTime > 0) {
        [self setTitle:[NSString stringWithFormat:@"%ld秒后重试", (long)self.remainingTime] forState:UIControlStateNormal];
        self.remainingTime--;
    } else {
        [self stopCountDown];
    }
}

- (void)stopCountDown {
    [self.countDownTimer invalidate];
    self.countDownTimer = nil;
    self.enabled = YES;
    [self setTitle:@"获取验证码" forState:UIControlStateNormal];
}

- (void)dealloc {
    [self.countDownTimer invalidate];
    self.countDownTimer = nil;
}

@end

代码解析

1. 初始化方法：在initWithFrame:方法中调用setup方法进行初始化设置。
2. setup方法：设置默认的倒计时时间、按钮样式，并添加点击事件。
3. buttonClicked方法：按钮点击后调用startCountDown方法开始倒计时。
4. startCountDown方法：设置剩余时间，禁用按钮，并启动定时器。
5. updateCountDown方法：定时器每秒调用一次，更新按钮标题显示剩余时间。
6. stopCountDown方法：停止定时器，恢复按钮状态。
7. dealloc方法：在对象销毁时停止定时器，防止内存泄漏。

4. 使用HLCountDownButton

在需要使用倒计时按钮的地方，我们可以直接创建HLCountDownButton实例并添加到视图中。

#import "ViewController.h"
#import "HLCountDownButton.h"

@interface ViewController ()

@end

@implementation ViewController

- (void)viewDidLoad {
    [super viewDidLoad];
    
    HLCountDownButton *countDownButton = [[HLCountDownButton alloc] initWithFrame:CGRectMake(100, 100, 200, 40)];
    countDownButton.countDownTime = 120; // 设置倒计时时间为120秒
    [self.view addSubview:countDownButton];
}

@end

代码解析

1. 创建HLCountDownButton实例：在viewDidLoad方法中创建HLCountDownButton实例，并设置倒计时时间为120秒。
2. 添加到视图：将按钮添加到当前视图控制器的视图中。

5. 自定义按钮样式

HLCountDownButton继承自UIButton，因此我们可以通过UIButton的属性来自定义按钮的样式。例如，设置按钮的背景颜色、文字颜色、圆角等。

countDownButton.backgroundColor = [UIColor redColor];
[countDownButton setTitleColor:[UIColor blackColor] forState:UIControlStateNormal];
countDownButton.layer.cornerRadius = 10.0;

6. 总结

通过封装HLCountDownButton，我们可以轻松地在iOS应用中实现倒计时按钮功能。该控件具有良好的复用性和可扩展性，开发者可以根据实际需求进一步定制按钮的样式和行为。

优点

• 代码复用：将倒计时功能封装成独立的控件，可以在多个项目中复用。
• 易于维护：倒计时逻辑集中在HLCountDownButton类中，便于维护和修改。
• 灵活性：允许外部设置倒计时时间和按钮样式，适应不同的业务需求。

扩展

在实际开发中，我们还可以进一步扩展HLCountDownButton的功能，例如：

• 支持倒计时结束回调：在倒计时结束时执行自定义操作。
• 支持暂停和恢复倒计时：在特定情况下暂停倒计时，稍后恢复。
• 支持多种倒计时格式：例如显示“分:秒”格式的倒计时。

通过不断优化和扩展，HLCountDownButton可以成为一个功能强大且灵活的倒计时按钮控件。