你的位置:首页 > 操作系统

[操作系统]使用dispatch_group实现并封装分组并发网络请求


在实际开发中我们通常会遇到这样一种需求:某个页面加载时通过网络请求获得相应的数据,再做某些操作。有时候加载的内容需要通过好几个请求的数据组合而成,比如有两个请求A和B,我们通常为了省事,会将B请求放在A请求成功的回调中发起,在B的成功回调中将数据组合起来,这样做有明显的问题:

1.请求如果多了,需要写许多嵌套的请求

2.如果在除了最后一个请求前的某个请求失败了,就不会执行后面的请求,数据无法加载

3.请求变成同步的,这是最大的问题,在网络差的情况下,如果有n个请求,意味着用户要等待n倍于并发请求的时间才能看到内容

 

同步请求这么low的方式当然是不可接受的,所以我们要并发这些请求,在所有请求都执行完成功回调后,再做加载内容或其他操作,考虑再三,选择用GCD的dispatch_group。

dispatch_group通常有两种用法,一种是

dispatch_group_async(<#dispatch_group_t group#>, <#dispatch_queue_t queue#>, <#^(void)block#>)

创建一个dispatch_group_t, 将并发的操作放在block中,在

dispatch_group_notify(<#dispatch_group_t group#>, <#dispatch_queue_t queue#>, <#^(void)block#>)

的block中执行多组block执行完毕后的操作,对于网络请求来说,在请求发出时他就算执行完毕了,并不会等待回调,所以不满足我们的需求。

所以采用另一种用法:

使用dispatch_group_enter和dispatch_group_leave,这种方式使用更为灵活,enter和leave必须配合使用,有几次enter就要有几次leave,否则group会一直存在。当所有enter的block都leave后,会执行dispatch_group_notify的block。

我们当然可以在网络请求前enter,在执行完每个请求的成功回调后leave,再在notify中执行内容加载,这样看来问题就解决了,但还是有点小小不爽,每次发起组请求我都得创建group,写一堆的enter和leave,既麻烦也不利于复用,很自然我们想到把他封装一下,最好能做到将一个网络请求加到组里,而不用修改原先的网络请求代码,就像这样:

[[MDXuexiBaoAPI sharedInstance] postForGroup:^{    request1 success:^(id responseObject) {          } failure:^(NSError *error) {          }];        request2 success:^(id responseObject) {    } failure:^(NSError *error) {          }];  } success:^{    // group success  } failure:^(NSArray *errorArray) {    // group failure  }]

如果我想做到这种效果,肯定要到网络单例层去做些修改,但我又不想改变现有的底层请求方法,所以我采用了

method_exchangeImplementations(<#Method m1#>, <#Method m2#>) 这个函数,基于现有的底层请求方法,实现一套组的请求方法,替换掉原先的方法,在组请求都发送完毕后,再换回原先的方法。

但这里有一些可怕的坑要处理,因为使用方法替换是很危险的。

1.我做了替换后,正常的非组网络请求也会走替换后的方法,但我不需要他走替换后的方法。

2.假如我同时发起了多个组请求,组和组之间要如何区分,不同的组是不应该相互影响的。

 

一开始我考虑给请求一个mark,标记他是属于哪个group的,但这需要你已经把请求封装成了一个对象,如果你的项目和我的一样,只是执行一个方法,是不好给他加标记的。

在一阵头脑风暴后,我决定用队列来区分每个gorup。

具体做法就是创建每个group时,开启一个队列,给队列动态添加group属性,一个队列对应一个group。在队列中替换方法,发起组里的请求,再替换回原先的方法,这样在replaced method里只需要拿到当前的队列,就可以拿到group,如果group是nil,说明是正常的非组请求,执行original method;如果group不是nil,根据group来enter和leave,每个group也能区分开。

创建group时,给group添加了一个errorArray,用来记录组里请求的error,只要errorArray不为空,就会走组失败的block。

下面上代码:

 

typedef void(^BlockAction)();typedef void(^GroupResponseFailure)(NSArray * errorArray);static char groupErrorKey;static char queueGroupKey;

- (void)sendPOSTRequestInGroup:(NSString *)strURL withData:(NSDictionary *)data paramForm:(ParamForm)paramForm withTimeout:(NSTimeInterval)timeout showAlert:(BOOL)show success:(BlockResponse)success failure:(BlockResponseFailure)failure {    dispatch_group_t group = objc_getAssociatedObject([NSOperationQueue currentQueue], &queueGroupKey);    // 如果是非组请求  if (group == nil) {    // 执行original method    [self sendPOSTRequestInGroup:strURL withData:data paramForm:paramForm withTimeout:timeout showAlert:show success:success failure:failure];    return;  }    dispatch_group_enter(group);  // 执行original method  [self sendPOSTRequestInGroup:strURL withData:data paramForm:paramForm withTimeout:timeout showAlert:show success:^(id responseObject) {    success(responseObject);    dispatch_group_leave(group);  } failure:^(NSError *error) {    NSMutableArray *arrayM = objc_getAssociatedObject(group, &groupErrorKey);    [arrayM addObject:error];    failure(error);    dispatch_group_leave(group);  }];}- (void)sendGroupPostRequest:(BlockAction)requests success:(BlockAction)success failure:(GroupResponseFailure)failure {  if (requests == nil) {    return;  }    dispatch_group_t group = dispatch_group_create();  objc_setAssociatedObject(group, &groupErrorKey, [NSMutableArray array], OBJC_ASSOCIATION_RETAIN_NONATOMIC);    Method originalPost = class_getInstanceMethod(self.class, @selector(sendPOSTRequest:withData:paramForm:withTimeout:showAlert:success:failure:));  Method groupPost = class_getInstanceMethod(self.class, @selector(sendPOSTRequestInGroup:withData:paramForm:withTimeout:showAlert:success:failure:));    NSOperationQueue *queue = [[NSOperationQueue alloc] init];  objc_setAssociatedObject(queue, &queueGroupKey, group, OBJC_ASSOCIATION_RETAIN_NONATOMIC);  queue.qualityOfService = NSQualityOfServiceUserInitiated;  queue.maxConcurrentOperationCount = 3;    [queue addOperationWithBlock:^{        method_exchangeImplementations(originalPost, groupPost);    requests();    // 发出请求后就可以替换回original method,不必等待回调,尽量减小替换的时间窗口    method_exchangeImplementations(originalPost, groupPost);        dispatch_group_notify(group, dispatch_get_main_queue(), ^{      NSMutableArray *arrayM = objc_getAssociatedObject(group, &groupErrorKey);      if (arrayM.count > 0) {        if (failure) {          failure(arrayM.copy);        }      } else if(success) {        success();      }          });  }];}

 

- (void)sendPOSTRequestInGroup:(NSString *)strURL withData:(NSDictionary *)data paramForm:(ParamForm)paramForm withTimeout:(NSTimeInterval)timeout showAlert:(BOOL)show success:(BlockResponse)success failure:(BlockResponseFailure)failure
替换网络工具层的底层post请求

sendPOSTRequest:withData:paramForm:withTimeout:showAlert:success:failure:

这样在 

- (void)sendGroupPostRequest:(BlockAction)requests success:(BlockAction)success failure:(GroupResponseFailure)failure
requests block中,把网络请求扔进去,单个请求本身的success和failure都能执行,success执行组成功的代码,failure中可以拿到每个请求的error,作相应处理。