如何在Go中将[]byte转换为io.Reader
在 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中将[]byte转换为io.Reader的文章就介绍到这了,更多相关Go []byte转换为io.Reader内容请搜索猪先飞以前的文章或继续浏览下面的相关文章希望大家以后多多支持猪先飞!
原文出处:https://www.cnblogs.com/alwaysbeta/p/15744469.html
相关文章
- 在程序员中,尤其是go新手,经常听到的一个讨论话题是:如何处理错误,这篇文章主要给大家介绍了关于Go应用中优雅处理Error的一些相关技巧,需要的朋友可以参考下...2021-09-08
Django def clean()函数对表单中的数据进行验证操作
这篇文章主要介绍了Django def clean()函数对表单中的数据进行验证操作,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2020-07-09- 这篇文章主要介绍了golang官方嵌入文件到可执行程序,本文通过示例代码给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下...2021-02-20
- 这篇文章主要介绍了go浮点数转字符串保留小数点后N位解决办法,本文给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下...2020-05-11
- 这篇文章主要介绍了Go语言使用读写OPC详解,图文讲解的很清晰,有感兴趣的同学可以学习下...2021-03-05
- 这篇文章主要介绍了Go项目的目录结构,对基础目录做了讲解,对项目开发中的其它目录也一并做了介绍,需要的朋友可以参考下...2020-05-01
- string与[]byte经常需要互相转化,普通转化会发生底层数据的复制,下面这篇文章主要给大家介绍了关于Go中string与[]byte高效互转的相关资料,文中通过示例代码介绍的非常详细,需要的朋友可以参考下...2021-09-20
- Go 语言提供的基础容器,免不了要查询容器中的数据,那么是如何实现遍历的呢?本文将会介绍几种常用容易的遍历及其使用。感兴趣的可以了解一下...2021-06-13
- 这篇文章主要介绍了创建第一个Go语言程序Hello,Go!本文详细的给出项目创建、代码编写的过程,同时讲解了GOPATH、Go install等内容,需要的朋友可以参考下...2020-05-01
- 这篇文章主要介绍了在Django中使用MQTT的方法,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2021-05-10
- 这篇文章主要介绍了go语言中的Carbon库时间处理,本文给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下...2021-02-05
- 这篇文章主要介绍了go嵌套匿名结构体的初始化详解,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2020-12-16
解决导入django_filters不成功问题No module named 'django_filter'
这篇文章主要介绍了解决导入django_filters不成功问题No module named 'django_filter',具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2020-07-15- 本文主要介绍了Django项目连接MongoDB的三种方法,文中通过示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下...2021-09-27
详解如何使用Docker部署Django+MySQL8开发环境
这篇文章主要介绍了详解如何使用Docker部署Django+MySQL8开发环境,文中通过示例代码以及图文介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面来一起学习学习吧...2020-07-19- 这篇文章主要介绍了Go语言json编码驼峰转下划线、下划线转驼峰的实现,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2020-06-09
- 这篇文章主要介绍了Django 解决由save方法引发的错误,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2020-05-21
- 这篇文章主要介绍了django前端页面下拉选择框默认值设置方式,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2020-08-10
- 这篇文章主要介绍了django数据模型中null和blank的区别说明,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2020-09-03
- 这篇文章主要介绍了Go 自定义package包设置与导入操作,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2021-05-06