如何在 Go 中测试 io.Reader 上的 EOF？

Question

Go 的 io.Reader 文档指出 Read() 可以同时 return 非零值 n 和 io.EOF。不幸的是，File 的 Read() 方法无法做到这一点。

当到达 EOF 时仍然可以读取一些字节，文件 return 的读取方法非零 n 和 nil 错误。只有当我们已经在文件末尾尝试读取时，我们才返回零 n 和 io.EOF 作为错误。

我找不到一个简单的方法来测试是否在不尝试从文件中读取数据的情况下达到 EOF。如果我们使用 0 字节的缓冲区执行 Read()，我们会返回零 n 和 nil 错误，尽管我们在文件末尾。

为了避免最后一次读取，我找到的唯一解决方案是自己跟踪文件中剩余要读取的字节数。有更简单的解决方案吗？

Answer 1

您可以创建一个新类型，用于跟踪到目前为止读取的字节数。然后，在 EOF 检查时，您可以将预期读取的字节数与实际读取的字节数进行比较。这是一个示例实现。 eofReader 跟踪读取的字节数并将其与文件大小进行比较，以防基础类型是文件：

package main

// ... imports 

// eofReader can be checked for EOF, without a Read. 
type eofReader struct {
    r     io.Reader
    count uint64
}

// AtEOF returns true, if the number of bytes read equals the file size.
func (r *eofReader) AtEOF() (bool, error) {
    f, ok := r.r.(*os.File)
    if !ok {
        return false, nil
    }
    fi, err := f.Stat()
    if err != nil {
        return false, err
    }
    return r.Count() == uint64(fi.Size()), nil
}

// Read reads and counts.
func (r *eofReader) Read(buf []byte) (int, error) {
    n, err := r.r.Read(buf)
    atomic.AddUint64(&r.count, uint64(n))
    return n, err
}

// Count returns the count.
func (r *eofReader) Count() uint64 {
    return atomic.LoadUint64(&r.count)
}

您可以通过将任何 reader 包装在 eofReader 中来使用此类型：

func main() {
    f, err := os.Open("main.go")
    if err != nil {
        log.Fatal(err)
    }

    r := &eofReader{r: f}
    log.Println(r.AtEOF())

    if _, err = ioutil.ReadAll(r); err != nil {
        log.Fatal(err)
    }

    log.Println(r.AtEOF())
}

// 2016/12/19 03:49:35 false <nil>
// 2016/12/19 03:49:35 true <nil>

编码为gist.

如何在 Go 中测试 io.Reader 上的 EOF？

How to test EOF on io.Reader in Go?

io

file

go

eof