关联：objc_getAssociatedObject和objc_setAssociatedObject使用

最新推荐文章于 2023-04-16 21:43:48 发布

转载最新推荐文章于 2023-04-16 21:43:48 发布 · 93 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

原文链接：http://www.cnblogs.com/On1Key/p/5144896.html

标签

#runtime

收录于

本文介绍如何通过Objective-C的运行时机制为UIButton类别添加额外的属性，包括子标题、前景颜色及自定义标签等，实现更灵活的按钮定制。

为UIButton的category添加属性

UIButton+subTitle.h

#import <UIKit/UIKit.h>

#import <objc/runtime.h>

 

@interface UIButton (subTitle)

@property (nonatomic, copy) NSString *subTitle;

@property (nonatomic, strong) UIColor *foreColor;

//此处添加int类型属性不能使用，因为objc需要用id类型，此处留着以后修改

@property (nonatomic) int tagAdd;

@end

UIButton+subTitle.m

#import "UIButton+subTitle.h"

 

@implementation UIButton (subTitle)

 

static char oooo;

 

- (NSString *)subTitle{

 

    return objc_getAssociatedObject(self, @selector(subTitle));

}

- (void)setSubTitle:(NSString *)subTitle{

    [self setTitle:subTitle forState:UIControlStateNormal];

    objc_setAssociatedObject(self, @selector(subTitle), subTitle, OBJC_ASSOCIATION_COPY_NONATOMIC);

}

- (UIColor *)foreColor{

    return objc_getAssociatedObject(self, @selector(foreColor));

}

- (void)setForeColor:(UIColor *)foreColor{

    self.backgroundColor = foreColor;

    objc_setAssociatedObject(self, @selector(foreColor), foreColor, OBJC_ASSOCIATION_RETAIN_NONATOMIC);

}

 

- (void)setTagAdd:(int)tagAdd{

    objc_setAssociatedObject(self, &oooo, @(tagAdd), OBJC_ASSOCIATION_ASSIGN);

}

- (int)tagAdd{

    return (int)objc_getAssociatedObject(self, &oooo);

}