TaskCounter로 병렬 작업 완료 계수 파악하기

 

💡 여러 이미지를 한번에 가져와야하는 작업이 있었다. 이미지들을 가져오는 Task를 비동기로 가져오는 작업을 TaskGroup으로 진행했다. 이미지들을 가져오는 모든 작업 완료 후, 뷰에 보여주는 방식을 사용했고 기다리는 과정에서 Progress를 보여주려 했다. 각각의 작업이 완료된 계수를 추적하는 방법을 찾지 못했고 카운터를 직접 만들어 이미지를 가져오는 작업 뿐만 아니라 다른 작업도 병렬 처리에 사용할 수 있는 Task(Group)Counter를 만들었다.

사용 기술

• Actor → 필수, TaskCounter는 actor 인스턴스
• Combine → RxSwift로 구성해도 상관 없을 것 같다.
• Concurrency

구현 매커니즘

저장 프로퍼티

• count: 작업 완료한 계수를 카운트
• maxCount: 병렬 처리할 작업의 계수
• completed: 모든 작업이 완료 되었는지 알려주는 PassthroughSubject → rxSwift의 PublishSubject

카운팅 관련 메서드) private level

• increment
• reset
• changeMax
• failed()

Task 처리 관련 메서드

• progress: 작업 완료 계수가 올라가면 보내는 Publisher → rxSwift의 Observer
• run: 실제 작업할 데이터 그룹과 각각의 그룹에 실행할 작업을 입력하는 async 메서드

전체 코드

import UIKit
import Combine
actor TaskCounter{
    @Published private(set) var count = 0
    @Published private(set) var maxCount: Int = 0
    private(set) var completed = PassthroughSubject<Bool,Never>()
    private var subscription = Set<AnyCancellable>()
    func changeMax(_ max:Int){ self.maxCount = max }
    private func increment(){
        count += 1
        if maxCount == count{
            count = 0
            completed.send(true)
        }
    }
    private func reset(){ count = 0 }
    private func failed(){ completed.send(false) }
    var progress: AnyPublisher<(Int,Int),Never>{
        $count.combineLatest($maxCount).eraseToAnyPublisher()
    }
}
extension TaskCounter{
    func run<T>(_ results: [T],action:@escaping ((T) async throws ->Void)) async throws{
        subscription.removeAll()
        self.changeMax(results.count)
        self.reset()
        try await withThrowingTaskGroup(of: Void.self) { group in
            for result in results{
                group.addTask {
                    do{
                        try await action(result)
                        await self.increment()
                    }catch{
                        await self.failed()
                    }
                }
            }
            self.completed.sink(receiveValue: {[group] val in
                if !val{ group.cancelAll() }
            }).store(in: &subscription)
            try await group.waitForAll()
        }
    }
}

✅ 병렬로 진행하는 작업에서 Return 하는 값이 존재한다면 return 값의 타입 계수에 맞게 제네릭 타입을 추가한 run 메서드를 작성해서 처리할 수 있을 것이다.