[iOS] 기존 비동기 함수를 Swift Concurrency로 변경하기

Continuation은 프로그램 상태의 불투명한 표현입니다. 비동기 코드에서 연속(continuation)을 만들려면 withCheckedContinuation(function:_:), withCheckedThrowingContinuation(function:_:) 와 같은 코드를 호출합니다. 비동기 태스크를 재개하려면, resume(returning:), resume(throwing:), resume(with:), resume() 메소드를 호출합니다.

continue에서 두 번 이상 resume을 호출하면 되지 않습니다. 또한, resume을 호출하지 않으면 Task가 무기한 일시 중단된 상태로 유됩니다.

Continuation를 이용하여 기존 비동기 코드(Closure)를 async로 변경해봅시다.

func request(completion: ((String) -> Void)) {
    completion("Hello")
}

func run() {
    request(completion: {
        print($0)
    })
}

위와 같은 Closure 코드를 withCheckedContinuation(function:_:) 함수를 이용해 async 코드로 만들어봅시다.

func requestAsync() async -> String {
    return await withCheckedContinuation { continuation in
        request(completion: {
            continuation.resume(returning: $0)
            // or 
            // continuation.resume(with: .success($0))
        })
    }
}

func run() {
    Task {
        let result = await requestAsync()
        print(result)
    }
}

/** Output
Hello
*/

만약 에러를 반환해야 하는 경우, 두 가지 방안이 있습니다. 에러를 던지거나, Result에 에러를 보내는 방식입니다.

첫 번째로, 에러를 던지는 방식의 코드를 작성해봅시다.

func requestAsync() async throws -> String {
    return try await withCheckedThrowingContinuation{ continuation in
        request(completion: { _ in
            continuation.resume(throwing: NSError(domain: "sample", code: 1))
        })
    }
}

func run() {
    Task {
        do {
            let result = try await requestAsync()
            print(result)
        } catch {
            print(error)
        }
    }
}
/** Output
Error Domain=error Code=1 "(null)"
*/

두 번째로, Result에 에러를 담아서 보내는 방식입니다.

func requestAsync() async -> Result<String, NSError> {
    return await withCheckedContinuation { continuation in
        request(completion: { result in
            continuation.resume(returning: .failure(NSError(domain: "sample", code: 1)))
        })
    }
}

func run() {
    Task {
        let result = await requestAsync()
        switch result {
        case .success(let result): print(result)
        case .failure(let error): print(error)
        }
    }
}
/** Output
Error Domain=error Code=1 "(null)"
*/