如何正确实现NWConnection.receive()以通过UDP套接字接收数据？

一、递归调用导致内存崩溃的解决 #

递归调用NWConnection.receive()时，闭包会强引用self形成循环引用，加上每次递归都会生成新的闭包实例，内存无法及时释放，最终导致崩溃。

解决方式：用弱引用+闭包内重复发起接收的非递归写法，避免循环引用和内存堆积。核心是在接收完成的闭包中，用[weak self]捕获self，处理完数据后再次调用接收方法，实现持续监听。

二、NWConnection.receiveMessage()的正确用法 #

根据苹果文档，receiveMessage()确实是调用一次就等待一条完整UDP消息，触发完成处理程序。但要持续接收的话，必须在处理完当前消息后，再次调用receiveMessage()，否则只会接收一次。

之前尝试无效大概率是因为只调用了一次，没有在处理闭包内循环发起接收请求。

三、控制台警告解析 #

• Error creating the CFMessagePort needed to communicate with PPT.：和你的UDP连接逻辑无关，是系统组件与PPT通信的端口创建失败，直接忽略即可。
• nw_connection_set_queue called after nw_connection_start：这个是关键错误！你在调用connection.start()之后才设置队列（setQueue()），必须在启动连接之前完成队列配置。

修复后的UDPConnection示例代码 #

import Network

class UDPConnection {
    private let connection: NWConnection
    private let queue = DispatchQueue(label: "UDPConnectionQueue")
    
    init(host: String, port: UInt16) {
        let hostEndpoint = NWEndpoint.hostPort(host: NWEndpoint.Host(host), port: NWEndpoint.Port(rawValue: port)!)
        self.connection = NWConnection(host: hostEndpoint.host, port: hostEndpoint.port, using: .udp)
        
        // 必须在start之前配置状态监听
        connection.stateUpdateHandler = { [weak self] state in
            guard let self = self else { return }
            switch state {
            case .ready:
                print("UDP连接已就绪")
                self.listen()
            case .failed(let error):
                print("UDP连接失败: \(error)")
            case .cancelled:
                print("UDP连接已取消")
            default:
                break
            }
        }
        
        // 启动连接并绑定队列
        connection.start(queue: queue)
    }
    
    private func listen() {
        connection.receiveMessage { [weak self] data, context, isComplete, error in
            guard let self = self else { return }
            
            if let error = error {
                print("接收数据错误: \(error)")
                // 出错后尝试重新监听
                self.listen()
                return
            }
            
            if let data = data, !data.isEmpty {
                print("收到UDP数据: \(String(data: data, encoding: .utf8) ?? "无法解析")")
            }
            
            // 处理完当前消息后，再次发起接收实现持续监听
            self.listen()
        }
    }
    
    func send(data: Data) {
        connection.send(content: data, completion: .contentProcessed { error in
            if let error = error {
                print("发送数据错误: \(error)")
            }
        })
    }
    
    func cancel() {
        connection.cancel()
    }
}

关键修复点 #

1. 队列初始化后，在connection.start()之前完成状态监听配置，通过start(queue:)直接绑定队列，避免启动后再设置队列的错误。
2. listen()方法中用[weak self]捕获self，彻底避免循环引用。
3. 接收完成闭包末尾再次调用listen()，实现持续接收消息，替代递归写法，内存可正常释放。

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