YFButton.swift

import UIKit
import Foundation

typealias HoldActionHandler = () -> Void

class YFButton: UIButton {
    
    var holdAction: HoldActionHandler = { }
    
    private var timer: Timer?
    private var longPressGesture = UILongPressGestureRecognizer()
    
    override func awakeFromNib() {
        super.awakeFromNib()
        self.setupGesture()
    }
    
    deinit {
        self.resetTimer()
        self.removeGestureRecognizer(self.longPressGesture)
    }
    
    private func setupGesture() {
        self.longPressGesture = UILongPressGestureRecognizer(target: self,
                                                             action: #selector(handleLongPress(_:)))
        self.addGestureRecognizer(self.longPressGesture)
    }
    
    @objc private func handleLongPress(_ gesture: UILongPressGestureRecognizer) {
     
        switch gesture.state {
        case .began:
            self.timer = Timer.scheduledTimer(timeInterval: 0.1,
                                              target: self,
                                              selector: #selector(handleTimer(_:)),
                                              userInfo: nil,
                                              repeats: true)
            case .ended, .cancelled:
                self.resetTimer()
        default:
            break
        }
    }
    
    private func resetTimer() {
        self.timer?.invalidate()
        self.timer = nil
    }
    
    @objc func handleTimer(_ timer: Timer) {
        self.holdAction()
    }
    
}