iOS Socket InputStream阻塞UI及streamDelegate主线程调用问题咨询

嘿，这个坑我之前踩过！NSStream的Delegate回调默认是绑定到创建它的线程的RunLoop上的——如果你是在主线程初始化的流，哪怕后续把相关操作扔到队列里，Delegate还是会乖乖跑回主线程，大量数据过来时自然会卡UI。给你几个实用的解决思路，按复杂度从低到高排序：

1. 最简单：把Delegate里的耗时操作扔去后台 #

如果不想折腾线程和RunLoop，最快速的办法就是让主线程的Delegate只做“触发”，把真正的数据读取、处理逻辑丢到后台队列：

func stream(_ aStream: Stream, handle eventCode: Stream.Event) {
    switch eventCode {
    case .hasBytesAvailable:
        // 主线程只负责启动异步任务，立刻交还控制权
        DispatchQueue.global(qos: .background).async {
            guard let inputStream = aStream as? InputStream else { return }
            let bufferSize = 1024
            var buffer = [UInt8](repeating: 0, count: bufferSize)
            
            // 后台读取并处理数据，不会卡主线程
            while inputStream.hasBytesAvailable {
                let bytesRead = inputStream.read(&buffer, maxLength: bufferSize)
                guard bytesRead > 0 else { break }
                
                // 这里做数据解析、存储等耗时操作...
                
                // 要更新UI的话，再切回主线程
                DispatchQueue.main.async {
                    // 比如更新进度条、显示已接收数据量
                }
            }
        }
    // 处理其他事件（比如流关闭、错误）
    default:
        break
    }
}

这个方法的好处是改动最小，不用动流的初始化逻辑，只要把核心耗时逻辑异步化就行。

2. 更彻底：给Stream单独分配后台线程+RunLoop #

如果希望整个流的生命周期都在后台跑（连Delegate回调都不碰主线程），可以创建一个独立的后台线程，把流绑定到这个线程的RunLoop上：

class StreamManager: NSObject, StreamDelegate {
    private var streamThread: Thread?
    private var inputStream: InputStream?

    func startStream(with url: URL) {
        // 创建后台线程
        streamThread = Thread { [weak self] in
            guard let self = self else { return }
            // 在后台线程初始化流
            self.inputStream = InputStream(url: url)!
            self.inputStream?.delegate = self
            // 把流调度到当前线程的RunLoop
            self.inputStream?.schedule(in: RunLoop.current, forMode: .default)
            self.inputStream?.open()
            
            // 启动RunLoop，让线程保持存活（否则线程会立刻退出）
            RunLoop.current.run()
        }
        streamThread?.start()
    }

    func stopStream() {
        inputStream?.close()
        inputStream?.remove(from: RunLoop.current, forMode: .default)
        inputStream = nil
        // 停止RunLoop，让线程退出
        streamThread?.cancel()
        streamThread = nil
    }

    // Delegate回调会在后台线程执行，完全不影响主线程
    func stream(_ aStream: Stream, handle eventCode: Stream.Event) {
        switch eventCode {
        case .hasBytesAvailable:
            // 直接在后台处理数据，不用额外异步
            guard let inputStream = aStream as? InputStream else { return }
            var buffer = [UInt8](repeating: 0, count: 1024)
            while inputStream.hasBytesAvailable {
                let bytesRead = inputStream.read(&buffer, maxLength: buffer.count)
                // 处理数据...
                
                // 更新UI仍需切回主线程
                DispatchQueue.main.async {
                    // UI更新操作
                }
            }
        default:
            break
        }
    }
}

注意：这个方法要记得在停止流时正确取消RunLoop，避免内存泄漏；另外访问共享数据时要注意线程安全（比如用串行队列同步，或者加锁）。

3. 最优解：改用现代网络API代替NSStream #

NSStream其实是比较老旧的API了，苹果现在更推荐用URLSession——它本身就是异步设计，完全不用手动处理线程和RunLoop，还能自动管理后台任务：

class NetworkManager: NSObject, URLSessionDataDelegate {
    private var session: URLSession!

    override init() {
        super.init()
        // 给Delegate指定后台队列，回调直接在后台执行
        let delegateQueue = OperationQueue()
        delegateQueue.qualityOfService = .background
        session = URLSession(configuration: .default, delegate: self, delegateQueue: delegateQueue)
    }

    func startDownload(with url: URL) {
        let task = session.dataTask(with: url)
        task.resume()
    }

    // 回调在我们指定的后台队列执行，主线程无压力
    func urlSession(_ session: URLSession, dataTask: URLSessionDataTask, didReceive data: Data) {
        // 处理接收到的数据...
        
        // 更新UI切回主线程
        DispatchQueue.main.async {
            // UI更新
        }
    }
}

URLSession不仅线程处理更省心，还支持断点续传、后台下载等高级功能，比NSStream更稳定高效，建议优先考虑迁移。

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