5.5 make 和 new

当我们想要在 Go 语言中初始化一个结构时,可能会用到两个不同的关键字 — makenew。因为它们的功能相似,所以初学者可能会对这两个关键字的作用感到困惑1,但是它们两者能够初始化的却有较大的不同。

  • make 的作用是初始化内置的数据结构,也就是我们在前面提到的切片、哈希表和 Channel2
  • new 的作用是根据传入的类型在堆上分配一片内存空间并返回指向这片内存空间的指针3

我们在代码中往往都会使用如下所示的语句初始化这三类基本类型,这三个语句分别返回了不同类型的数据结构:

  1. slice := make([]int, 0, 100)
  2. hash := make(map[int]bool, 10)
  3. ch := make(chan int, 5)
  • slice 是一个包含 datacaplen 的私有结构体 internal/reflectlite.sliceHeader
  • hash 是一个指向 runtime.hmap 结构体的指针;
  • ch 是一个指向 runtime.hchan 结构体的指针;相比与复杂的 make 关键字,new 的功能就很简单了,它只能接收一个类型作为参数然后返回一个指向该类型的指针:
  1. i := new(int)
  2. var v int
  3. i := &v

上述代码片段中的两种不同初始化方法是等价的,它们都会创建一个指向 int 零值的指针。

golang-make-and-ne

图 5-14 make 和 new 初始化的类型

接下来我们将分别介绍 makenew 在初始化不同数据结构时的过程,我们会从编译期间和运行时两个不同阶段理解这两个关键字的原理,不过由于前面的章节已经详细地分析过 make 的原理,所以这里会将重点放在另一个关键字 new 上。

5.5.1 make

在前面的章节中我们已经谈到过 make 在创建切片、哈希表和 Channel 的具体过程,所以在这一小节,我们只是会简单提及 make 相关的数据结构的初始化原理。

golang-make-typecheck

图 5-15 make 关键字的类型检查

在编译期间的类型检查阶段,Go 语言就将代表 make 关键字的 OMAKE 节点根据参数类型的不同转换成了 OMAKESLICEOMAKEMAPOMAKECHAN 三种不同类型的节点,这些节点会调用不同的运行时函数来初始化相应的数据结构。

5.5.2 new

编译器会在中间代码生成阶段通过以下两个函数处理该关键字:

  1. func callnew(t *types.Type) *Node {
  2. ...
  3. n := nod(ONEWOBJ, typename(t), nil)
  4. ...
  5. return n
  6. }
  7. func (s *state) expr(n *Node) *ssa.Value {
  8. switch n.Op {
  9. case ONEWOBJ:
  10. if n.Type.Elem().Size() == 0 {
  11. return s.newValue1A(ssa.OpAddr, n.Type, zerobaseSym, s.sb)
  12. }
  13. typ := s.expr(n.Left)
  14. vv := s.rtcall(newobject, true, []*types.Type{n.Type}, typ)
  15. return vv[0]
  16. }
  17. }

需要注意的是,无论是直接使用 new,还是使用 var 初始化变量,它们在编译器看来就是 ONEWODCL 节点。这些节点在这一阶段都会被 cmd/compile/internal/gc.walkstmt 转换成通过 runtime.newobject 函数在堆上申请内存:

  1. func walkstmt(n *Node) *Node {
  2. switch n.Op {
  3. case ODCL:
  4. v := n.Left
  5. if v.Class() == PAUTOHEAP {
  6. if prealloc[v] == nil {
  7. prealloc[v] = callnew(v.Type)
  8. }
  9. nn := nod(OAS, v.Name.Param.Heapaddr, prealloc[v])
  10. nn.SetColas(true)
  11. nn = typecheck(nn, ctxStmt)
  12. return walkstmt(nn)
  13. }
  14. case ONEW:
  15. if n.Esc == EscNone {
  16. r := temp(n.Type.Elem())
  17. r = nod(OAS, r, nil)
  18. r = typecheck(r, ctxStmt)
  19. init.Append(r)
  20. r = nod(OADDR, r.Left, nil)
  21. r = typecheck(r, ctxExpr)
  22. n = r
  23. } else {
  24. n = callnew(n.Type.Elem())
  25. }
  26. }
  27. }

不过这也不是绝对的,如果通过 var 或者 new 创建的变量不需要在当前作用域外生存,例如不用作为返回值返回给调用方,那么就不需要初始化在堆上。

runtime.newobject 函数会是获取传入类型占用空间的大小,调用 runtime.mallocgc 在堆上申请一片内存空间并返回指向这片内存空间的指针:

  1. func newobject(typ *_type) unsafe.Pointer {
  2. return mallocgc(typ.size, typ, true)
  3. }

runtime.mallocgc 函数的实现大概有 200 多行代码,我们会在后面的章节中详细分析 Go 语言的内存管理机制。

5.5.3 小结

到了最后,简单总结一下 Go 语言中 makenew 关键字的实现原理,make 关键字的作用是创建切片、哈希表和 Channel 等内置的数据结构,而 new 的作用是为类型申请一片内存空间,并返回指向这片内存的指针。


wechat-account-qrcode

本作品采用知识共享署名-非商业性使用-禁止演绎 4.0 国际许可协议进行许可。