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 的方式:

1. 先利用runtime获的UITabBarButton的对象名称(具体方法可以查看这篇博客:三分钟教会你runtime获取属性和成员变量), 最后打印的结果为(因打印的内容太多,这里只贴出最关键的结果): 打印结果: UITabBarItem内的成员变量类型为: @"UITabBarButton"，名字为: _view
2. 再使用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对象的,不过有了上面的