本文介绍了在Go语言中，如何高效地将未知数量的字节数据追加到一个动态数组中，并最终获取一个字节切片。核心方法是使用[]byte切片进行动态追加，避免了container/vector包的复杂性，并提供了一种将整数编码为变长字节序列的实用示例。
在Go语言中，处理未知长度的字节序列时，通常需要动态地向一个数据结构中添加字节，并在最后获取一个[]byte切片。虽然container/vector包提供了一种动态数组的实现，但使用内置的[]byte切片通常更为简洁和高效。
使用[]byte切片动态追加字节
Go语言的切片本身就支持动态增长。可以使用append函数向切片中追加元素。这避免了使用container/vector包的必要，简化了代码并提高了性能。
以下是一个示例，展示了如何使用[]byte切片来动态构建字节序列：
立即学习“go语言免费学习笔记（深入）”；
package main
import (
"fmt"
)
func main() {
var byteSlice []byte
// 动态添加字节
byteSlice = append(byteSlice, 0x01)
byteSlice = append(byteSlice, 0x02)
byteSlice = append(byteSlice, 0x03)
fmt.Printf("Byte Slice: %v\n", byteSlice) // 输出: Byte Slice: [1 2 3]
}变长字节编码示例
以下是一个将无符号整数编码为变长字节序列的示例。这种编码方式使用较少的字节来表示较小的数字，而使用更多的字节来表示较大的数字，从而节省存储空间。
package main
import (
"fmt"
)
func vbEncodeNumber(n uint) []byte {
var bytes []byte
for {
bytes = append(bytes, byte(n%128))
if n < 128 {
break
}
n /= 128
}
bytes[len(bytes)-1] |= 128 // 设置最高位，表示这是最后一个字节
// 反转bytes
for i, j := 0, len(bytes)-1; i < j; i, j = i+1, j-1 {
bytes[i], bytes[j] = bytes[j], bytes[i]
}
return bytes
}
func main() {
encoded := vbEncodeNumber(10000)
fmt.Printf("Encoded: %v\n", encoded) // 输出: Encoded: [49 32] (十六进制表示为 [31 20])
encoded = vbEncodeNumber(127)
fmt.Printf("Encoded: %v\n", encoded) // Output: Encoded: [255]
encoded = vbEncodeNumber(128)
fmt.Printf("Encoded: %v\n", encoded) // Output: Encoded: [1 128]
}代码解释:
- vbEncodeNumber(n uint) []byte 函数: 接受一个无符号整数 n 作为输入，返回一个字节切片。
- 循环: 循环地将 n 除以 128，并将余数（0-127）添加到 bytes 切片中。
- 最高位设置: 当 n
- 反转切片: 因为我们是从最低位开始编码的，所以需要将切片反转，以保证编码后的字节序列的顺序是正确的。
- append函数: 使用内置的 append 函数向 bytes 切片中动态添加字节。
- main函数: 测试函数，分别编码了10000, 127, 128这三个数。
注意事项
- 内存分配: append 函数在底层可能会重新分配内存，如果需要频繁地追加字节，可以预先分配足够的容量，以减少内存分配的次数。可以使用 make([]byte, 0, capacity) 创建一个具有指定容量的切片。
- 错误处理: 在实际应用中，应考虑错误处理，例如，检查整数是否过大，导致编码后的字节序列过长。
- 性能优化: 对于性能要求较高的场景，可以考虑使用 bytes.Buffer 类型，它提供了更高效的字节缓冲功能。
总结
使用Go语言内置的[]byte切片，可以方便地动态构建字节序列。结合append函数，可以高效地向切片中添加字节。对于需要变长字节编码的场景，可以参考上述示例代码，并根据实际需求进行调整。避免使用container/vector包，可以简化代码并提高性能。
