在 Go 中传递响应体 (response.Body) 的有效方法是什么？

Question

如果我有一些代码，例如下面的示例，它从链接中获取图像然后将其保存到磁盘，那么传递图像数据的最佳方法是什么？

我想过使用ioutil.ReadAll(res.Body)将它转换为[]byte但传递它似乎很昂贵，尽管我无法从文档中判断它是否返回切片或数组。我也尝试返回一个指针res.Body，一个*io.ReadCloser类型，但我无法弄清楚如何正确地调用.Close()上的指示的方法来实现。

我知道将保存代码移到FetchImage函数中可能是解决此问题的最简单方法，但如果可能，我希望将这些部分分开。

type ImageData struct {

    Data      io.ReadCloser

    Name      string

}

func FetchImage(url string) (io.ReadCloser, error) {

    res, err := http.Get(url)

    if err != nil {

        return nil, err

    }

    return res.Body, nil

}

func Save(data *ImageData) error {

    defer data.Data.Close()

    file, err := os.Create(data.Name)

    defer file.Close()

    if err != nil {

        return err

    }

    _, err = io.Copy(file, data.Data)

    if err != nil {

        return err

    }

    return nil

}

func main() {

    body, err := fetcher.FetchImage("https://imgur.com/asdf.jpg")

    if err != nil {

        panic(err)

    }

    imageData := ImageData{body, "asdf.jpg"}

    saver := Saver{config.BaseDir, 1}

    err = saver.Save(&imageData)

    if err != nil {

        panic(err)

    }

}

此外，我对 Go 很陌生，所以如果这段代码中有任何看起来不好的地方，请告诉我。

Answer 1

使用ioutil.ReadAll。该函数返回一个字节片。切片可以有效地传递。切片是指向后备数组、长度和容量的指针。type ImageData struct {    Data      []byte    Name      string}func FetchImage(url string) ([]byte, error) {    res, err := http.Get(url)    if err != nil {        return nil, err    }    defer resp.Body.Close()    if res.StatusCode != 200 {        return nil, fmt.Errorf("%s: %d", url, res.StatusCode)    }    return ioutil.ReadAll(resp.Body)}func Save(data *ImageData) error {    file, err := os.Create(data.Name)    if err != nil {        return err    }    defer file.Close()    _, err := file.Write(data.Data)    return err}您也可以传递响应主体，但要小心。响应主体必须关闭才能释放底层连接。问题中的代码确实关闭了响应主体，但很难看到，因为响应主体传递到关闭它的函数。

Answer 2

在 Go 中，所有数组都有一个大小说明符，看起来像[8]byte. 切片不会有这个，看起来像[]byte。切片内部只存储一个指向数据的指针、切片的长度和切片的容量。在 64 位系统上这只是 24 个字节，因此您不必担心传递它。但是，我建议返回一个*ImageDatafromFetchImage因为它已经有像图像名称这样的元数据。至于为什么不能使用指向接口的指针，SO 上有一篇文章解释了原因。另外，在Save你defer file.Close()检查错误之前。您应该交换它，因为如果出现错误文件将为零，并且可能会出现段错误。