文档建议反馈控制台
首页
学习
活动
专区
圈层
工具
MCP广场
文章/答案/技术大牛
发布
社区首页 >专栏 >Golang 语言标准库 io 详解

Golang 语言标准库 io 详解

作者头像
frank.
发布于 2023-03-13 10:38:56
发布于 2023-03-13 10:38:56
7160
举报
文章被收录于专栏：Golang语言开发栈Golang语言开发栈

大家好，我是 frank。 欢迎大家点击标题下方蓝色文字「Golang 语言开发栈」关注公众号。

01

介绍

Go 语言标准库 io 提供 I/O 原语的基本接口。所谓 I/O，顾名思义 input 和 output，即数据输入和输出。

本文我们通过阅读源码介绍 golang 语言标准库 io，限于篇幅，我们以介绍 io.Reader 接口为例。

02

标准库 io 的接口类型 Reader

我们阅读一下标准库 io 的 Reader 源码：

Reader 源码：

代码语言：javascript
复制
type Reader interface {
 Read(p []byte) (n int, err error)
}

阅读源码，我们可以发现标准库 io 的 Reader 接口包含一个函数 Read(p []byte) (n int, err error)。

在 Go 语言中，任意类型实现某一接口中的所有方法，即代表该类型实现了该接口。

所以，任意实现 Read(p []byte) (n int, err error) 的类型，即代表实现了 io.Reader 接口。

03

标准库中io.Reader 的实现类型

在 Golang 标准库 strings、bytes、bufio、和 os 中，都有实现 io.Reader 的类型。

标准库 strings

标准库 strings 的 Reader 类型，实现了标准库 io 的 Reader 接口，源码：

代码语言：javascript
复制
type Reader struct {
 s        string
 i        int64 // current reading index
 prevRune int   // index of previous rune; or < 0
}
// 省略代码 ...
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
}

阅读源码，我们可以发现 strings.Reader 实现了 Read(b []byte) (n int, err error)，即代表实现了 io.Reader 接口。

使用 strings 包提供的 NewReader 函数，可以创建一个 strings.Reader 类型的变量。

代码语言：javascript
复制
func NewReader(s string) *Reader { return &Reader{s, 0, -1} }

示例：

代码语言：javascript
复制
r := strings.NewReader("some io.Reader stream to be read\n")
if _, err := io.Copy(os.Stdout, r); err != nil {
    log.Fatal(err)
}

输出:

代码语言：javascript
复制
some io.Reader stream to be read

标准库 bytes

标准库 bytes 的 Reader 类型，也实现了标准库 io 的 Reader 接口，源码：

代码语言：javascript
复制
type Reader struct {
 s        []byte
 i        int64 // current reading index
 prevRune int   // index of previous rune; or < 0
}
// 省略代码 ...
// 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
}

阅读源码，我们可以发现 bytes.Reader 实现了 Read(b []byte) (n int, err error)，即代表实现了 io.Reader 接口。

使用 bytes 包提供的 NewReader 函数，可以创建一个 bytes.Reader 类型的变量。

代码语言：javascript
复制
// NewReader returns a new Reader reading from b.
func NewReader(b []byte) *Reader { return &Reader{b, 0, -1} }

示例：

代码语言：javascript
复制
r := bytes.NewReader([]byte("some io.Reader stream to be read"))
if _, err := io.Copy(os.Stdout, r); err != nil {
    log.Fatal(err)
}

输出:

代码语言：javascript
复制
some io.Reader stream to be read

标准库 bufio

标准库 bufio 的函数 NewReader，创建一个新的其缓冲区 Reader.buf 有默认大小的 *Reader，源码：

代码语言：javascript
复制
const (
 defaultBufSize = 4096
)
// 省略代码 ...
// Reader implements buffering for an io.Reader object.
type Reader struct {
 buf          []byte
 rd           io.Reader // reader provided by the client
 r, w         int       // buf read and write positions
 err          error
 lastByte     int // last byte read for UnreadByte; -1 means invalid
 lastRuneSize int // size of last rune read for UnreadRune; -1 means invalid
}
// 省略代码 ...
func NewReaderSize(rd io.Reader, size int) *Reader {
 // Is it already a Reader?
 b, ok := rd.(*Reader)
 if ok && len(b.buf) >= size {
  return b
 }
 if size < minReadBufferSize {
  size = minReadBufferSize
 }
 r := new(Reader)
 r.reset(make([]byte, size), rd)
 return r
}

// NewReader returns a new Reader whose buffer has the default size.
func NewReader(rd io.Reader) *Reader {
 return NewReaderSize(rd, defaultBufSize)
}

阅读源码，我们可以发现函数 NewReader 创建一个新的其缓冲区 Reader.buf 有默认大小的 *Reader。

示例：

代码语言：javascript
复制
rd := strings.NewReader("some io.Reader stream to be read")
r := bufio.NewReader(rd)
if _, err := io.Copy(os.Stdout, r); err != nil {
    log.Fatal(err)
}

输出:

代码语言：javascript
复制
some io.Reader stream to be read

标准库 os

标准库 os 的 os.File 类型，也实现了标准库 io 的 Reader 接口，源码：

代码语言：javascript
复制
// File represents an open file descriptor.
type File struct {
 *file // os specific
}
// 省略代码 ...
// Read reads up to len(b) bytes from the File.
// It returns the number of bytes read and any error encountered.
// At end of file, Read returns 0, io.EOF.
func (f *File) Read(b []byte) (n int, err error) {
 if err := f.checkValid("read"); err != nil {
  return 0, err
 }
 n, e := f.read(b)
 return n, f.wrapErr("read", e)
}

阅读源码，我们可以发现 os.File 实现了 Read(b []byte) (n int, err error)，即代表实现了 io.Reader 接口。

使用 os 包提供的 Open 函数，可以创建一个 os.File 类型的变量。

代码语言：javascript
复制
func Open(name string) (*File, error) {
 return OpenFile(name, O_RDONLY, 0)
}

示例：

代码语言：javascript
复制
f, _ := os.Open("service.log")
if _, err := io.Copy(os.Stdout, f); err != nil {
    log.Fatal(err)
}

输出：

代码语言：javascript
复制
some io.Reader stream to be read

04

总结

Go 语言标准库 io 提供 I/O 原语的基本接口。在 Go 语言标准库 strings、bytes、bufio、和 os 中，都有实现 io.Reader 的类型。

本文我们介绍了在 Go 语言标准库中实现 io.Reader 的类型的源码，及简单的使用示例。

建议感兴趣的读者朋友们，继续阅读 Go 语言标准库中实现 io.Writer 和 io.Closer 的类型的源码，并思考其使用场景。

通过阅读标准库 io 的代码，不仅可以熟悉标准库 io 的使用方式，还可以更加深入理解 golang 中接口的使用方式。

推荐阅读：

  1. Go 微服务框架 go-micro 使用客户端 RPC 调用服务端方法返回 408 怎么解决？
  2. Go 1.18 新增三大功能之一“模糊测试”使用方式
  3. Go 语言怎么解决编译器错误“err is shadowed during return”？
  4. Golang 语言中 map 的键值类型选择，它是并发安全的吗？
  5. Golang 语言标准库 context 包控制 goroutine

参考资料：

  1. https://pkg.go.dev/io
  2. https://pkg.go.dev/strings
  3. https://pkg.go.dev/bytes
  4. https://pkg.go.dev/bufio
  5. https://pkg.go.dev/os
本文参与 腾讯云自媒体同步曝光计划，分享自微信公众号。
原始发表：2023-02-19，如有侵权请联系 cloudcommunity@tencent.com 删除
网络安全
https
go

本文分享自 Golang语言开发栈 微信公众号，前往查看

如有侵权，请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ，欢迎热爱写作的你一起参与！

网络安全
https
go
评论
登录后参与评论
0 条评论
热度
最新
登录 后参与评论
推荐阅读
领券

腾讯云开发者

扫码关注腾讯云开发者

扫码关注腾讯云开发者

领取腾讯云代金券

热门产品

热门推荐

更多推荐

Copyright © 2013 - 2025 Tencent Cloud. All Rights Reserved. 腾讯云 版权所有 

深圳市腾讯计算机系统有限公司 ICP备案/许可证号：粤B2-20090059 粤公网安备44030502008569号

腾讯云计算（北京）有限责任公司 京ICP证150476号 |  京ICP备11018762号

问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档

Copyright © 2013 - 2025 Tencent Cloud.

All Rights Reserved. 腾讯云 版权所有

登录 后参与评论