Golang ioutil.ReadAll无尽循环

问题描述 投票:-3回答:1

我是Golang的新手。

我试图了解如何使用ioutil.ReadAll(非HTTP响应)。根据源代码(和文档):

// ReadAll reads from r until an error or EOF and returns the data it read.
// A successful call returns err == nil, not err == EOF. Because ReadAll is
// defined to read from src until EOF, it does not treat an EOF from Read
// as an error to be reported.
func ReadAll(r io.Reader) ([]byte, error) {
    return readAll(r, bytes.MinRead)
}

在此示例中,我已经在此处实现了io.Reader(在Go Playground上也为here:]

// go version go1.13.5 darwin/amd64

package main

import (
    "bytes"
    "fmt"
    "io/ioutil"
)

// Thing contains a body
type Thing struct {
    Body []byte
}

// Read reads from body into p
func (t Thing) Read(dst []byte) (n int, err error) {
    // Note: bytes.Reader does return io.EOF
    // https://golang.org/src/bytes/reader.go?s=1154:1204#L30
    reader := bytes.NewReader(t.Body)
    return reader.Read(dst)
}

func main() {
    fmt.Println("Testing bytes")

    thing := new(Thing)
    thing.Body = []byte("Hello World")

    fmt.Println("thing.Body:", string(thing.Body))

    // This works
    buf := make([]byte, len(thing.Body))
    n, err := thing.Read(buf)
    fmt.Println("Amount read:", n)
    if err != nil {
        fmt.Println("Error: ", err)
    }
    fmt.Println("buf:", string(buf))

    // ReadAll runs forever....why?
    buf2, err := ioutil.ReadAll(thing)
    if err != nil {
        fmt.Println("Error:", err)
    }
    fmt.Println("buf2:", buf2)

}

以上,Read实现工作正常。它只是调用bytes.NewReader()并从中读取。但是,在结构上使用ioutil.ReadAll时,它会永远运行(超时),我不明白为什么。最初我以为也许那里没有EOF,但是字节读取器source code确实在这里返回io.EOF:

// Read implements the io.Reader interface.
func (r *Reader) Read(b []byte) (n int, err error) {
    if r.i >= int64(len(r.s)) {
        return 0, io.EOF
    }
    r.prevRune = -1
    n = copy(b, r.s[r.i:])
    r.i += int64(n)
    return
}

我已经在http响应主体上看到了此实现的其他实现,在这些实现中,它们必须在读取主体后明确关闭主体,但我看不到字节读取器上的任何方法来关闭主体。

有人可以帮助我了解这种情况下的情况吗?预先感谢。

go
1个回答
0
投票

每次调用bytes.Reader时,您正在创建一个新的Thing.Read()。每当获得相同的11个字节时,ReadAll()只会继续调用您的Thing.Read()函数。

从您的评论中,我认为您只想创建一次,例如,在“构造函数”中,您不需要存储Body,因为它存储在Bytes.Reader中。

type Thing struct {
    r io.Reader
}

func NewThing(s string) (t Thing) {
    t.r = bytes.NewReader([]byte(s))
    return
}

func (t Thing) Read(dst []byte) (n int, err error) {
    return t.r.Read(dst)
}

但是Thing只是io.Reader的包装,没有任何用途。您也可以直接使用`io.Reader'。

© www.soinside.com 2019 - 2024. All rights reserved.