给UIButton添加块(Block)
今天简单的记录一下怎么用块来实现Button的点击事件
首先，我想要的做的效果是直接通过类方法来初始化Button，然后同时把点击事件的操作放在块中，最后返回创建好的Button。
- 第一步，创建一个UIButton的扩展(category)
- 第二步，在h文件中添加初始化Button的类方法声明
#import <UIKit/UIKit.h>
typedef void (^ActionBlock)(UIButton *button);
@interface UIButton (Block)
@property (nonatomic,copy) ActionBlock actionBlock;
/**
通过block对Button的点击事件进行封装
@param frame frame大小
@param title 内容
@param titleColor 内容颜色
@param bgImgName 背景图片
@param completion 点击事件
@return Button
*/
+ (UIButton *)createButtonWithFrame:(CGRect)frame
title:(NSString *)title
titleColor:(UIColor *)titleColor
bgImageName:(NSString *)bgImgName
actionBlock:(void(^)(UIButton *sender))completion;
/**
通过block对Button的点击事件进行封装
@return 返回初始化后的button
*/
+ (UIButton *)button;
@end
- 第三步，在m文件中实现方法
#import "UIButton+Block.h"
#import <objc/runtime.h>
static NSString *keyWithMethod = @"keyWithMethod"; //关联对象的key
static NSString *keyWithBlock = @"keyWithBlock";
@implementation UIButton (Block)
+ (UIButton *)createButtonWithFrame:(CGRect)frame
title:(NSString *)title
titleColor:(UIColor *)titleColor
bgImageName:(NSString *)bgImgName
actionBlock:(void (^)(UIButton *sender))completion {
UIButton *button = [UIButton buttonWithType:UIButtonTypeRoundedRect];
button.frame = frame;
[button setTitle:title forState:UIControlStateNormal];
[button setTitleColor:titleColor forState:UIControlStateNormal];
[button setBackgroundImage:[UIImage imageNamed:bgImgName] forState:UIControlStateNormal];
[button addTarget:button action:@selector(buttonTapAction:) forControlEvents:UIControlEventTouchUpInside];
/*
*用runtime中的函数通过key关联对象
*
*objc_setAssociatedObject(id _Nonnull object, const void * _Nonnull key, id _Nullable value, objc_AssociationPolicy policy)
*id object 表示关联者，是一个对象，变量名也是object
*const void *key 获取被关联者的索引
*id value 被关联者，这里是一个block
*objc_AssociationPolicy policy 关联时采用的协议，有assign，retain，copy等协议，一般使用OBJC_ASSOCIATION_RETAIN_NONATOMIC
*/
objc_setAssociatedObject(button, &keyWithMethod, completion, OBJC_ASSOCIATION_COPY_NONATOMIC);
return button;
}
+ (UIButton *)button {
UIButton *button = [UIButton buttonWithType:UIButtonTypeCustom];
[button addTarget:button action:@selector(buttonTapAction:) forControlEvents:UIControlEventTouchUpInside];
return button;
}
- (void)setActionBlock:(ActionBlock)actionBlock{
objc_setAssociatedObject(self, &keyWithBlock, actionBlock, OBJC_ASSOCIATION_COPY_NONATOMIC );
}
- (ActionBlock)actionBlock{
return objc_getAssociatedObject(self ,&keyWithBlock);
}
- (void)buttonTapAction:(UIButton *)button {
//通过key获取被关联对象
//objc_getAssociatedObject(id object, const void *key)
void (^tapBlock)(UIButton *) = objc_getAssociatedObject(button, &keyWithMethod);
if (tapBlock) {
tapBlock(button);
}
ActionBlock block2 = (ActionBlock)objc_getAssociatedObject(button, &keyWithBlock);
if(block2){
block2(button);
}
}
@end
- 第四步，简单的使用
UIButton *button = [UIButton createButtonWithFrame:CGRectMake(20, 100, 200, 200) title:@"带块的button" titleColor:[UIColor redColor] bgImageName:@"" actionBlock:^(UIButton *button) {
NSString *str = [button titleForState:UIControlStateNormal];
NSLog(@"%@",str);
}];
[self.view addSubview:button];
UIButton *btn = [UIButton button];
btn.frame = CGRectMake(20, 100, 200, 200);
[btn setTitle:@"1234" forState:UIControlStateNormal];
btn.actionBlock = ^(UIButton *button) {
NSString *str = [button titleForState:UIControlStateNormal];
NSLog(@"%@",str);
};
[self.view addSubview:btn];
注意：由于扩展不能直接添加属性，所以要用运行时来自己添加属性的get、set方法。
最后，简单的了解一下objc_setAssociatedObject，objc_getAssociatedObject方法
Objective-C有两个扩展机制：Associative和Category。Category用来扩展类方法，Associative用于扩展属性。Associative机制的原理是把两个对象关联起来，让一个对象成为另外一个对象的一部分。它可以在不修改类的定义的前提下为其对象增加存储空间，这在我们无法访问类的源码时(例如给UILable添加一个selected的BOOL属性)是非常有用的。Associative基于关键字的，因此我们可以使用不同的关键字为任何对象添加任意多的Associative。Associative可以保证被关联的对象在对象的整个生命周期都是可用的。Associative基于runtime，是运行时里的东西,所以头文件需要引用#im port<objc/runtime.h>文件。
Associative提供了3个方法
- objc_setAssociatedObject(id object, const void *key, id value, objc_AssociationPolicy policy)
- objc_getAssociatedObject(id object, const void *key)
- objc_removeAssociatedObjects(id object)
第一个用于给关联对象赋值：
objc_setAssociatedObject(id object, const void *key, id value, objc_AssociationPolicy policy)注意的是返回值类型为Object类型，注意一些不是Object类型的例如：BOOL是结构类型。当碰到这种情况可以考虑通过中间类型来转换，如设置BOOL类型属性的时候可以转换为NSNumber类型，获取的时候再转换成BOOL类型即可。
四个参数分别是：源对象、关键字、关联对象和关联策略。
关键字是一个void类型的指针，例如static NSString *keyWithMethod = @"keyWithMethod"; //关联对象的key每一个关联的关键字必须是唯一的。
关联策略是枚举类型，如下：
typedef OBJC_ENUM(uintptr_t, objc_AssociationPolicy) {
OBJC_ASSOCIATION_ASSIGN = 0, /**< Specifies a weak reference to the associated object. */
OBJC_ASSOCIATION_RETAIN_NONATOMIC = 1, /**< Specifies a strong reference to the associated object.
* The association is not made atomically. */
OBJC_ASSOCIATION_COPY_NONATOMIC = 3, /**< Specifies that the associated object is copied.
* The association is not made atomically. */
OBJC_ASSOCIATION_RETAIN = 01401, /**< Specifies a strong reference to the associated object.
* The association is made atomically. */
OBJC_ASSOCIATION_COPY = 01403 /**< Specifies that the associated object is copied.
* The association is made atomically. */
};
第二个用于获取关联的对象:
objc_getAssociatedObject(id object, const void *key)用于获取关联对象的值。这里需要注意的是返回值类型为Object类型，注意一些不是Object类型的例如：BOOL是结构类型。
第三个用于断开关联的对象：
objc_removeAssociatedObjects(id object) 是断开关联，需要注意的是它会断开所有关联，所以不推荐这种方式。需要断开关联的时候使用objc_setAssociatedObject函数，传入nil值即可。
断开关联是使用objc_setAssociatedObject函数，传入nil值即可。
使用函数objc_removeAssociatedObjects可以断开所有关联
慢慢来，一步一个巴掌印~~~
