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,

}