在Vapor Swift应用WebSocket关闭回调中是否需使用unowned socket？

好问题！咱们来仔细拆解这个场景，搞清楚要不要用[unowned socket]以及会不会产生内存泄漏：

先理清楚引用关系 #

首先看你的代码结构：

app.webSocket("route") { _, socket in 
    socket.onClose.whenComplete { _ in 
        removeFromWhatever(socket) 
    } 
}

这里的引用链是：socket 持有 onClose（一个Promise对象），onClose 持有 whenComplete 的闭包，而闭包又捕获了 socket——看起来像是形成了循环引用？但别急，关键要看这个循环会不会被打破。

对比onText的设计逻辑 #

你提到onText回调会传入WebSocket参数来避免内存泄漏，这是因为onText的闭包如果直接捕获外部的socket，同样会形成循环，但通过传入参数的方式，闭包里使用的是局部的socket引用，不会捕获外部的socket实例，自然就不会有循环问题。

onClose场景的结论：不需要用[unowned socket] #

回到你的问题，这里不需要添加[unowned socket]，也不会造成长期的内存泄漏，原因有两个：

• 当WebSocket关闭时，Vapor的内部机制会自动清理onClose这个Promise的相关引用——也就是说，onClose会释放对闭包的持有，原本的循环引用链会被打破，socket实例最终会被ARC回收。
• 退一步讲，就算闭包执行前存在短暂的循环引用，当WebSocket关闭后，这个socket实例本身已经不再被系统需要，循环引用带来的内存也会在闭包执行完成后被自动处理，不会造成持续的内存泄漏。

额外建议：如果担心可以用weak替代unowned #

如果实在想避免任何潜在的循环风险，更安全的做法是用[weak socket]，然后在闭包里可选绑定：

socket.onClose.whenComplete { [weak socket] _ in 
    guard let socket = socket else { return }
    removeFromWhatever(socket) 
}

相比unowned，weak是可选类型，不会因为socket提前释放导致崩溃，安全性更高——不过在这个场景下，onClose的闭包只会在socket关闭时触发，此时socket实例还处于有效状态，unowned也不会崩溃，但完全没必要多此一举。

内容的提问来源于stack exchange，提问作者swift-lynx