本文将介绍如何在不修改标准库源码的情况下，扩展Go语言标准库类型的功能，以bufio.Reader为例，演示如何通过类型嵌入和方法重写或新增方法，实现自定义的读取字节功能，从而满足特定的需求。
在Go语言中，我们经常需要使用标准库提供的类型和方法。但有时，标准库提供的功能可能无法完全满足我们的特定需求。在这种情况下，我们可以通过一些技巧来扩展标准库类型的功能，而无需修改标准库的源代码。本文将以bufio.Reader为例，介绍如何通过类型嵌入和方法重写或新增方法，实现自定义的读取字节功能。
类型嵌入和方法重写/新增
一种常用的方法是使用类型嵌入（也称为组合）和方法重写或新增。这种方法的核心思想是创建一个新的类型，该类型嵌入了标准库类型，并重写或新增我们需要的方法。
下面是一个示例，演示如何扩展bufio.Reader，使其能够读取直到遇到多个分隔符中的任何一个字节为止：
立即学习“go语言免费学习笔记（深入）”；
PHP是一种功能强大的网络程序设计语言，而且易学易用，移植性和可扩展性也都非常优秀，本书将为读者详细介绍PHP编程。 全书分为预备篇、开始篇和加速篇三大部分，共9章。预备篇主要介绍一些学习PHP语言的预备知识以及PHP运行平台的架设；开始篇则较为详细地向读者介绍PKP语言的基本语法和常用函数，以及用PHP如何对MySQL数据库进行操作；加速篇则通过对典型实例的介绍来使读者全面掌握PHP。 本书
package main
import (
"bufio"
"fmt"
"io"
"strings"
)
type reader struct {
*bufio.Reader // 'reader' 嵌入 bufio.Reader，继承其所有方法
}
func newReader(rd io.Reader) reader {
return reader{bufio.NewReader(rd)}
}
// Add a new method to bufio.Reader
func (r reader) ReadBytesEx(delims []byte) (line []byte, err error) {
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 := "hello world\tthis is a test\n"
r := newReader(strings.NewReader(input))
delims := []byte{' ', '\t', '\n'}
line, err := r.ReadBytesEx(delims)
if err != nil && err != io.EOF {
fmt.Println("Error:", err)
return
}
fmt.Printf("Read line: %s\n", string(line))
}代码解释：
- 定义新的类型 reader： reader 类型嵌入了 bufio.Reader，这意味着 reader 类型自动继承了 bufio.Reader 的所有方法。
- newReader 函数： 这是一个辅助函数，用于创建一个新的 reader 实例。它接收一个 io.Reader 接口作为参数，并返回一个嵌入了 bufio.Reader 的 reader 实例。
- ReadBytesEx 方法： 这是我们新增的方法，它实现了读取直到遇到 delims 切片中任何一个字节的功能。它循环读取字节，直到遇到分隔符或发生错误。
- main 函数： 创建一个包含字符串的reader，然后调用ReadBytesEx方法，读取到指定的分隔符为止。
注意事项：
- 这种方法无法访问原始包的非导出实体（即小写字母开头的变量或函数）。
- 如果需要重写 bufio.Reader 的现有方法，只需在 reader 类型上定义一个同名方法即可。
总结：
通过类型嵌入和方法重写/新增，我们可以方便地扩展标准库类型的功能，而无需修改标准库的源代码。这种方法可以使我们的代码更加灵活和可维护。在实际开发中，可以根据具体的需求选择合适的方法来扩展标准库类型。例如，如果只需要新增功能，则可以使用类型嵌入和新增方法；如果需要修改现有功能，则可以使用类型嵌入和方法重写。
