Swift中给UIView添加Badge

extension UIView{
//任意UIView添加badge
func showBadgeValue(#strBadgeValue: String) -> Void{

let tabBar = UITabBar(frame: CGRectMake(0, 0, 320, 50))
let item = UITabBarItem(title: "", image: nil, tag: 0)
item.badgeValue = strBadgeValue
let array = [item]

tabBar.items = array
for viewTab in tabBar.subviews{
for subview in viewTab.subviews{
let strClassName = String(UTF8String: object_getClassName(subview))
if strClassName == "UITabBarButtonBadge" || strClassName == "_UIBadgeView"{
let theSubView = subview as! UIView
theSubView.removeFromSuperview()
self.addSubview(theSubView)
theSubView.frame = CGRectMake(self.frame.size.width - theSubView.frame.size.width, 0, theSubView.frame.size.width, theSubView.frame.size.height)

}

}
}
}

//删除UIView的badge
func removeBadge() -> Void{
for subview in self.subviews{
let strClassName = String(UTF8String: object_getClassName(subview))
if strClassName == "UITabBarButtonBadge" || strClassName == "_UIBadgeView"{
let theSubView = subview as! UIView
theSubView.removeFromSuperview()
}
}
}
}