Use BadgeContainer Protocol

BadgeViewUsage.swift

//custom class
//Don't forget to assigne that custom class to your UIButton views in the storyboard file
class BadgeButton: UIButton, BadgeContainer {
    var badgeTimer: Timer?
    var badgeView: UIView?
    var badgeLabel: UILabel?
}

//here is how to use it in simple view controller
class MyViewController: UIViewController {
   @IBOutlet weak var button: BadgeButton!

    override func viewDidLoad() {
        super.viewDidLoad()

        button.showBadge(blink: false, text: "swift")
        button.badgeView?.backgroundColor = .blue
        button.badgeLabel?.textColor = .orange
    }
    //here is an example how to hide the badge
    func buttonHideBadge() {
        button.hideBadge()
    }
}