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: %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 不同了。
在一些多个点共同对一个点享有修改权的场景中,指针类型很有用。

    原文作者:厚土火烟
    原文地址: https://www.jianshu.com/p/16fe66f47a61
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞