前言

IO 操作是我们在编程中不可避免会遇到的，例如读写文件，Go语言的 io 包中提供了相关的接口，定义了相应的规范，不同的数据类型可以根据规范去实现相应的方法，提供更加丰富的功能。

Go 语言提倡小接口 + 接口组合的方式，来扩展程序的行为以及增加程序的灵活性。io代码包恰恰就可以作为这样的一个标杆，它可以成为我们运用这种技巧时的一个参考标准。

上一篇文章，我们学习了io包中的四个核心接口，本篇文章我们来学习下多个基本接口！

ReaderFrom

io.ReaderFrom 接口定义了 ReadFrom 方法，用于从一个 Reader 中读取数据

  • 入参：Reader r
  • 返回值：n为读取的字节数，err 为读取过程中遇到的error
1
2
3

type ReaderFrom interface {
	ReadFrom(r Reader) (n int64, err error)
}

方法功能详解

  1. 该方法从 Reader r 中读取数据，直到读到文件末尾（EOF）或者 遇到 其他 error
  2. 返回的 n 就是读取的字节数

WriterTo

io.WriteTo 接口定义了 WriteTo 方法，用于将数据写入到一个 Writer 中

  • 入参：Writer w
  • 返回值：n为写入的字节数，err 为写入过程中遇到的 error
1
2
3

type WriterTo interface {
   WriteTo(w Writer) (n int64, err error)
}

方法功能详解

  1. 该方法将数据写入到 w 中，直到没有数据可写或者遇到error
  2. 返回的 n 就是写入的字节数

ReaderAt

io.ReaderAt 接口定义了 ReadAt 方法，用于从指定位置读取数据

  • 入参：读取的起始位置 off，读取后写入字节数组 p
  • 返回值：n 为成功读取的字节数，err为读取过程中遇到的 error
1
2
3

type ReaderAt interface {
	ReadAt(p []byte, off int64) (n int, err error)
}

方法功能详解

  1. 方法从指定的偏移量位置 off 开始读取指定的数据，然后至多读取 len(p) 个字节，写入字节数组 p
  2. 返回成功读取的字节量 n，以及遇到的 error
  3. 当该方法返回的 n < len(p) 时，即读取的数据没有将字节数组p填充满，返回的 err != nil，在这一点上，ReadAt方法要比 Read方法严格（Read方法规定遇到n<len(p)，本次调用可以返回 err=nil，然后在下一次调用时返回err!=nil ）
  4. 即使方法返回的 n < len(p)，也可能会占用整个字节数组 p 作为暂存空间
  5. 如果数据可用，但是不够 len（p）个字节，ReadAt 方法会阻塞，直到所有数据可用 或者 遇到 error，这一点也与 Read 方法不同
  6. 如果方法返回的 n=len(p)，并且此时正好读取到了文件末尾，ReadAt 可以返回 err = EOF 或者 err = nil。如果返回的 err 为 nil的话，下次就一定返回 EOF 了。

不能影响seek offset

  1. 如果 ReadAt 方法的入参 off 是由 seek 方法计算得到的，ReadAt 方法不能修改 off 的值

可并行

  1. 对于同一个数据源，可以并行调用 ReadAt 方法读取数据

方法实现需注意

  1. 实现该方法后，一定不要持有字节数组p（不能保存 p 的地址，用于其他地方）

WriteAt

io.WriterAt 接口定义了 WriteAt 方法，将字节数组 p 中的数据，从文件 off 偏移量开始写入到文件中

  • 入参：字节数组p，文件开始写处的偏移量 off
  • 返回值：成功写入的字节数 n，写入过程中遇到的error err
1
2
3

type WriterAt interface {
	WriteAt(p []byte, off int64) (n int, err error)
}

方法功能详解

  1. 方法将 字节数组 p 中的 len(p) 个字节，从文件 off 位置开始，写入到文件中。
  2. 方法返回成功写入的字节个数n (0 <= n <= len(p)) ，以及是否遇到 error
  3. 如果 n < len(p)，方法必须返回 err!=nil

不能影响offset

  1. 如果 WriteAt 方法的入参 off 是由 seek 方法计算得到的，ReadAt 方法不能修改 off 的值

可并行

  1. 对于同一个数据源，如果写入的范围不重叠，可以并行调用 WriteAt 方法写数据

方法实现需注意

  1. 实现该方法后，一定不要持有字节数组p（不能保存 p 的地址，用于其他地方）

ByteReader

io.ByteReader 接口定义了 ReadByte 方法，用于读取一个字节

  • 返回值：读取的一个字节，以及可能遇到的 error
1
2
3

type ByteReader interface {
   ReadByte() (byte, error)
}

方法功能详解

  1. 方法读取底层数据流的下一个字节，然后返回该字节，同时返回对应的error
  2. 如果方法读取过程中遇到error，不会消费底层数据的一个字节，相当于当前读取的位置不会变

ByteScanner

ByteScanner 是 ByteReader 接口 和 UnreadByte 方法的组合，因此该组合接口包含了两个方法，一个 ReadByte，一个 UnreadByte。从名字我们也可以看出来，ReadByte 是读取一个字节，UnreadByte 是回退一个字节。

1
2
3
4

type ByteScanner interface {
   ByteReader
   UnreadByte() error
}

方法功能详解

  1. 在调用一次 ReadByte 之后，如果调用一次 UnreadByte，那么下次再调用 ReadByte 方法得到的值，和第一次调用 ReadByte 应该一样，也就是回退了一个位置。
  2. 如果连续两次调用 UnreadByte，中间没有调用 ReadByte，可能会产生 error (这里说的可能，比如从文件开始读取一次，然后两次回退，那么回退到的位置可能就不合法了，具体还是依赖实现方法定义)

ByteWriter

io.ByteWriter 接口定义了 WriteByte 方法，用于向底层文件写入一个字节，然后返回写入过程产生的 error

1
2
3

type ByteWriter interface {
	WriteByte(c byte) error
}

RuneReader

io.RuneReader 接口定义了 ReadRune 方法，读取一个 UTF-8 字符。UTF-8 是一种变长编码规则，从 1 到 4 个字节不等，比如一个字母占一个字节，中文每个字符占用 3 个字节。ReadRune 每次读出的是一个完整的编码字符，比如读汉字，每次就会读出一个汉字。

1
2
3

type RuneReader interface {
	ReadRune() (r rune, size int, err error)
}

RuneScanner

类似 ByteScanner，RuneScanner 也是个组合接口，提供了 UnreadRune 方法，可以回退一个 rune。如果连续两次调用 UnreadRune 方法，可能会产生error。

1
2
3
4

type RuneScanner interface {
	RuneReader
	UnreadRune() error
}

StringWriter

io.StringWriter 接口定义了 WriteString 方法，用于将字符串 s 写入底层的数据流中，返回写入成功的字节数 n，以及可能产生的 error

1
2
3
4

// StringWriter is the interface that wraps the WriteString method.
type StringWriter interface {
   WriteString(s string) (n int, err error)
}

总结

本篇文章介绍了 io包 中的十个基本接口及对应的方法：

  • ReaderFrom: 定义了 ReadFrom 方法，从 Reader 中读取数据
  • WriterTo: 定义了 WriteTo 方法，向 Writer 中写入数据
  • ReaderAt: 定义了 ReadAt 方法，从指定位置读取数据
  • WriterAt: 定义了 WriteAt 方法，从指定位置写入数据
  • ByteReader: 定义了 ReadByte 方法，读取一个字节
  • ByteScanner: 组合接口，除了ReadByte 方法，定义了 UnreadByte 方法，用于回退一个字节
  • ByteWriter： 定义了 WriteByte 方法，向文件写入一个字节
  • RuneReader: 定义了 ReadRune 方法，读取一个 UTF-8 字符
  • RuneScanner: 组合接口，除了 ReadRune 方法，定义了 UnreadRune 方法，用于回退一个 rune
  • StringWriter: 定义了 WriteString 方法，用于向文件写入一个字符串

更多

个人博客: https://lifelmy.github.io/

微信公众号：CodePlayer