迹忆客 专注技术分享

在本文中，我们将了解如何在 Go (Golang) 中将字节切片转换为 io.Reader。 字节切片只是 Go 中字节的“动态”数组（切片），例如可以表示内存中的任何数据，如文件内容、通过网络发送的数据等……将字节切片转换为可能很有用 一个 io.Reader 因为 io.Reader 允许我们将它与程序的其他部分和 Go 标准库组合起来，因为它实现了 Read 方法。

字节切片到 io.Reader

让我们看一个关于如何在 Go 中将字节切片转换为 io.Reader 的示例

package main

import (
  "bytes"
  "log"
)

func main() {
  data := []byte("this is some data stored as a byte slice in Go Lang!")
  
  // convert byte slice to io.Reader
  reader := bytes.NewReader(data)
  
  // read only 4 byte from our io.Reader
  buf := make([]byte, 4)
  n, err := reader.Read(buf)
  if err != nil {
    log.Fatal(err)
  }
  log.Println(string(buf[:n]))
}

预期输出如下

2023/01/03 07:53:52 this

尝试自己在 Go Playground 上运行它 https://play.golang.org/p/pzTj9J8aJyL

io.Reader 到字节片

让我们看一个例子，看看我们现在如何在 Go 中将 io.Reader 转换为字节切片

package main

import (
  "bytes"
  "log"
  "strings"
)

func main() {
  ioReaderData := strings.NewReader("this is some data stored as a byte slice in Go Lang!")
  
  // creates a bytes.Buffer and read from io.Reader
  buf := &bytes.Buffer{}
  buf.ReadFrom(ioReaderData)
  
  // retrieve a byte slice from bytes.Buffer
  data := buf.Bytes()
  
  // only read the first 4 bytes
  log.Println(string(data[:4]))
}

预期输出如下

2023/01/03 07:55:15 this

尝试自己在 Go Playground 上运行它 https://play.golang.org/p/TPVm7Y7XIus