Go bufio.Writer.Available 怎样判断缓冲区还剩多少空间

在 Go 里，bufio.Writer.Available() 返回的是内部缓冲区当前还没有使用的字节数。它适合回答“这一小段数据还有没有机会先放进缓冲区”，不适合回答“底层文件或网络现在还能不能写”。最容易混淆的三个值是：Size() 看总容量，Buffered() 看已经缓冲的字节，Available() 看剩余空间；正常情况下它们满足 Size() = Buffered() + Available()。

因此，标题中的问题可以直接这样处理：在写入前调用 Available()，把它和待写数据的字节长度比较；写入后重新读取它，不要把旧值当成当前状态。数据最终是否到达底层目标，还要由 Flush() 的返回错误来确认。

先把三个容量方法分清楚

bufio.NewWriterSize 创建的是带内部缓冲区的 Writer。Size() 是缓冲区总容量，Buffered() 是已经写入缓冲区、尚未刷到底层 Writer 的数据量，而 Available() 是尚未使用的空间。它们描述的是同一个缓冲区的不同切面。

package main

import (
    "bufio"
    "bytes"
    "fmt"
)

func main() {
    var dst bytes.Buffer
    w := bufio.NewWriterSize(&dst, 8)

    fmt.Println(w.Size(), w.Buffered(), w.Available())

    // 只写入内部缓冲区，观察已用空间和剩余空间的变化。
    _, err := w.WriteString("go")
    if err != nil {
        // 先处理写入错误，避免继续用一个状态未知的 Writer。
        panic(err)
    }
    fmt.Println(w.Size(), w.Buffered(), w.Available())

    // Flush 才会把缓冲数据交给底层 bytes.Buffer。
    if err := w.Flush(); err != nil {
        // Flush 的错误代表底层写入没有完成，不能只看 Buffered()。
        panic(err)
    }
    fmt.Println(dst.String())
}

对于容量为 8、已经缓冲 2 个字节的 Writer，Available() 通常就是 6。调用 Flush() 后，缓冲区回到空闲状态，剩余空间又接近总容量；底层 bytes.Buffer 的内容则变成已经刷出的数据。

写入前后怎样读取 Available

如果业务希望把小消息尽量合并后再刷新，可以用 Available() 做一个“是否适合暂存”的判断。注意要用 len(data)，因为 Go 的字符串长度和字节切片长度都是字节数，不是中文字数。

func writeSmall(w *bufio.Writer, data []byte) error {
    // Available 的单位是字节，和 len(data) 保持一致。
    if len(data) 

这里的比较只是在控制“是否先刷新”，不是在做容量许可。Write 仍然可能接受比剩余空间更大的数据；具体路径由 bufio.Writer 根据缓冲区状态决定，所以不要把 Available() 直接当成写入失败。

空间不足时，为什么不能简单认为写不进去

Available() 只描述内部缓冲区，不描述底层 Writer 的容量。对于大块写入，bufio.Writer 在特定条件下可以把数据直接交给底层 Writer，避免先复制到内部缓冲区再复制出去。于是“剩余空间不足”和“这次 Write 必然失败”是两件不同的事。

更稳妥的决策是：小数据用 Available 判断是否暂存；大数据直接检查 Write 返回的 n 和 err；批量流程结束时统一调用 Flush 并检查错误。

AvailableBuffer 与 Available 不是一回事

较新的 Go 版本还提供 AvailableBuffer()。它返回一个长度为 0、容量可利用的字节切片，适合配合 append 组装一条即将写入的数据。此时不能用 len(buf) 判断剩余空间，因为它一开始就是 0；如果要看容量，应使用 cap(buf)，或者继续调用 Writer 的 Available()。

func writeNumber(w *bufio.Writer, n int) error {
    // 返回切片的长度为 0，容量来自 Writer 的剩余缓冲空间。
    buf := w.AvailableBuffer()
    buf = append(buf, 'i', 'd', '=')
    buf = appendInt(buf, n)

    // AvailableBuffer 只保证这次紧接着的 Write 使用场景。
    _, err := w.Write(buf)
    return err
}

func appendInt(dst []byte, n int) []byte {
    if n == 0 {
        return append(dst, '0')
    }
    if n  0 {
        i--
        digits[i] = byte('0' + n%10)
        n /= 10
    }
    return append(dst, digits[i:]...)
}

示例中的辅助函数只为了展示切片如何追加；生产代码也可以使用 strconv.AppendInt。重点是：拿到 AvailableBuffer() 后应尽快完成追加并紧接着调用 Write，不要跨越下一次写操作长期保存这个切片。

四个边界情况用一张表判断

结论

记住三个公式和一个边界即可：Size() 是总容量，Buffered() 是已占用字节，Available() 是剩余字节；在没有异常状态时，三者满足 Size() = Buffered() + Available()。Available 适合优化小写入的暂存与刷新时机，不是底层可写能力探针，更不能替代 Write 和 Flush 的错误处理。

相关问题

• Available() 返回 0 就一定不能 Write 吗？不一定，大写入可能走底层 Writer；仍应以 Write 返回值和错误为准。
• Buffered() 和 Available() 能互相替代吗？不能，一个表示已用空间，一个表示未用空间。
• 为什么 Flush 成功后还要保留错误处理？Flush 负责把缓冲数据写到底层，底层文件或网络失败只能通过它的返回错误暴露。