Golang 语言 method 接收者使用值类型和指针类型的区别
01
介绍
在 Golang 语言中,function 的参数和 method 的接收者都可以选择使用值传递和指针传递(“引用传递”),需要注意的是,其中指针传递是传递的指针值的副本,而不是指针指向的数据的副本。也就是说 Golang 语言和 C 系的所有语言相同,一切传递都是值传递。本文我们主要介绍 method 的接收者怎么选择使用值类型和指针类型。
02
method 接收者的类型选择
在使用关键字 type 定义的类型上定义 method,method 的接收者也可以作为 method 的参数,类似于 function 的参数,所以 method 的接收者和 function 参数一样,我们也需要考虑选择使用值类型和指针类型。
关于这个问题,我们通常会从两方面去考虑,一是如果该 method 需要修改接收者,那么接收者必须使用指针类型;二是如果接收者占用的内存大小较大,出于性能考虑,我们也会选择使用指针类型的接收者。
除此之外,我们还需考虑一致性。也就是说,如果该类型的某些 method 必须使用指针类型的接收者,其他 method 也应该使用指针类型的接收者。因此无论如何使用该类型,它的方法集都是一致的。
最后,如果接收者是基本类型,切片和小结构体,他们的值类型的内存占用较低,并且易读。所以,该情况下除非 method 的语义需要必须使用指针类型的接收者,否则,我们可以选择使用值类型的接收者。
type User struct {
name string
}
func (u User) SetNameValueType(str string) {
fmt.Printf("SetNameValueType() pointer:%p\n", &u) // SetNameValueType() pointer:0xc000096240
u.name = str
}
func (u *User) SetNamePointerType(str string) {
fmt.Printf("SetNamePointerType() pointer:%p\n", u) // SetNamePointerType() pointer:0xc000096220
u.name = str
}
func main () {
user1 := &User{}
fmt.Printf("pointer:%p\n", user1) // pointer:0xc000096220
fmt.Println(user1) // &{}
user1.SetNameValueType("lucy")
fmt.Println(user1) // &{}
user1.SetNamePointerType("lily")
fmt.Println(user1) // &{lily}
}
阅读上面这段代码,我们可以发现值类型的接收者,调用方拷贝了副本;指针类型的接收者,调用方未拷贝副本。
03
复合类型
map 和 slice 值类似于指针:它们是包含指向底层 map 或 slice 数据的指针的描述符。复制 map 或 slice 值不会复制它指向的数据。需要注意的是,如果超过 slice 的容量,运行时会重新分配一个新内存地址。
map 源码:
type hmap struct {
count int // # live cells == size of map. Must be first (used by len() builtin)
flags uint8
B uint8 // log_2 of # of buckets (can hold up to loadFactor * 2^B items)
noverflow uint16 // approximate number of overflow buckets; see incrnoverflow for details
hash0 uint32 // hash seed
buckets unsafe.Pointer // array of 2^B Buckets. may be nil if count==0.
oldbuckets unsafe.Pointer // previous bucket array of half the size, non-nil only when growing
nevacuate uintptr // progress counter for evacuation (buckets less than this have been evacuated)
extra *mapextra // optional fields
}
slice 源码:
type slice struct {
array unsafe.Pointer
len int
cap int
}
示例代码:
func main () {
user1 := &User{}
fmt.Printf("pointer:%p\n", user1) // pointer:0xc000096220
fmt.Println(user1) // &{}
user1.SetNameValueType("lucy")
fmt.Println(user1) // &{}
user1.SetNamePointerType("lily")
fmt.Println(user1) // &{lily}
// m := make(map[int]int)
m := map[int]int{}
fmt.Printf("map pointer:%p\n", m) // map pointer:0xc000100180
m[0] = 1
fmt.Printf("map pointer:%p\n", m) // map pointer:0xc000100180
m[1] = 2
s := make([]int, 0, 1)
fmt.Printf("slice pointer:%p\n", s) // slice pointer:0xc00001c0a0
s = append(s, 1)
fmt.Printf("slice pointer:%p\n", s) // slice pointer:0xc00001c0a0
s = append(s, 2)
fmt.Printf("slice pointer:%p\n", s) // slice pointer:0xc00001c0b0
}
阅读上面这段代码,我们可以发现 map 类型未分配新内存地址,使用 append 函数向 slice 中追加元素,当元素个数未超出其容量之前,slice 也未分配新内存地址。
关于接口类型,复制接口值将复制存储在接口值中的对象。如果接口值持有一个结构体,则复制接口值会复制该结构体。如果接口值持有指针,则复制接口值会复制指针,但不会复制它指向的数据。
04
值类型怎么避免拷贝副本
阅读到这里,读者朋友可能会简单认为使用值类型会拷贝副本,使用指针类型不会拷贝副本。实际上,我们可以通过优化代码,在不改变语义的前提下,实现使用值类型也不会拷贝副本。
示例代码:
type User struct {
name string
}
func (u User) SetNameValueType(str string) {
fmt.Printf("SetNameValueType() pointer:%p\n", &u) // SetNameValueType() pointer:0xc000096240
u.name = str
}
func (u User) ValueSetName(str string) User {
u.name = str
return u
}
func main () {
user2 := &User{}
fmt.Printf("user2 pointer:%p\n", user2) // user2 pointer:0xc000010290
user2.SetNameValueType("tom") // SetNameValueType() pointer:0xc0000102a0
user3 := &User{}
fmt.Printf("user3 pointer:%p\n", user3) // user3 pointer:0xc0000102b0
user3.ValueSetName("bob")
fmt.Printf("pointer:%p\n", user3) // pointer:0xc0000102b0
}
阅读上面这段代码,我们发现 User 的 SetNameValueType 方法和 ValueSetName 方法,二者都是值传递,但是 SetNameValueType 方法会拷贝副本,ValueSetName 方法不会拷贝副本。原因是我们给 ValueSetName 方法定义了一个 User 类型的返回值,从而避免了 ValueSetName 方法拷贝副本。
05
总结
本文我们主要介绍了 method 的接收者使用值传递和指针传递的区别,并且讲述了选择使用值传递和指针传递需要考虑的决定因素,也指出了复合类型与值类型的区别。最后,使用一个简单示例演示了通过优化代码,在不改变语义的前提下,怎么实现使用值类型也不会拷贝副本。