Go 专栏｜复合数据类型：字典 map 和 结构体 struct

楼下新开了一家重庆砂锅肥肠，扩音喇叭一直在放：正宗的老重庆砂锅肥肠，麻辣可口，老巴适了。

正不正宗不知道，反正听口音，我以为我回东北了。

本篇介绍复合数据类型的最后一篇：字典和结构体。内容很重要，编程时用的也多，需要熟练掌握才行。

本文所有代码基于 go1.16.6 编写。

字典

字典是一种非常常用的数据结构，Go 中用关键词 map 表示，类型是 map[K]V。K 和 V 分别是字典的键和值的数据类型，其中键必须支持相等运算符，比如数字，字符串等。

创建字典

有两种方式可以创建字典，第一种是直接使用字面量创建；第二种使用内置函数 make。

字面量方式创建：

// 字面量方式创建
var m = map[string]int{"a": 1, "b": 2}
fmt.Println(m) // map[a:1 b:2]

使用 make 创建：

// 使用 make 创建
m1 := make(map[string]int)
fmt.Println(m1)

还可以初始化字典的长度。在已知字典长度的情况下，直接指定长度可以提升程序的执行效率。

// 指定长度
m2 := make(map[string]int, 10)
fmt.Println(m2)

字典的零值是 nil，对值是 nil 的字典赋值会报错。

// 零值是 nil
var m3 map[string]int
fmt.Println(m3 == nil, len(m3) == 0) // true true
// nil 赋值报错
// m3["a"] = 1
// fmt.Println(m3)    // panic: assignment to entry in nil map

使用字典

赋值：

// 赋值
m["c"] = 3
m["d"] = 4
fmt.Println(m) // map[a:1 b:2 c:3 d:4]

取值：

// 取值
fmt.Println(m["a"], m["d"]) // 1 4
fmt.Println(m["k"])         // 0

即使在 Key 不存在的情况下，也是不报错的。而是返回对应类型的零值。

删除元素：

// 删除
delete(m, "c")
delete(m, "f") // key 不存在也不报错
fmt.Println(m) // map[a:1 b:2 d:4]

获取长度：

// 获取长度
fmt.Println(len(m)) // 3

判断键是否存在：

// 判断键是否存在
if value, ok := m["d"]; ok {
    fmt.Println(value) // 4
}

和 Python 对比起来看，这个用起来就很爽。

遍历：

// 遍历
for k, v := range m {
    fmt.Println(k, v)
}

引用类型

map 是引用类型，所以在函数间传递时，也不会制造一个映射的副本，这点和切片类似，都很高效。

package main

import "fmt"

func main() {
    ...

    // 传参
    modify(m)
    fmt.Println("main: ", m) // main:  map[a:1 b:2 d:4 e:10]
}

func modify(a map[string]int) {
    a["e"] = 10
    fmt.Println("modify: ", a) //   modify:  map[a:1 b:2 d:4 e:10]
}

结构体

结构体是一种聚合类型，包含零个或多个任意类型的命名变量，每个变量叫做结构体的成员。

创建结构体

首先使用 type 来自定义一个结构体类型 user，里面有两个成员变量，分别是：name 和 age。

// 声明结构体
type user struct {
    name string
    age  int
}

结构体的初始化有两种方式：

第一种是按照声明字段的顺序逐个赋值，这里需要注意，字段的顺序要严格一致。

// 初始化
u1 := user{"zhangsan", 18}
fmt.Println(u1) // {zhangsan 18}

这样做的缺点很明显，如果字段顺便变了，那么凡是涉及到这个结构初始化的部分都要跟着变。

所以，更推荐使用第二种方式，按照字段名字来初始化。

// 更好的方式
// u := user{
//     age: 20,
// }
// fmt.Println(u)    // { 20}
u := user{
    name: "zhangsan",
    age:  18,
}
fmt.Println(u) // {zhangsan 18}

未初始化的字段会赋值相应类型的零值。

使用结构体

使用点号 . 来访问和赋值成员变量。

// 访问结构体成员
fmt.Println(u.name, u.age) // zhangsan 18
u.name = "lisi"
fmt.Println(u.name, u.age) // lisi 18

如果结构体的成员变量是可比较的，那么结构体也是可比较的。

// 结构体比较
u2 := user{
    age:  18,
    name: "zhangsan",
}
fmt.Println(u1 == u)  // false
fmt.Println(u1 == u2) // true

结构体嵌套

现在我们已经定义一个 user 结构体了，假设我们再定义两个结构体 admin 和 leader，如下：

type admin struct {
    name    string
    age     int
    isAdmin bool
}

type leader struct {
    name     string
    age      int
    isLeader bool
}

那么问题就来了，有两个字段 name 和 age 被重复定义了多次。

懒是程序员的必修课。有没有什么办法可以复用这两个字段呢？答案就是结构体嵌套。

使用嵌套方式优化后变成了这样：

type admin struct {
    u       user
    isAdmin bool
}

type leader struct {
    u        user
    isLeader bool
}

代码看起来简洁了很多。

匿名成员

但这样依然不是很完美，每次访问嵌套结构体的成员变量时还是有点麻烦。

// 结构体嵌套
a := admin{
    u:       u,
    isAdmin: true,
}
fmt.Println(a) // {{lisi 18} true}
a.u.name = "wangwu"
fmt.Println(a.u.name)  // wangwu
fmt.Println(a.u.age)   // 18
fmt.Println(a.isAdmin) // true

这个时候就需要匿名成员登场了，不指定名称，只指定类型。

type admin1 struct {
    user
    isAdmin bool
}

通过这种方式可以省略掉中间变量，直接访问我们需要的成员变量。

// 匿名成员
a1 := admin1{
    user:    u,
    isAdmin: true,
}
a1.age = 20
a1.isAdmin = false

fmt.Println(a1)         // {{lisi 20} false}
fmt.Println(a1.name)    // lisi
fmt.Println(a1.age)     // 20
fmt.Println(a1.isAdmin) // false

总结

本文介绍了字典和结构体，两种很常用的数据类型。虽然篇幅不长，但基本操作也都包括，写代码肯定是没有问题的。更底层的原理和更灵活的用法就需要大家自己去探索和发现了。

当然，我也会在写完基础专栏之后，分享一些更深层的文章，欢迎大家关注，交流。

到目前为止，数据类型就都介绍完了。

先是学习了基础数据类型，包括整型，浮点型，复数类型，布尔型和字符串型。然后是复合数据类型，包括数组，切片，字典和结构体。

这些都是 Go 的基础，一定要多多练习，熟练掌握。文中的代码我都已经上传到 Github 了，有需要的同学可以点击文末地址，自行下载。

文章中的脑图和源码都上传到了 GitHub，有需要的同学可自行下载。

地址： https://github.com/yongxinz/gopher/tree/main/sc

关注公众号 AlwaysBeta，回复「goebook」领取 Go 编程经典书籍。

Go 专栏文章列表：