iOS 给UIAlertView添加block回调

在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 

typedef void(^CopletionBlock)(NSInteger buttonIndex);

@interface UIAlertView (Extension)

- (void)showWithBlock:(CopletionBlock)block;

@end

//  UIAlertView+Extension.m
#import "UIAlertView+Extension.h"
#import 

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 

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处理下。

有错误的地方或更改的实现方式，请留言讨论，谢谢！