本文深入探讨了在 go 语言中如何高效且符合惯例地预分配和填充切片，特别是包含指针类型的切片。文章阐明了使用 `make` 函数初始化切片时长度与容量的区别，指出了直接使用 `append` 填充已指定长度切片的常见误区。通过对比两种核心方法——先分配长度后赋值，以及先分配容量后追加——文章提供了清晰的示例代码和专业指导，帮助开发者避免运行时错误并优化内存使用。
理解 Go 切片的长度与容量
在 Go 语言中，切片（slice）是一个对底层数组的引用。创建切片时，make 函数允许我们指定其长度（length）和容量（capacity）。
长度（length）：切片当前包含的元素数量。容量（capacity）：切片底层数组能容纳的最大元素数量。
例如，make([]*UselessStruct, 5) 会创建一个长度为 5、容量为 5 的切片。这意味着它已经包含了 5 个元素，这些元素在初始化时是其零值。对于指针类型 *UselessStruct，其零值是 nil。
常见误区：对已指定长度的切片使用 append
许多开发者在预分配切片时，可能会错误地认为 append 操作会替换切片中已有的零值元素。考虑以下代码示例：
package mainimport "fmt"type UselessStruct struct { a int b int}func main() { mySlice := make([]*UselessStruct, 5) // 创建一个长度为5的切片，包含5个nil指针 for i := 0; i != 5; i++ { mySlice = append(mySlice, &UselessStruct{}) // 每次append都会增加切片长度 } fmt.Println(mySlice)}
这段代码的输出将是 [ 0xc… 0xc… 0xc… 0xc… 0xc…]。可以看到，前 5 个元素仍然是 nil，而 append 操作在切片的末尾又添加了 5 个新的 UselessStruct 实例的指针。这是因为 append 函数总是增加切片的长度，而不是替换现有元素。如果切片容量不足，append 还会触发底层数组的重新分配。
对于值类型切片也存在类似问题：
package mainimport "fmt"type UselessStruct struct { a int b int}func main() { mySlice := make([]UselessStruct, 5) // 创建一个长度为5的切片，包含5个零值UselessStruct for i := 0; i != 5; i++ { mySlice = append(mySlice, UselessStruct{}) // 再次append，增加长度 } fmt.Println(mySlice)}
输出将是 [{0 0} {0 0} {0 0} {0 0} {0 0} {0 0} {0 0} {0 0} {0 0} {0 0}]，同样是重复的零值元素。
惯用的预分配与填充方法
根据不同的使用场景，Go 提供了两种惯用的方法来预分配和填充切片。
1. 已知确切长度：预分配长度并直接赋值
当你知道切片最终需要包含的元素数量时，最直接且惯用的方法是使用 make 函数预先指定切片的长度，然后通过索引直接为每个元素赋值。
package mainimport "fmt"type UselessStruct struct { a int b int}func main() { // 创建一个长度为 5 的切片，包含 5 个 nil 指针 mySlice := make([]*UselessStruct, 5) // 遍历切片，为每个索引位置赋值新的 UselessStruct 实例的指针 for i := range mySlice { mySlice[i] = new(UselessStruct) // 使用 new() 分配内存并返回指针 // 或者 mySlice[i] = &UselessStruct{} // 使用复合字面量分配内存并返回指针 } fmt.Println(mySlice) // 预期输出: [0xc... 0xc... 0xc... 0xc... 0xc...] (5个不同的结构体指针)}
这种方法直接替换了切片中原有的零值元素，避免了 append 带来的额外长度增长和潜在的重复元素。它适用于需要精确控制切片内容且长度固定的场景。
2. 长度不确定或动态增长：预分配容量并使用 append
当切片的最终长度不确定，或者需要动态地向切片中添加元素时，最佳实践是预先指定切片的容量，而将长度初始化为 0。这样，append 操作可以在不触发底层数组重新分配的情况下，高效地向切片中添加元素，直到容量耗尽。
package mainimport "fmt"type UselessStruct struct { a int b int}func main() { // 创建一个长度为 0，但容量为 5 的切片 mySlice := make([]*UselessStruct, 0, 5) // 使用 append 添加元素。由于容量已预设，不会立即触发重新分配 for i := 0; i < 5; i++ { mySlice = append(mySlice, &UselessStruct{}) } fmt.Println(mySlice) // 预期输出: [0xc... 0xc... 0xc... 0xc... 0xc...] (5个不同的结构体指针) fmt.Printf("Length: %d, Capacity: %dn", len(mySlice), cap(mySlice)) // 预期输出: Length: 5, Capacity: 5}
这种方法在 append 操作的循环中，切片的长度会从 0 逐渐增加到 5，而不会出现 nil 元素。当需要添加的元素数量超过预设容量时，Go 运行时会自动进行底层数组的重新分配，通常会以当前容量的 2 倍进行扩容，这可能会带来一定的性能开销。因此，尽可能准确地预估容量可以提高程序的效率。
new(Type) 与 &Type{} 的选择
在填充指针切片时，new(UselessStruct) 和 &UselessStruct{} 都可以用来创建结构体实例并返回其地址。
new(Type)：返回一个指向 Type 类型零值的指针。&Type{}：这是一个复合字面量，创建一个 Type 类型的零值（或指定字段值），并返回其地址。
在大多数情况下，这两种方式的效果是等价的。&Type{} 形式更灵活，可以直接在创建时初始化结构体字段，例如 &UselessStruct{a: 1, b: 2}。在教程的示例中，由于我们只需要零值结构体，两者均可。
注意事项与总结
理解 len 和 cap：始终清楚切片的当前长度和底层容量。len 决定了可以访问的元素范围，cap 决定了在不重新分配内存的情况下可以追加多少元素。避免不必要的重新分配：如果能预估切片大小，尽量使用 make([]Type, 0, capacity) 或 make([]Type, length) 来预分配内存，以减少 append 操作可能引起的底层数组重新分配，这对于性能敏感的应用尤为重要。指针切片的初始化：当切片元素是指针类型时，make([]*Type, length) 会填充 length 个 nil 指针。需要通过循环迭代并赋值 new(Type) 或 &Type{} 来填充实际的结构体实例。惯用风格：对于已知长度的切片，直接通过索引赋值（mySlice[i] = …）通常被认为是更清晰和惯用的方式。对于需要动态增长的切片，预分配容量并使用 append 是标准做法。
通过掌握这些 Go 语言切片预分配和填充的惯用方法，开发者可以编写出更高效、更健壮的代码，避免常见的内存管理陷阱。
以上就是Go 语言中切片指针的预分配与填充：最佳实践的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/1413172.html
微信扫一扫 
支付宝扫一扫 
