在iOS开发中，经常用到警告视图与用户交互，虽然在iOS8.0以后推荐使用UIAlertController了，但是对于兼容iOS7.x的项目，有时还是要用UIAlterView。系统是使用了委托进行了回调，把创建视图与处理点击事件的代码分开。当处理事件的逻辑较复杂时，这非常不错。但是当处理逻辑较复杂时，我们一般会抽出某个方法去执行。平时是这样使用UIAlterView的
- (void)someEvent {
UIAlertView *alert = [[UIAlertView alloc] initWithTitle:@"title" message:@"message" delegate:self cancelButtonTitle:@"cancel" otherButtonTitles:@"done", nil];
[alert show];
}
- (void)alertView:(UIAlertView *)alertView clickedButtonAtIndex:(NSInteger)buttonIndex {
if (buttonIndex == 1) {
// done
} else {
//cancel
}
}
//当然要遵守UIAlertViewDelegate协议
如果想在一个类中处理多个警告视图，那么代码就会变得较为复杂。我们必须在delegate中检查传入的alertView参数，并据此选用相应的处理逻辑。如果能在创建视图的时候，用block进行回调就简单多了。
接下来，我们要做的就是给UIAlertView添加block回调。
1.使用关联对象(AssociatedObject)实现
在Objective-C中，可以使用以下几个方法进行关联对象操作。
-
void objc_setAssociatedObject(id object, void*key, id value, objc_AssociationPolicy policy)
此方法以给定的键和策略为某对象设置关联对象值 -
id objc_getAssociatedObject(id object, const void *key)
此方法根据给定的键从某对象中获取相应的关联对象值 -
void objc_removeAssociatedObjects(id object)
此方法移除指定对象的全部关联对象
实现一个UIAlertView的category，UIAlertView+Extension，定义一个block，并实现show方法。代码如下：
// UIAlertView+Extension.h
#import <UIKit/UIKit.h>
typedef void(^CopletionBlock)(NSInteger buttonIndex);
@interface UIAlertView (Extension)
- (void)showWithBlock:(CopletionBlock)block;
@end
// UIAlertView+Extension.m
#import "UIAlertView+Extension.h"
#import <objc/runtime.h>
static void *MyAlterViewKey = "MyAlterViewKey";
@implementation UIAlertView (Extension)
- (void)showWithBlock:(CopletionBlock)copletionBlock {
if (copletionBlock) {
objc_removeAssociatedObjects(self);
objc_setAssociatedObject(self, MyAlterViewKey, copletionBlock, OBJC_ASSOCIATION_COPY);
self.delegate = self;
}
[self show];
}
- (void)alertView:(UIAlertView *)alertView clickedButtonAtIndex:(NSInteger)buttonIndex {
CopletionBlock copletionBlock = objc_getAssociatedObject(self, MyAlterViewKey);
if (copletionBlock) {
copletionBlock(buttonIndex);
}
}
@end
在show的时候将对象与block进行关联，在delegate中获取该block。
使用时，导入UIAlertView+Extension.h文件
UIAlertView *alert = [[UIAlertView alloc] initWithTitle:@"title" message:@"This is message!" delegate:nil cancelButtonTitle:@"cancel" otherButtonTitles:@"done", nil];
[alert showWithBlock:^(NSInteger index) {
if (index == 0) {
// cancel
} else {
// done
}
}];
2.使用继承实现
继承自UIAlertView实现自定义的子类，在子类中设置block为子类的属性。贴下代码：
// WSAlterView.h
#import <UIKit/UIKit.h>
typedef void(^CopletionBlock)(NSInteger buttonIndex);
@interface WSAlterView : UIAlertView
- (void)showWithBlock:(CopletionBlock)copletionBlock;
@end
#import "WSAlterView.h"
@interface WSAlterView ()
@property (nonatomic, copy) CopletionBlock copletionBlock;
@end
@implementation WSAlterView
- (void)showWithBlock:(CopletionBlock)copletionBlock {
if (copletionBlock) {
self.copletionBlock = copletionBlock;
self.delegate = self;
}
[self show];
}
- (void)alertView:(UIAlertView *)alertView clickedButtonAtIndex:(NSInteger)buttonIndex {
if (self.copletionBlock) {
self.copletionBlock(buttonIndex);
}
}
@end
使用时，先导入WSAlterView.h。使用代码如下：
WSAlterView *alert = [[WSAlterView alloc] initWithTitle:@"title" message:@"This is message!" delegate:nil cancelButtonTitle:@"cancel" otherButtonTitles:@"done", nil];
[alert showWithBlock:^(NSInteger index) {
if (index == 0) {
// cancel
} else {
// done
}
}];
至此，为UIAlertView添加block回调就用两种方式实现了。
Demo在这里
注意：
需要注意的是，如果在类中强引用了UIAlertView，而警告视图的block回调中又强引用这个类的话，就会形成循环引用。比如在某个控制器中强引用一个UIAlertView，循环引用是这样的
VC—>alertView—>block—>VC
这种情况需要用__weak处理下。
有错误的地方或更改的实现方式，请留言讨论，谢谢！
