如何避免Go语言中bufio.Reader或bufio.Writer的嵌套使用？

方法一：直接类型断言检查 #

利用Go的类型断言，判断传入的io.Reader是否已经是*bufio.Reader实例，如果是就直接复用，否则再创建新的缓冲读取器：

import (
    "bufio"
    "io"
)

func ReadSomething(r io.Reader) {
    bufReader, ok := r.(*bufio.Reader)
    if !ok {
        bufReader = bufio.NewReader(r)
    }
    buf := make([]byte, 1024)
    _, err := bufReader.Read(buf)
    if err != nil {
        // 记得处理读取错误
        return
    }
    // 后续逻辑...
}

注意：bufio.Reader通常以指针形式使用，所以断言要针对*bufio.Reader而非值类型。

方法二：通过接口匹配通用缓冲读取器 #

如果担心未来会有其他带缓冲的io.Reader实现，可以定义一个包含缓冲特征方法的接口，只要传入的Reader实现了该接口，就直接复用：

import (
    "bufio"
    "io"
)

// 定义包含Buffered方法的接口，bufio.Reader天然实现这个接口
type bufferedReader interface {
    io.Reader
    Buffered() int
}

func ReadSomething(r io.Reader) {
    bufReader, ok := r.(bufferedReader)
    if !ok {
        bufReader = bufio.NewReader(r)
    }
    buf := make([]byte, 1024)
    _, err := bufReader.Read(buf)
    if err != nil {
        // 处理错误
        return
    }
    // 后续逻辑...
}

这种方式更灵活，兼容所有自带缓冲能力的Reader实现。

为什么不建议把参数改为bufio.Reader？ #

如果强制参数为bufio.Reader，会大幅降低函数的通用性——上层调用者如果使用的是os.File、strings.Reader这类原生io.Reader实现，必须手动包装后才能传入，违背了Go接口设计的“依赖抽象而非具体”原则。保持参数为io.Reader能让你的函数适配更多场景。

内容的提问来源于stack exchange，提问作者Vibbit