厚土Go学习笔记 | 28. go语言没有类却可以在结构体或任意类型定义方法

在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: %vn", v, v.Abs())
    v.Scale(5)
    fmt.Printf("After scaling: %+v, ABS: %vn", 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 不同了。

在一些多个点共同对一个点享有修改权的场景中，指针类型很有用。

厚土Go学习笔记 | 28. go语言没有类 却可以在结构体或任意类型定义方法

厚土Go学习笔记 | 28. go语言没有类却可以在结构体或任意类型定义方法