Stripe-iOS 中 SwiftUI 调用 PaymentSheet 时的内存泄漏问题解析

问题背景

在 Stripe-iOS 项目中，开发者使用 SwiftUI 集成支付功能时遇到了一个关于任务延续(Task Continuation)的内存泄漏问题。具体表现为当调用 paymentConfirmationSheet 方法时，控制台会输出警告："SWIFT TASK CONTINUATION MISUSE: fetchIntentClientSecretFromMerchant(intentConfig:paymentMethod:shouldSavePaymentMethod:) leaked its continuation!"。

技术分析

这个问题本质上是因为在使用 Swift 的 withCheckedThrowingContinuation 时，没有正确处理异步回调的完成情况。在 Stripe 的支付流程中，开发者需要实现一个确认处理器(confirmHandler)，这个处理器必须确保在所有执行路径上都调用 intentCreationCallback 回调。

核心问题点

1. 任务延续泄漏：当使用 Swift 并发模型中的 withCheckedThrowingContinuation 时，系统会创建一个临时的延续对象。如果这个延续没有被正确恢复(resume)或取消，就会导致内存泄漏。

2. 条件分支遗漏：在示例代码中，确认处理器只处理了 if let cardId, payMoney > 1 的情况，而没有处理其他分支路径。这意味着在某些条件下，intentCreationCallback 不会被调用，导致延续泄漏。

解决方案

正确的实现方式应该确保在所有可能的执行路径上都调用 intentCreationCallback。以下是关键改进点：

func confirmHandler(_ paymentMethod: STPPaymentMethod,
                   _ shouldSavePaymentMethod: Bool,
                   _ intentCreationCallback: @escaping (Result<String, Error>) -> Void) {
    guard let cardId = paymentMethod.stripeId, payMoney > 1 else {
        intentCreationCallback(.failure(ConfirmHandlerError.invalidParameters))
        return
    }
    
    Task {
        do {
            let result = try await createPaymentIntent(cardId: cardId, amount: payMoney)
            intentCreationCallback(.success(result))
        } catch {
            intentCreationCallback(.failure(error))
        }
    }
}

最佳实践建议

1. 全面覆盖所有分支：确保确认处理器中的每个可能的执行路径都会调用回调函数。

2. 错误处理：为不同的错误情况定义明确的错误类型，便于调试和问题追踪。

3. 异步操作封装：使用 Task 包装异步操作时，确保正确处理成功和失败两种情况。

4. 参数验证：在处理器开始处验证所有必要参数，尽早失败并返回明确的错误信息。

总结

这个问题展示了在使用 Swift 并发模型与第三方支付 SDK 集成时需要特别注意的点。Stripe 的支付流程依赖于开发者正确实现回调机制，任何遗漏都可能导致不可预期的行为。通过确保所有执行路径都正确处理回调，可以避免任务延续泄漏的问题，同时提高代码的健壮性和可维护性。

对于开发者来说，理解 Swift 并发模型的工作原理以及第三方 SDK 的回调机制是避免这类问题的关键。在集成支付功能时，建议仔细阅读文档并充分测试各种边界情况。