Swift中如何将stdout重定向至字符串？现有实现遇阻求助

你的代码核心问题在于异步读取的时机没同步，加上管道处理和stdout恢复的方式有误，导致无法捕获到print("Captured")的输出，具体问题和修复方案如下：

原代码的核心问题 #

1. 异步回调未执行就读取结果：readabilityHandler是在后台队列异步触发的，你在print("Captured")后立刻恢复stdout并打印outString，此时回调还没来得及处理管道数据，所以outString还是初始值。
2. 管道写端未关闭：管道的写端一直处于打开状态时，读取端无法判断数据流是否结束，可能不会触发完整的读取回调。
3. 强制解包风险：String(data:encoding:)可能返回nil（比如数据不是UTF-8编码），强制解包会导致程序崩溃。
4. stdout恢复方式不可靠：用freopen恢复stdout容易和之前的文件描述符操作冲突，不如直接保存并恢复原始的文件描述符。

修正后的代码 #

import Foundation

let outPipe = Pipe()
var outString = "Initial"

// 安全处理管道数据读取，避免强制解包
let semaphore = DispatchSemaphore(value: 0)
outPipe.fileHandleForReading.readabilityHandler = { fileHandle in
    guard let dataString = String(data: fileHandle.availableData, encoding: .utf8) else {
        print("无法将管道数据转换为字符串")
        fileHandle.closeFile()
        semaphore.signal()
        return
    }
    outString += dataString
    
    // 无更多数据时关闭读取端并触发信号量
    if fileHandle.availableData.isEmpty {
        fileHandle.closeFile()
        semaphore.signal()
    }
}

print("Starting")

// 关闭stdout缓冲，确保输出实时写入管道
setvbuf(stdout, nil, _IONBF, 0)
// 保存原始stdout描述符，用于后续恢复
let originalStdout = dup(STDOUT_FILENO)
// 将stdout重定向到管道写端
dup2(outPipe.fileHandleForWriting.fileDescriptor, STDOUT_FILENO)

// 输出需要捕获的内容
print("Captured")
print("测试多行捕获")

// 关闭管道写端，告知读取端数据流结束
outPipe.fileHandleForWriting.closeFile()

// 等待异步读取完成
semaphore.wait()

// 恢复原始stdout
dup2(originalStdout, STDOUT_FILENO)
close(originalStdout)

// 打印最终捕获的内容
print(outString)

关键修改说明 #

• 用信号量同步异步操作：通过DispatchSemaphore等待管道读取完成，确保所有输出都被捕获后再执行后续逻辑。
• 保存并恢复原始stdout：用dup复制原始的STDOUT文件描述符，后续通过dup2恢复，比freopen更可靠。
• 关闭管道写端：写完需要捕获的内容后关闭写端，让读取端明确知道数据流结束，触发完整的读取回调。
• 安全处理字符串转换：添加可选值绑定，处理编码转换失败的情况，避免程序崩溃。

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