Swift Concurrency で async 関数とキャンセル処理を実装した際の試行錯誤と結果

// 基本的な構造
class UserDataStoreImpl: UserDataStore {
  func fetchUser(id: Int) async throws -> User {
      let response = try await networkRequest(route: .getUser(id: id), parameters: nil)
      return try processResponse<User>(from: response)
  }
}

private func networkRequest<P: Encodable>(
    route: APIRouter,
    parameters: P?
) async throws -> AFDataResponse<Data> {
    return try await withCheckedThrowingContinuation { continuation in
        let encoder = URLEncodedFormParameterEncoder(
            encoder: URLEncodedFormEncoder(nilEncoding: .dropKey)
        )

        AF.request(
            route,
            method: route.method,
            parameters: parameters,
            encoder: encoder
        )
        .responseData { response in
            continuation.resume(with: .success(response))
        }
    }
}

private func networkRequest<P: Encodable>(
        route: APIRouter,
        parameters: P?
    ) async throws -> AFDataResponse<Data> {

        var dataRequest: DataRequest?

        return try await withTaskCancellationHandler {
            try await withCheckedThrowingContinuation { continuation in

                // 省略：encoderの設定

                // リクエスト参照を保持
                dataRequest = AF.request(...)

                dataRequest.responseData { response in
                    // 省略：キャンセルエラー処理

                    continuation.resume(returning: response)
                }
            }
        } onCancel: {
            // `Task.cancel()` が呼ばれた瞬間に実行される

            dataRequest.cancel() // <- ワーニング: Reference to captured var 'dataRequest' in concurrently-executing code; this is an error in the Swift 6 language mode
        }
    }

Reference to captured var 'dataRequest' in concurrently-executing code; 
this is an error in the Swift 6 language mode

// ワーニングが出たコード（Swift 6 でエラーになる）
private func networkRequest() async throws -> AFDataResponse<Data> {
    var dataRequest: DataRequest?  // ← 可変値変数(var)

    return try await withTaskCancellationHandler {
        // コンテキスト1: メイン処理
        dataRequest = AF.request(...)  // ← ここでキャプチャ
    } onCancel: {
        // コンテキスト2: キャンセル処理
        dataRequest?.cancel()  // ← 異なるコンテキストから同じ変数にアクセス、データ競合の可能性ある
    }
}

// エラーが発生せず、正常に実行できるパターン
private func networkRequest<P: Encodable>(
    route: APIRouter,
    parameters: P?
) async throws -> AFDataResponse<Data> {
    let encoder = URLEncodedFormParameterEncoder(
        encoder: URLEncodedFormEncoder(nilEncoding: .dropKey)
    )

    let headers = HTTPHeaders([
        HTTPHeader(name: "Content-Type", value: "multipart/form-data")
    ])

    // 不変参照 = データ競合なし
    let dataRequest = AF.request(
        route,
        method: route.method,
        parameters: parameters,
        encoder: encoder,
        headers: headers
    )

    return try await withTaskCancellationHandler {
        try await withCheckedThrowingContinuation { continuation in
            dataRequest.responseData { response in
                // 省略：キャンセルエラー処理
                continuation.resume(returning: response)
            }
        }
    } onCancel: {
        dataRequest.cancel()
    }
}

extension DataRequest {
    /// async/await対応メソッド（Taskのキャンセル対応）
    func responseDataAsync() async throws -> AFDataResponse<Data> {
        return try await withTaskCancellationHandler {
            try await withCheckedThrowingContinuation { continuation in
                self.responseData { response in
                    // キャンセルエラーはシステム制御フローの一部なのでここで処理する
                    if let error = response.error,
                       error.isExplicitlyCancelledError {
                        continuation.resume(throwing: CancellationError())
                        return
                    }

                    // パースエラーなどのビジネスロジックエラーは呼び出し元で処理する
                    continuation.resume(returning: response)
                }
            }
        } onCancel: {
            self.cancel()
        }
    }
}

private func networkRequest<P: Encodable>(
    route: APIRouter,
    parameters: P?
) async throws -> AFDataResponse<Data> {
    let request = AF.request(...)

    return try await request.responseDataAsync()  // ← 共通化された処理
}

// 使用例
func fetchUser(id: Int) async throws -> User {
    let response = try await apiRequest(
        route: .getUser(id: id),
        parameters: nil
    )
    return try processResponse<User>(from: response)  // ← 既存のデコード処理
}

class UserListPresenter {
    private var loadingTask: Task<Void, Never>?

    func loadUsers() {
        // 前のタスクをキャンセル
        loadingTask?.cancel()

        loadingTask = Task {
            do {
                let users = try await userRepository.fetchUsers()

                // キャンセルされてたら画面更新しない
                // 明示的にチェックしなければキャンセルは無視される
                guard !Task.isCancelled else { return }

                await MainActor.run {
                    view.showUsers(users)
                }
            } catch {
                // キャンセルエラーは特別扱い
                guard !Task.isCancelled else { return }

                await MainActor.run {
                    view.showError(error)
                }
            }
        }
    }
}