go并发编程

来源：https://www.cnblogs.com/cheng-sir/p/18004471
go的GMP并发模型，让go天然支持高并发，先了解一下GMP模型吧

GMP G协程，M工作线程、P处理器，M必须持有P才可以执行G
P维护着一个协程G队列，P依次将G调度到M中运行
if M0中G0发生系统调用，M0将释放P，冗余的M1获取P，继续执行P队列中剩余的G。（只要P不空闲就充分利用了CPU）
G0系统调用结束后，如果有空闲的P，则获取P继续执行G0，否则将G0放入全局队列，M0进入缓存池睡眠。（全局队列中的G主要来自从系统调用中恢复的G）

下面介绍一下编程常用的同步(synchronize)原语

互斥锁 mutex rwmutex，要了解自旋和饥饿模式
自旋最多4次，cpu核数要大于1，Processor大于1
饥饿模式阻塞超过1ms，饥饿模式下不会启动自旋过程

channel go的精华所在，注意不能关闭已经关闭的channel，不能向已关闭的channel写数据，会panic

Once#

多用于实现单例模式

饿汉模式，一般是直接创建一个包级变量直接使用即可，注意既然是单例模式，就不能让他人随意创建，类型要是私有的，使用接口暴露方法，让外部获得私有变量

懒汉模式，在第一次使用时创建，这里需要注意并发安全，可以使用sync.Once来保证并发安全

	type Singleton interface {
	Work() string
	}
	type singleton2 struct{}
	func (s *singleton2) Work() string {
	return "singleton2 is working"
	}
	func newSingleton2() *singleton2 {
	return &singleton2{}
	}
	var (
	instance *singleton2
	once sync.Once
	)
	// GetSingleton2 用于获取单例模式对象
	func GetSingleton2() Singleton {
	once.Do(func() {
	instance = newSingleton2()
	})
	return instance
	}

Pool#

sync.Pool 是 Go 语言标准库中的一个并发安全的对象池，用于缓存和重用临时对象，以提高性能。通常用于减少内存分配和垃圾回收的开销。PS：一般将变量Put回Pool中，需要将变量重置

	var studentPool = sync.Pool{
	New: func() any {
	return new(Student)
	},
	}
	func BenchmarkUnmarshalWithPool(b *testing.B) {
	for n := 0; n < b.N; n++ {
	stu := studentPool.Get().(*Student)
	json.Unmarshal(buf, stu)
	studentPool.Put(stu)
	}
	}

Cond#

Cond是条件变量，可以让一组goroutine等待某个条件的发生。当条件发生时，调用Broadcast或者Signal来通知所有等待的goroutine继续执行。
需要注意的是 sync.Cond 都要在构造的时候绑定一个 sync.Mutex。Wait() 和 Signal() 函数必须在锁保护下的临界区中执行。Wait()一般放在for循环中，因为可能会出现虚假唤醒

	var mu = &sync.Mutex{}
	var done = false
	func read(name string, c *sync.Cond) {
	c.L.Lock()
	for !done {
	c.Wait()
	}
	log.Println(name, "reading")
	c.L.Unlock()
	}
	func write(name string, c *sync.Cond) {
	time.Sleep(1 * time.Second)
	c.L.Lock()
	done = true
	c.L.Unlock()
	log.Println(name, "writting finish")
	c.Broadcast()
	}
	func main() {
	cond := sync.NewCond(mu)
	go read("reader1", cond)
	go read("reader2", cond)
	go read("reader3", cond)
	write("writer", cond)
	time.Sleep(time.Second * 3)
	}

WaitGroup#

注意Wait后不能再Add，否则会panic

	// WaitGroup 完美写法
	func main() {
	taskNum := 3
	ch := make(chan any)
	go func() {
	group := &sync.WaitGroup{}
	for i := 1; i <= taskNum; i++ {
	group.Add(1)
	go func(i int) {
	defer group.Done()
	ch <- i
	}(i)
	}
	// 确保所有取数据的协程都完成了工作，才关闭 ch
	group.Wait()
	close(ch)
	}()
	for i := range ch {
	log.Println("goroutine ", i)
	}
	log.Println("finish")
	}

扩展：errgroup，可以捕获goroutine中的panic，只要有一个goroutine出错，就会取消所有的goroutine

atomic#

原子操作，用于解决并发问题，比如计数器，锁等

	func main() {
	// 定义一个共享的计数器，使用 int64 类型
	var counter int64
	// 使用 WaitGroup 来等待所有 goroutine 完成
	var wg sync.WaitGroup
	// 启动多个 goroutine 来增加计数器的值
	for i := 0; i < 5; i++ {
	wg.Add(1)
	go func() {
	// 在每个 goroutine 中原子地增加计数器的值
	for j := 0; j < 100000; j++ {
	atomic.AddInt64(&counter, 1)
	}
	wg.Done()
	}()
	}
	// 等待所有 goroutine 完成
	wg.Wait()
	// 输出最终的计数器值
	fmt.Println("Final Counter Value:", atomic.LoadInt64(&counter))
	}

context#

WithCancel#

context.WithCancel() 创建可取消的 Context 对象，即可以主动通知子协程退出。

	func reqTask(ctx context.Context, name string) {
	for {
	select {
	case <-ctx.Done():
	fmt.Println("stop", name)
	return
	default:
	fmt.Println(name, "send request")
	time.Sleep(1 * time.Second)
	}
	}
	}
	func main() {
	ctx, cancel := context.WithCancel(context.Background())
	go reqTask(ctx, "worker1")
	time.Sleep(3 * time.Second)
	cancel()
	time.Sleep(3 * time.Second)
	}

WithValue#

如果需要往子协程中传递参数，可以使用 context.WithValue()。

	type Options struct{ Interval time.Duration }
	func reqTask(ctx context.Context, name string) {
	for {
	select {
	case <-ctx.Done():
	fmt.Println("stop", name)
	return
	default:
	fmt.Println(name, "send request")
	op := ctx.Value("options").(*Options)
	time.Sleep(op.Interval * time.Second)
	}
	}
	}
	func main() {
	ctx, cancel := context.WithCancel(context.Background())
	vCtx := context.WithValue(ctx, "options", &Options{1})
	go reqTask(vCtx, "worker1")
	go reqTask(vCtx, "worker2")
	time.Sleep(3 * time.Second)
	cancel()
	time.Sleep(3 * time.Second)
	}

WithTimeout#

如果需要控制子协程的执行时间，可以使用 context.WithTimeout 创建具有超时通知机制的 Context 对象。

ctx, cancel := context.WithTimeout(context.Background(), 1*time.Second)

WithDeadline 与 WithTimeout 类似，不同的是 WithDeadline 可以指定一个具体的时间点。

ctx, cancel := context.WithDeadline(context.Background(), time.Now().Add(1*time.Second))

semaphore#

用于限制对一组共享资源的访问。也可以手动用channel实现

	// 创建一个具有初始计数为 2 的信号量
	sem := semaphore.NewWeighted(2)
	// 启动多个 goroutine 模拟并发访问
	for i := 1; i <= 5; i++ {
	go func(id int) {
	fmt.Printf("Goroutine %d is trying to acquire the semaphore\n", id)
	// 尝试获取信号量，如果已经达到限制，则阻塞
	err := sem.Acquire(context.Background(), 1)
	if err != nil {
	fmt.Printf("Error acquiring semaphore in goroutine %d: %v\n", id, err)
	return
	}
	fmt.Printf("Goroutine %d has acquired the semaphore\n", id)
	time.Sleep(2 * time.Second)
	// 释放信号量
	sem.Release(1)
	fmt.Printf("Goroutine %d has released the semaphore\n", id)
	}(i)
	}
	// 等待所有 goroutine 完成
	time.Sleep(10 * time.Second)

singleflight#

golang.org/x/sync/singleflight 防止缓存击穿

	var sf singleflight.Group
	var wg sync.WaitGroup
	for i := 1; i <= 5; i++ {
	wg.Add(1)
	go func(id int) {
	defer wg.Done()
	// 以 "key" 为参数调用 Do 方法
	val, err, _ := sf.Do("key", func() (interface{}, error) {
	fmt.Printf("Goroutine %d is performing the operation\n", id)
	time.Sleep(2 * time.Second)
	return fmt.Sprintf("Result from goroutine %d", id), nil
	})
	if err != nil {
	fmt.Printf("Goroutine %d encountered an error: %v\n", id, err)
	} else {
	fmt.Printf("Goroutine %d got result: %v\n", id, val)
	}
	}(i)
	}
	wg.Wait()