如何搜索嵌套数组的自定义对象的NSArray？

// 定义一个自定义对象
class CustomObject {
    var name: String
    
    init(name: String) {
        self.name = name
    }
}

// 递归搜索嵌套数组的函数
func searchCustomObject(in array: [Any]) -> [CustomObject] {
    var result: [CustomObject] = []
    
    for item in array {
        if let customObject = item as? CustomObject {
            result.append(customObject)
        } else if let nestedArray = item as? [Any] {
            let nestedResult = searchCustomObject(in: nestedArray)
            result.append(contentsOf: nestedResult)
        }
    }
    
    return result
}

// 示例用法
let array: [Any] = [
    CustomObject(name: "Object 1"),
    CustomObject(name: "Object 2"),
    [
        CustomObject(name: "Object 3"),
        CustomObject(name: "Object 4"),
        [
            CustomObject(name: "Object 5"),
            CustomObject(name: "Object 6")
        ]
    ]
]

let result = searchCustomObject(in: array)
for object in result {
    print(object.name)
}