在项目开发中经常需要用到倒计时的功能，比如注册部分获取验证码时，这里通过extension（UIButton）的方式来实现倒计时的功能
直接上代码
var codeTimer = DispatchSource.makeTimerSource(queue:DispatchQueue.global())
extension UIButton {
//倒计时启动
func countDown(count: Int){
// 倒计时开始,禁止点击事件
isEnabled = false
var remainingCount: Int = count {
willSet {
setTitle("\(newValue)秒重发", for: .normal)
if newValue <= 0 {
setTitle("获取验证码", for: .normal)
}
}
}
if codeTimer.isCancelled {
codeTimer = DispatchSource.makeTimerSource(queue:DispatchQueue.global())
}
// 设定这个时间源是每秒循环一次，立即开始
codeTimer.scheduleRepeating(deadline: .now(), interval: .seconds(1))
// 设定时间源的触发事件
codeTimer.setEventHandler(handler: {
// 返回主线程处理一些事件，更新UI等等
DispatchQueue.main.async {
// 每秒计时一次
remainingCount -= 1
// 时间到了取消时间源
if remainingCount <= 0 {
self.isEnabled = true
codeTimer.cancel()
}
}
})
// 启动时间源
codeTimer.resume()
}
//取消倒计时
func countdownCancel() {
if !codeTimer.isCancelled {
codeTimer.cancel()
}
// 返回主线程
DispatchQueue.main.async {
self.isEnabled = true
if self.titleLabel?.text?.count != 0
{
self.setTitle("获取验证码", for: .normal)
}
}
}
}
在需要的地方直接调用即可
//启动倒计时
self.smsCodeBtn.countDown(count: 60)
//取消倒计时
self.smsCodeBtn.countdownCancel()