直接上代码,具体的说明,请看代码里面的注释:
package cmd
import (
"fmt"
"sync"
"time"
"github.com/spf13/cobra"
)
func init() {
rootCmd.AddCommand(syncCmd)
}
type Lock struct{}
func (l *Lock) Lock() {}
func (l *Lock) Unlock() {}
var syncCmd = &cobra.Command{
Use: "sync",
Run: func(cmd *cobra.Command, args []string) {
wg := &sync.WaitGroup{}
lock := &sync.RWMutex{} // 用这个方式定义的lock,在consumer函数里面,必须调用Lock()和Unlock(),不然会panic
// lock := &Lock{} // 自己定义的lock,在consumer里面可以不用调用Lock()和Unlock(),也不会panic
c := sync.NewCond(lock)
for _, n := range []int{1, 2, 3, 4, 5, 6, 7, 8, 9, 10} {
wg.Add(1)
consumer(n, c, wg)
}
producer(c, wg)
wg.Wait()
},
}
func consumer(name int, c *sync.Cond, wg *sync.WaitGroup) {
go func() {
defer wg.Done()
for {
fmt.Printf("lock[%d]\n", name)
c.L.Lock()
fmt.Printf("wait[%d]\n", name)
// Wait()中,模块会执行Unlock()的操作,所以其它的协程也会睡在这
c.Wait() // 所有协程都会睡(阻塞)在这儿,等待信号的到来
fmt.Printf("unlk[%d]\n", name)
c.L.Unlock()
}
}()
}
func producer(c *sync.Cond, wg *sync.WaitGroup) {
go func() {
defer func() {
wg.Done()
}()
for {
// fmt.Println("send signal")
// 有一个需要注意的就是:
// 如果发一个唤醒信号出去,如果此时没有协程处于wait()状态,
// 这个发出去的信号是没有任何效果的,直接会被忽略掉。
// 信号之后,再进入wait()状态的协程,要重新再次发信号才可以唤醒它
// c.Signal() // 唤醒一个协程
c.Broadcast() // 唤醒所有协程
//
// break
// time.Sleep(5 * time.Second)
}
}()
}
下面这网友写得挺清楚的:
https://blog.csdn.net/skh2015java/article/details/102730802