iOSruntime运行时机制03(设置tabbarButton的badgeView)

if ([badgeViewChild isKindOfClass:NSClassFromString(&＃64;"_UIBadgeBackground")]) {//NSLog(&＃64;"终于找到你,还好没放弃");unsigned int count;//获取完成之后,count的值就代码当前类身上成员变量的个数//获取类身上的成员变量Ivar *vars &＃61; class_copyIvarList(NSClassFromString(&＃64;"_UIBadgeBackground"), &count);//遍历查看for (int i&＃61;0; i }

&＃43; (void)load{static dispatch_once_t onceToken;dispatch_once(&onceToken, ^{Class clazz &＃61; [self class];// 获取到原来的方法与需要交互的SELSEL originalSelector &＃61; &＃64;selector(setBadgeValue:);SEL swizzledSelector &＃61; &＃64;selector(wb_setBadgeValue:); // 通过SEL 获取到类身上的Method Method originalMethod &＃61; class_getInstanceMethod(clazz, originalSelector); Method swizzledMethod &＃61; class_getInstanceMethod(clazz, swizzledSelector); // 向类身上添加方法 BOOL result &＃61; class_addMethod(clazz, swizzledSelector, method_getImplementation(swizzledMethod), method_getTypeEncoding(swizzledMethod)); if (result) { // 如果添加成功就替换方法实现 class_replaceMethod(clazz, originalSelector, method_getImplementation(swizzledlMethod), method_getTypeEncoding(swizzledMethod)); }else{ // 没有添加成功&＃xff0c;直接交换方法实现 method_exchangeImplementations(originalMethod, swizzledMethod); } }); }

// 设置关联属性
- (void)setBadgeImageName:(NSString *)badgeImageName{objc_setAssociatedObject(self, &＃64;"haha", badgeImageName, OBJC_ASSOCIATION_COPY_NONATOMIC);
}- (NSString *)badgeImageName{return objc_getAssociatedObject(self, &＃64;"haha");
}

//
// UITabBarItem&＃43;BadgeValueImage.h//
// Created by ji on 15/8/20.
// Copyright (c) 2015年 ji. All rights reserved.
//#import &＃64;interface UITabBarItem (BadgeValueImage)
&＃64;property (nonatomic,copy) NSString *badgeImageName;
&＃64;end

//
// UITabBarItem&＃43;BadgeValueImage.m//
// Created by ji on 15/8/20.
// Copyright (c) 2015年 ji. All rights reserved.
//#import "UITabBarItem&＃43;BadgeValueImage.h"
#import &＃64;implementation UITabBarItem (BadgeValueImage)&＃43; (void)load{static dispatch_once_t onceToken;dispatch_once(&onceToken, ^{Class clazz &＃61; [self class];// 获取到原来的方法与需要交互的SELSEL originalSelector &＃61; &＃64;selector(setBadgeValue:); SEL swizzledSelector &＃61; &＃64;selector(wb_setBadgeValue:); // 通过SEL 获取到类身上的Method Method originalMethod &＃61; class_getInstanceMethod(clazz, originalSelector); Method swizzledMethod &＃61; class_getInstanceMethod(clazz, swizzledSelector); // 向类身上添加方法 BOOL result &＃61; class_addMethod(clazz, swizzledSelector, method_getImplementation(swizzledMethod), method_getTypeEncoding(swizzledMethod)); if (result) { // 如果添加成功就替换方法实现 class_replaceMethod(clazz, originalSelector, method_getImplementation(swizzledMethod), method_getTypeEncoding(swizzledMethod)); }else{ // 没有添加成功&＃xff0c;直接交换方法实现 method_exchangeImplementations(originalMethod, swizzledMethod); } }); } - (void)wb_setBadgeValue:(NSString *)badgeValue{ //调用父类 [self wb_setBadgeValue:badgeValue]; // 实现逻辑 // 如果badgeValue为nil 的话。直接返回 if(!badgeValue){ return; } // 我们直接能够拿到的就是WBTabBar UITabBarController *target &＃61; [self valueForKeyPath:&＃64;"_target"]; for (UIView *tabBarChild in target.tabBar.subviews) { //找UITabBarButton if ([tabBarChild isKindOfClass:NSClassFromString(&＃64;"UITabBarButton")]) { for (UIView *tabBarButtonChild in tabBarChild.subviews) { if ([tabBarButtonChild isKindOfClass:NSClassFromString(&＃64;"_UIBadgeView")]) { for (UIView *badgeViewChild in tabBarButtonChild.subviews) { if ([badgeViewChild isKindOfClass:NSClassFromString(&＃64;"_UIBadgeBackground")]) { //NSLog(&＃64;"终于找到你,还好没放弃"); unsigned int count;//获取完成之后,count的值就代码当前类身上成员变量的个数 //获取类身上的成员变量 Ivar *vars &＃61; class_copyIvarList(NSClassFromString(&＃64;"_UIBadgeBackground"), &count); //遍历查看 for (int i&＃61;0; i