iOS: 教你给 UI 控件添加 Badge(消息提醒小圆点)

本文主要是介绍iOS: 教你给 UI 控件添加 Badge(消息提醒小圆点)，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

 PPBadgeView
1、前言
最近项目的一个需求是在UIView, UITabBarItem, UIBarButtonItem 这三种类型的控件上添加消息提醒小圆点(Badge),一开始找了一个星星很多的Badge库 RKNotificationHub,但发现其不能很好的满足项目需求(在UITabBarItem表现不是很好), 于是就自己动手写了PPBadgeView,可以很方便的为UIView, UITabBarItem, UIBarButtonItem及其子类添加Badge, 支持 Objective-C与Swift .
2、原理
PPBadgeView的实现原理很简单:  将一个UILabel控件(Badge,也可为其它)添加到UIView(及子类)的对象上 , 所以我们的关键点是要找到这个UIView对象! 像UIControl、UILabel、UIButton、UIImageView...这些控件就不用说了,都是继承UIView,可直接在它们的身上添加Badge,但是  UITabBarItem ,  UIBarButtonItem 就不同了,它们两个并不是继承的UIView,该怎么办?
2.1、给UITabBarItem添加Badge
通过系统的API可以看到 UITabBarItem 继承关系:
  UIBarButtonItem --> UIBarItem --> NSObject ,
并没有看见UIView(及子类) , UITabBarItem/UIBarItem的公开属性里也没发现可用的UIView...不急,还记得Xcode自带的UI视图调试神器吗?

运行Demo后点击打开,可清楚的看到底部栏的UITabBarItem内有一个UITabBarButton,其下属还有一个UITabBarSwappableImageView的图片控件,我们要找的就是这个UITabBarSwappableImageView
接下来就是要获取这个UITabBarSwappableImageView,我们可以使用Runtime + KVC 的方式: 
先利用runtime获的UITabBarButton的对象名称(具体方法可以查看这篇博客:三分钟教会你runtime获取属性和成员变量), 最后打印的结果为(因打印的内容太多,这里只贴出最关键的结果):
打印结果: UITabBarItem内的成员变量类型为: @"UITabBarButton"，名字为: _view
再使用KVC取出这个UITabBarButton对象,遍历出UITabBarSwappableImageView对象 UIView *tabBarButton = [tabBarItem valueForKey:@"_view"];for (UIView *subView in tabBarButton.subviews) {if ([subView isKindOfClass:NSClassFromString(@"UITabBarSwappableImageView")]) {return subView;}}
找到了Badge可以依靠的UIView, 剩下添加Badge的工作是不是变得容易很多了,这里不多说, 在PPBadgeView里有详细代码.
2.2、给UIBarButtonItem添加Badge
通过系统的API可以看到 UIBarButtonItem 继承关系:
  UIBarButtonItem --> UIBarItem --> NSObject 
和UITabBarItem一样其公开的属性也是没有可用的UIView对象的,不过有了上面的