swift - 如何从终端检测 FileHandle.standardInput 的文件结尾

Question

我有一个从 FileHandle.standardInput 读取的 Swift 程序(在 Objective-C 中，这将是 +[NSFileHandle fileHandleWithStandardInput]。它应该在它命中时终止读取输入流上的文件结尾，但是当我使用终端(在 macOS Sierra 上)作为输入运行它时，当我按 Ctrl+D 时它没有检测到文件结尾。

这是我正在做的一个简化示例。这个程序简单地从标准输入中读取并将读取的内容写入标准输出:

#!/usr/bin/swift

import Foundation

let input = FileHandle.standardInput
let output = FileHandle.standardOutput

let bufferSize = 1024

var data = input.readData(ofLength: bufferSize)
while data.count > 0 {
    output.write(data)
    data = output.readData(ofLength: bufferSize)
}

我希望 readData(ofLength:) 在到达文件末尾时返回一个 Data 对象，其 count 为零。

当我使用重定向到标准输入的文件运行时，如下所示:

./echo.swift < foo.txt

它写出文件并终止。

但是，如果我这样运行它:

./echo.swift

然后键入一些文本并按 Ctrl+D，我希望它将 Ctrl+D 视为文件结尾并终止。但它不会那样做。它只是不断地运行和回声线。如果我一遍又一遍地按 Ctrl+D，它最终会终止，但这不是我想要的。

更改 bufferSize 似乎没有帮助。如果将其设置为 1，我会得到相同的行为。

我怀疑我需要在 stdin 文件描述符或终端设备上设置某种缓冲参数，或者捕捉信号或其他东西，但我不知道是什么。

我知道我可以改用 C stdio fread() API，它可以从终端正确检测文件结束条件，或者我可以使用 Swift 的 readLine(_ :) 从标准输入中读取而不用担心文件句柄/描述符，但我想知道是否有一种方法可以使用 FileHandle 或原始文件描述符来执行此操作而无需重新实现C 标准输入输出。

---

更新:在花了一个小时查看 Apple 的 LibC source 之后，我得出结论“这很复杂”，所以现在我只是在我的程序中使用 fread(..., stdin) 。但我仍然想知道是否有一些简单的方法可以让它与 FileHandle 一起工作。

Answer 1

今天，Swift 5，是这样的……

while (FileHandle.standardInput.availableData.count > 0) {
  FileHandle.standardOutput.write(FileHandle.standardInput.availableData)
}

简单并且几乎像猫程序一样工作，需要为此调整一些东西。

正确的行为更好:

var data: Data
repeat {
  data = FileHandle.standardInput.availableData
  FileHandle.standardOutput.write(data)
} while (data.count > 0)

关于它的文档:

The data currently available through the receiver, up to the the maximum size that can be represented by an NSData object.

If the receiver is a file, this method returns the data obtained by reading the file from the current file pointer to the end of the file. If the receiver is a communications channel, this method reads up to a buffer of data and returns it; if no data is available, the method blocks. Returns an empty data object if the end of file is reached. This method raises NSFileHandleOperationException if attempts to determine the file-handle type fail or if attempts to read from the file or channel fail