Swift倒计时按钮

在需要使用倒计时按钮的地方黏贴以下代码

 //倒计时验证码
    func countDown(_ timeOut: Int, btn: UIButton){
           //倒计时时间
           var timeout = timeOut
           let queue:DispatchQueue = DispatchQueue.global(qos: DispatchQoS.QoSClass.default)
           let _timer:DispatchSource = DispatchSource.makeTimerSource(flags: [], queue: queue) as! DispatchSource
           _timer.scheduleRepeating(wallDeadline: DispatchWallTime.now(), interval: .seconds(1))
           //每秒执行
           _timer.setEventHandler(handler: { () -> Void in
               if(timeout<=0){ //倒计时结束，关闭
                   _timer.cancel();
                   DispatchQueue.main.sync(execute: { () -> Void in
                       btn.setTitle("重新获取", for: .normal)
                       btn.isEnabled = true
                       btn.layer.backgroundColor = UIColor.orange.cgColor
                   })
               }else{//正在倒计时
                   let seconds = timeout
                   DispatchQueue.main.sync(execute: { () -> Void in
                       let str = String(describing: seconds)
                       btn.setTitle("\(str)秒后获取", for: .normal)
                       btn.isEnabled = false
                       btn.layer.backgroundColor = UIColor.gray.cgColor
                   })
                   timeout -= 1;
               }
           })
           _timer.resume()
       }

使用的地方使用以下代码：

  // 启动倒计时
   countDown(10, btn: self.sendSMSButton)