首页 > 代码库 > go语言sync包的学习(Mutex、WaitGroup、Cond)
go语言sync包的学习(Mutex、WaitGroup、Cond)
package main;import ( "fmt" "sync" "runtime" "time")//加锁,注意锁要以指针的形式传进来,不然只是拷贝func total1(num *int, mu *sync.Mutex, ch chan bool) { mu.Lock(); for i := 0; i < 1000; i++ { *num += i; } ch <- true; mu.Unlock();}//不加锁func total2(num *int, ch chan bool) { for i := 0; i < 1000; i++ { *num += i; } ch <- true;}//Lock、Unlock与RLock、RUnlock不能嵌套使用func total3(num *int, rwmu *sync.RWMutex, ch chan bool) { for i := 0; i < 1000; i++ { rwmu.Lock(); *num += i; rwmu.Unlock(); if(i == 500) { //读锁定 rwmu.RLock(); fmt.Print(*num, " "); rwmu.RUnlock(); } } ch <- true;}func printNum(num int, cond *sync.Cond) { cond.L.Lock(); if num < 5 { //num小于5时,进入等待状态 cond.Wait(); } //大于5的正常输出 fmt.Println(num); cond.L.Unlock();}func main() { //Once.Do()保证多次调用只执行一次 once := sync.Once{}; ch := make(chan bool, 3); for i := 0; i < 3; i++ { go func(n int) { once.Do(func() { //只会执行一次,因为闭包引用了变量n,最后的值为2 fmt.Println(n) }); //给chan发送true,表示执行完成 ch <- true; }(i); } for i := 0; i < 3; i++ { //读取三次chan,如果上面三次没执行完会一直阻塞 <-ch; } //互斥锁,保证某一时刻只能有一个访问对象 mutex := sync.Mutex{}; ch2 := make(chan bool, 20); //使用多核,不然下面的结果会一样 runtime.GOMAXPROCS(runtime.NumCPU()); num1 := 0; num2 := 0; for i := 0; i < 10; i++ { go total1(&num1, &mutex, ch2); } for i := 0; i < 10; i++ { go total2(&num2, ch2); } for i := 0; i < 20; i++ { <-ch2; } //会发现num1与num2计算出的结果不一样 //而num1的结果才是正确的,因为total2没有加锁,导致多个goroutine操作num时发生数据混乱 fmt.Println(num1, num2); //读写锁,多了读锁定,和读解锁,让多个goroutine同时读取对象 rwmutex := sync.RWMutex{}; ch3 := make(chan bool, 10); num3 := 0; for i := 0; i < 10; i++ { go total3(&num3, &rwmutex, ch3); } for i := 0; i < 10; i++ { <-ch3; } fmt.Println(num3); //组等待,等待一组goroutine的结束 wg := sync.WaitGroup{}; //增加计数器 wg.Add(10); for i:= 0; i< 10; i++ { go func(n int) { fmt.Print(n, " "); //这里表示该goroutine执行完成 wg.Done(); }(i); } //等待所有线程执行完成 wg.Wait(); fmt.Println(""); //条件等待 mutex2 := sync.Mutex{}; //使用锁创建一个条件等待 cond := sync.NewCond(&mutex2); for i := 0; i < 10; i++ { go printNum(i, cond); } time.Sleep(time.Second * 1); //等待一秒后,我们先唤醒一个等待,输出一个数字 cond.L.Lock() cond.Signal(); cond.L.Unlock(); time.Sleep(time.Second * 1); //再次待待一秒后,唤醒所有,输出余下四个数字 cond.L.Lock() cond.Broadcast(); cond.L.Unlock(); time.Sleep(time.Second * 1);}
go语言sync包的学习(Mutex、WaitGroup、Cond)
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。