iOS 等待for循环里的异步任务完成再进行其他操作的一个解决办法 -- 信号量+串行队列卡for小循环

转自：https://blog.csdn.net/qq_34417314/article/details/80449484

for循环里的异步操作

开发中经常会遇到这样一些情况，比如：
1.登录失败后的多次登录重连场景。
2.在一个for循环遍历里，有多种异步操作，需要在所有的异步操作完成后，也就是for循环的遍历结束后，再去执行其他操作，但是不能卡主线程，这时候就需要用其他方法了。

我遇到的需求是，在一个for循环里有数据库的查询操作以及网络请求操作，然后将数据库的查询和网络请求的结果添加到一个临时数组中，最后等for循环结束后，将临时数组通过block回传出去。
解决办法如下：

• 串行队列配合信号量

因为for循环不能卡主线程，所以我们首先要创建一个串行的队列(并发的不可以)，然后通过信号量来控制for循环，下面是模拟的一个操作：

printf("处理前创建信号量，开始循环异步请求操作\n");

// 1.创建一个串行队列，保证for循环依次执行
        dispatch_queue_t serialQueue = dispatch_queue_create("serialQueue", DISPATCH_QUEUE_SERIAL);

// 2.异步执行任务
dispatch_async(serialQueue, ^{
    // 3.创建一个数目为1的信号量，用于“卡”for循环，等上次循环结束在执行下一次的for循环
    dispatch_semaphore_t sema = dispatch_semaphore_create(0);
   
    for (int i = 0; i<5; i++) {
        // 开始执行for循环，让信号量-1，这样下次操作须等信号量>=0才会继续,否则下次操作将永久停止
    
        printf("信号量等待中\n");
        // 模拟一个异步任务
        NSMutableURLRequest *urlRequest = [[NSMutableURLRequest alloc] initWithURL:[NSURL URLWithString:@"https://github.com"]];
        NSURLSession *session = [NSURLSession sharedSession];
        NSURLSessionDataTask *dataTask = [session dataTaskWithRequest:urlRequest completionHandler:^(NSData * _Nullable data, NSURLResponse * _Nullable response, NSError * _Nullable error) {
        // 本次for循环的异步任务执行完毕，这时候要发一个信号，若不发，下次操作将永远不会触发
            [tampArray addObject:@(i)];
            NSLog(@"本次耗时操作完成，信号量+1 %@\n",[NSThread currentThread]);
            dispatch_semaphore_signal(sema);
            
        }];
        [dataTask resume];
        dispatch_semaphore_wait(sema, DISPATCH_TIME_FOREVER);
    }
    
    NSLog(@"其他操作");
    for (NSNumber *num in tampArray) {
        NSLog(@"所有操作完成后的操作--->   %@\n",num);
    }
    
});

以上就是一个简单的卡for循环的实例

• 实际场景中的应用

我遇到场景：
1.从相册里取若干原图，先获取到缩略图的数组，后根据缩略图信息取原图，这是一个可同步可异步的操作，鉴于卡顿考虑，采用异步取原图的操作,

 NSMutableArray *tmpPhotoes = [NSMutableArray array];
            // 这里用信号量卡 for 循环 来获取原图，等全部获取完成再刷新UI
            dispatch_queue_t serialQueue = dispatch_queue_create("getOriginalQueue", DISPATCH_QUEUE_SERIAL);
            
            dispatch_async(serialQueue, ^{
                
                dispatch_semaphore_t sema = dispatch_semaphore_create(0);
                
                [assets enumerateObjectsUsingBlock:^(PHAsset *tmpAsset, NSUInteger idx, BOOL * _Nonnull stop) {
                    
                    /* 这里换成系统的方法也是一样的，[[PHImageManager defaultManager] requestImageForAsset:asset targetSize:PHImageManagerMaximumSize contentMode:PHImageContentModeAspectFit options:option resultHandler:^(UIImage *result, NSDictionary *info) {
    }];*/ 
                        __block int time = 0;  // 由于获取原图会有2次回调，PHImageResultIsDegradedKey为0时才为原图
                    [[TZImageManager manager] getOriginalPhotoWithAsset:tmpAsset newCompletion:^(UIImage *photo, NSDictionary *info, BOOL isDegraded) {
                        // 原图
                        if ([[info jk_stringForKey:PHImageResultIsDegradedKey] isEqualToString:@"0"]) {
                            [tmpPhotoes addObject:photo];
                        }
                        time++;
                        if (time == 2) {
                            dispatch_semaphore_signal(sema);
                        }
                    }];
                    
                    dispatch_semaphore_wait(sema, DISPATCH_TIME_FOREVER);
                    
                }];

                dispatch_async(dispatch_get_main_queue(), ^{
                    if (tmpPhotoes.count) {
                        [self refreshImageDataWithPhotos:tmpPhotoes assets:assets];
                    }
                });
            });

2.外界多个渠道随时可能来查询一个会话列表数组的数据，但是原始的数组里没有包含想要的东西，需要包装一下再供外界使用。首先需要创建一个条件锁，防止这个列表数据被多方访问造成资源抢夺，其次循环遍历原始数组，通过数据库查询和网络请求操作重新包装会话列表里的数据，最后等待for循环结束，将包装过的数组回传给外界以供使用。

 

/** 会话列表 */
-(void)getConversationList:(void(^)(NSArray *conversationList))handler {
// 条件锁，若当前资源可获取则进行下一步，否则等待条件锁完成才能访问
if ([self.lock obtain]) {
    BBLog(@"BigBang- 获取资源中");
    // 会话列表
    NSArray *oriList = [[BBConversationManager manager] getConversationList];
    
    dispatch_queue_t serailQueue = dispatch_queue_create("conversationQueue", DISPATCH_QUEUE_SERIAL);
    dispatch_async(serailQueue, ^{
        
        // 每次重置，防止 dispose 
        self.sema = nil;
        self.sema = dispatch_semaphore_create(1);
        BBLog(@"BigBang- CrashInfo -- 信号量创建 当前线程---%@",[NSThread currentThread]);
        // 临时数组 存储变换后的会话模型
        __block NSMutableArray *temList = [NSMutableArray array];
        if (oriList.count > 0) {
            [oriList enumerateObjectsUsingBlock:^(RCConversation *conversation, NSUInteger idx, BOOL * _Nonnull stop) {
                
                NSString *identifier = conversation.targetId;
                RCMessage *message = [[RCIMClient sharedRCIMClient] getMessage:conversation.lastestMessageId];
                
                if ([conversation conversationType] == ConversationType_PRIVATE || [conversation conversationType] == ConversationType_SYSTEM) {
                    
                    BBLog(@"BigBang- CrashInfo 信号量-1");
                    dispatch_semaphore_wait(self.sema, DISPATCH_TIME_FOREVER);
                    
                    // 本地存在，直接从本地数据库取 这里是一个数据库的查询操作，是在子线程中进行的
                    if ([BBUserTable isUserExist:identifier]) {
                        
                        BBConversationListModel *model = [[BBConversationListModel alloc] initWithRCMessage:message userInfo:[BBUserTable gerUserWithID:identifier]];
                        model.unreadMessageCount = conversation.unreadMessageCount;
                        if ([conversation conversationType] == ConversationType_SYSTEM) {
                            [[BBConversationManager manager] setConversationToTop:ConversationType_SYSTEM targetId:identifier isTop:YES];
                        }
                        
                        [temList addObject:model];
                        // 从数据库获取完成 发送信号 继续下一次取值
                        dispatch_semaphore_signal(self.sema);
                        BBLog(@"BigBang- CrashInfo 数据库 信号量+1");
                        
                    }else {
                        
                        //  本地不存在 调接口请求数据 这里是一个网络请求操作，无论请求成功与否，都要将信号量恢复
                        [[BBDataManager manager] asyncGetUserInfoWithUserID:identifier result:^(NSDictionary *userInfo) {
                            
                            BBUserInfo *newUserInfo = [[BBUserInfo alloc] initWithUserID:identifier userName:[NSString stringWithFormat:@"%@",userInfo[@"nickname"]] avatar:[NSString stringWithFormat:@"%@",userInfo[@"avatar"]] level:[NSString stringWithFormat:@"%@",userInfo[@"level"]]];
                            BBConversationListModel *model = [[BBConversationListModel alloc] initWithRCMessage:message userInfo:newUserInfo];
                            model.unreadMessageCount = conversation.unreadMessageCount;
                            
                            if ([conversation conversationType] == ConversationType_SYSTEM) {
                                [[BBConversationManager manager] setConversationToTop:ConversationType_SYSTEM targetId:identifier isTop:YES];
                            }
                            
                            [temList addObject:model];
                            
                            BBLog(@"BigBang- CrashInfo 网络请求 信号量+1");
                            dispatch_semaphore_signal(self.sema);
                            [BBUserTable saveUserInfoWithModel:newUserInfo];
                            
                        } error:^(NSError *error){
                            
                            BBLog(@"BigBang- CrashInfo 网络请求失败 信号量+1");
                            dispatch_semaphore_signal(self.sema);
                            
                        }];
                    }
                }
            }];
            // 结束操作后，将锁打开，以供后面的访问
            if (handler) {
                if (temList.count) {
                    dispatch_async(dispatch_get_main_queue(), ^{
                        handler(temList);
                    });
                    BBLog(@"BigBang- CrashInfo 获取数据完毕，回传显示  当前线程--%@ ",[NSThread currentThread]);
                }
            }
            [self.lock finish];
        }else {
            if (handler) {
                handler(nil);
            }
            [self.lock finish];
        }
    });
    
}else {
    BBLog(@"BigBang- 抢夺资源中，等待当前获取资源结束继续进行");
}
　}

 

上面的条件锁的文件如下： .h 文件 这里的 < > 不能用，请自行替换 ‘ 为 < >
使用方法为：

self.lock = [[BBResourceLock alloc] init];
if ([self.lock obtain]) {
// 各种任务
…

// 结束后将锁关闭
[self.lock finish];

}

具体的文件如下：
import ‘Foundation/Foundation.h’

@interface BBResourceLock : NSObject

/**

• 获取资源，如果资源已经被获取过则返回 NO，否则返回 YES
• @return 如果资源已经被获取过则返回 NO，否则返回 YES
  */
  -(BOOL)obtain;

/**
等待资源可用
如果资源没有被获取过 则获取资源并返回YES
如果资源被其他线程获取则等待，直到资源可用并返回 YES
如果资源是被同一线程获取则不会等待，并且返回NO
@return YES/NO
*/
-(BOOL)wait;

/**
释放占用的资源
*/
-(void)finish;

@end

.m文件
#import “BBResourceLock.h”

@interface BBResourceLock ()
/** 当前线程 /
@property (nonatomic,strong) NSThread mOccupiedThread;
/ 条件 */
@property (nonatomic,strong) NSCondition *mCondition;
@end

@implementation BBResourceLock

• (instancetype)init
  {
  self = [super init];
  if (self) {
  self.mCondition = [[NSCondition alloc] init];
  }
  return self;
  }

/**

• 获取资源，如果资源已经被获取过则返回 NO，否则返回 YES

• @return 如果资源已经被获取过则返回 NO，否则返回 YES
  */
  -(BOOL)obtain {

  @synchronized(self.mCondition) {
  if (self.mOccupiedThread) {
  return NO;
  }else {
  self.mOccupiedThread = [NSThread currentThread];
  return YES;
  }
  }
  }

/**
等待资源可用
如果资源没有被获取过 则获取资源并返回YES
如果资源被其他线程获取则等待，直到资源可用并返回 YES
如果资源是被同一线程获取则不会等待，并且返回NO
@return YES/NO
*/
-(BOOL)wait {
@synchronized (self.mCondition) {
if (!self.mOccupiedThread) {
return [self obtain];
}
if (self.mOccupiedThread == [NSThread currentThread]) {
return YES;
}
[self.mCondition wait];
return [self obtain];
}
}

/**
释放占用的资源
*/
-(void)finish {
@synchronized(self.mCondition) {
self.mOccupiedThread = nil;
[self.mCondition signal];
}
}

@end

以上就是关于for循环的异步任务处理