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

PPBadgeView

一、前言

最近项目的一个需求是在UIView, UITabBarItem, UIBarButtonItem 这三种类型的控件上添加消息提醒小圆点(Badge),一开始找了一个星星不少的Badge库 RKNotificationHub,但发现其不能很好的知足项目需求(在UITabBarItem表现不是很好), 因而就本身动手写了PPBadgeView,能够很方便的为UIView, UITabBarItem, UIBarButtonItem及其子类添加Badge, 支持 Objective-C与Swift 上效果图:

二、原理

PPBadgeView的实现原理很简单: 将一个UILabel控件(Badge,也可为其它)添加到UIView(及子类)的对象上 , 因此咱们的关键点是要找到这个UIView对象! 像UIControl、UILabel、UIButton、UIImageView...这些控件就不用说了,都是继承UIView,可直接在它们的身上添加Badge,可是 UITabBarItem , UIBarButtonItem 就不一样了,它们两个并非继承的UIView,该怎么办?

2.一、给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.二、给UIBarButtonItem添加Badge

经过系统的API能够看到 UIBarButtonItem 继承关系:
UIBarButtonItem --> UIBarItem --> NSObject
和UITabBarItem同样其公开的属性也是没有可用的UIView对象的,不过有了上面的👆经验,获取UIBarButtonItem中的UIView(及子类)也是一样的作法, 看图:
咱们要找的就是UINavigationButton中的UIImageView属性了

很奇怪,利用runtime获取出来UINavigationButton类名称为"UIView"
打印结果: UIBarButtonItem内的成员变量类型为: @"UIView"，名字为: _view
这里须要注意的是, UINavigationButton中的UIImageView对象的layer默认masksToBounds = YES,在取出的时候须要设置masksToBounds = NO:

UIView *navigationButton = [barButtonItem valueForKey:@"_view"];
    for (UIView *subView in navigationButton.subviews) {
        if ([subView isKindOfClass:NSClassFromString(@"UIImageView")]) {
            subView.layer.masksToBounds = NO;
            return subView;
        }
    }复制代码

固然,你也能够直接使用navigationButton来作Badge的父视图...

=========================

2017-06-27更新: UITabBarItem分类/拓展移除掉直接比对"私有属性API:UITabBarSwappableImageView"的代码,规避上架审核被拒的风险.很是感谢@iOS程序犭袁 大大的指出的这个问题

=========================

结束

PPBadgeView现已托管到GitHub维护,有Objective-C和Swift双版本,支持CocoaPods导入,地址: github.com/jkpang/PPBa…

个人GitHub主页: github.com/jkpang