在 Go 语言开发中，我们经常会使用标准库提供的类型和方法。但有时，标准库的功能可能无法完全满足我们的需求，这时就需要考虑如何扩展标准库类型。本文将以 bufio.Reader 为例，介绍一种常用的扩展方法：类型嵌入。
正如上述摘要所言，本文的核心在于通过类型嵌入的方式，创建自定义的 Reader 类型，并在此基础上添加新的方法或覆盖现有方法，从而实现对标准库功能的增强。
类型嵌入：扩展 bufio.Reader 的关键
类型嵌入是 Go 语言中一种强大的特性，它允许我们将一个类型嵌入到另一个类型中，从而使外部类型拥有嵌入类型的所有字段和方法。通过这种方式，我们可以基于现有的 bufio.Reader 创建一个新的 Reader 类型，并在此基础上进行扩展。
以下代码展示了如何通过类型嵌入创建一个新的 Reader 类型：
package main
import (
"bufio"
"io"
"fmt"
"strings"
)
type reader struct {
*bufio.Reader // 'reader' 嵌入了 bufio.Reader
}
func newReader(rd io.Reader) reader {
return reader{bufio.NewReader(rd)}
}
func main() {
input := strings.NewReader("hello world\nthis is a test")
myReader := newReader(input)
line, _ := myReader.ReadString('\n')
fmt.Println(line)
}在上面的代码中，我们定义了一个名为 reader 的类型，它嵌入了 *bufio.Reader。这意味着 reader 类型拥有了 bufio.Reader 的所有方法，例如 ReadByte、ReadBytes、ReadLine 等。
覆盖现有方法
除了继承 bufio.Reader 的方法外，我们还可以覆盖现有方法，以改变其行为。例如，我们可以覆盖 ReadBytes 方法，使其能够读取到多个分隔符中的任意一个为止。
package main
import (
"bufio"
"io"
"fmt"
)
type reader struct {
*bufio.Reader
}
func newReader(rd io.Reader) reader {
return reader{bufio.NewReader(rd)}
}
// 覆盖 bufio.Reader.ReadBytes
func (r reader) ReadBytes(delim byte) (line []byte, err error) {
line = make([]byte, 0)
for {
b, err := r.ReadByte()
if err != nil {
return line, err
}
line = append(line, b)
if b == delim {
return line, nil
}
}
}
func main() {
input := strings.NewReader("hello world\nthis is a test")
myReader := newReader(input)
line, _ := myReader.ReadBytes('\n')
fmt.Println(string(line))
}在上面的代码中，我们覆盖了 ReadBytes 方法，使其能够自定义读取逻辑。
添加新的方法
除了覆盖现有方法外，我们还可以向 reader 类型添加新的方法，以扩展其功能。例如，我们可以添加一个 ReadBytesEx 方法，使其能够读取到多个分隔符中的任意一个为止。
package main
import (
"bufio"
"io"
"fmt"
"strings"
)
type reader struct {
*bufio.Reader
}
func newReader(rd io.Reader) reader {
return reader{bufio.NewReader(rd)}
}
// 添加一个新的方法 ReadBytesEx
func (r reader) ReadBytesEx(delims []byte) (line []byte, err error) {
line = make([]byte, 0)
for {
b, err := r.ReadByte()
if err != nil {
return line, err
}
line = append(line, b)
for _, delim := range delims {
if b == delim {
return line, nil
}
}
}
}
func main() {
input := strings.NewReader("hello world\nthis is a test")
myReader := newReader(input)
line, _ := myReader.ReadBytesEx([]byte{'\n', ' '})
fmt.Println(string(line))
}在上面的代码中，我们添加了一个名为 ReadBytesEx 的新方法，它可以读取到多个分隔符中的任意一个为止。
注意事项
- 访问内部状态： 需要注意的是，通过类型嵌入的方式，我们无法访问原始 bufio.Reader 类型的非导出字段（即小写字母开头的字段）。这意味着我们无法直接操作 bufio.Reader 的内部缓冲区。
- 方法冲突： 如果嵌入的类型和外部类型具有相同的方法名，则外部类型的方法会覆盖嵌入类型的方法。
- 代码可维护性： 在覆盖标准库方法时，请务必谨慎，确保你的修改不会破坏原始方法的行为，并添加充分的注释，以提高代码的可维护性。
总结
通过类型嵌入的方式，我们可以方便地扩展 Go 标准库类型，例如 bufio.Reader。我们可以覆盖现有方法或添加新的方法，以满足特定的需求。但需要注意的是，我们无法访问原始类型的非导出字段，并且在覆盖方法时需要谨慎。这种方法在不修改标准库源码的情况下，提供了灵活的定制能力，是 Go 语言中一种常用的扩展技巧。
