TaskBag

TaskBag.swift

public actor TaskBag {
    private var tasks: [UUID: Task<Void, Never>] = [:]
    func add(_ task: Task<Void, Never>) {
        let id = UUID()
        tasks[id] = Task { [weak self] in
            await task.value
            await self?.remove(id: id)
        }
    }

    private func remove(id: UUID) {
        tasks.removeValue(forKey: id)
    }

    private static func shutdown(tasks: some Collection<Task<Void, Never>>) async {
        for task in tasks {
            task.cancel()
        }
        for task in tasks {
            await task.value
        }
    }
    public func shutdown() async {
        await Self.shutdown(tasks: tasks.values)
    }

    deinit {
        Task { [tasks] in
            await Self.shutdown(tasks: tasks.values)
        }
    }
}