乐趣区

关于golang:如何在-Go-中将-byte-转换为-ioReader

jiezi

原文链接： [如何在 Go 中将 []byte 转换为 io.Reader？](https://mp.weixin.qq.com/s/nF…)

在 stackoverflow 上看到一个问题，题主进行了一个网络申请，接口返回的是 []byte。如果想要将其转换成 io.Reader，须要怎么做呢？

这个问题解决起来并不简单，简略几行代码就能够轻松将其转换胜利。不仅如此，还能够再通过几行代码反向转换回来。

上面听我缓缓给你吹，首先间接看两段代码。

[]byte 转 io.Reader

package main

import (
    "bytes"
    "fmt"
    "log"
)

func main() {data := []byte("Hello AlwaysBeta")

    // byte slice to bytes.Reader, which implements the io.Reader interface
    reader := bytes.NewReader(data)

    // read the data from reader
    buf := make([]byte, len(data))
    if _, err := reader.Read(buf); err != nil {log.Fatal(err)
    }

    fmt.Println(string(buf))
}

输入：

Hello AlwaysBeta

这段代码先将 []byte 数据转换到 reader 中，而后再从 reader 中读取数据，并打印输出。

io.Reader 转 []byte

package main

import (
    "bytes"
    "fmt"
    "strings"
)

func main() {ioReaderData := strings.NewReader("Hello AlwaysBeta")

    // 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 left bytes from 6
    fmt.Println(string(data[6:]))
}

输入：

AlwaysBeta

这段代码先创立了一个 reader，而后读取数据到 buf，最初打印输出。

以上两段代码就是 []byte 和 io.Reader 相互转换的过程。比照这两段代码不难发现，都有 NewReader 的身影。而且在转换过程中，都起到了关键作用。

那么问题来了，这个 NewReader 到底是什么呢？接下来咱们通过源码来一探到底。

源码解析

Go 的 io 包提供了最根本的 IO 接口，其中 io.Reader 和 io.Writer 两个接口最为要害，很多原生构造都是围绕这两个接口开展的。

上面就来别离说说这两个接口：

Reader 接口

io.Reader 示意一个读取器，它将数据从某个资源读取到传输缓冲区。在缓冲区中，数据能够被流式传输和应用。

接口定义如下：

type Reader interface {Read(p []byte) (n int, err error)
}

Read() 办法将 len(p) 个字节读取到 p 中。它返回读取的字节数 n，以及产生谬误时的错误信息。

举一个例子：

package main

import (
    "fmt"
    "io"
    "os"
    "strings"
)

func main() {reader := strings.NewReader("Clear is better than clever")
    p := make([]byte, 4)

    for {n, err := reader.Read(p)
        if err != nil {
            if err == io.EOF {fmt.Println("EOF:", n)
                break
            }
            fmt.Println(err)
            os.Exit(1)
        }
        fmt.Println(n, string(p[:n]))
    }
}

输入：

4 Clea
4 r is
4  bet
4 ter
4 than
4  cle
3 ver
EOF: 0

这段代码从 reader 一直读取数据，每次读 4 个字节，而后打印输出，直到结尾。

最初一次返回的 n 值有可能小于缓冲区大小。

Writer 接口

io.Writer 示意一个编写器，它从缓冲区读取数据，并将数据写入指标资源。

type Writer interface {Write(p []byte) (n int, err error)
}

Write 办法将 len(p) 个字节从 p 中写入到对象数据流中。它返回从 p 中被写入的字节数 n，以及产生谬误时返回的错误信息。

举一个例子：

package main

import (
    "bytes"
    "fmt"
    "os"
)

func main() {
    // 创立 Buffer 暂存空间，并将一个字符串写入 Buffer
    // 应用 io.Writer 的 Write 办法写入
    var buf bytes.Buffer
    buf.Write([]byte("hello world ,"))

    // 用 Fprintf 将一个字符串拼接到 Buffer 里
    fmt.Fprintf(&buf, "welcome to golang !")

    // 将 Buffer 的内容输入到规范输出设备
    buf.WriteTo(os.Stdout)
}

输入：

hello world ,  welcome to golang !

bytes.Buffer 是一个构造体类型，用来暂存写入的数据，其实现了 io.Writer 接口的 Write 办法。

WriteTo 办法定义：

func (b *Buffer) WriteTo(w io.Writer) (n int64, err error)

WriteTo 办法第一个参数是 io.Writer 接口类型。

转换原理

再说回文章结尾的转换问题。

只有某个实例实现了接口 io.Reader 里的办法 Read()，就满足了接口 io.Reader。

bytes 和 strings 包都实现了 Read() 办法。

// src/bytes/reader.go

// NewReader returns a new Reader reading from b.
func NewReader(b []byte) *Reader {return &Reader{b, 0, -1} }
// src/strings/reader.go

// NewReader returns a new Reader reading from s.
// It is similar to bytes.NewBufferString but more efficient and read-only.
func NewReader(s string) *Reader {return &Reader{s, 0, -1} }

在调用 NewReader 的时候，会返回了对应的 T.Reader 类型，而它们都是通过 io.Reader 扩大而来的，所以也就实现了转换。

总结

在开发过程中，防止不了要进行一些 IO 操作，包含打印输出，文件读写，网络连接等。

在 Go 语言中，也提供了一系列规范库来应答这些操作，次要封装在以下几个包中：

  • io：根本的 IO 操作接口。
  • io/ioutil：封装了一些实用的 IO 函数。
  • fmt：实现了 IO 格式化操作。
  • bufio：实现了带缓冲的 IO。
  • net.Conn：网络读写。
  • os.Stdin，os.Stdout：零碎规范输入输出。
  • os.File: 系统文件操作。
  • bytes：字节相干 IO 操作。

除了 io.Reader 和 io.Writer 之外，io 包还封装了很多其余根本接口，比方 ReaderAt，WriterAt，ReaderFrom 和 WriterTo 等，这里就不一一介绍了。这部分代码并不简单，读起来很轻松，而且还能加深对接口的了解，举荐大家看看。

好了，本文就到这里吧。关注我，带你通过问题读 Go 源码。

举荐浏览：

  • 开始读 Go 源码了

激情举荐：

  • 计算机经典书籍（含下载方式）
  • 技术博客： 硬核后端技术干货，内容包含 Python、Django、Docker、Go、Redis、ElasticSearch、Kafka、Linux 等。
  • Go 程序员： Go 学习路线图，包含根底专栏，进阶专栏，源码浏览，实战开发，面试刷题，必读书单等一系列资源。
  • 面试题汇总： 包含 Python、Go、Redis、MySQL、Kafka、数据结构、算法、编程、网络等各种常考题。

参考文章：

  • https://books.studygolang.com…
  • https://www.cnblogs.com/jiuju…
  • https://segmentfault.com/a/11…
退出移动版