Go语言的标准库提供了丰富的文件操作接口，但默认的文件写入都是基于字节单位的，当我们需要处理位级别的数据，比如将零散的位数据按顺序写入文件时，就需要自己实现对应的逻辑。位级文件写入的核心是先缓存不足一个字节的位数据，凑满8位后再以字节形式写入文件，同时处理最后剩余不足8位的情况。
位级文件写入的核心思路
位级写入和常规字节写入的最大区别在于，写入的基本单位是位而不是字节，因此我们需要解决两个核心问题：
- 如何缓存零散的位数据，直到凑满一个完整的字节
- 如何处理最后剩余不足8位的位数据，避免数据丢失
位缓存的设计
我们可以用两个变量来实现位缓存：一个变量存储当前已经缓存的位数据，另一个变量记录当前缓存的位数量。每当有新的位数据加入时，先将其拼接到缓存中，当缓存的位数量达到8时，就将这8位转换为一个字节写入文件，然后清空缓存。
位拼接的逻辑
假设当前缓存的位是cache，已有位数量是cacheLen，新加入的位是bit（值为0或1），那么拼接后的缓存计算方式为：
cache = (cache << 1) | bit，同时cacheLen加1。当cacheLen等于8时，就可以将cache作为一个字节写入文件，之后重置cache为0，cacheLen为0。
完整实现示例
下面是一个完整的位级文件写入工具的实现，支持逐个写入位数据，最后自动处理剩余位：
package main
import (
"os"
)
// BitWriter 位级文件写入器
type BitWriter struct {
file *os.File // 目标文件
cache byte // 位缓存，存储当前凑的位
cacheLen int // 缓存中已有的位数量
}
// NewBitWriter 创建位级写入器
func NewBitWriter(filePath string) (*BitWriter, error) {
f, err := os.OpenFile(filePath, os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0644)
if err != nil {
return nil, err
}
return &BitWriter{
file: f,
cache: 0,
cacheLen: 0,
}, nil
}
// WriteBit 写入一个位，bit只能是0或1
func (b *BitWriter) WriteBit(bit byte) error {
if bit != 0 && bit != 1 {
return nil // 可自定义错误处理，这里简单忽略非法位
}
// 将新位拼接到缓存的高位
b.cache = (b.cache << 1) | bit
b.cacheLen++
// 凑满8位就写入一个字节
if b.cacheLen == 8 {
_, err := b.file.Write([]byte{b.cache})
if err != nil {
return err
}
// 重置缓存
b.cache = 0
b.cacheLen = 0
}
return nil
}
// WriteBits 批量写入多个位，bits是0和1组成的切片
func (b *BitWriter) WriteBits(bits []byte) error {
for _, bit := range bits {
if err := b.WriteBit(bit); err != nil {
return err
}
}
return nil
}
// Flush 将缓存中剩余的不足8位的位写入文件，不足的高位补0
func (b *BitWriter) Flush() error {
if b.cacheLen > 0 {
// 将剩余位左移到字节的高位，低位补0
b.cache = b.cache << (8 - b.cacheLen)
_, err := b.file.Write([]byte{b.cache})
if err != nil {
return err
}
b.cache = 0
b.cacheLen = 0
}
return nil
}
// Close 关闭写入器，会自动调用Flush
func (b *BitWriter) Close() error {
if err := b.Flush(); err != nil {
return err
}
return b.file.Close()
}
// 使用示例
func main() {
writer, err := NewBitWriter("test_bit.bin")
if err != nil {
panic(err)
}
defer writer.Close()
// 写入10个位，分别是1,0,1,1,0,0,1,0,1,1
bits := []byte{1, 0, 1, 1, 0, 0, 1, 0, 1, 1}
writer.WriteBits(bits)
// 最后会Flush，剩余2位会补0后写入，最终文件内容是 10110010 11000000 即 0xB2 0xC0
}
注意事项
在使用位级写入时需要注意几个问题：
- Flush方法的逻辑是根据需求调整的，如果需要剩余位不补0而是丢弃，可以修改Flush的实现
- 写入的位顺序可以根据需求调整，上述示例是将先写入的位放在字节的高位，也可以调整为放在低位
- 位级写入通常用于特定的二进制协议、压缩算法等场景，常规文本写入不需要使用该方式
位级写入的核心是对位运算的灵活运用，理解左移、或运算的作用就能快速掌握实现逻辑，实际开发中可以根据具体需求调整缓存和拼接的规则。