在go语言中没有类。可是,是有方法的。
给结构体定义方法,在对应的 func 和方法名之间,加上方法的接收者就可以了。
比如,我们定义了一个结构体
type Vertex struct {
X, Y float64
}
希望 Vertex 有一个 abs() 方法,就这样写
func (v *Vertex) Abs() float64 {
return math.Sqrt(v.X * v.X + v.Y * v.Y)
}
注意结构体的方法接收者是指针时,调用需要在前面加上 & 符号。
(&Vertex{3, 4}).Abs()
这样写有点冗长,在结构的方法有多个的时候,调用也不方便。你可以初始化一个变量,然后再调用。
v := &Vertex{3, 4}
v.Abs()
完整代码看一下
package main
import(
"fmt"
"math"
)
type Vertex struct {
X, Y float64
}
func (v *Vertex) Abs() float64 {
return math.Sqrt(v.X * v.X + v.Y * v.Y)
}
func main() {
v := &Vertex{3, 4}
fmt.Println(v.Abs())
}
这个示例输出的结果是 5
除了结构体,还可以对自己包中的任意类型,定义任意方法。(对来自其他包的类型或基础类型是不能定义方法的。)
比如,你可以创建一个类型
type MyFloat float64
然后给这个 MyFloat 定义方法
func (f MyFloat) Abs() float64 {
if f < 0 {
return float64(-f)
}
return float64(f)
}
完整示例
package main
import ("fmt"
"math"
)
type MyFloat float64
func (f MyFloat) Abs() float64 {
if f < 0 {
return float64(-f)
}
return float64(f)
}
func main() {
f := MyFloat(-math.Sqrt2)
fmt.Println(f.Abs())
}
运行结果
1.4142135623730951
上面两个例子中,分别实现了两个 Abs() 。一个指针类型,一个值类型。
使用指针是为了避免在每个方法调用时都进行值拷贝(如果类型是大型结构体的话,会更有效率。);其次,指针方法可以修改接收者指向的值。
我们定义一个 Scale 方法,用指针做类型做接收者。会发现 Scale 之后,Vertex 的值发生了改变,并影响以后的运算结果。
package main
import(
"fmt"
"math"
)
type Vertex struct {
X, Y float64
}
func (v *Vertex) Scale(f float64) {
v.X = v.X * f
v.Y = v.Y * f
}
func (v Vertex) Abs() float64 {
return math.Sqrt(v.X * v.X + v.Y * v.Y)
}
func main() {
v := Vertex{3, 4}
fmt.Printf("Before scaling: %+v, ABS: %v\n", v, v.Abs())
v.Scale(5)
fmt.Printf("After scaling: %+v, ABS: %v\n", v, v.Abs())
}
运行结果是
Before scaling: {X:3 Y:4}, ABS: 5
After scaling: {X:15 Y:20}, ABS: 25
之所以 Before 和 After 的结果不同,是因为 func (v *Vertex) Scale(f float64) 对 X 和 Y 做了修改,由于使用的是指针类型,所以事实上修改了 Vertex 结构体中 X 和 Y 的值。
然后 After 的输出 v 和 v.Abs() 就和 Before 不同了。
在一些多个点共同对一个点享有修改权的场景中,指针类型很有用。