最小化go map的内存分配

Question

我正在挖掘优化一些高度使用的代码。 我的问题归结为以下代码片段（使用pprof list命令的内存分配复制）。 如您所见，所有分配都在填充地图的行中完成（第959行）。

ROUTINE ======================== cart.BenchmarkMapWithOutCapacityFixVal in /.../cart_test.go
   3328966    3328966 (flat, cum) 15.50% of Total
         .          .    954:
         .          .    955:func BenchmarkMapWithOutCapacityFixVal(b *testing.B) {
         .          .    956:   for i := 0; i < b.N; i++ {
         .          .    957:       m := make(map[int]float32)
         .          .    958:       for k := 0; k < 10; k++ {
   3328966    3328966    959:           m[k] = 0.434295723423
         .          .    960:       }
         .          .    961:   }
         .          .    962:}

这是我想要做的：我试图在（内部）循环之前分配内存，因此没有不必要的分配发生：

ROUTINE ======================== cart.BenchmarkMapWithCapacityFixVal in /.../cart_test.go
   3214263    3214263 (flat, cum) 14.97% of Total
         .          .    963:
         .          .    964:func BenchmarkMapWithCapacityFixVal(b *testing.B) {
         .          .    965:   for i := 0; i < b.N; i++ {
   3048075    3048075    966:       m := make(map[int]float32, 10)
         .          .    967:       for k := 0; k < 10; k++ {
    166188     166188    968:           m[k] = 0.434295723423
         .          .    969:       }
         .          .    970:   }
         .          .    971:}

为什么在第968行（第二个样本）中仍然会发生分配？如何在内循环之前正确分配映射？

Answer 1

地图不是数组。 您无法在其中预先分配空间，因为您无法知道要在地图中插入元素的位置。 在make(map..., X) X只是一个容量提示 ，它不会绑定地图，并且绝对不能保证键完全散列到它中。 因此，它将尽最大努力减少未来分配的数量，但没有办法消除所有这些分配。

在这个特定示例中，如果要完全控制分配，则应该使用数组而不是地图。 使用正确的数组，您只需要一次分配。