iOS GCD之dispatch_group_enter和dispatch_group_leave使用

Posted on by

正文

在實際開發中，經常需要在幾個任務全部執行完成之後，在執行後續操作，在 iOS 中，我們可以通過 NSOperation 等達到這一目的。在本篇文章中，我們會介紹如何通過 dispatch_group_enter 和 dispatch_group_leave 來實現這一功能，以及使用過程中遇到的坑。

如何使用

通過一個例子來看下如何使用 dispatch_group_enter 和 dispatch_group_leave：

{
    // 首先 需要創建一個線程組
    dispatch_group_t group = dispatch_group_create();
    // 任務1
    dispatch_group_enter(group);
    NSURLSessionDataTask *task1 = [[NSURLSession sharedSession] dataTaskWithURL:[NSURL URLWithString:@""] completionHandler:^(NSData * _Nullable data, NSURLResponse * _Nullable response, NSError * _Nullable error) {
        NSLog(@"任務1完成");
        dispatch_group_leave(group);
    }];
    [task1 resume];
     // 任務2
    dispatch_group_enter(group);
    NSURLSessionDataTask *task2 = [[NSURLSession sharedSession] dataTaskWithURL:[NSURL URLWithString:@""] completionHandler:^(NSData * _Nullable data, NSURLResponse * _Nullable response, NSError * _Nullable error) {
        NSLog(@"任務2完成");
        dispatch_group_leave(group);
    }];
    [task2 resume];
    // 全部完成
    dispatch_group_notify(group, dispatch_get_main_queue(), ^(){
        NSLog(@"全部完成");
    });
}

任務1和任務2執行完成之後，才會執行全部完成中的任務。

**註意：**在使用時，dispatch_group_enter 和 dispatch_group_leave 需要成對出現，如果 dispatch_group_leave 的調用次數多於 dispatch_group_enter 的調用次數，程序會 crash。相反，雖然不會發生 crash ， 但可能不會達到預期效果。

crash 場景分析

使用場景是，需要異步獲取多個圖片封面，所有都獲取完成後，在執行指定任務，代碼示例如下：

- (void)fetchCovers {
    dispatch_queue_t queue = dispatch_queue_create("com.demo.xxx", DISPATCH_QUEUE_CONCURRENT);
    dispatch_group_t group = dispatch_group_create();
    for (int i = 0; i < 40; ++i) {
        dispatch_group_enter(group);
        dispatch_async(queue, ^{
            [self fetchCoverByPHAsset:asset targetSize:CGSizeMake(200, 200) resultHandler:^(UIImage * _Nonnull, NSDictionary * _Nonnull, BOOL) {
            dispatch_group_leave(group);
        }];
        });
    }
    dispatch_group_notify(group, dispatch_get_main_queue(), ^(){
        // 全部完成後 執行指定任務
    });
}
- (void)fetchCover:(PHAsset *)asset targetSize:(CGSize)targeSize resultHandler:(void (^)(UIImage * _Nonnull, NSDictionary * _Nonnull, BOOL))resultHandler {
    PHImageRequestOptions *options = [[PHImageRequestOptions alloc] init];
    if (@available(iOS 14.0, *)) {
        options.version = PHImageRequestOptionsVersionCurrent;
    }
    options.networkAccessAllowed = YES;
    [[PHImageManager defaultManager] requestImageForAsset:asset
                                               targetSize:targeSize
                                              contentMode:PHImageContentModeAspectFill
                                                  options:options
                                            resultHandler:^(UIImage * _Nullable result, NSDictionary * _Nullable info) {
        if (resultHandler) {
            resultHandler(result, info, [[info objectForKey:PHImageResultIsDegradedKey] boolValue]);
        }
    }];
}

這裡有個小 tips : 在使用 requestImageForAsset 獲取圖片時，如果 options 的 deliveryMode 屬性使用默認值，在異步獲取圖片時，其回調可能會走2次。解決方案是將其顯示設置為 PHImageRequestOptionsDeliveryModeHighQualityFormat 或 PHImageRequestOptionsDeliveryModeFastFormat。

    options.deliveryMode = PHImageRequestOptionsDeliveryModeFastFormat; // 或 PHImageRequestOptionsDeliveryModeHighQualityFormat

因為獲取圖片封面的回調可能會走 2 次，從而導致 dispatch_group_leave 調用次數多於 dispatch_group_enter 的調用次數，因此可能會發生 crash。

源碼實現

  • dispatch_group_enter
void 
dispatch_group_enter(dispatch_group_t dg)
{
    uint32_t old_bits = os_atomic_sub_orig2o(dg, dg_bits,
    		DISPATCH_GROUP_VALUE_INTERVAL, acquire);
    uint32_t old_value = old_bits & DISPATCH_GROUP_VALUE_MASK;
    if (unlikely(old_value == 0)) {
    	_dispatch_retain(dg); // <rdar://problem/22318411>
    }
    if (unlikely(old_value == DISPATCH_GROUP_VALUE_MAX)) {
    	DISPATCH_CLIENT_CRASH(old_bits,
    			"Too many nested calls to dispatch_group_enter()");
    }
}
  • dispatch_group_leave
void 
dispatch_group_leave(dispatch_group_t dg)
{   
    uint64_t new_state, old_state = os_atomic_add_orig2o(dg, dg_state,
    		DISPATCH_GROUP_VALUE_INTERVAL, release);
    uint32_t old_value = (uint32_t)(old_state &amp; DISPATCH_GROUP_VALUE_MASK);
    if (unlikely(old_value == DISPATCH_GROUP_VALUE_1)) {
    	old_state += DISPATCH_GROUP_VALUE_INTERVAL;
    	do {
    		new_state = old_state;
    		if ((old_state &amp; DISPATCH_GROUP_VALUE_MASK) == 0) {
    			new_state &amp;= ~DISPATCH_GROUP_HAS_WAITERS;
    			new_state &amp;= ~DISPATCH_GROUP_HAS_NOTIFS;
    		} else {
    			new_state &amp;= ~DISPATCH_GROUP_HAS_NOTIFS;
    		}
    		if (old_state == new_state) break;
    	} while (unlikely(!os_atomic_cmpxchgv2o(dg, dg_state,
    			old_state, new_state, &amp;old_state, relaxed)));
    	return _dispatch_group_wake(dg, old_state, true);
    }
    if (unlikely(old_value == 0)) {
    	DISPATCH_CLIENT_CRASH((uintptr_t)old_value,
    			"Unbalanced call to dispatch_group_leave()");
    }
}

以上就是iOS GCD之dispatch_group_enter和dispatch_group_leave使用的詳細內容，更多關於iOS GCD使用的資料請關註WalkonNet其它相關文章！

推薦閱讀：