Swift中如何优雅实现多类型子节点N叉树的状态化导航？

针对你遇到的N叉树目录导航类型问题，这里有几个更优雅的解决方案：

1. 给Directory添加专属的目录查找方法 #

在Directory类中新增一个专门查找子目录的方法，直接返回Directory?类型，避免后续类型转换：

class Directory: Node {
    // ... 原有代码 ...

    // 新增：专门查找子目录
    func findDirectory(name: String) -> Directory? {
        return findBy(name: name) as? Directory
    }
}

使用时直接调用这个方法，无需额外转换：

var pwd: Directory? = // 初始目录
pwd = pwd?.findDirectory(name: "myFancyName")

2. 泛型化查找方法 #

重载findBy方法，通过泛型参数指定返回类型，让方法更通用灵活：

class Directory: Node {
    // ... 原有代码 ...

    // 泛型重载：根据指定类型查找子节点
    func findBy<T: Node>(name: String, type: T.Type) -> T? {
        return findBy(name: name) as? T
    }
}

调用时可以指定要查找的类型，比如查找目录：

var pwd: Directory? = // 初始目录
pwd = pwd?.findBy(name: "myFancyName", type: Directory.self)

这种方式还能用来查找文件：let file = pwd?.findBy(name: "note.txt", type: File.self)

3. 安全的可选绑定+类型转换 #

如果不想修改原有方法，推荐使用可选绑定结合类型转换的写法，既安全又清晰，避免强制转换的崩溃风险：

var pwd: Directory? = // 初始目录
if let targetDir = pwd?.findBy(name: "myFancyName") as? Directory {
    pwd = targetDir
}

注意事项 #

• 永远避免使用as!强制转换，除非你100%确定目标节点是Directory类型，否则会在运行时崩溃。
• 上述方案都保持了类型安全，同时简化了导航代码的写法。

内容的提问来源于stack exchange，提问作者Morpheu5