Go语言入门记录：从基础到变量、函数、控制语句、包引用、interface、panic、go协程、Channel、sync下的waitGroup和Once等

喂-不吃素的熊宝宝

于 2023-08-27 20:38:42 发布

阅读量682

点赞数

文章标签： golang 开发语言后端

本文链接：https://blog.csdn.net/weisubao/article/details/132515336

版权

程序入口文件的包名必须是main，但主程序文件所在文件夹名称不必须是main，即我们下图hello_world.go在main中，所以感觉package main写顺理成章，但是如果我们把main目录名称改成随便的名字如filename也是可以运行的，所以迷思就在于写在文件开头的那个package main和java中不是一个概念。主程序中函数是固定的。运行这个文件用go run hello_world.go，使用go build hello_world.go会在同目录下生成一个执行文件，在windows上就是一个同名的exe文件，如hello_world.exe，使用.\hello_world.exe也可以执行得到结果。
main函数没有参数，但它可直接通过os.Args获取，os.Args是个数组，如以下代码：

package main

import (
	"fmt"
	"os"
)

func main() {
	fmt.Println(os.Args)
	if len(os.Args) > 1 {
		fmt.Println("Hello Go...", os.Args[1])
	}
	os.Exit(0)
}

快速测试的方法，单元测试。文件名用_test.go结尾，方法用TestXXX开头。包名和其他不论。在VS Code中左边会有一个按钮，点击即可测试，在其他编辑器中，可能点击保存就会运行。
以下是变量使用方法。go支持变量类型推断，即可以省略类型定义。

func TestFib(t *testing.T) {
	var a int
	a = 1
	var b int = 1

	// var (
	// 	a int
	// 	b int = 1
	// )

	t.Log(a)
	for i := 1; i < 10; i++ {
		t.Log(b)
		tmp := a
		a = b
		b = tmp + a
	}
}

以下这个一句语句对多个变量赋值也是常用交换功能了：

func TestSwap(t *testing.T) {
	a := 1
	b := 2
	a, b = b, a
	t.Log(a, b)
}

常量使用。注意iota是个语法糖，出现时会被设置位0，代码运行到下一行时iota自动加1，所以以下用到它的代码定义的常量是左移操作，每次乘以2。

const (
	Monday  = 1
	Tuesday = 2
)

const (
	Friday = 1 << iota
	Saturday
	Sunday
)

const ttt int = 1

func TestWeekDay(t *testing.T) {
	t.Log(ttt)
	t.Log(Monday, Tuesday)
	t.Log(Friday, Saturday, Sunday)
	a := 7 // 0111
	t.Log(a&Friday == Friday, a&Saturday == Saturday, a&Sunday == Sunday)
}

基本数据类型后面会添加位数，C#和其他语言中也会有部分影子。注意的是go不支持数据类型隐式转换。

bool
string
uint8 uint16 uint32 uint64 int8 int16 int32 int64
float32、float64
complex64 complex128
byte 类似 uint8
rune 类似 int32
uint：32 或 64 位
uintptr：无符号整型，用于存放一个指针

// 非要转换的话，就显示转换
b = int64(a)

go语言的指针不支持运算

func TestPointer(t *testing.T) {
	a := 1
	aPtr := &a
	// aPtr = aPtr + 1 // 不支持指针运算
	t.Log(a, aPtr) //1 0xc000018320
	t.Logf("%T %T", a, aPtr) //int *int
}

string默认是空字符串，而不是null或nil。

func TestString(t *testing.T) {
	var s string
	t.Log("*" + s + "*") //**
	t.Log(s == "")       //true
}

go中没有前置的自增和自减，之后后置的a++和a--。go中数组的比较不是比较引用地址，而是直接比较数组长度和值，只有数组长度相同才能比较，不然直接编译报错，其次每个元素都相等，数组才相等。

func TestArray(t *testing.T) {
	a := [...]int{1, 2, 3, 4}
	//b := [...]int{1, 2, 3, 4, 5}
	c := [...]int{1, 2, 3, 4}
	d := [...]int{1, 2, 3, 5}

	//t.Log(a == b) // 直接编译报错
	t.Log(a == c) //true
	t.Log(a == d) //false
}

有个按位清零的运算符&^，即当右边运算数的位为1时结果直接清零，如果是0则左边运算数的位是什么结果就是什么。相当于右边操作数对左边操作数定向位清零。

a := 7 // 0111
Readable := 1 << 0 // 00000001
a = a &^ Readable // 结果是0110赋值给a

控制语句。go语言关键字很少大概20多个，这体现在它的循环只有for循环一种。

n := 0
for n < 5 {
	n++
	...
}

// 无限循环
for {
	...
}

// 条件语句
if a == b {
} else if c == d {
} else {
}
// 比较特殊的是if后面可以有赋值和条件，如前面定义个a后面条件就用这个a
if res,err := someFunc(), err == nil {
	// 如果错误为空，即返回正确则...
} else {
}

// switch不局限于常量或整数，还可以充当if的角色，case后面可以写多个匹配项，以省略break。
switch i {
	case 0,2: // 可以多个项
		...   // 可以省略break
	case 1:
		...
	default:
		...
}
switch {
	case i > 1 && i < 3: // 类似if判断语句
		...
	default:
		....
}

数组相关。

func TestArrayLearning(t *testing.T) {
	var a [3]int
	a[0] = 1
	t.Log(a) // [1 0 0]

	arr1 := [4]int{1, 2, 3}
	t.Log(arr1) // [1 2 3 0] 没赋值的初始化为0

	arr2 := [...]int{1, 3, 5, 7}
	t.Log(arr2) //[...]表示数组长度会按照后面的值得数量初始化

	for i := 0; i < len(arr2); i++ {
		t.Log(arr2[i])
	}
	// 快捷方法
	for idx, v := range arr2 {
		t.Log(idx, v)
	}
	// 如果不要索引，则用下划线占位，不能直接去掉，不然报错
	for _, v := range arr2 {
		t.Log(v)
	}
	// 包含开始，不包含结束
	t.Log(arr2[1:3]) //[3 5]
	t.Log(arr2[1:])  //[3 5 7]
	t.Log(arr2[:3])  //[1 3 5]
}

切片相关。

func TestSlice(t *testing.T) {
	var s0 []int
	t.Log(len(s0), cap(s0)) //0
	s0 = append(s0, 1)
	t.Log(len(s0), cap(s0)) //1 1

	s1 := []int{1, 2, 3, 4}
	t.Log(len(s1), cap(s1)) //4 4

	s2 := make([]int, 3, 5)
	//t.Log(s2[0], s2[1], s2[2], s2[3])// 会报错，因为虽然初始化容量是5，但是只有3个元素可访问
	s2 = append(s2, 1)
	t.Log(s2[0], s2[1], s2[2], s2[3]) //0 0 0 1
}

切片其实是共享了存储空间，即如果两个切片有重叠元素，那么修改时会相互影响。
在这里插入图片描述

func TestSliceShare(t *testing.T) {
	var s = []int{1, 2, 3, 4, 5, 6, 7, 8, 9}
	s1 := s[3:6]
	s2 := s[5:8]
	t.Log(s1, len(s1), cap(s1)) // [4 5 6] 3 6
	t.Log(s2, len(s2), cap(s2)) // [6 7 8] 3 4
	s2[0] = 100
	t.Log(s1) // [4 5 100]
	//t.Log(s1 == s2) // 切片不能互相比较，和数组不同
	t.Log(s1 == nil)  // 切片只能和nil比较 
}

map，主要注意的是判断key是否存在。遍历也是用range得到k，v。

func TestMap(t *testing.T) {
	m1 := map[int]int{1: 100, 2: 200, 3: 300}
	t.Log(m1[1]) //100

	m2 := map[string]string{"name": "eric", "gender": "male"}
	t.Logf("len m2 = %d", len(m2)) //len m2 = 2

	m3 := map[string]int{}
	m3["math"] = 59
	m3["language"] = 61
	t.Logf("len m3 = %d", len(m3)) //len m3 = 2

	m4 := make(map[int]int, 10)    // 10是capacity
	t.Logf("len m4 = %d", len(m4)) //len m4 = 0

	// 如果key不存在则根据value类型反馈对应的初始化的值，如int的0或string的空字符串
	t.Log(m1[4])     // 0
	t.Log(m2["age"]) // 空字符串
	// 所以无法判断原先就是0值和空字符串，go提供了判断方法
	if v, ok := m1[4]; ok {
		t.Log("value is %i", v)
	} else {
		t.Log("key is not existing")
	}

	// 遍历map，也是range
	for k, v := range m2 {
		t.Logf("%s : %s", k, v)
	}
}

比较牛的是map的value还可以是函数：

m8 := map[int]func(num int) int{}
m8[1] = func(num int) int { return num }
m8[2] = func(num int) int { return num * num }
m8[3] = func(num int) int { return num * num * num }
t.Log(m8[1](2)) //2
t.Log(m8[2](2)) //4
t.Log(m8[3](2)) //8

go集合没有set，可以用map[type]bool来变相实现。

m9 := map[int]bool{}
m9[1] = true
t.Log(m9[1] == true) // true
delete(m9, 1)
t.Log(m9[1] == true) // false

字符串。

func TestString1(t *testing.T) {
	var s1 string
	s1 = "hello"
	t.Log(len(s1)) // 5

	//s1[1] = "k" // 字符串其实是不可更改的切片

	s2 := "\x53\xbb"
	t.Log(s2)      // 是个乱码
	t.Log(len(s2)) // 2，不管实际写的是什么，长度最终看看byte的长度

	s3 := "中"
	t.Log(len(s3))
	c := []rune(s3)            //3
	t.Logf("Unicode:%x", c[0]) //Unicode:4e2d
	t.Logf("UTF8:%x", s3)      //UTF8:e4b8ad

	s4 := "\xe4\xb8\xad"
	t.Log(len(s4)) //3
	t.Log(s4)      //中

	// range返回的是rune，但是我们可以通过格式化输出进行转换成字符或者二进制等形式
	// 中 4e2d 20013
	// 华 534e 21326
	// 民 6c11 27665
	// 族 65cf 26063
	s5 := "中华民族"
	for _, c := range s5 {
		t.Logf("%[1]c %[1]x %[1]d", c)
	}
}

两个库用户扩展字符串操作的：

import (
	"strings"
	"strconv"
)
func TestOperateString(t *testing.T) {
	s1 := "1,2,3,4"
	parts := strings.Split(s1, ",")
	for _, part := range parts {
		t.Log(part)
	}
	t.Log(strings.Join(parts, "-")) //1-2-3-4

	t.Logf("string is %s", strconv.Itoa(100)) //string is 100
	if v, err := strconv.Atoi("100"); err == nil {
		t.Logf("result is %d", 100+v) //result is 200
	}
}

函数，一等公民。

可返回多个值
所有参数都是值传递
函数可以作为变量的值，作为参数值，可以作为返回值
如果多个返回值，不需要多个时，用下划线代替，之前有演示过

func ComputeFunction(op1 int, op2 int) (int, int, int, int) {
	return op1 + op2, op1 - op2, op1 * op2, op1 / op2
}

func TestCompute(t *testing.T) {
	t.Log(ComputeFunction(3, 5)) //8 -2 15 0
}

再看看函数作为参数和返回值的情况：

func timeSpent(innerFunc func(op int) int) func(op int) int {
	return func(n int) int {
		start := time.Now()
		ret := innerFunc(n)
		fmt.Println("time spent seconds:", time.Since(start).Seconds())
		return ret
	}
}

func slowFunc(n int) int {
	time.Sleep(time.Second * 2)
	return n
}

func TestFunc(t *testing.T) {
	resFunc := timeSpent(slowFunc) // 得到了一个给slowFunc增强的函数，即增加了打印计时功能
	t.Log(resFunc(100))            // 100，得到的函数可以直接使用
}

再来看看可变参数，这个其他语言也有：

func sum(ops ...int) int {
	s := 0
	for _, v := range ops {
		s += v
	}
	return s
}

func TestSum(t *testing.T) {
	t.Log(sum(1, 2, 3, 4, 5)) //15
	t.Log(sum(1, 2, 3))       //6
}

再看看延迟执行：

func ClearResource() {
	fmt.Println("Clear resource...")//2.不管有无报错，最后都会先输出Clear resource...
}

func TestDeferFunc(t *testing.T) {
	defer ClearResource() // defer相当于try-catch中的finally，函数最后会执行它，尽管有panic报错也会执行它
	fmt.Println("Start") // 1.先输出Start
	panic("error")
}

最后补充自定义类型type，可以简化代码：

// 自定义一个类型，下面就能用这个，相当于定一个类型别名
type convFunc func(op int) int
func timeSpent(innerFunc convFunc ) convFunc  {
	return func(n int) int {
		start := time.Now()
		ret := innerFunc(n)
		fmt.Println("time spent seconds:", time.Since(start).Seconds())
		return ret
	}
}

go语言不支持继承（黑魔法写法除外），建议使用复合。go语言中定义结构和行为如下：

type User struct {
	Id   string
	Name string
	Age  int
}

func TestStruct(t *testing.T) {
	e1 := User{"1", "Eric", 18}
	t.Log(e1) //{1 Eric 18}
	e2 := User{Id: "2", Name: "Tom"}
	t.Log(e2.Name)  //Tom
	e3 := new(User) // 这种其实返回的是一个引用/指针
	e3.Id = "3"
	e3.Name = "Jerry"
	t.Log(e3.Age) //0

	t.Logf("%T", e1) //test.User
	t.Logf("%T", e3) //*test.User
}

定义行为和普通函数不太一样，需要注意一下，建议使用引用方式，即结构体参数使用user *User之类的格式：

unc (user User) StringFmt1() string {
	fmt.Printf("StringFmt1:The Name Address is %x\n", unsafe.Pointer(&user.Name))
	return fmt.Sprintf("%s-%s-%d", user.Id, user.Name, user.Age)
}

// 建议使用这种引用方法，避免复制产生的消耗
func (user *User) StringFmt2() string {
	fmt.Printf("StringFmt2: The Name Address is %x\n", unsafe.Pointer(&user.Name))
	return fmt.Sprintf("%s/%s/%d", user.Id, user.Name, user.Age)
}

// 这是普通函数
func StringFmt3(user User) string {
	fmt.Printf("StringFmt3: The Name Address is %x\n", unsafe.Pointer(&user.Name))
	return fmt.Sprintf("%s/%s/%d", user.Id, user.Name, user.Age)
}

func TestStructFunc(t *testing.T) {
	u := User{Id: "100", Name: "Robin", Age: 99}
	fmt.Printf("The Name Address is %x\n", unsafe.Pointer(&u.Name)) //The Name Address is c000114910
	t.Log(u.StringFmt1())                                           //StringFmt1:The Name Address is c000114940
	t.Log(u.StringFmt2())                                           //StringFmt2: The Name Address is c000114910
	t.Log(StringFmt3(u))                                            //StringFmt3: The Name Address is c0001149a0
}

go的接口比较解耦，不会强相互依赖，这可以从下面的例子看出来。下面的例子先写了一个普通的结构体并定义了一个行为，这时候没有任何接口。如果我们愿意的话，可以把这个行为单独提取出来形成一个接口。这个过程我们发现，提取接口完全不需要修改原先行为的代码和使用。只需要方法签名一致就行。

type Employee struct {
	Age int
}

func (e *Employee) AgeAddOneYear() int {
	return e.Age + 1
}

func TestInterface(t *testing.T) {
	e := Employee{Age: 18}
	t.Log(e.AgeAddOneYear()) // 19
}

然后在文件任何位置提取一个接口，保证签名一致即可。原先代码正常运行。我们称这种接口为没有侵入性。我们的实现也不依赖这个接口定义，即没有这个接口的时候我们也能正常运行使用。

type MyInterface interface {
	AgeAddOneYear() int
}

复合。

go有没有继承，只要使用父类去定义子类即可发现var pet Pet := new(Dog)是不可以的，连强制转换也是不行的。
所以一个变量在初始化的时候就决定了它的类型是Pet还是Dog，那么它接下来的行为就好说了：以为Dog复合了Pet的数据和行为，如果Dog自己有就调用自己的，自己没有就调用Pet的。所以下面的func (d *Dog) SpeakTo(host string)是关键。

type Pet struct{}

type Dog struct {
	Pet
}

func (p *Pet) Speak() {
	fmt.Printf("Pet is speaking")
}

func (p *Pet) SpeakTo(host string) {
	p.Speak()
	fmt.Print(":", host)
}

func (d *Dog) Speak() {
	fmt.Printf("Dog is speaking")
}

// 如果没有这个行为，则输出的是Pet的行为，因为复合了Pet的代码：Pet is speaking:Eric
// 如果有这个行为，则输出的是Dog的行为：Dog is speaking:Eric
func (d *Dog) SpeakTo(host string) {
	d.Speak()
	fmt.Print(":", host)
}

func TestInherit(t *testing.T) {
	d := new(Dog)
	d.SpeakTo("Eric")
}

多态的实现，主要在于那个方法的参数使用接口参数。

type Duck struct{}
type Goose struct{}

func (d *Duck) bark() string {
	return "gua gua gua..."
}

func (g *Goose) bark() string {
	return "e e e..."
}

// 这个方法是关键，使用了接口作为参数，这样就能实现多态
func diffrentbark(interf MyInterface2) string {
	return interf.bark()
}

func TestBark(t *testing.T) {
	duck := new(Duck)
	t.Log(diffrentbark(duck)) //gua gua gua...
	goose := new(Goose)
	t.Log(diffrentbark(goose)) //e e e...
}

接口的常见使用方法。

按照上述的多态的情况，定义函数的时候参数我们经常用接口，但更常用的是我们使用空接口
配合断言判断不同情况
好的习惯是让接口尽量小，一般包含一个方法，如果需要大的接口则使用小接口组合而成

// 注意这里的参数，是一个空接口，然后在代码里判断这个接口实际传过来的可能的值或者类型，做相应的行为
func UseEmptyInterface(p interface{}) {
	// if v, ok := p.(int); ok {
	// 	fmt.Println("int is", v)
	// } else if v, ok := p.(string); ok {
	// 	fmt.Println("string is", v)
	// }
	switch v := p.(type) {
	case int:
		fmt.Println("int is", v)
	case string:
		fmt.Println("string is", v)
	default:
		fmt.Println("unknown")
	}
}

func TestEmptyInterface(t *testing.T) {
	UseEmptyInterface(100)   //int is 100
	UseEmptyInterface("100") //string is 100
}

错误的处理方法，正常情况下使用errors.New("")即可。如果错误类型比较多，可以定义错误类型var xxx = errors.New()。这里需要注意的是，要习惯于go这种返回多个值得形式，并利用多个参数这个将返回值和提示信息一并返回。

var LessThanTwo = errors.New("The num should not less that 2.")

func GetFib(n int) ([]int, error) {
	if n < 2 {
		return nil, LessThanTwo
	}
	if n > 100 {
		return nil, errors.New("The num should not more than 100.")
	}
	fibList := []int{1, 1}
	for i := 2; i < n; i++ {
		fibList = append(fibList, fibList[i-2], fibList[i-1])
	}
	return fibList, nil
}

func TestError(t *testing.T) {
	if lst, err := GetFib(2); err != nil {
		t.Log(err)
	} else {
		t.Log(lst)
	}
}

看一下recover的使用方法，配合defer使用，可以获得错误信息，在这里释放资源或者做一些其他操作：

func TestRecover(t *testing.T) {
	defer func() {
		if err := recover(); err != nil {
			t.Log("recover from error:", err) //recover from error: crashed...
		}
	}()
	t.Log("start...")
	panic(errors.New("crashed..."))
}

需要注意的是如果使用os.Exist(0)退出的话，是不会执行defer的。

包的引用方式不同版本的go不一样，早期的go项目通过设置GOPATH，所有项目共用一套GOOATH以及包版本，所有的代码写在src文件夹中。后来有go mod的模块管理工具，每个项目都可以写在不同位置，只要先使用go mod init projectName初始化项目即可。后面就可以互相使用本地包，从项目名开始写，比如下面初始化了一个goProjects项目。

第三方包的引用和使用：

先下载
后引用

// 去官网查询包https://pkg.go.dev
// go get下载 -u强制更新
go get -u github.com/google/go-cmp/cmp

// 在项目中使用
import (
	"testing"
	cmp "github.com/google/go-cmp/cmp"
)

func TestImportThirdPackage(t *testing.T) {
	t.Log(cmp.Equal(1, 2)) // false
}

这个时候项目中有2个文件：

go.mod，里面记录了依赖版本的全部信息
go.sum，记录了所有依赖module的校验信息

注意一些历史小知识：最早的没有模块管理和包管理，后来有了vendor+利用第三方glide或dep来管理，它们会生成一个配置文件，在这个配置文件里记录了项目使用的依赖包信息。再后来官方退出了module功能。这就是简单的go项目管理和依赖管理的发展。

协程。关于goroutine协程们可以参考这边文章GMP 原理与调度。原先只有线程M和协程G，虽然实现了M对N的关系。但是仍然存在问题，后来增加了处理器P即调度器。

在 Go 中，线程是运行 goroutine 的实体，调度器的功能是把可运行的 goroutine 分配到工作线程上。
1、全局队列（Global Queue）：存放等待运行的 G。
2、P 的本地队列：同全局队列类似，存放的也是等待运行的 G，存的数量有限，不超过 256 个。新建 G’时，G’优先加入到 P 的本> 地队列，如果队列满了，则会把本地队列中一半的 G 移动到全局队列。
3、P 列表：所有的 P 都在程序启动时创建，并保存在数组中，最多有 GOMAXPROCS(可配置) 个。
4、M：线程想运行任务就得获取 P，从 P 的本地队列获取 G，P 队列为空时，M 也会尝试从全局队列拿一批 G 放到 P 的本地队> 列，或从其他 P 的本地队列偷一半放到自己 P 的本地队列。M 运行 G，G 执行之后，M 会从 P 获取下一个 G，不断重复下去。

在这里插入图片描述
以下的协程代码，在函数前加go就进入协程了。

func TestRoutine(t *testing.T) {
	for i := 0; i < 10; i++ {
		go func(num int) {
			fmt.Println(num)
		}(i)
	}
	time.Sleep(time.Second * 1)
}

紧接着就是不同协程之间的数据安全，即锁。还有waitgroup。

// 没有锁
func TestNotLock(t *testing.T) {
	counter := 0
	for i := 0; i < 5000; i++ {
		go func() {
			counter++
		}()
	}
	time.Sleep(time.Second * 2)
	fmt.Println(counter)
}
// 有锁，注意defer中释放
func TestLock(t *testing.T) {
	var mut sync.Mutex
	counter := 0
	for i := 0; i < 5000; i++ {
		go func() {
			defer func() {
				mut.Unlock()
			}()
			mut.Lock()
			counter++
		}()
	}
	time.Sleep(time.Second * 2)
	fmt.Println(counter)
}

waitgroup使用wait的时候就是等所有的都执行完，所有指的多少个，就是add的数量。使用done相当于减1。

func TestLockWaitGroup(t *testing.T) {
	var mut sync.Mutex
	var wg sync.WaitGroup
	counter := 0
	for i := 0; i < 5000; i++ {
		wg.Add(1)
		go func() {
			defer func() {
				mut.Unlock()
				wg.Done()
			}()
			mut.Lock()
			counter++
		}()
	}
	wg.Wait()
	fmt.Println(counter)
}

csp并发机制：channel。分为两种一种是一对一交换数据，只有双方发送和接收都完成了才能往下进行否则阻塞，即这个指需要从channel中拿走才行。第二种是给channel加buffer，即发送和接收解耦，不会阻塞。注意channel的符号<-往channel中放数据和取数据。

func doTask() chan string {
	// 如果使用这种一对一的channel，那么只有当fmt.Println(<-retCh)执行时相当于接收数据后，这个retCh <- "init data..."才取消阻塞继续执行
	// 所以fmt.Println("doing task done")总是在fmt.Println(<-retCh)后执行
	//retCh := make(chan string)
	
	// 如果使用这种channel，1就是buffer，这时候不用阻塞，retCh <- "init data..."执行完直接执行fmt.Println("doing task done")，不用管最后fmt.Println(<-retCh)何时从channel中接收数据
	retCh := make(chan string, 1)
	go func() {
		fmt.Println("doing task")
		retCh <- "init data..."
		fmt.Println("doing task done")
	}()
	return retCh
}

func doOtherTask() {
	fmt.Println("doing other task")
	time.Sleep(time.Second * 3)
	fmt.Println("doing other task is done")
}

func TestAsyncChannel(t *testing.T) {
	retCh := doTask()
	doOtherTask()
	fmt.Println(<-retCh)
}

另外有一个select多路选择器，了解一下，类似于switch，可以从不同channel中获取结果进行处理：

func TestSelect(t *testing.T) {
	select {
	case retCh := <-doTask():
		fmt.Print(retCh)
	case <-time.After(time.Second * 1):
		fmt.Print("time out")
	}
}

channel还有一个close方法。假如有个生产者和消费者，因为消费者不知道生产者生产数量，所以我们用无限for循环，但如果一直没有生产，那么就会一直阻塞在消费者获取数据处，如果生产结束时给一个关闭通道的信号，消费者判断是否已经传送结束，那么就能针对性处理。以下就是通过取值时不仅可以取值还能获取状态，如果是true就正常，如果是false意味着关闭了：

如果通道已关闭，无法传数据，会报错
如果通道关闭，则会唤醒所有的订阅者，并且传状态为false，可以用于信号订阅发布

上面说的利用这个close实现订阅发布比如关闭所有协程。

// 比如我们这边判断要不要关闭协程，则可以用
func isCanceled(ch chan struct{}) bool {
	select {
	// 只要收到消息，就会走case <-ch，当然这个消息我们一般用于关闭
	case <-ch:
		return true
	default:
		return false
	}
}
// 此时，在程序中使用channel，有需要时就关闭这个channel，在需要用到isCanceled判断的地方就能判断已关闭，然后做相应操作

context和取消。如果遇到一个嵌套的任务，取消一个节点时需要取消其子节点，就可以用context实现。

func isCanceled(ctx context.Context) bool {
	select {
	case <-ctx.Done():
		return true
	default:
		return false
	}
}

func TestContextCancel(t *testing.T) {
	ctx, cancel := context.WithCancel(context.Background())

	for i := 0; i < 5; i++ {
		go func(i int, ctx context.Context) {
			for {
				if isCanceled(ctx) {
					break
				}
				time.Sleep(time.Millisecond * 10)
			}
			fmt.Println(i, "canceled")
		}(i, ctx)
	}
	cancel()
	time.Sleep(time.Second * 1)
}
// 输出结果：
4 canceled
1 canceled
3 canceled
2 canceled
0 canceled

还是sync下的一个Once，类似懒汉模式，只运行一次的功能。

type Singleton struct{}

var once sync.Once
var singleInstance *Singleton

func GetSingleton() *Singleton {
	// 这里代码调用多次，只执行一次
	once.Do(func() {
		fmt.Println("create instance ...")
		singleInstance = new(Singleton)
	})
	return singleInstance
}

func TestSyncOnce(t *testing.T) {
	var wg sync.WaitGroup
	for i := 0; i < 5; i++ {
		wg.Add(1)
		go func() {
			obj := GetSingleton()
			fmt.Printf("obj is %x\n", unsafe.Pointer(obj))
			wg.Done()
		}()
	}
	wg.Wait()
}
// 输出
create instance ...
obj is ec2ba0
obj is ec2ba0
obj is ec2ba0
obj is ec2ba0
obj is ec2ba0

喂-不吃素的熊宝宝

关注

0
点赞
踩
0

收藏

觉得还不错? 一键收藏
0
评论
Go语言入门记录：从基础到变量、函数、控制语句、包引用、interface、panic、go协程、Channel、sync下的waitGroup和Once等

假如有个生产者和消费者，因为消费者不知道生产者生产数量，所以我们用无限for循环，但如果一直没有生产，那么就会一直阻塞在消费者获取数据处，如果生产结束时给一个关闭通道的信号，消费者判断是否已经传送结束，那么就能针对性处理。4、M：线程想运行任务就得获取 P，从 P 的本地队列获取 G，P 队列为空时，M 也会尝试从全局队列拿一批 G 放到 P 的本地队> 列，或从其他 P 的本地队列偷一半放到自己 P 的本地队列。waitgroup使用wait的时候就是等所有的都执行完，所有指的多少个，就是add的数量。
复制链接

扫一扫